Update app.py

app.py

@@ -82,19 +82,19 @@ def query_documents(user_query):
 
     return response
 
-# Gradio interface to upload PDFs
-pdf_input = gr.inputs.File(file_count="multiple", label="Upload up to 10 PDF files")
-query_input = gr.inputs.Textbox(label="Enter your query", placeholder="Type a question here...")
-
-# Gradio output
-output = gr.outputs.Textbox()
 
 # Gradio interface combining document upload and query features
-gr.Interface(
-    fn=[process_documents, query_documents], 
-    inputs=[pdf_input, query_input], 
-    outputs=[output, output],
-    title="PDF Document Embedding and Query",
-    description="Upload PDF files to embed them and then query to retrieve relevant documents.",
-    live=True
-).launch()
+with gr.Blocks() as demo:
+    pdf_input = gr.File(file_count="multiple", label="Upload up to 10 PDF files")
+    query_input = gr.Textbox(label="Enter your query", placeholder="Type a question here...")
+    
+    process_btn = gr.Button("Process PDFs")
+    query_btn = gr.Button("Query Documents")
+    
+    process_output = gr.Textbox()
+    query_output = gr.Textbox()
+    
+    process_btn.click(process_documents, inputs=[pdf_input], outputs=[process_output])
+    query_btn.click(query_documents, inputs=[query_input], outputs=[query_output])
+
+demo.launch()