Update app.py

app.py

@@ -58,7 +58,6 @@
 
 
 
-
 import gradio as gr
 import subprocess
 import os
@@ -102,7 +101,7 @@ def get_random_images():
         generate_images()
         image_files = [f for f in os.listdir(OUTPUT_DIR) if f.endswith(".png")]
     random_images = random.sample(image_files, min(10, len(image_files)))
-    return [Image.open(os.path.join(OUTPUT_DIR, img)) for img in random_images], image_files
+    return [Image.open(os.path.join(OUTPUT_DIR, img)) for img in random_images], random_images
 
 # === Save Image Function ===
 def save_image(image_filename):
@@ -115,8 +114,8 @@ def save_image(image_filename):
 # === Gradio Interface ===
 def generate_and_display():
     images, filenames = get_random_images()
-    buttons = [gr.Button("Save", elem_id=filename) for filename in filenames]
-    return images, buttons
+    save_buttons = [gr.Button("Save", elem_id=filename) for filename in filenames]
+    return images, save_buttons
 
 def save_button_clicked(image_filename):
     return save_image(image_filename)
@@ -125,9 +124,10 @@ with gr.Blocks() as demo:
     gr.Markdown("# 🎨 AI-Generated Clothing Designs - Dresses")
     generate_button = gr.Button("Generate New Designs")
     output_gallery = gr.Gallery(label="Generated Designs", columns=5)
-    generate_button.click(fn=generate_and_display, outputs=output_gallery)
-
-    # Add save buttons to each image dynamically
+    
+    generate_button.click(fn=generate_and_display, outputs=[output_gallery, generate_button])
+    
+    # Add save buttons for each image in the gallery
     output_gallery.select(fn=save_button_clicked, inputs=output_gallery, outputs=None)
 
 if __name__ == "__main__":