Update app.py

app.py

@@ -7,10 +7,9 @@ from code import (initialize_session,
                   visualize_all       
                   )
 
-def df_to_csv(dataframe):
-    csv_path = "/mnt/data/modified_dataframe.csv"
-    dataframe.to_csv(csv_path, index=False)
-    return gr.File.update(value=filepath, label="Download Data")
+def export_csv(df):
+    df.to_csv("output.csv")
+    return gr.File.update(value="output.csv", visible=True)
     
 with gr.Blocks(gr.themes.Soft(primary_hue="emerald",
                               font=[gr.themes.GoogleFont("Quicksand"),"ui-sans-serif", "system-ui", "sans-serif"],
@@ -70,9 +69,10 @@ with gr.Blocks(gr.themes.Soft(primary_hue="emerald",
                 with gr.Column(scale=1):
                     combine_button = gr.Button("Confirm")
                 with gr.Column(scale=1):
-                    download_button = gr.DownloadButton(label="Download Data", visible=True)
-                with gr.Column(scale=1):
-                    receipt_count_text = gr.Textbox(label="Uploaded files:", interactive=False)
+                    download_button = gr.Button("Export")
+                    csv = gr.File(interactive=False, visible=False)
+                #with gr.Column(scale=1):
+                    #receipt_count_text = gr.Textbox(label="Uploaded files:", interactive=False)
 
             gr.Markdown("""
             # Visualize Expenses
@@ -109,7 +109,7 @@ with gr.Blocks(gr.themes.Soft(primary_hue="emerald",
     upload_button.click(image_to_df, inputs=[image_input, api_key_input], outputs=image_output)
     upload_button.click(fn=lambda files: [file.name for file in files], inputs=image_input, outputs=uploaded_images)
     combine_button.click(fn=update_combined_df, inputs=[image_output, session_id], outputs=[combined_df, receipt_count_text])
-    download_button.click(fn=df_to_csv, inputs=combined_df, outputs=gr.File())
+    download_button.click(fn=export_csv, inputs=combined_df, outputs=csv)
     visualize_button.click(fn=visualize_all, inputs=[combined_df, month, budget], outputs=[fig1, fig2, fig3, fig4, fig5, fig6])
 
 demo.launch(debug=True)