vishwask commited on
Commit
1733de6
·
verified ·
1 Parent(s): f5719ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
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#, response_source1, response_source1_page, response_source2, response_source2_page, response_source3, response_source3_page
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)