Update app.py

app.py

@@ -140,29 +140,31 @@ def process_and_query(state, question):
     
 # Initialize an empty state variable to store conversation history
 state = []
+
 # Create Gradio interface
-with gr.Blocks() as demo:
+with gr.Interface(fn=None, live=True, capture_session=True) 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)
+        upload_button.click(upload_files, inputs=upload, outputs=upload_output)
     
     with gr.Tab("Query"):
         query_input = gr.Textbox(label="Enter your query")
-        query_button = gr.Button("Search")
         query_output = gr.Textbox()
 
-        # Define function to handle query
-        def query_text():
-            return query_input, query_output
+        # Define function to handle query and update state
+        def query_handler():
+            question = query_input.value  # Get user's question from the input box
+            state = []  # Initialize or update state as needed
+            response = process_and_query(state, question)  # Process query and retrieve response
+            query_output.value = response.get("message", "Error: No response")  # Update output textbox
 
         # Setup the click event with correct inputs and outputs
-        query_button.click(fn=process_and_query, inputs=query_text, outputs=query_output)
-
-demo.launch()
+        query_button = gr.Button("Search", click=query_handler)
+        demo.Interface(layout="vertical").launch()