Update app.py

app.py

@@ -26,33 +26,26 @@ model.setup()
 CACHE_DIR = "gradio_cached_examples"
 
 #for local cache
-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
+def load_cached_example_outputs(example_index: int) -> Tuple[Union[str, None], str]:
+    cached_dir = os.path.join(CACHE_DIR, str(example_index)) 
     cached_image_path = os.path.join(cached_dir, "processed_image.png")
     cached_audio_path = os.path.join(cached_dir, "audio.wav")
-    cached_video_path = os.path.join(cached_dir, "processed_video.mp4")  # Support video output too
-
-    # Ensure cached files exist (either image or video)
-    if os.path.exists(cached_image_path) and os.path.exists(cached_audio_path):
-        return cached_image_path, cached_audio_path
-    elif os.path.exists(cached_video_path) and os.path.exists(cached_audio_path):
-        return cached_video_path, cached_audio_path
+    cached_video_path = os.path.join(cached_dir, "processed_video.mp4")
+    
+    if os.path.exists(cached_video_path) and os.path.exists(cached_audio_path):
+        return cached_video_path, cached_audio_path  # Return video output
+    elif os.path.exists(cached_image_path) and os.path.exists(cached_audio_path):
+        return cached_image_path, cached_audio_path  # Return image output
     else:
         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]:
+    return load_cached_example_outputs(example_index)
 
-# detect whether the input is an image or a video
-def determine_input_type(filepath: str):
-    if filepath.endswith((".png", ".jpg", ".jpeg")):
-        return gr.Image
-    elif filepath.endswith(".mp4"):
-        return gr.Video
-    else:
-        raise ValueError("Unsupported file type")
-
-# Function 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
+# # 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
 
 
 @spaces.GPU(duration=280)
@@ -121,20 +114,30 @@ with gr.Blocks(css=css) as demo:
                     show_controls=True,
                 ),
             )
-
-    def load_example_by_input_type(file_input, *args, example_index):
-        input_type = determine_input_type(file_input)
-        if input_type == gr.Video:
-            return gr.Video(label="Video Example"), on_example_click(example_index=example_index)
+    
+     # 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]
+    ]
+
+    # Function to dynamically switch output visibility based on file type
+    def handle_example_click(example):
+        output, audio = on_example_click(int(example[0][-5]))  # Example index based on file path
+        if output.endswith('.mp4'):
+            return None, output, audio  # Show video and hide image
         else:
-            return gr.Image(label="Image Example"), on_example_click(example_index=example_index)
+            return output, None, audio  # Show image and hide video
 
+    # Add the examples to Gradio
     gr.Examples(
-        examples=example_inputs,
-        inputs=[gr.File(label="Select image or video")],
-        outputs=[processed_media, generated_audio],
-        cache_examples=True,
-        fn=lambda *args, **kwargs: load_example_by_input_type(args[0], example_index=args[0][0]),  # Dynamically detect and process
+        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
     )
 
     gr.on(