Update app.py

app.py

@@ -2,8 +2,6 @@ import gradio as gr
 import pandas as pd
 from functools import partial
 
-import os
-
 def save_chatbot_dialogue(chat_tutor, save_type):
 
     formatted_convo = pd.DataFrame(chat_tutor.conversation_memory, columns=['user', 'chatbot'])
@@ -26,6 +24,7 @@ def save_chatbot_dialogue(chat_tutor, save_type):
 
 save_json = partial(save_chatbot_dialogue, save_type='json')
 save_txt = partial(save_chatbot_dialogue, save_type='txt')
+save_csv = partial(save_chatbot_dialogue, save_type='csv')
 
 
 # history is a list of list  
@@ -99,14 +98,16 @@ with gr.Blocks() as demo:
         Export your chat history as a .json, .txt, or .csv file
         """)
         with gr.Row():
-          export_dialogue_button_json = gr.Button("JSON")
-          export_dialogue_button_txt = gr.Button("TXT")
+            export_dialogue_button_json = gr.Button("JSON")
+            export_dialogue_button_txt = gr.Button("TXT")
+            export_dialogue_button_csv = gr.Button("CSV")
         
         file_download = gr.Files(label="Download here",
-                                file_types=['.txt', '.json'], type="file", visible=False)
+                                 file_types=['.json', '.txt', '.csv'], type="file", visible=False)
         
         export_dialogue_button_json.click(save_json, study_tutor, file_download, show_progress=True)
         export_dialogue_button_txt.click(save_txt, study_tutor, file_download, show_progress=True)
+        export_dialogue_button_csv.click(save_csv, study_tutor, file_download, show_progress=True)
     
 demo.queue()
 demo.launch()