Add NBA API static players data tool

app.py

@@ -4,9 +4,52 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from nba_api.stats.static import players
+from nba_api.stats.endpoints import commonplayerinfo
+
 
 from Gradio_UI import GradioUI
 
+
+def get_player(name):
+    player = players.find_players_by_full_name(name)
+    if player:
+        return player[0]
+    return None
+
+@tool
+def get_player_info(player_name:str): -> dict
+    """
+    A tool that gets details about an NBA player using the NBA API.
+    the details that are returned in the dictionary are:
+    Name, Height, Weight, Birth Date, College, Country, Draft Year, Draft Round, Draft Number, Years Pro, Current Team
+    Args:
+        player_name: the full name of the NBA player (e.g. Kobe Bryant)
+    """
+    # Fetch player bio details
+    player_id = get_player(player_name)['id']
+    player_info = commonplayerinfo.CommonPlayerInfo(player_id=player_id)
+    data = player_info.get_dict()
+    info = data['resultSets'][0]['rowSet'][0]  # The first (and only) row
+
+    # Dictionary mapping for extracted values
+    player_details = {
+        "Name": info[3],  # Full name
+        "Height": info[11],  # Height in feet-inches
+        "Weight": info[12],  # Weight in pounds
+        "Birth Date": info[7].split('T')[0],  # Date of birth
+        "College": info[8],  # College name
+        "Country": info[9],  # Country of birth
+        "Draft Year": info[29],  # Year drafted
+        "Draft Round": info[30],  # Draft round
+        "Draft Number": info[31],  # Pick number
+        "Years Pro": info[13],  # Years in the NBA
+        "Team": info[20] if info[20] != '' else 'NA'   # Current team
+    }
+    return player_details
+
+
+
 # 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
@@ -55,7 +98,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, get_player_info], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,