Update Gradio_UI.py

Gradio_UI.py

@@ -2,84 +2,85 @@ import gradio as gr
 
 class GradioUI:
     def __init__(self, agent):
+        # The agent object passed from app.py
         self.agent = agent
 
+    def process_query_with_agent(self, query):
+        """
+        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.
+        """
+        # 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)}"
+
     def launch(self):
+        """
+        Launches the Gradio user interface for the agent.
+        """
+        # Use gr.Blocks for a more flexible layout
         with gr.Blocks() as demo:
             gr.Markdown("# Multi-Tool AI Agent")
-            gr.Markdown("Use the tabs below to interact with different tools:")
-
-            with gr.Tabs():
-
-                # Tab 1: Weather
-                with gr.TabItem("Weather"):
-                    place_input = gr.Textbox(label="Enter place name (e.g., London)", placeholder="City or location")
-                    weather_output = gr.Textbox(label="Current Weather", interactive=False)
-                    get_weather_btn = gr.Button("Get Weather")
-
-                    def get_weather(place):
-                        if not place.strip():
-                            return "Please enter a valid place name."
-                        return self.agent.tools[1](place)
-
-                    get_weather_btn.click(get_weather, inputs=place_input, outputs=weather_output)
-
-                # Tab 2: Local Time
-                with gr.TabItem("Local Time"):
-                    timezone_input = gr.Textbox(label="Enter timezone (e.g., America/New_York)", placeholder="Timezone string")
-                    time_output = gr.Textbox(label="Current Time", interactive=False)
-                    get_time_btn = gr.Button("Get Local Time")
-
-                    def get_time(tz):
-                        if not tz.strip():
-                            return "Please enter a valid timezone."
-                        return self.agent.tools[0](tz)
-
-                    get_time_btn.click(get_time, inputs=timezone_input, outputs=time_output)
-
-                # Tab 3: Image Generation
-                with gr.TabItem("Image Generation"):
-                    image_prompt = gr.Textbox(label="Enter image description prompt", lines=2)
-                    image_output = gr.Image(label="Generated Image")
-                    gen_image_btn = gr.Button("Generate Image")
-
-                    def gen_image(prompt):
-                        if not prompt.strip():
-                            return None
-                        # The image generation tool might return an image URL or PIL.Image.
-                        # Adjust this depending on your tool's output format.
-                        result = self.agent.tools[2](prompt)
-                        return result
-
-                    gen_image_btn.click(gen_image, inputs=image_prompt, outputs=image_output)
-
-                # Tab 4: Web Search
-                with gr.TabItem("Web Search"):
-                    search_query = gr.Textbox(label="Enter search query")
-                    search_output = gr.Textbox(label="Search Results", interactive=False)
-                    search_btn = gr.Button("Search")
-
-                    def search(q):
-                        if not q.strip():
-                            return "Please enter a search query."
-                        return self.agent.tools[3](q)
-
-                    search_btn.click(search, inputs=search_query, outputs=search_output)
-
-                # Tab 5: Document Q&A
-                with gr.TabItem("Document Q&A"):
-                    pdf_upload = gr.File(label="Upload PDF Document", file_types=[".pdf"])
-                    question_input = gr.Textbox(label="Enter your question about the document")
-                    answer_output = gr.Textbox(label="Answer", interactive=False)
-                    docqa_btn = gr.Button("Get Answer")
-
-                    def doc_qa(pdf_file, question):
-                        if pdf_file is None:
-                            return "Please upload a PDF file."
-                        if not question.strip():
-                            return "Please enter a question."
-                        return self.agent.tools[4](pdf_file.name, question)
-
-                    docqa_btn.click(doc_qa, inputs=[pdf_upload, question_input], outputs=answer_output)
-
-            demo.launch()
+            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."
+            )
+
+            # Single input field for the user's natural language request
+            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)",
+                lines=3, # Provides more space for typing longer queries
+                interactive=True # User can type in this box
+            )
+
+            # Single output field to display the agent's final response
+            agent_output = gr.Textbox(
+                label="Agent's Response:",
+                interactive=False, # User cannot type in the output box
+                lines=10, # Provides more space to display potentially long responses
+                autoscroll=True # Automatically scrolls to the latest output
+            )
+
+            # Button to trigger the agent when clicked
+            submit_btn = gr.Button("Submit to Agent")
+
+            # 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.
+            submit_btn.click(
+                fn=self.process_query_with_agent,
+                inputs=query_input,
+                outputs=agent_output
+            )
+
+            # Optional: Add examples to show users what they can ask
+            # You would need to add example queries relevant to your tools
+            # 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
+            # ]
+            # gr.Examples(examples=examples, inputs=query_input)
+
+
+        # Launch the Gradio interface
+        # Setting share=True creates a public URL (useful for demos, be mindful of security/costs)
+        # Setting inline=False opens the app in a new browser tab
+        demo.launch(share=False, inline=False)