Update app.py

app.py

@@ -178,14 +178,15 @@ def save_csv(*args):
             })
     df = pd.DataFrame(data)
     try:
-        # Generate timestamp
+        # Generate timestamped CSV filename
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         csv_filename = f'edited_metadata_{timestamp}.csv'
         df.to_csv(csv_filename, index=False)
-        return f"Data saved successfully to '{csv_filename}'!"
+        return csv_filename  # Return CSV path for download
     except Exception as e:
         return f"Error saving data: {e}"
 
+
 # Functions to update character and keyword counts
 def update_title_info(text):
     return f"{len(text)} characters"
@@ -273,6 +274,15 @@ with gr.Blocks(css="""
         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
+
+# Update click events
+    save_csv_button.click(
+        fn=save_csv,
+        inputs=all_outputs,
+        outputs=[csv_output, save_output]  # Return both the file and status
+    )
+
 
     # Collect all output components for extraction and saving
     all_outputs = []