render_example: handle recipe specific media rendering

recipes/BulkEval.py

@@ -276,7 +276,7 @@ def render_agg_inputs(key: str, del_key: str, d: AggFunction):
     # def render_settings(self):
     #     language_model_settings()
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         render_documents(state)
 
     def render_output(self):

recipes/BulkRunner.py

@@ -253,7 +253,7 @@ def render_form_v2(self):
             flatten_dict_key="url",
         )
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         render_documents(state)
 
     def render_output(self):

recipes/ChyronPlant.py

@@ -126,7 +126,7 @@ def run_chyron(self, state: dict):
                 return text
         return ""
 
-    def render_example(self, state):
+    def render_run_preview_output(self, state):
         col1, col2 = gui.columns(2)
         with col1:
             gui.write(state.get("midi_translation", ""))

recipes/CompareLLM.py

@@ -122,7 +122,7 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
     def render_output(self):
         _render_outputs(gui.session_state, 450)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         col1, col2 = gui.columns(2)
         with col1:
             gui.write("**Prompt**")

recipes/CompareText2Img.py

@@ -251,7 +251,7 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
                     )
                 ]
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         col1, col2 = gui.columns(2)
         with col1:
             gui.markdown("```properties\n" + state.get("text_prompt", "") + "\n```")

recipes/CompareUpscaler.py

@@ -156,7 +156,7 @@ def render_description(self):
     def render_output(self):
         _render_outputs(gui.session_state)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         col1, col2 = gui.columns(2)
         with col1:
             gui.image(state.get("input_image"), caption="Input Image")

recipes/DeforumSD.py

@@ -263,12 +263,15 @@ def estimate_run_duration(self):
         # in seconds
         return gui.session_state.get("max_frames", 100) * MODEL_ESTIMATED_TIME_PER_FRAME
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         display = self.preview_input(state)
         gui.markdown("```lua\n" + display + "\n```")
 
         gui.video(state.get("output_video"), autoplay=True)
 
+    def render_example_preview_media(self, published_run: dict):
+        gui.video(published_run.saved_run.state.get("output_video"))
+
     @classmethod
     def preview_input(cls, state: dict) -> str:
         input_prompt = state.get("input_prompt")

recipes/DocExtract.py

@@ -153,7 +153,7 @@ def validate_form_v2(self):
     def preview_description(self, state: dict) -> str:
         return "Transcribe YouTube videos in any language with Whisper, Google Chirp & more, run your own GPT4 prompt on each transcript and save it all to a Google Sheet. Perfect for making a YouTube-based dataset to create your own chatbot or enterprise copilot (ie. just add the finished Google sheet url to the doc section in https://gooey.ai/copilot)."
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         if sheet_url := state.get("sheet_url"):
             render_documents(state, label="**Input Documents**")
             gui.write("**Google Sheets URL**")

recipes/DocSearch.py

@@ -118,7 +118,7 @@ def render_output(self):
         refs = gui.session_state.get("references", [])
         render_sources_widget(refs)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         render_documents(state)
         gui.html("**Search Query**")
         gui.write("```properties\n" + state.get("search_query", "") + "\n```")

recipes/DocSummary.py

@@ -128,7 +128,7 @@ def validate_form_v2(self):
     def render_output(self):
         render_output_with_refs(gui.session_state)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         render_documents(state)
         gui.write("**Instructions**")
         gui.write("```properties\n" + state.get("task_instructions", "") + "\n```")

recipes/EmailFaceInpainting.py

@@ -333,7 +333,7 @@ def _get_email_body(
 
         return email_body
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         if state.get("email_address"):
             gui.write("**Input Email** -", state.get("email_address"))
         elif state.get("twitter_handle"):

recipes/FaceInpainting.py

@@ -323,7 +323,7 @@ def related_workflows(self) -> list:
             SEOSummaryPage,
         ]
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         col1, col2 = gui.columns(2)
         with col2:
             output_images = state.get("output_images")

recipes/GoogleGPT.py

@@ -151,7 +151,7 @@ def render_output(self):
         refs = gui.session_state.get("references", [])
         render_sources_widget(refs)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         gui.write("**Search Query**")
         gui.write("```properties\n" + state.get("search_query", "") + "\n```")
         site_filter = state.get("site_filter")

recipes/GoogleImageGen.py

@@ -226,7 +226,7 @@ def render_steps(self):
         else:
             gui.div()
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         gui.write(
             f"""
             **Google Search Query** `{state.get("search_query", '')}` \\

recipes/ImageSegmentation.py

@@ -298,7 +298,7 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
         yield
 
     def render_output(self):
-        self.render_example(gui.session_state)
+        self.render_run_preview_output(gui.session_state)
 
     def render_steps(self):
         col1, col2, col3, col4 = gui.columns(4)
@@ -337,7 +337,7 @@ def render_steps(self):
             else:
                 gui.div()
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         col1, col2 = gui.columns(2)
 
         with col1:

recipes/Img2Img.py

@@ -5,7 +5,7 @@
 from pydantic import BaseModel
 
 import gooey_gui as gui
-from bots.models import Workflow
+from bots.models import PublishedRun, Workflow
 from daras_ai_v2.base import BasePage
 from daras_ai_v2.img_model_settings_widgets import img_model_settings
 from daras_ai_v2.loom_video_widget import youtube_video
@@ -137,7 +137,7 @@ def render_output(self):
         for img in output_images:
             gui.image(img, show_download_button=True)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         col1, col2 = gui.columns(2)
         with col2:
             output_images = state.get("output_images", [])
@@ -149,6 +149,14 @@ def render_example(self, state: dict):
             gui.write("**Prompt**")
             gui.write("```properties\n" + state.get("text_prompt", "") + "\n```")
 
+    def render_example_preview_media(self, published_run: PublishedRun):
+        state = published_run.saved_run.state
+        input_image = state.get("input_image", None)
+        output_images = state.get("output_images", [])
+        with gui.div(className="d-flex gap-2"):
+            gui.image(input_image)
+            gui.image(output_images[0])
+
     def run(self, state: dict) -> typing.Iterator[str | None]:
         request: Img2ImgPage.RequestModel = self.RequestModel.parse_obj(state)
 

recipes/Lipsync.py

@@ -106,7 +106,7 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
                     max_frames=max_frames,
                 )
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         output_video = state.get("output_video")
         if output_video:
             gui.write("#### Output Video")
@@ -115,7 +115,7 @@ def render_example(self, state: dict):
             gui.div()
 
     def render_output(self):
-        self.render_example(gui.session_state)
+        self.render_run_preview_output(gui.session_state)
 
     def related_workflows(self) -> list:
         from recipes.DeforumSD import DeforumSDPage

recipes/LipsyncTTS.py

@@ -127,7 +127,7 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
         state["input_audio"] = state["audio_url"]
         yield from LipsyncPage(request=self.request).run(state)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         output_video = state.get("output_video")
         if output_video:
             gui.video(
@@ -140,7 +140,7 @@ def render_example(self, state: dict):
             gui.div()
 
     def render_output(self):
-        self.render_example(gui.session_state)
+        self.render_run_preview_output(gui.session_state)
 
     def get_raw_price(self, state: dict):
         # _get_tts_provider comes from TextToSpeechPage

recipes/ObjectInpainting.py

@@ -292,7 +292,7 @@ def run(self, state: dict):
         )
         state["output_images"] = diffusion_images
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         col1, col2 = gui.columns(2)
         with col2:
             output_images = state.get("output_images")

recipes/QRCodeGenerator.py

@@ -13,7 +13,7 @@
 
 import gooey_gui as gui
 from app_users.models import AppUser
-from bots.models import Workflow
+from bots.models import PublishedRun, Workflow
 from daras_ai.image_input import (
     upload_file_from_bytes,
     bytes_to_cv2_img,
@@ -438,7 +438,12 @@ def render_output(self):
         state = gui.session_state
         self._render_outputs(state)
 
-    def render_example(self, state: dict):
+    def render_example_preview_media(self, published_run: PublishedRun):
+        output_images = list(published_run.saved_run.state.get("output_images", []))
+        if output_images:
+            gui.image(output_images[0])
+
+    def render_run_preview_output(self, state: dict):
         self._render_outputs(state, max_count=1)
 
     def _render_outputs(self, state: dict, max_count: int | None = None):

recipes/RelatedQnA.py

@@ -80,7 +80,7 @@ def render_steps(self):
     def render_output(self):
         render_qna_outputs(gui.session_state, 300)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         gui.write("**Search Query**")
         gui.write("```properties\n" + state.get("search_query", "") + "\n```")
         site_filter = state.get("site_filter")

recipes/RelatedQnADoc.py

@@ -59,7 +59,7 @@ def validate_form_v2(self):
     def render_output(self):
         render_qna_outputs(gui.session_state)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         gui.write("**Search Query**")
         gui.write("```properties\n" + state.get("search_query", "") + "\n```")
         site_filter = state.get("site_filter")

recipes/SEOSummary.py

@@ -245,7 +245,7 @@ def render_steps(self):
         else:
             gui.div()
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         gui.write(
             f"""
             Search Query `{state.get('search_query', '')}` \\

recipes/SmartGPT.py

@@ -183,7 +183,7 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
     def render_output(self):
         render_output_with_refs(gui.session_state)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         gui.write("**Prompt**")
         gui.write("```properties\n" + state.get("input_prompt", "") + "\n```")
         render_output_with_refs(state, 200)

recipes/SocialLookupEmail.py

@@ -217,7 +217,7 @@ def render_steps(self):
         else:
             gui.div()
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         gui.write("**Email Address**")
         gui.write(state.get("email_address", ""))
         gui.write("**Email Body Output**")

recipes/Text2Audio.py

@@ -132,7 +132,7 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
     def render_output(self):
         _render_output(gui.session_state)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         col1, col2 = gui.columns(2)
         with col1:
             gui.markdown("```properties\n" + state.get("text_prompt", "") + "\n```")

recipes/TextToSpeech.py

@@ -463,7 +463,7 @@ def related_workflows(self) -> list:
             CompareText2ImgPage,
         ]
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         col1, col2 = gui.columns(2)
         with col1:
             text = state.get("text_prompt")

recipes/Translation.py

@@ -149,7 +149,7 @@ def validate_form_v2(self):
     def render_output(self):
         text_outputs("**Translations**", key="output_texts", height=300)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         text_outputs("**Translations**", value=state.get("output_texts", []))
 
     def render_steps(self):

recipes/VideoBots.py

@@ -624,7 +624,7 @@ def run_as_api_tab(self):
     def get_example_preferred_fields(cls, state: dict) -> list[str]:
         return ["input_prompt", "messages"]
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         input_prompt = state.get("input_prompt")
         if input_prompt:
             gui.write(

recipes/asr_page.py

@@ -219,7 +219,7 @@ def validate_form_v2(self):
     def render_output(self):
         text_outputs("**Transcription**", key="output_text", height=300)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         render_documents(state)
         text_outputs("**Transcription**", value=state.get("output_text"))
 

recipes/embeddings_page.py

@@ -49,7 +49,7 @@ def render_output(self):
             gui.write(f"##### `embeddings[{i}]`")
             gui.json(embedding, depth=0)
 
-    def render_example(self, state: dict):
+    def render_run_preview_output(self, state: dict):
         texts = gui.session_state.setdefault("texts", [""])
         for i, text in enumerate(texts):
             texts[i] = gui.text_area(f"`texts[{i}]`", value=text, disabled=True)

routers/account.py

@@ -295,9 +295,7 @@ def _render_run(pr: PublishedRun):
         workflow = Workflow(pr.workflow)
         visibility = PublishedRunVisibility(pr.visibility)
 
-        workflow.page_cls().render_published_run_full_width(
-            pr, workflow_pill=workflow.short_title
-        )
+        workflow.page_cls().render_example(pr, workflow_pill=workflow.short_title)
 
     gui.write("# Saved Workflows")
     explore_path = get_route_path(explore_page)