Update app.py
Browse files
app.py
CHANGED
@@ -212,13 +212,16 @@ def conversation(qa_chain, message, history):
|
|
212 |
#response_source1_page = response_sources[0].metadata["page"] + 1
|
213 |
#response_source2_page = response_sources[1].metadata["page"] + 1
|
214 |
#response_source3_page = response_sources[2].metadata["page"] + 1
|
|
|
|
|
|
|
215 |
# print ('chat response: ', response_answer)
|
216 |
# print('DB source', response_sources)
|
217 |
|
218 |
# Append user message and response to chat history
|
219 |
new_history = history + [(message, response_answer)]
|
220 |
# return gr.update(value=""), new_history, response_sources[0], response_sources[1]
|
221 |
-
return qa_chain, gr.update(value=""), new_history
|
222 |
|
223 |
|
224 |
def upload_file(file_obj):
|
@@ -322,28 +325,24 @@ def demo():
|
|
322 |
inputs=[slider_temperature, slider_maxtokens, slider_topk, vector_db], \
|
323 |
outputs=[qa_chain, llm_progress]).then(lambda:[None,"",0,"",0,"",0], \
|
324 |
inputs=None, \
|
325 |
-
outputs=[chatbot
|
326 |
-
#, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page
|
327 |
], \
|
328 |
queue=False)
|
329 |
|
330 |
# Chatbot events
|
331 |
msg.submit(conversation, \
|
332 |
inputs=[qa_chain, msg, chatbot], \
|
333 |
-
outputs=[qa_chain, msg, chatbot
|
334 |
-
#, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page
|
335 |
], \
|
336 |
queue=False)
|
337 |
submit_btn.click(conversation, \
|
338 |
inputs=[qa_chain, msg, chatbot], \
|
339 |
-
outputs=[qa_chain, msg, chatbot
|
340 |
-
#, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page
|
341 |
], \
|
342 |
queue=False)
|
343 |
clear_btn.click(lambda:[None,"",0,"",0,"",0], \
|
344 |
inputs=None, \
|
345 |
-
outputs=[chatbot
|
346 |
-
#, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page
|
347 |
], \
|
348 |
queue=False)
|
349 |
demo.queue().launch(debug=True)
|
|
|
212 |
#response_source1_page = response_sources[0].metadata["page"] + 1
|
213 |
#response_source2_page = response_sources[1].metadata["page"] + 1
|
214 |
#response_source3_page = response_sources[2].metadata["page"] + 1
|
215 |
+
response_source1_page = 0
|
216 |
+
response_source2_page = 0
|
217 |
+
response_source3_page = 0
|
218 |
# print ('chat response: ', response_answer)
|
219 |
# print('DB source', response_sources)
|
220 |
|
221 |
# Append user message and response to chat history
|
222 |
new_history = history + [(message, response_answer)]
|
223 |
# return gr.update(value=""), new_history, response_sources[0], response_sources[1]
|
224 |
+
return qa_chain, gr.update(value=""), new_history, response_source1, response_source1_page, response_source2, response_source2_page, response_source3, response_source3_page
|
225 |
|
226 |
|
227 |
def upload_file(file_obj):
|
|
|
325 |
inputs=[slider_temperature, slider_maxtokens, slider_topk, vector_db], \
|
326 |
outputs=[qa_chain, llm_progress]).then(lambda:[None,"",0,"",0,"",0], \
|
327 |
inputs=None, \
|
328 |
+
outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page
|
|
|
329 |
], \
|
330 |
queue=False)
|
331 |
|
332 |
# Chatbot events
|
333 |
msg.submit(conversation, \
|
334 |
inputs=[qa_chain, msg, chatbot], \
|
335 |
+
outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page
|
|
|
336 |
], \
|
337 |
queue=False)
|
338 |
submit_btn.click(conversation, \
|
339 |
inputs=[qa_chain, msg, chatbot], \
|
340 |
+
outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page
|
|
|
341 |
], \
|
342 |
queue=False)
|
343 |
clear_btn.click(lambda:[None,"",0,"",0,"",0], \
|
344 |
inputs=None, \
|
345 |
+
outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page
|
|
|
346 |
], \
|
347 |
queue=False)
|
348 |
demo.queue().launch(debug=True)
|