Update app.py

app.py

@@ -84,11 +84,20 @@ def get_box_score_data(links: list) -> dict:
 
 @tool
 def get_stats_from_boxScore_data(url: str, stat: str) -> dict:
-    """A tool that fetches the player names and box score statistic for the provided box score url 
+    """A tool that fetches the player names and box score statistic for the provided box score url.
     Args:
-        url: A string representing the URL to the box score for a game
-        stat: A string representing the statistic that this function will return for each player the box score
+        url: A string representing the URL to the box score for a game.
+        stat: A string representing the statistic that this function will return for each player in the box score.
+               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:
+        return {"Error": f"Invalid stat '{stat}'. Allowed values are: {', '.join(allowed_stats)}"}
     try:
         box_scores = get_box_score_data(url)
         stats = {}
@@ -98,7 +107,7 @@ def get_stats_from_boxScore_data(url: str, stat: str) -> dict:
                 if stat_key in player:
                     if player[stat_key] is not None:
                         if player[stat_key].replace('.', '').isdigit():
-                            stats[player[list(player.keys())[0]]]=pd.to_numeric(player[stat_key], errors='coerce')
+                            stats[player[list(player.keys())[0]]] = pd.to_numeric(player[stat_key], errors='coerce')
         return stats
     except Exception as e:
         return {"Error": f"Error fetching boxScore data for given statistic: {str(e)}"}