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()