adding chatbot

app.py

@@ -1,6 +1,6 @@
 import os 
 import gradio as gr
-
+import time 
 from langchain.document_loaders import PDFMinerLoader,CSVLoader ,UnstructuredWordDocumentLoader,TextLoader,OnlinePDFLoader
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.embeddings import SentenceTransformerEmbeddings
@@ -98,6 +98,30 @@ def process_pdf_document(document_file):
     document = loader.load()
     return document
 
+def infer(question, history):
+    
+    res = []
+    for human, ai in history[:-1]:
+        pair = (human, ai)
+        res.append(pair)
+    
+    chat_history = res
+    query = question
+    result = qa({"question": query, "chat_history": chat_history})
+    return result["answer"]
+
+def bot(history):
+    response = infer(history[-1][0], history)
+    history[-1][1] = ""
+    
+    for character in response:     
+        history[-1][1] += character
+        time.sleep(0.05)
+        yield history
+
+def add_text(history, text):
+    history = history + [(text, None)]
+    return history, ""
 
 
 css="""
@@ -134,7 +158,7 @@ with gr.Blocks(css=css) as demo:
         # chatbot = gr.Chatbot()
         # question = gr.Textbox(label="Question", placeholder="Type your question and hit Enter")
         # submit_button = gr.Button("Send Message")
-        
+
     load_pdf.click(loading_file, None, langchain_status, queue=False)    
     load_pdf.click(document_loader, inputs=[pdf_doc,API_key,file_extension,LLM_option], outputs=[langchain_status], queue=False)
         
@@ -144,7 +168,9 @@ with gr.Blocks(css=css) as demo:
             sources = gr.HTML(value = "Source paragraphs where I looked for answers will appear here", height=300)
 
         with gr.Row():
-            message = gr.Textbox(label="Type your question?",lines=1).style(full_width=False)
-        submit_query = gr.Button(value="Send message", variant="secondary", scale = 1)
+            question = gr.Textbox(label="Type your question?",lines=1).style(full_width=False)
+            submit_btn = gr.Button(value="Send message", variant="secondary", scale = 1)
+        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)
 
 demo.launch()