Update app.py

app.py

@@ -25,6 +25,15 @@ def pdf_changes(pdf_doc):
     qa = RetrievalQA.from_chain_type(llm=flan_ul2, chain_type="stuff", retriever=retriever, return_source_documents=True)
     return "Ready"
 
+def add_text(history, text):
+    history = history + [(text, None)]
+    return history, ""
+
+def bot(history):
+    response = infer(history):
+    history[-1][1] = response
+    return history
+
 def infer(question):
     
     query = question
@@ -37,13 +46,16 @@ with gr.Blocks() as demo:
         pdf_doc = gr.File(label="Load a pdf", file_types=['.pdf'], type="file")
         langchain_status = gr.Textbox()
         load_pdf = gr.Button("Load pdf to langchain")
-        question = gr.Textbox(label="Your Question")
-        answer = gr.Textbox(label="Anwser")
         
-        submit_button = gr.Button("Send Question")
+        chatbot = gr.Chatbot([], elem_id="chatbot").style(height=350)
+        question = gr.Textbox(label="Question")
+        
     
     load_pdf.click(pdf_changes, pdf_doc, langchain_status, queue=False)
-    submit_button.click(infer, inputs=[question], outputs=[answer])
-        
+    
+
+    question.submit(add_text, [chatbot, txt], [chatbot, txt]).then(
+        bot, chatbot, chatbot
+    )
 
 demo.launch()