Update Gradio_UI.py

Gradio_UI.py

@@ -1,52 +1,101 @@
 import gradio as gr
+import os # Import os to potentially handle file paths if needed, though Gradio provides the path
 
 class GradioUI:
     def __init__(self, agent):
         # The agent object passed from app.py
         self.agent = agent
 
-    def process_query_with_agent(self, query):
+    def process_input(self, query: str, pdf_file: gr.File = None):
         """
-        Processes the user query by passing it to the agent's run method.
-        The agent will then decide which tool(s) to use based on the query.
+        Processes user input, which can include a text query and optionally a PDF file.
+        If a PDF is uploaded, it assumes the query is a question about it and calls
+        the Document Q&A tool directly. Otherwise, it sends the query to the agent.run() method.
+
+        Args:
+            query (str): The text query from the user.
+            pdf_file (gr.File, optional): The uploaded PDF file object provided by Gradio. Defaults to None.
+
+        Returns:
+            str: The response from the agent or tool.
         """
-        # Check if the query is empty or only whitespace
-        if not query or not query.strip():
-            return "Please enter your request for the agent."
-
-        print(f"User query received by Gradio UI: {query}")
-        try:
-            # Call the agent's run method with the user's query.
-            # The agent internally figures out tool selection and execution.
-            response = self.agent.run(query)
-            print(f"Agent execution finished. Response generated.")
-            return response
-        except Exception as e:
-            # Catch any exceptions during the agent's run
-            print(f"Error during agent execution: {e}")
-            return f"An error occurred while processing your request: {str(e)}"
+        # Check if a PDF file is uploaded AND there is a text query (assumed to be the question)
+        if pdf_file is not None and query and query.strip():
+            print(f"PDF file uploaded: {pdf_file.name}")
+            print(f"Query (assumed question for PDF): {query}")
+            try:
+                # --- Call the Document Q&A tool directly ---
+                # We assume the document_qna_tool is at index 4 as per your app.py initialization
+                # Gradio's gr.File object has a .name attribute which is the file path
+                # The document_qna_tool expects a file path string.
+                print("Detected PDF upload and query. Calling document_qna_tool...")
+                # The tool signature is document_qna_tool(pdf_path: str, question: str)
+                response = self.agent.tools[4](pdf_file.name, query)
+                print("Document Q&A tool finished.")
+                # Clean up the uploaded file after processing (optional but good practice)
+                # if os.path.exists(pdf_file.name):
+                #     os.remove(pdf_file.name)
+                return response
+
+            except IndexError:
+                 return "Error: Document Q&A tool not found at the expected index (4). Check agent tool setup in app.py."
+            except Exception as e:
+                print(f"Error during Document Q&A tool execution: {e}")
+                # Clean up the uploaded file on error too
+                # if os.path.exists(pdf_file.name):
+                #     os.remove(pdf_file.name)
+                return f"An error occurred during Document Q&A: {str(e)}"
+
+        # If no PDF file, or query is empty when file is present, handle as a general agent query
+        elif query and query.strip():
+            print(f"No PDF file or query is for general task. Processing with agent.run(): {query}")
+            try:
+                # --- Call the agent's run method for general queries ---
+                response = self.agent.run(query)
+                print("Agent.run finished for general query.")
+                return response
+            except Exception as e:
+                print(f"Error during agent.run: {e}")
+                return f"An error occurred while processing your request: {str(e)}"
+
+        # Handle cases where only a PDF is uploaded without a question, or no input at all
+        elif pdf_file is not None:
+             # Clean up the uploaded file if no question was provided
+            # if os.path.exists(pdf_file.name):
+            #     os.remove(pdf_file.name)
+            return "Please enter a question in the textbox to ask about the uploaded PDF."
+        else:
+            return "Please enter a request or upload a document for analysis."
+
 
     def launch(self):
         """
-        Launches the Gradio user interface for the agent.
+        Launches the Gradio user interface for the agent with file upload capability.
         """
         # Use gr.Blocks for a more flexible layout
         with gr.Blocks() as demo:
-            gr.Markdown("# Multi-Tool AI Agent")
+            gr.Markdown("# Multi-Tool AI Agent with Document Upload")
             gr.Markdown(
-                "Enter your request below. The agent will use its available tools "
-                "(Weather, Time, Search, Image Generation, Document Q&A, etc.) "
-                "to understand and fulfill your request."
+                "Enter your request and optionally upload a PDF document. "
+                "If you upload a PDF, the text box should contain your question about the document. "
+                "Otherwise, use the text box for general requests (weather, search, etc.)."
             )
 
-            # Single input field for the user's natural language request
+            # Text input for the user's query or question
             query_input = gr.Textbox(
-                label="Enter your request here:",
-                placeholder="e.g., What is the weather in London? Search for the latest news about AI. Tell me the time in New York. Generate an image of a cat wearing a hat. What does this document say about [topic]? (For Document Q&A, ensure the PDF is handled by your app.py's tool setup if needed)",
+                label="Enter your request or question:",
+                placeholder="e.g., What is the weather in London? Search for the latest news about AI. What does this document say about [topic]? Summarize the document.",
                 lines=3, # Provides more space for typing longer queries
                 interactive=True # User can type in this box
             )
 
+            # File upload component specifically for PDF documents
+            pdf_upload = gr.File(
+                label="Upload a PDF Document (Optional - for Document Q&A)",
+                file_types=[".pdf"], # Restrict file types to PDF
+                interactive=True # User can upload files
+            )
+
             # Single output field to display the agent's final response
             agent_output = gr.Textbox(
                 label="Agent's Response:",
@@ -55,29 +104,27 @@ class GradioUI:
                 autoscroll=True # Automatically scrolls to the latest output
             )
 
-            # Button to trigger the agent when clicked
-            submit_btn = gr.Button("Submit to Agent")
+            # Button to trigger the processing function
+            submit_btn = gr.Button("Submit")
 
-            # Link the button click event to the process_query_with_agent function
-            # The value from query_input will be passed as the 'query' argument.
-            # The return value of the function will be displayed in agent_output.
+            # Link the button click event to the process_input function
+            # Pass both query_input (text) and pdf_upload (file) as inputs
             submit_btn.click(
-                fn=self.process_query_with_agent,
-                inputs=query_input,
-                outputs=agent_output
+                fn=self.process_input,
+                inputs=[query_input, pdf_upload], # This list specifies all inputs
+                outputs=agent_output # This specifies the output where the return value goes
             )
 
-            # Optional: Add examples to show users what they can ask
-            # You would need to add example queries relevant to your tools
+            # Optional examples - adjust these based on your tools and whether you have default files
+            # You would need to add example queries and potentially example files if you use them.
             # examples = [
-            #     ["What is the weather in Tokyo?"],
-            #     ["Tell me the current time in Paris."],
-            #     ["Generate an image of a space dog."],
-            #     ["Search for the history of the internet."],
-            #     # Add an example for document Q&A if you have a default document loaded
-            #     # ["What is the main topic of the document?", "path/to/your/default.pdf"] # Example with file input if you re-add it
+            #     ["What is the time in Berlin?"],
+            #     ["Generate an image of a robot cooking pasta."],
+            #     # Example for document Q&A - requires a default file path accessible by the tool
+            #     # For file examples, you need to provide the path to the example file(s)
+            #     # ["Summarize the introduction section", "path/to/your/sample_document.pdf"]
             # ]
-            # gr.Examples(examples=examples, inputs=query_input)
+            # gr.Examples(examples=examples, inputs=[query_input, pdf_upload]) # Examples take same inputs as function
 
 
         # Launch the Gradio interface