Update app.py

app.py

@@ -37,31 +37,36 @@ def get_current_time_in_timezone(timezone: str) -> str:
 @tool
 def get_nba_matches() -> str:
     """
-    A tool that retrieves upcoming NBA games via TheRundown API.
+    A tool that retrieves upcoming NBA matches using TheRundown API.
     
     Returns:
-        A string containing a human-readable list of upcoming games.
+        A human-readable string listing the upcoming NBA matches.
     """
-    url = "https://api.apilayer.com/therundown/sports/4/events"  # Replace with the NBA-specific endpoint if available.
+    import requests
+    # Replace the endpoint below with the correct NBA events endpoint if available.
+    # Here we assume sport_id 4 corresponds to NBA and that the '/events' endpoint returns the matches.
+    url = "https://api.apilayer.com/therundown/sports/4/events"
     headers = {
         "apikey": "7k4hKSUeWkbigKxZiNV5CQ8RSlEd72Cj"
     }
+    
     response = requests.get(url, headers=headers)
     if response.status_code == 200:
-        events = response.json()
-        # Suppose 'events' is a dictionary containing a list of games under the key "events"
-        games = []
-        for event in events.get("events", []):
-            # Extract relevant information: date, teams, etc.
-            date = event.get("start_time", "Unknown date")
-            teams = event.get("teams", "Game info not defined")
-            games.append(f"{date} : {teams}")
-        if games:
-            return "\n".join(games)
+        data = response.json()
+        matches = []
+        # Parse the JSON based on the API's structure.
+        # (Adjust the keys below according to the actual response format.)
+        for event in data.get("events", []):
+            start_time = event.get("start_time", "Unknown time")
+            home_team = event.get("home_team", "Home team not available")
+            away_team = event.get("away_team", "Away team not available")
+            matches.append(f"{start_time}: {home_team} vs {away_team}")
+        if matches:
+            return "\n".join(matches)
         else:
-            return "No NBA games found in the response."
+            return "No NBA matches found."
     else:
-        return f"Error retrieving NBA games: {response.status_code}"
+        return f"Error retrieving NBA matches: {response.status_code}"
 
 @tool
 def predict_nba_match(match_info: str) -> str:
@@ -69,19 +74,18 @@ def predict_nba_match(match_info: str) -> str:
     A tool that generates a prediction for an NBA match.
     
     Args:
-        match_info: A string containing the match details in the format "TeamA vs TeamB".
+        match_info: A string containing match details in the format "TeamA vs TeamB".
     
     Returns:
-        A string with the prediction (e.g., "The prediction is that TeamA will win.").
+        A string with the prediction, e.g. "The prediction is that TeamA will win.".
     """
     import random
-    # Assume match_info is in the format "TeamA vs TeamB"
     teams = match_info.split(" vs ")
     if len(teams) == 2:
         prediction = random.choice(teams)
         return f"The prediction is that {prediction} will win."
     else:
-        return "Incorrect match_info format. Please provide teams in the format 'TeamA vs TeamB'."
+        return "Invalid match format. Please provide match details in the format 'TeamA vs TeamB'."
 
 
 final_answer = FinalAnswerTool()