Yoxas commited on
Commit
94f8386
·
verified ·
1 Parent(s): 411ec57

Update src/pdfchatbot.py

Browse files
Files changed (1) hide show
  1. src/pdfchatbot.py +1 -23
src/pdfchatbot.py CHANGED
@@ -170,26 +170,4 @@ class PDFChatBot:
170
  if not text:
171
  raise gr.Error('Enter text')
172
  history.append((text, ''))
173
- return history
174
-
175
- def download_chat_history(self, history):
176
- """
177
- Download the chat history as a CSV file.
178
-
179
- Parameters:
180
- history (list): List of chat history tuples.
181
- Returns:
182
- gr.outputs.File: A CSV file containing the chat history.
183
- """
184
- import csv
185
- import io
186
-
187
- output = io.StringIO()
188
- writer = csv.writer(output)
189
- writer.writerow(["User Input", "Bot Response"]) # header row
190
-
191
- for user_input, bot_response in history:
192
- writer.writerow([user_input, bot_response])
193
-
194
- output.seek(0)
195
- return gr.outputs.File("chat_history.csv", output.read(), "text/csv")
 
170
  if not text:
171
  raise gr.Error('Enter text')
172
  history.append((text, ''))
173
+ return history