Update Gradio_UI.py

Gradio_UI.py

@@ -1,9 +1,8 @@
 import gradio as gr
-import os # Import os to potentially handle file paths if needed, though Gradio provides the path
+import os
 
 class GradioUI:
     def __init__(self, agent):
-        # The agent object passed from app.py
         self.agent = agent
 
     def process_input(self, query: str, pdf_file: gr.File = None):
@@ -33,6 +32,7 @@ class GradioUI:
                 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)
+                # import os
                 # if os.path.exists(pdf_file.name):
                 #     os.remove(pdf_file.name)
                 return response
@@ -42,6 +42,7 @@ class GradioUI:
             except Exception as e:
                 print(f"Error during Document Q&A tool execution: {e}")
                 # Clean up the uploaded file on error too
+                # import os
                 # if os.path.exists(pdf_file.name):
                 #     os.remove(pdf_file.name)
                 return f"An error occurred during Document Q&A: {str(e)}"
@@ -61,6 +62,7 @@ class GradioUI:
         # 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
+            # import os
             # 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."
@@ -70,9 +72,8 @@ class GradioUI:
 
     def launch(self):
         """
-        Launches the Gradio user interface for the agent with file upload capability.
+        Launches the Gradio user interface with layout adjustments.
         """
-        # Use gr.Blocks for a more flexible layout
         with gr.Blocks() as demo:
             gr.Markdown("# Multi-Tool AI Agent with Document Upload")
             gr.Markdown(
@@ -81,33 +82,40 @@ class GradioUI:
                 "Otherwise, use the text box for general requests (weather, search, etc.)."
             )
 
-            # Text input for the user's query or question
-            query_input = gr.Textbox(
-                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
-            )
+            # Use a gr.Row to place input components side-by-side (or stacked on small screens)
+            with gr.Row():
+                # Text input for the user's query or question
+                # Give it more space (e.g., scale=3) compared to the file upload
+                query_input = gr.Textbox(
+                    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,
+                    interactive=True,
+                    scale=3 # This component will take 3 parts of the available width in the row
+                )
+
+                # File upload component for PDF documents
+                # Give it less space (e.g., scale=1 or scale=0)
+                pdf_upload = gr.File(
+                    label="Upload PDF (Optional)", # Can also shorten the label
+                    file_types=[".pdf"], # Restrict file types to PDF
+                    interactive=True, # User can upload files
+                    scale=1 # This component will take 1 part of the available width in the row
+                    # Using scale=0 would make it take the minimum possible space
+                )
 
             # 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
+                autoscroll=True
             )
 
             # Button to trigger the processing function
             submit_btn = gr.Button("Submit")
 
-            # Link the button click event to the process_input function
+            # Link the button click to the process_input function
             # Pass both query_input (text) and pdf_upload (file) as inputs
             submit_btn.click(
                 fn=self.process_input,
@@ -116,7 +124,6 @@ class GradioUI:
             )
 
             # 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 time in Berlin?"],
             #     ["Generate an image of a robot cooking pasta."],
@@ -126,7 +133,7 @@ class GradioUI:
             # ]
             # gr.Examples(examples=examples, inputs=[query_input, pdf_upload]) # Examples take same inputs as function
 
- 
+
         # 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