Update app.py

app.py

@@ -57,30 +57,20 @@ def launch_app():
             progress_bar = gr.Progress()
 
         with gr.Row():
-            download_path = gr.State("")
-            download_button = gr.Button("Download Cleaned CSV", visible=False)
+            cleaned_file_output = gr.File(label="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_prepare_download(file):
+        def process_and_show_results(file):
             cleaned_csv_path, image_files = clean_and_visualize(file, progress=progress_bar)
-            return cleaned_csv_path, image_files, gr.Button(visible=True)
-
-        def trigger_download(path):
-            return gr.File.update(value=path, visible=True)
+            return cleaned_csv_path, image_files
 
         clean_button.click(
-            fn=process_and_prepare_download,
+            fn=process_and_show_results,
             inputs=file_input,
-            outputs=[download_path, output_gallery, download_button]
-        )
-
-        download_button.click(
-            fn=trigger_download,
-            inputs=[download_path],
-            outputs=[gr.File(label="Cleaned CSV", visible=False)]
+            outputs=[cleaned_file_output, output_gallery]
         )
 
     app.launch()