Update app.py

app.py

@@ -41,10 +41,13 @@ def local_query(query, context):
     
     return tokenizer.batch_decode(outputs, skip_special_tokens=True)
 
-def run_query(query):
+def run_query(history, query):
+    
     context = get_context(query)
     result = local_query(query, context)
-    return result
+
+    history = history.append(query)
+    return history, result
 
 
 def load_document(pdf_filename):
@@ -74,8 +77,7 @@ def load_document(pdf_filename):
     return 'Success'
 
 
-import gradio as gr
-import os
+
 
 def upload_pdf(file):
     try:
@@ -99,9 +101,11 @@ def upload_pdf(file):
     
  
 with gr.Blocks() as demo:  
-    btn = gr.UploadButton("📁 Upload a PDF", file_types=[".pdf"])
+    
+    btn = gr.UploadButton("Upload a PDF", file_types=[".pdf"])
     output = gr.Textbox(label="Output Box")
     chatbot = gr.Chatbot(value=[], elem_id="chatbot")
+    
     with gr.Row():
         with gr.Column(scale=0.70):
             txt = gr.Textbox(
@@ -111,8 +115,10 @@ with gr.Blocks() as demo:
 
  
     # Event handler for uploading a PDF
-            
     btn.upload(fn=upload_pdf, inputs=[btn], outputs=[output])
+    txt.submit(run_query, [chatbot, txt], [chatbot, txt], queue=False)
+    #.then(
+            # generate_response, inputs =[chatbot,],outputs = chatbot,)