Update app.py

app.py

@@ -45,8 +45,9 @@ def infer(color_prompt, phone_type_prompt, design_prompt):
 
 # Save the design
 def save_design(image):
-    file_path = "saved_designs/saved_design.png"
-    os.makedirs(os.path.dirname(file_path), exist_ok=True)
+    saved_designs_dir = "saved_designs"
+    os.makedirs(saved_designs_dir, exist_ok=True)
+    file_path = os.path.join(saved_designs_dir, f"design_{int(time.time())}.png")
     image.save(file_path)
     return f"Design saved as {file_path}!"
 
@@ -119,13 +120,21 @@ with gr.Blocks(css=custom_css) as interface:
     with gr.Tabs():
         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")
             
-            # Load the saved designs dynamically when the tab is rendered
-            def display_saved_designs():
+            # Gallery to display saved designs
+            saved_images_gallery = gr.Gallery(label="Saved Designs").style(grid=[2], height="400px")
+            
+            # Button to refresh gallery
+            refresh_button = gr.Button("Refresh Gallery")
+            
+            def refresh_gallery():
                 return fetch_saved_designs()
             
-            saved_images_gallery.render(value=fetch_saved_designs())
+            refresh_button.click(
+                refresh_gallery,
+                inputs=[],
+                outputs=[saved_images_gallery]
+            )
 
         with gr.Tab("Design"):
             with gr.Row():
@@ -167,4 +176,4 @@ with gr.Blocks(css=custom_css) as interface:
             """)
 
 # Launch the app
-interface.launch(debug=True)
+interface.launch(debug=True)