Update app.py

app.py

@@ -128,14 +128,21 @@ def process_and_query(state, question):
         return {"message": generated_text, "conversation": state}
 
     return {"error": "No question provided"}
-
+    
 # Create Gradio interface
-iface = gr.Interface(
-    fn=upload_files,
-    inputs=gr.File(label="Upload PDF or DOCX file"),
-    outputs="json",
-    live=True,
-    capture_session=True
-)
-
-iface.launch()
+with gr.Blocks() as demo:
+    gr.Markdown("## Document Upload and Query System")
+    
+    with gr.Tab("Upload Files"):
+        upload = gr.File(file_count="multiple", label="Upload PDF or DOCX files")
+        upload_button = gr.Button("Upload")
+        upload_output = gr.Textbox()
+        upload_button.click(fn=upload_files, inputs=upload, outputs=upload_output)
+    
+    with gr.Tab("Query"):
+        query = gr.Textbox(label="Enter your query")
+        query_button = gr.Button("Search")
+        query_output = gr.Textbox()
+        query_button.click(fn=query_text, inputs=query, outputs=query_output)
+
+demo.launch()