Update Gradio_UI.py

Gradio_UI.py

@@ -1,68 +1,73 @@
 import gradio as gr
+from smolagents import CodeAgent
 
 class GradioUI:
-    def __init__(self, agent):
-        self.agent = agent
-        self.interface = self._create_interface()
-
-    def _create_interface(self):
-        # Define custom CSS to hide the resize arrows and style the output
-        custom_css = """
-        /* Hide the resize arrows */
-        .output-container .svelte-1g9btlg { display: none !important; }
+    def __init__(self, agent: CodeAgent):
+        """Initialize the Gradio UI with the provided agent.
         
-        /* Style the output panel */
-        #output-panel {
-            height: 400px !important;  /* Fixed height */
-            overflow-y: auto;          /* Add scrollbar if content overflows */
-            border: 1px solid #ccc;    /* Add a subtle border */
-            border-radius: 5px;        /* Rounded corners */
-            padding: 10px;             /* Inner padding */
-            background-color: #f9f9f9; /* Light background */
-        }
+        Args:
+            agent: An instance of CodeAgent from smolagents.
+        """
+        self.agent = agent
+    
+    def run_agent(self, player_name: str, role: str) -> str:
+        """Run the agent with the cricketer analysis tool.
         
-        /* Style the links */
-        #output-panel a {
-            color: #1a73e8;            /* Google-blue link color */
-            text-decoration: none;     /* Remove underline */
-            font-weight: bold;         /* Bold links */
-        }
-        #output-panel a:hover {
-            text-decoration: underline; /* Underline on hover */
-        }
+        Args:
+            player_name: The name of the cricketer to analyze.
+            role: The role of the cricketer (e.g., 'batter', 'bowler').
         
-        /* Style the list items */
-        #output-panel ul {
-            list-style-type: none;     /* Remove default bullets */
-            padding-left: 0;           /* Remove default padding */
-        }
-        #output-panel li {
-            margin-bottom: 10px;       /* Space between list items */
-        }
+        Returns:
+            A string with the analysis result or an error message.
         """
-
-        with gr.Blocks(title="Landsat Image Fetcher", css=custom_css) as demo:
-            gr.Markdown("# Landsat Image Fetcher")
-            gr.Markdown("Enter coordinates and a time range to fetch Landsat images from USGS.")
+        # Construct the query for the agent
+        query = f"Analyze the form of cricketer '{player_name}' as a '{role}'"
+        try:
+            # Execute the agent's run method with the query
+            response = self.agent.run(query)
+            return response
+        except Exception as e:
+            return f"Error running analysis: {str(e)}"
+    
+    def launch(self):
+        """Launch the Gradio interface."""
+        # Define the UI layout using Gradio Blocks
+        with gr.Blocks(title="Cricketer Form Analyzer") as interface:
+            # Header
+            gr.Markdown("# Cricketer Form Analyzer")
+            gr.Markdown("Enter a cricketer's name and select their role to analyze their recent form and career stats.")
+            
+            # Input section
             with gr.Row():
-                with gr.Column():
-                    lat_min = gr.Number(label="Min Latitude", value=37.75)
-                    lon_min = gr.Number(label="Min Longitude", value=-122.35)
-                    lat_max = gr.Number(label="Max Latitude", value=37.85)
-                    lon_max = gr.Number(label="Max Longitude", value=-122.25)
-                    start_date = gr.Textbox(label="Start Date (YYYY-MM-DD)", value="2023-01-01")
-                    end_date = gr.Textbox(label="End Date (YYYY-MM-DD)", value="2023-12-31")
-                    submit_btn = gr.Button("Fetch Images")
-                with gr.Column():
-                    output = gr.HTML(label="Download Links", elem_id="output-panel")
+                player_input = gr.Textbox(
+                    label="Player Name",
+                    placeholder="e.g., Virat Kohli",
+                    lines=1
+                )
+                role_input = gr.Dropdown(
+                    choices=["batter", "bowler", "wicket_keeper", "fielder"],
+                    label="Role",
+                    value="batter"  # Default selection
+                )
+            
+            # Submit button
+            submit_btn = gr.Button("Analyze")
+            
+            # Output section
+            output = gr.Textbox(
+                label="Analysis Result",
+                lines=10,
+                placeholder="Analysis will appear here..."
+            )
+            
+            # Connect the button to the run_agent function
             submit_btn.click(
-                fn=lambda lm, ln, lx, lo, sd, ed: self.agent.run(
-                    f"fetch_landsat_image({lm}, {ln}, {lx}, {lo}, '{sd}', '{ed}')"
-                ),
-                inputs=[lat_min, lon_min, lat_max, lon_max, start_date, end_date],
+                fn=self.run_agent,
+                inputs=[player_input, role_input],
                 outputs=output
             )
-        return demo
+        
+        # Launch the interface
+        interface.launch()
 
-    def launch(self, **kwargs):
-        self.interface.launch(**kwargs)
+# Note: This is imported and used in app.py as `GradioUI(agent).launch()`