Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from v1_functions import *
 from v2_functions import *
+from google_drive import * 
+
 
 with gr.Blocks(title="Engage Quotation Chatbot") as demo:
     gr.Markdown("# Engage Quotation Chatbot")
@@ -64,6 +66,7 @@ with gr.Blocks(title="Engage Quotation Chatbot") as demo:
                     generate_btn = gr.Button("1. Generate Quotation")
                     analyse_quotation_btn = gr.Button("2. Analyse Quotation")
                     save_csv_btn = gr.Button("3. Export All")
+                    upload_btn = gr.Button("4. Upload to Google Drive")
                     units_output = gr.Textbox(label="Cost Summary", lines=3, interactive=False)
                     progress_update = gr.Textbox(label="Progress Update", lines=2, interactive=False)
                     recalc_btn = gr.Button("Recalculate")
@@ -81,6 +84,11 @@ with gr.Blocks(title="Engage Quotation Chatbot") as demo:
                 outputs=[quantity_table,progress_update, units_output]
             )
 
+            upload_btn.click(
+                fn=v1_upload_to_gdrive,
+                inputs=[prd_box,tasks_table,quantity_table,merged_table,units_output,analysis_box],
+                outputs=progress_update
+            )
             save_csv_btn.click(
                 fn=v1_save_csv,
                 inputs=[prd_box,tasks_table,quantity_table,merged_table,units_output],
@@ -131,6 +139,7 @@ with gr.Blocks(title="Engage Quotation Chatbot") as demo:
                     engage_generate_quotation_btn = gr.Button("3. Generate Final Quotation")  # New button
                     engage_analyse_quotation_btn = gr.Button("4. Analyse Quotation")  # New button
                     engage_save_all_btn = gr.Button("5. Save All Output")  # New Save CSV button
+                    engage_upload_btn = gr.Button("6. Upload to Google Drive")  # New Save CSV button
 
                     engage_units_output = gr.Textbox(label="Cost Summary", lines=3, interactive=False)
                     engage_progress_update = gr.Textbox(label="Progress Update", lines=2, interactive=False)
@@ -176,6 +185,13 @@ with gr.Blocks(title="Engage Quotation Chatbot") as demo:
                         outputs=engage_progress_update  # You can adjust the output as needed
                     )
 
+                    engage_upload_btn.click(
+                        fn=v2_upload_to_gdrive,
+                        inputs=[engage_prd_box_v2, engage_component_table_v2, engage_task_table_v2, engage_task_unit_table_v2, engage_quantity_table_v2, engage_analysis_box, engage_units_output],
+                        outputs=engage_progress_update  # You can adjust the output as needed
+                    )
+
+
     # Replace single textbox with separate components
     with gr.Tab(label= "Load Project"):
         gr.Markdown("### Past submissions")