add export csv functionality

app.py

@@ -38,34 +38,11 @@ with gr.Blocks() as main_app:
                    "Context Prompt", "Final Response GPT", "Create Clean Message", "Create Audio Elevenlabs", "Create Video D-iD", "Final Time"]
         table_times = gr.DataFrame(headers=columns, visible=False, interactive=False)
 
-        export_button = gr.Button(value="Export CSV")
-
-
-
-    # with gr.Tab('Ventana Pruebas'):
-    #     with gr.Row():
-    #         with gr.Column():
-    #             button_show = gr.Button(value="Mostrar texto")
-    #         with gr.Column():
-    #             button_hidden = gr.Button(value="Ocultar texto")
-        
-    #     with gr.Row(visible=False) as first_row:
-    #         text = gr.Textbox(value="Hola mundo")
-
+        with gr.Column():
+            with gr.Row(visible=False) as row_export_csv:
+                export_button = gr.Button(value="Export CSV")
+                csv = gr.File(interactive=False, visible=False)
 
-    # Actions
-        
-    # button_show.click(
-    #     fn=make_visible,
-    #     inputs=None,
-    #     outputs=first_row
-    # )
-    
-    # button_hidden.click(
-    #     fn=make_invisible,
-    #     inputs=None,
-    #     outputs=first_row
-    # )
 
     text.submit(
         fn=get_answer,
@@ -73,6 +50,10 @@ with gr.Blocks() as main_app:
         outputs=[chat, output_audio, output_video, table_times]
     ).then(
         lambda: None, None, [text]
+    ).then(
+        fn=make_visible,
+        inputs=None,
+        outputs=row_export_csv
     )
 
     button_text.click(
@@ -81,11 +62,16 @@ with gr.Blocks() as main_app:
         outputs=[chat, output_audio, output_video, table_times]
     ).then(
         lambda: None, None, [text]
+    ).then(
+        fn=make_visible,
+        inputs=None,
+        outputs=row_export_csv
     )
 
     export_button.click(
         fn=export_dataframe,
         inputs=table_times,
+        outputs=csv
     )
 
     main_app.load(init_greeting, inputs=[chat, messages], outputs=[chat, messages])

backend_functions.py

@@ -8,13 +8,13 @@ from pinecone import Pinecone
 import uuid
 import re
 import pandas as pd
-import tensorflow as tf
 from google.cloud import storage
 from elevenlabs.client import ElevenLabs, AsyncElevenLabs
 from elevenlabs import play, save, Voice, stream
 from pymongo.mongo_client import MongoClient
 from utils import create_folders
 from gcp import download_credentials
+from csv import writer
 from dotenv import load_dotenv
 load_dotenv()
 
@@ -382,4 +382,6 @@ def init_greeting(chatbot, history_messages):
 def export_dataframe(df):
     final_df = pd.DataFrame(df)
     final_df = final_df.iloc[1:]
-    final_df.to_csv("./csv_times/csv_times.csv", index=False, encoding='utf-8')
+    final_df.to_csv("./csv_times/csv_times.csv", index=False, encoding='utf-8')
+    
+    return gr.File(value="./csv_times/csv_times.csv", visible=True)