Update app.py

app.py

@@ -27,18 +27,23 @@ 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
-    cached_image_path = os.path.join(cached_dir, "processed_image.png")
-    cached_audio_path = os.path.join(cached_dir, "audio.wav")
-
-    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}")
+    # 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")
+
+                # 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}")
+
+                # 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):