Update app.py

app.py

@@ -1,88 +1,88 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool, VisitWebpageTool, UserInputTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool, VisitWebpageTool, UserInputTool
 import datetime
 import requests
 import pytz
 import yaml
-import matplotlib.pyplot
-import bs4
+import matplotlib.pyplot as plt
+import pandas as pd
+import re
+from bs4 import BeautifulSoup  # Fixed Import
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
-@tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
 @tool
 def get_box_score_links() -> list:
-    """A tool that fetches the URLs to the boxscores for each of last nights nba games
-    Args:
-    """
-    # Base URL of the main page containing box score links
+    """A tool that fetches the URLs to the boxscores for each of last night's NBA games."""
     BASE_URL = "https://www.basketball-reference.com"
     MAIN_URL = f"{BASE_URL}/boxscores/"
-    response = requests.get(MAIN_URL)
     try:
+        response = requests.get(MAIN_URL)
+        response.raise_for_status()  # Raise exception for HTTP errors
+        
         soup = BeautifulSoup(response.text, 'html.parser')
         box_score_links = []
+
         # Find all box score links
         for link in soup.find_all('a', href=True):
             href = link['href']
             if '/boxscores/' in href and href.endswith('.html') and 's/202' in href:
                 box_score_links.append(BASE_URL + href)
-        return list(set(box_score_links))
-    except Exception as e:
-        return f"Error fetching boxScore links: {str(e)}"
+        
+        # Return unique links while preserving order
+        return list(dict.fromkeys(box_score_links))
+    
+    except requests.exceptions.RequestException as e:
+        # Return error message as a list to maintain consistent return type
+        return [f"Error fetching boxScore links: {str(e)}"]
 
 @tool
 def get_box_score_data(links: list) -> dict:
-    """A tool that fetches the URLs to the boxscores for each of last nights nba games
+    """A tool that fetches the boxscores data from the provided list of URLs.
     Args:
-        links: A list of strings representing the URLs to the box score of each of last nights games
+        links: A list of strings representing the URLs to the box score of each of last night's games.
     """
-    # Base URL of the main page containing box score links
-    BASE_URL = "https://www.basketball-reference.com"
-    MAIN_URL = f"{BASE_URL}/boxscores/"
-    response = requests.get(MAIN_URL)
     try:
         box_scores = {}
-        for _ in links:
-            response = requests.get(_)
+        for url in links:
+            response = requests.get(url)
+            response.raise_for_status()  # Raise exception for HTTP errors
+            
             soup = BeautifulSoup(response.text, 'html.parser')
             pattern = r"<h1>(.*?) at (.*?) Box Score"
             match = re.search(pattern, str(soup.find('div', id="content").find('h1')))
             
-            box_scores[match.group(1)] = pd.read_html(_)[0].to_dict(orient='records')
-            box_scores[match.group(2)] = pd.read_html(_)[8].to_dict(orient='records')
+            if match:
+                team1 = match.group(1)
+                team2 = match.group(2)
+                
+                # Read HTML tables
+                tables = pd.read_html(url)
+                
+                # Check if the expected tables exist before accessing
+                if len(tables) > 0:
+                    df_team1 = tables[0].to_dict(orient='records')
+                else:
+                    df_team1 = [{"Error": "Team 1 data not found"}]
+
+                if len(tables) > 8:
+                    df_team2 = tables[8].to_dict(orient='records')
+                else:
+                    df_team2 = [{"Error": "Team 2 data not found"}]
+                
+                # Store box score data
+                box_scores[team1] = df_team1
+                box_scores[team2] = df_team2
+            
+            else:
+                # If regex pattern did not match
+                box_scores[url] = [{"Error": "Team names not found in the page title"}]
+        
         return box_scores
+    
     except Exception as e:
-        return f"Error fetching boxScore data: {str(e)}"
-
+        return {"Error": f"Error fetching boxScore data: {str(e)}"}
 
-box_score_links = get_box_score_links()
-box_score_data = get_box_score_data()
 final_answer = FinalAnswerTool()
 search_tool = DuckDuckGoSearchTool()
 visit_webpage_tool = VisitWebpageTool()
@@ -92,32 +92,40 @@ user_input_tool = UserInputTool()
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # It is possible that this model may be overloaded
+    custom_role_conversions=None,
 )
 
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-
+# Load prompt templates
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
+# Setup the Agent
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool, search_tool, visit_webpage_tool, user_input_tool, get_box_score_links, get_box_score_data], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer, 
+        image_generation_tool, 
+        search_tool, 
+        visit_webpage_tool, 
+        user_input_tool, 
+        get_box_score_links, 
+        get_box_score_data
+    ],  # Add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
-    name=None,
-    description=None,
+    name="NBA Box Scores Agent",
+    description="Fetches NBA box scores and player points from last night's games.",
     prompt_templates=prompt_templates, 
     additional_authorized_imports=["matplotlib"]
 )
 
-
-GradioUI(agent).launch()
+# Launch Gradio UI
+GradioUI(agent).launch()