Update app.py

app.py

@@ -6,7 +6,7 @@ import yaml
 import matplotlib.pyplot as plt
 import pandas as pd
 import re
-from bs4 import BeautifulSoup  
+from bs4 import BeautifulSoup  # Fixed Import
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -30,7 +30,7 @@ def get_box_score_links() -> list:
                 box_score_links.append(BASE_URL + href)
         
         # Return unique links while preserving order
-        return list(dict.fromkeys(box_score_links))[:3]  # Limit to first 3 games
+        return list(dict.fromkeys(box_score_links))
     
     except requests.exceptions.RequestException as e:
         # Return error message as a list to maintain consistent return type
@@ -61,18 +61,12 @@ def get_box_score_data(links: list) -> dict:
                 
                 # Check if the expected tables exist before accessing
                 if len(tables) > 0:
-                    df_team1 = tables[0]
-                    # Filter to relevant columns only if they exist
-                    relevant_cols = ['Starters', 'PTS', 'AST', 'TRB']
-                    df_team1 = df_team1[df_team1.columns.intersection(relevant_cols)]
-                    df_team1 = df_team1.dropna().to_dict(orient='records')
+                    df_team1 = tables[0].to_dict(orient='records')
                 else:
                     df_team1 = [{"Error": "Team 1 data not found"}]
 
                 if len(tables) > 8:
-                    df_team2 = tables[8]
-                    df_team2 = df_team2[df_team2.columns.intersection(relevant_cols)]
-                    df_team2 = df_team2.dropna().to_dict(orient='records')
+                    df_team2 = tables[8].to_dict(orient='records')
                 else:
                     df_team2 = [{"Error": "Team 2 data not found"}]
                 
@@ -89,32 +83,6 @@ def get_box_score_data(links: list) -> dict:
     except Exception as e:
         return {"Error": f"Error fetching boxScore data: {str(e)}"}
 
-
-@tool
-def get_stats_from_boxScore_data(box_scores: dict, stat: str) -> dict:
-    """A tool that fetches the player names and box score statistic for the provided stat and from the provided box score data.
-    Args:
-        box_scores: A dictionary representing the box score data for all of last nights games.
-        stat: A string representing the statistic that this function will return for each player in box_scores.
-    """
-    try:
-        stats = {}
-        for team, players in box_scores.items():
-            team_stats = []
-            for player in players:
-                # Extract player name and requested stat if it exists
-                player_name = player.get('Starters', 'Unknown')
-                stat_value = player.get(stat, 'N/A')
-                team_stats.append({player_name: stat_value})
-            stats[team] = team_stats
-        
-        return stats
-    
-    except Exception as e:
-        return {"Error": f"Error fetching boxScore data for given statistic: {str(e)}"}
-
-
-
 final_answer = FinalAnswerTool()
 search_tool = DuckDuckGoSearchTool()
 visit_webpage_tool = VisitWebpageTool()
@@ -147,8 +115,7 @@ agent = CodeAgent(
         visit_webpage_tool, 
         user_input_tool, 
         get_box_score_links, 
-        get_box_score_data,
-        get_stats_from_boxScore_data
+        get_box_score_data
     ],  # Add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
@@ -161,4 +128,4 @@ agent = CodeAgent(
 )
 
 # Launch Gradio UI
-GradioUI(agent).launch()
+GradioUI(agent).launch()