Update Gradio_UI.py

Gradio_UI.py

@@ -1,57 +1,36 @@
 import gradio as gr
 import os
+# Import the themes module
+import gradio.themes as themes
 
 class GradioUI:
     def __init__(self, agent):
         self.agent = agent
 
+    # Keep the process_input method as is
     def process_input(self, query: str, pdf_file: gr.File = None):
         """
-        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.
+        Processes user input...
         """
-        # Check if a PDF file is uploaded AND there is a text query (assumed to be the question)
+        # ... (your existing process_input code)
         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)
-                # import os
-                # 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
-                # 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)}"
 
-        # 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
@@ -59,12 +38,7 @@ class GradioUI:
                 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
-            # 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."
         else:
             return "Please enter a request or upload a document for analysis."
@@ -72,9 +46,11 @@ class GradioUI:
 
     def launch(self):
         """
-        Launches the Gradio user interface with layout adjustments.
+        Launches the Gradio user interface with a theme.
         """
-        with gr.Blocks() as demo:
+        # Apply a theme here using the 'theme' argument
+        # You can try different themes like themes.Soft(), themes.Glass(), themes.Monochrome()
+        with gr.Blocks(theme=themes.Soft()) as demo: # <--- Added theme argument here
             gr.Markdown("# Multi-Tool AI Agent with Document Upload")
             gr.Markdown(
                 "Enter your request and optionally upload a PDF document. "
@@ -82,59 +58,36 @@ class GradioUI:
                 "Otherwise, use the text box for general requests (weather, search, etc.)."
             )
 
-            # 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
+                    scale=3
                 )
 
-                # 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
+                    label="Upload PDF (Optional)",
+                    file_types=[".pdf"],
+                    interactive=True,
+                    scale=1
                 )
 
-            # 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
+                interactive=False,
+                lines=10,
                 autoscroll=True
             )
 
-            # Button to trigger the processing function
             submit_btn = gr.Button("Submit")
 
-            # 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,
-                inputs=[query_input, pdf_upload], # This list specifies all inputs
-                outputs=agent_output # This specifies the output where the return value goes
+                inputs=[query_input, pdf_upload],
+                outputs=agent_output
             )
 
-            # Optional examples - adjust these based on your tools and whether you have default files
-            # examples = [
-            #     ["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, 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
         demo.launch(share=False, inline=False)