Update app.py

app.py

@@ -7,6 +7,7 @@ import matplotlib.pyplot as plt
 import pandas as pd
 import re
 import logging
+import inspect
 from bs4 import BeautifulSoup  # Fixed Import
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
@@ -18,7 +19,7 @@ def get_box_score_links() -> list:
     """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/"
-    logging.debug("Reached Line 21 - app")
+    logging.debug(f"[Line {inspect.currentframe().f_lineno}] Fetching box score links from: {MAIN_URL}")
     try:
         response = requests.get(MAIN_URL)
         response.raise_for_status()  # Raise exception for HTTP errors
@@ -32,22 +33,26 @@ def get_box_score_links() -> list:
             if '/boxscores/' in href and href.endswith('.html') and 's/202' in href:
                 box_score_links.append(BASE_URL + href)
         
+        logging.info(f"[Line {inspect.currentframe().f_lineno}] Found {len(box_score_links)} box score links.")
         # Return unique links while preserving order
         return list(dict.fromkeys(box_score_links))
     
     except requests.exceptions.RequestException as e:
+        logging.error(f"[Line {inspect.currentframe().f_lineno}] Error fetching boxScore links: {str(e)}")
         # Return error message as a list to maintain consistent return type
         return [f"Error fetching boxScore links: {str(e)}"]
 
-def get_box_score_data(url: list) -> dict:
+def get_box_score_data(url: str) -> dict:
     """A tool that fetches the boxscores data from a provided URL.
     Args:
         url: A string representing the URL to a box score of an nba game from last night.
     """
+    logging.debug(f"[Line {inspect.currentframe().f_lineno}] Fetching box score data from URL: {url}")
     try:
         box_scores = {}
         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')))
@@ -55,6 +60,7 @@ def get_box_score_data(url: list) -> dict:
         if match:
             team1 = match.group(1)
             team2 = match.group(2)
+            logging.info(f"[Line {inspect.currentframe().f_lineno}] Game found: {team1} vs {team2}")
             
             # Read HTML tables
             tables = pd.read_html(url)
@@ -76,11 +82,13 @@ def get_box_score_data(url: list) -> dict:
         
         else:
             # If regex pattern did not match
+            logging.warning(f"[Line {inspect.currentframe().f_lineno}] Team names not found in the page title for URL: {url}")
             box_scores[url] = [{"Error": "Team names not found in the page title"}]
         
         return box_scores
     
     except Exception as e:
+        logging.error(f"[Line {inspect.currentframe().f_lineno}] Error fetching boxScore data: {str(e)}")
         return {"Error": f"Error fetching boxScore data: {str(e)}"}
 
 @tool
@@ -92,47 +100,45 @@ def get_stats_from_boxScore_data(url: str, stat: str) -> dict:
                Must be one of: 'MP', 'FG', 'FGA', 'FG%', '3P', '3PA', '3P%', 'FT', 'FTA', 'FT%', 
                'ORB', 'DRB', 'TRB', 'AST', 'STL', 'BLK', 'TOV', 'PF', 'PTS', 'GmSc', '+/-'
     """
-    # Define the allowed stats
     allowed_stats = ['MP', 'FG', 'FGA', 'FG%', '3P', '3PA', '3P%', 'FT', 'FTA', 'FT%', 
                      'ORB', 'DRB', 'TRB', 'AST', 'STL', 'BLK', 'TOV', 'PF', 'PTS', 
                      'GmSc', '+/-']
+    
     # Check if stat is valid
     if stat not in allowed_stats:
+        logging.error(f"[Line {inspect.currentframe().f_lineno}] Invalid stat requested: {stat}")
         return {"Error": f"Invalid stat '{stat}'. Allowed values are: {', '.join(allowed_stats)}"}
+    
     try:
         box_scores = get_box_score_data(url)
-        logging.debug(f"Reached Line 106 , URL={url}, box_scores= {box_scores}")
+        logging.debug(f"[Line {inspect.currentframe().f_lineno}] Box Scores: {box_scores}")
         stats = {}
         stat_key = ('Basic Box Score Stats', stat)
         for teams in box_scores.keys():
-            logging.debug(f"Reached Line 110, Teams = {teams}, Box Score.keys() {box_scores.keys()}")
+            logging.debug(f"[Line {inspect.currentframe().f_lineno}] Processing team: {teams}")
             for player in box_scores[teams]:
-                logging.debug(f"Reached Line 112 (player = {player}, stat_key = {stat_key}")
                 if stat_key in player:
-                    logging.debug(f"Reached Line 114")
-                    if player[stat_key] is not None:
-                        logging.debug(f"Reached Line 116")
-                        if player[stat_key].replace('.', '').isdigit():
-                            logging.debug(f"Reached Line 118")
-                            if player[list(player.keys())[0]] != "Team Totals":
-                                stats[player[list(player.keys())[0]]] = pd.to_numeric(player[stat_key], errors='coerce')
-        logging.debug(f"Reached BoxScores return, stats = {str(stats)}")
+                    if player[stat_key] is not None and player[stat_key].replace('.', '').isdigit():
+                        if player[list(player.keys())[0]] != "Team Totals":
+                            stats[player[list(player.keys())[0]]] = pd.to_numeric(player[stat_key], errors='coerce')
+        
+        logging.info(f"[Line {inspect.currentframe().f_lineno}] Stats for {stat}: {stats}")
         return stats
+    
     except Exception as e:
+        logging.error(f"[Line {inspect.currentframe().f_lineno}] Error fetching boxScore data for given statistic: {str(e)}")
         return {"Error": f"Error fetching boxScore data for given statistic: {str(e)}"}
 
+# Instantiate Tools and Model
 final_answer = FinalAnswerTool()
 search_tool = DuckDuckGoSearchTool()
 visit_webpage_tool = VisitWebpageTool()
 user_input_tool = UserInputTool()
 
-# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
-# 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
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None,
 )
 
@@ -154,11 +160,9 @@ agent = CodeAgent(
         user_input_tool, 
         get_box_score_links, 
         get_stats_from_boxScore_data
-    ],  # Add your tools here (don't remove final answer)
+    ],
     max_steps=6,
     verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
     name="NBA Box Scores Agent",
     description="Fetches NBA box scores and player points from last night's games.",
     prompt_templates=prompt_templates, 
@@ -166,4 +170,4 @@ agent = CodeAgent(
 )
 
 # Launch Gradio UI
-GradioUI(agent).launch()
+GradioUI(agent).launch()