Update app.py

app.py

@@ -15,16 +15,33 @@ from langchain.vectorstores import Chroma
 
 from langchain.chains import RetrievalQA
 
-def infer(pdf_doc):
+def pdf_changes(pdf_doc):
     loader = OnlinePDFLoader(pdf_doc)
     documents = loader.load()
     texts = text_splitter.split_documents(documents)
     db = Chroma.from_documents(texts, embeddings)
     retriever = db.as_retriever()
     qa = RetrievalQA.from_chain_type(llm=flan_ul2, chain_type="stuff", retriever=retriever, return_source_documents=True)
-    query = "What is the title of this paper?"
+    return "Ready"
+
+def infer(question):
+    
+    query = question
     result = qa({"query": query})
 
     return result
 
-gr.Interface(fn=infer, inputs=[gr.Textbox(value="https://arxiv.org/pdf/2304.03757.pdf")], outputs=[gr.Textbox()]).launch()
+with gr.Blocks() as demo:
+    with gr.Column():
+        pdf_doc = gr.File(label="Load a pdf")
+        langchain_status = gr.Textbox()
+        pdf_doc.change(pdf_changes, pdf_doc, langchain_status, queue=False)
+        question = gr.Textbox(label="Your Question")
+        answer = gr.Textbox(label="Anwser")
+        submit_button = gr.Button("Send Question")
+    
+    pdf_doc.change(pdf_changes, pdf_doc, langchain_status, queue=False)
+    submit_button.click(infer, inputs=[question], outputs=[answer])
+        
+
+demo.launch()