Update app.py

app.py

@@ -121,7 +121,8 @@ def update_image_metadata(*args):
     success_count = 0
     error_count = 0
     image_paths = []
-    
+    edited_image_paths = []  # List to store paths of edited images
+
     # First, extract all image paths
     for i in range(100):
         idx = i * 6
@@ -141,10 +142,12 @@ def update_image_metadata(*args):
             keywords = args[idx + 5]
             if write_metadata_to_image(copied_path, title, description, keywords):
                 success_count += 1
+                edited_image_paths.append(copied_path)  # Add edited image path to the list
             else:
                 error_count += 1
 
-    return f"Images copied and metadata updated. Successful updates: {success_count}, Failed updates: {error_count}. Check the 'edited_images' folder for results."
+    return (f"Images copied and metadata updated. Successful updates: {success_count}, Failed updates: {error_count}.", edited_image_paths)
+
 
 def copy_images_to_project_dir(image_paths):
     copied_paths = []
@@ -272,10 +275,11 @@ with gr.Blocks(css="""
 
     # Save Buttons and Output
     with gr.Row():
+        save_metadata_button = gr.Button("Save Metadata to Images")
         save_csv_button = gr.Button("Save CSV")
         save_output = gr.Textbox(label="Save Status", interactive=False)
         csv_output = gr.File(label="Download CSV")  # File download component
-
+        edited_images_output = gr.File(label="Download Edited Images", file_count="multiple")  # Add this line
 
 
 
@@ -305,4 +309,11 @@ with gr.Blocks(css="""
         outputs=[csv_output, save_output]  # Return both the file and status
     )
 
+save_metadata_button.click(
+    fn=update_image_metadata,
+    inputs=all_outputs,
+    outputs=[save_output, gr.File(label="Download Edited Images", file_count="multiple")]  # Add file output for edited images
+)
+
+
 demo.launch()