Update app.py

app.py

@@ -47,19 +47,11 @@ def pdf_changes(pdf_doc):
     )
     return "Ready"
 
-def add_text(history, text):
-    history = history + [(text, None)]
-    return history, ""
-
-def bot(history):
-    response = infer(history[-1][0] , history)
-    history[-1][1] = response['answer']
-    return history
-
-def infer(question , history):
-    hist = history[:-1]
-    result = qa({'question': question , 'chat_history': hist})
-    return result
+def text(history, text):
+    result = qa_chain({'question': text, 'chat_history': history})
+    history.append((text, result['answer']))
+    
+    return history,""
 
 css="""
 #col-container {max-width: 700px; margin-left: auto; margin-right: auto;}
@@ -82,12 +74,8 @@ with gr.Blocks(css=css) as demo:
         question = gr.Textbox(label="Question", placeholder="Type your question and hit Enter ")
         submit_btn = gr.Button("Send message") 
     load_pdf.click(pdf_changes, inputs=[pdf_doc], outputs=[langchain_status], queue=False)
-    question.submit(add_text, [chatbot, question], [chatbot, question]).then(
-        bot, chatbot, chatbot
-    )
-    submit_btn.click(add_text, [chatbot, question], [chatbot, question]).then(
-        bot, chatbot, chatbot
-    )
+    question.submit(text, [chatbot, question], [chatbot, question])
+    submit_btn.click(text, [chatbot, question], [chatbot, question])
 
 demo.launch()