Update src/interface.py

src/interface.py

@@ -1,22 +1,20 @@
 import csv
 import gradio as gr
 
-# Define a function to export conversations to CSV
-def export_conversations(conversations):
-    # Create a CSV writer
-    csvfile = "conversations.csv"
-    with open(csvfile, "w", newline="") as f:
-        writer = csv.writer(f)
-        
-        # Write the header row
-        writer.writerow(["User Input", "Model Response"])
-        
-        # Iterate over the conversations and write each row
-        for conversation in conversations:
-            for i, (user_input, model_response) in enumerate(conversation):
-                writer.writerow([user_input, model_response])
-    
-    return csvfile
+#NEW
+# Function to update chat history and conversation history
+def update_chat_history(user_input, bot_response):
+        chat_history.update(value=chat_history.value + [(user_input, bot_response)])
+        conversation_history.append([user_input, bot_response])
+
+# Function to download conversation history
+def download_conversation_history():
+    with open('conversation_history.csv', 'w', newline='') as file:
+        writer = csv.writer(file)
+        writer.writerow(["User Input", "Bot Response"])
+        writer.writerows(conversation_history)
+    return "conversation_history.csv"
+#NEW
 
 # Gradio application setup
 def create_demo():
@@ -73,26 +71,22 @@ def create_demo():
                 submit_button = gr.Button('Send')
             with gr.Column(scale=0.20):
                 uploaded_pdf = gr.UploadButton("📁 Upload PDF", file_types=[".pdf"], elem_id='upload_pdf')
+            #NEW
+            with gr.Colum(scale=0.20):
+                download_button = gr.Button('Download Conversation History')
+            
+            # Conversation History
+            conversation_history = []
+            
+        # Event listeners
+        submit_button.click(fn=update_chat_history, inputs=[text_input, chat_history], outputs=chat_history)
+        download_button.click(fn=download_conversation_history, outputs="download")
+            #NEW
 
-        # Define the Gradio interface
-        iface = Interface(
-            fn=export_conversations,
-            inputs="conversations",
-            outputs=Output.File(type="csv"),
-            title="Export Conversations",
-            description="Export conversations as a CSV file"
-        )
-
-        # Add a download button to trigger the export
-        download_button = Button("Download Conversations as CSV")
-
-        # Add the components to the interface
-        iface.launch(inputs=["conversations"], outputs="file", api_name="export_conversations", api_open_browser=False, flagging_callback=None, analytics_enabled=False, enable_queue=True, show_error=True, allow_screenshot=True, allow_flagging=True)
-
-    
-    return demo, chat_history, show_img, text_input, submit_button, uploaded_pdf, slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k
+            
+        
 
 if __name__ == '__main__':
-    demo, chatbot, show_img, text_input, submit_button, uploaded_pdf, slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k = create_demo()
+    demo, chatbot, show_img, text_input, submit_button, uploaded_pdf, slider_chunk_size,slider_overlap_percentage,slider_temp,slider_k, download_button = create_demo()
     demo.queue()
     demo.launch()