Update app.py

app.py

@@ -40,9 +40,22 @@ def load_cached_example_outputs(example_index: int) -> Tuple[Union[str, None], s
         raise FileNotFoundError(f"Cached outputs not found for example {example_index}")
 
 # to handle the example click
-def on_example_click(example_index: int) -> Tuple[Union[str, None], str]:
+def load_cached_example_outputs(example_index: int) -> Tuple[str, str]:
+    cached_dir = os.path.join(CACHE_DIR, str(example_index))  # use the example index to find the directory
+    cached_image_path = os.path.join(cached_dir, "processed_image.png")
+    cached_audio_path = os.path.join(cached_dir, "audio.wav")
+
+    # Ensure cached files exist
+    if os.path.exists(cached_image_path) and os.path.exists(cached_audio_path):
+        return cached_image_path, cached_audio_path
+    else:
+        raise FileNotFoundError(f"Cached outputs not found for example {example_index}")
+
+# Function to handle the example click, it now accepts arbitrary arguments and returns the appropriate cached outputs
+def on_example_click(*args, example_index=1, **kwargs):
     return load_cached_example_outputs(example_index)
 
+
 # # to handle the example click, it now accepts arbitrary arguments
 # def on_example_click(*args, **kwargs):
 #     return load_cached_example_outputs(1)  # Always load example 1 for now
@@ -118,11 +131,11 @@ with gr.Blocks(css=css) as demo:
     
      # Example inputs, the last two are videos
     examples = [
-        ["examples/1.png", 3, "Example 1", 500],
-        ["examples/2.png", 3, "Example 2", 500],
-        ["examples/3.png", 3, "Example 3", 500],
-        ["examples/4.mp4", 3, "Example 4", 500],
-        ["examples/5.mp4", 3, "Example 5", 500]
+        ["gradio_cached_examples/1/processed_image.png", 3, "Example 1", 500],
+        ["gradio_cached_examples/2/processed_image.png", 3, "Example 2", 500],
+        ["gradio_cached_examples/3/processed_image.png", 3, "Example 3", 500],
+        # ["gradio_cached_examples/4/processed_image.png", 3, "Example 4", 500],
+        # ["gradio_cached_examples/4/processed_image.png", 3, "Example 5", 500]
     ]
 
     # Function to dynamically switch output visibility based on file type
@@ -135,10 +148,11 @@ with gr.Blocks(css=css) as demo:
 
     # Add the examples to Gradio
     gr.Examples(
-        examples=examples,
-        inputs=[gr.File(label="Select image or video")],  # Use gr.File to support both image and video inputs for examples
-        outputs=[processed_image, processed_video, generated_audio],
-        fn=handle_example_click
+        examples=example_inputs,
+        inputs=[image, num_audios, prompt, steps],
+        outputs=[processed_image, generated_audio],
+        cache_examples=True,
+        fn=lambda *args, **kwargs: on_example_click(example_index=args[0][0]),  # Select the correct example based on the index
     )
 
     gr.on(