Update app.py

app.py

@@ -36,17 +36,16 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def get_boxScores() -> list:
+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
     BASE_URL = "https://www.basketball-reference.com"
     MAIN_URL = f"{BASE_URL}/boxscores/"
+    response = requests.get(MAIN_URL)
     try:
-        response = requests.get(MAIN_URL)
-        soup = bs4.BeautifulSoup(response.text, 'html.parser')
+        soup = BeautifulSoup(response.text, 'html.parser')
         box_score_links = []
         # Find all box score links
         for link in soup.find_all('a', href=True):
@@ -55,10 +54,35 @@ def get_boxScores() -> list:
                 box_score_links.append(BASE_URL + href)
         return list(set(box_score_links))
     except Exception as e:
-        return f"Error fetching boxScores: {str(e)}"
+        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
+    Args:
+        links: A list of strings representing the URLs to the box score of each of last nights 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(_)
+            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')
+        return box_scores
+    except Exception as e:
+        return f"Error fetching boxScore data: {str(e)}"
 
 
-box_scores = get_boxScores()
+box_score_links = get_box_score_links()
+box_score_data = get_box_score_data()
 final_answer = FinalAnswerTool()
 search_tool = DuckDuckGoSearchTool()
 visit_webpage_tool = VisitWebpageTool()
@@ -84,7 +108,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool, search_tool, visit_webpage_tool, user_input_tool, get_boxScores], ## 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,