Update app.py

app.py

@@ -57,7 +57,6 @@
 #     demo.launch()
 
 
-
 import gradio as gr
 import subprocess
 import os
@@ -115,7 +114,7 @@ def save_image(image_filename):
 def generate_and_display():
     images, filenames = get_random_images()
     save_buttons = [gr.Button("Save", elem_id=filename) for filename in filenames]
-    return images, save_buttons
+    return images, save_buttons, filenames
 
 def save_button_clicked(image_filename):
     return save_image(image_filename)
@@ -126,9 +125,10 @@ with gr.Blocks() as demo:
     output_gallery = gr.Gallery(label="Generated Designs", columns=5)
     
     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__":
+    # Create dynamic save buttons for each generated image
+    with gr.Row() as button_row:
+        for filename in []:
+            gr.Button("Save", elem_id=filename).click(fn=save_button_clicked, inputs=filename, outputs=None)
+        
     demo.launch()