Update app.py

app.py

@@ -27,27 +27,19 @@ CACHE_DIR = "gradio_cached_examples"
 
 #for local cache
 def load_cached_example_outputs(example_index: int) -> Tuple[str, str]:
-    # Search for any folder that contains the example index in its name
-    for root, dirs, files in os.walk(CACHE_DIR):
-        for dir_name in dirs:
-            if dir_name.endswith(str(example_index)):  # Search for folder ending with the example index
-                cached_dir = os.path.join(root, dir_name)
-                cached_image_path = os.path.join(cached_dir, "processed_image.png")
-                cached_audio_path = os.path.join(cached_dir, "audio.wav")
+    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")
 
-                # Debugging: Print the paths being checked
-                print(f"Looking for cached image at: {cached_image_path}")
-                print(f"Looking for cached audio at: {cached_audio_path}")
+    # 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}")
 
-                # Check if the cached files exist
-                if os.path.exists(cached_image_path) and os.path.exists(cached_audio_path):
-                    return cached_image_path, cached_audio_path
-
-    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)
+# Function to handle the example click, based on index
+def on_example_click(index: int, *args, **kwargs):
+    return load_cached_example_outputs(index)
 
 # # to handle the example click, it now accepts arbitrary arguments
 # def on_example_click(*args, **kwargs):
@@ -142,13 +134,17 @@ with gr.Blocks(css=css) as demo:
         example_index = example_map.get(example_image, 1)  # Default to 1 if not found
         return load_cached_example_outputs(example_index)
 
-    # Add the examples to Gradio
     gr.Examples(
-        examples=example_inputs,
+        examples=[
+            ["examples/1.png", 3, "A scenic mountain view", 500],  # Example 1
+            ["examples/2.png", 2, "A forest with birds", 500],    # Example 2
+            ["examples/3.png", 1, "A crowded city", 500]          # Example 3
+        ],
         inputs=[image, num_audios, prompt, steps],
         outputs=[processed_image, generated_audio],
-        cache_examples=True,
-        fn=on_example_click
+        cache_examples=True,  # Cache examples to avoid running the model
+        fn=update_examples,   # Call function that loads example based on index
+        inputs=[index_selector]  # Use the dropdown to select the example index
     )
 
     gr.on(