Update app.py

app.py

@@ -54,7 +54,7 @@ def save_design(image):
 def fetch_saved_designs():
     saved_designs_dir = "saved_designs"
     if not os.path.exists(saved_designs_dir):
-        return ["No saved designs found."]
+        return []
     
     designs = []
     for file_name in os.listdir(saved_designs_dir):
@@ -120,7 +120,12 @@ with gr.Blocks(css=custom_css) as interface:
         with gr.Tab("Home"):
             gr.Markdown("Welcome to the **AI Phone Cover Designer**! Below are some of your saved designs.")
             saved_images_gallery = gr.Gallery(label="Saved Designs")
-            saved_images_gallery.update(value=fetch_saved_designs())
+            
+            # Load the saved designs dynamically when the tab is rendered
+            def display_saved_designs():
+                return fetch_saved_designs()
+            
+            saved_images_gallery.render(value=fetch_saved_designs())
 
         with gr.Tab("Design"):
             with gr.Row():
@@ -162,4 +167,4 @@ with gr.Blocks(css=custom_css) as interface:
             """)
 
 # Launch the app
-interface.launch(debug=True)
+interface.launch(debug=True)