Update app.py

app.py

@@ -26,19 +26,18 @@ model.setup()
 CACHE_DIR = "gradio_cached_examples"
 
 #for local cache
-def load_cached_example_outputs(example_index: int) -> Tuple[Union[str, None], str]:
-    cached_dir = os.path.join(CACHE_DIR, str(example_index)) 
+def load_cached_example_outputs(example_index: int) -> Tuple[str, 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")
-    
-    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
+
+    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}")
 
+
+
 # to handle the example click
 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
@@ -138,21 +137,16 @@ with gr.Blocks(css=css) as demo:
         # ["gradio_cached_examples/4/processed_image.png", 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 output, None, audio  # Show image and hide video
+        output, audio = on_example_click(int(example[0].split('/')[-2]))  # Extract index from path
+        return output, audio
 
     # Add the examples to Gradio
     gr.Examples(
         examples=examples,
         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
+        fn=handle_example_click
     )
 
     gr.on(