Update src/pdfchatbot.py

src/pdfchatbot.py

@@ -172,10 +172,24 @@ class PDFChatBot:
         history.append((text, ''))
         return history
 
-    # Function to download conversation history
-    def download_chat(history,text):
-        with open('chat_history.csv', 'w', newline='') as file:
-            writer = csv.writer(file)
-            writer.writerow(["User Input", "Bot Response"])
-            writer.writerows(chat_history)
-        return "chat_history.csv"
+    def download_chat_history(self, history):
+        """
+        Download the chat history as a CSV file.
+
+        Parameters:
+            history (list): List of chat history tuples.
+        Returns:
+            gr.outputs.File: A CSV file containing the chat history.
+        """
+        import csv
+        import io
+    
+        output = io.StringIO()
+        writer = csv.writer(output)
+        writer.writerow(["User Input", "Bot Response"])  # header row
+    
+        for user_input, bot_response in history:
+            writer.writerow([user_input, bot_response])
+    
+        output.seek(0)
+        return gr.outputs.File("chat_history.csv", output.read(), "text/csv")