Update app.py

app.py

@@ -57,20 +57,30 @@ def launch_app():
             progress_bar = gr.Progress()
 
         with gr.Row():
-            download_button = gr.DownloadButton(label="Download Cleaned CSV", visible=False)
+            download_path = gr.State("")
+            download_button = gr.Button("Download Cleaned CSV", visible=False)
 
         with gr.Row():
             output_gallery = gr.Gallery(label="Visualization Results", show_label=True, elem_id="gallery", columns=[2],
                                         rows=[2], object_fit="contain", height="auto")
 
-        def process_and_show_download(file):
+        def process_and_prepare_download(file):
             cleaned_csv_path, image_files = clean_and_visualize(file, progress=progress_bar)
-            return gr.DownloadButton.update(value=cleaned_csv_path, visible=True), image_files
+            return cleaned_csv_path, image_files, gr.Button(visible=True)
+
+        def trigger_download(path):
+            return gr.File.update(value=path, visible=True)
 
         clean_button.click(
-            fn=process_and_show_download,
+            fn=process_and_prepare_download,
             inputs=file_input,
-            outputs=[download_button, output_gallery]
+            outputs=[download_path, output_gallery, download_button]
+        )
+
+        download_button.click(
+            fn=trigger_download,
+            inputs=[download_path],
+            outputs=[gr.File(label="Cleaned CSV", visible=False)]
         )
 
     app.launch()