Fetch pl team match statics

app.py

@@ -7,16 +7,121 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-# 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
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+def get_pl_team_stats(pl_team_name: str = "Liverpool", season: str = "2023-24") -> dict:
+    """Fetches PL team latest Premier League statistics including:
+    - Average points
+    - Home/Away performance
+    - Performance against different table tiers
+    
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        pl_team_name: The Premier League football club's team name
+        season: The Premier League season to analyze (format: YYYY-YY)
+    
+    Returns:
+        Dictionary containing comprehensive PL team statistics
     """
-    return "What magic will you build ?"
+    try:
+        # First fetch current standings to determine team tiers
+        standings_url = "https://fantasy.premierleague.com/api/bootstrap-static/"
+        response = requests.get(standings_url)
+        response.raise_for_status()
+        data = response.json()
+        
+        # Get team IDs and positions
+        teams = {team['id']: team['name'] for team in data['teams']}
+        standings = {team['name']: team['position'] for team in data['teams']}
+        
+        # Determine table tiers
+        top6 = [name for name, pos in standings.items() if pos <= 6]
+        middle = [name for name, pos in standings.items() if 7 <= pos <= 14]
+        bottom6 = [name for name, pos in standings.items() if pos >= 15]
+        
+        # Now fetch team's fixtures
+        team_id = [id for id, name in teams.items() if "pl_team_name" in name][0]
+        fixtures_url = f"https://fantasy.premierleague.com/api/team/{team_id}/fixtures/"
+        response = requests.get(fixtures_url)
+        response.raise_for_status()
+        fixtures = response.json()
+        
+        # Filter for completed matches in specified season
+        completed_matches = [
+            f for f in fixtures 
+            if f['finished'] and f['season_name'] == season
+        ]
+        
+        # Initialize stats
+        stats = {
+            'total_matches': len(completed_matches),
+            'total_points': 0,
+            'home': {'matches': 0, 'points': 0},
+            'away': {'matches': 0, 'points': 0},
+            'vs_top6': {'matches': 0, 'points': 0},
+            'vs_middle': {'matches': 0, 'points': 0},
+            'vs_bottom6': {'matches': 0, 'points': 0}
+        }
+        
+        # Process each match
+        for match in completed_matches:
+            # Calculate points (3 for win, 1 for draw)
+            if match['team_h_score'] > match['team_a_score']:
+                points = 3 if match['team_h'] == team_id else 0
+            elif match['team_h_score'] == match['team_a_score']:
+                points = 1
+            else:
+                points = 3 if match['team_a'] == team_id else 0
+            
+            stats['total_points'] += points
+            
+            # Home/away stats
+            if match['team_h'] == team_id:
+                stats['home']['matches'] += 1
+                stats['home']['points'] += points
+                opponent = teams[match['team_a']]
+            else:
+                stats['away']['matches'] += 1
+                stats['away']['points'] += points
+                opponent = teams[match['team_h']]
+            
+            # Opposition tier stats
+            if opponent in top6:
+                stats['vs_top6']['matches'] += 1
+                stats['vs_top6']['points'] += points
+            elif opponent in middle:
+                stats['vs_middle']['matches'] += 1
+                stats['vs_middle']['points'] += points
+            elif opponent in bottom6:
+                stats['vs_bottom6']['matches'] += 1
+                stats['vs_bottom6']['points'] += points
+        
+        # Calculate averages
+        def safe_divide(a, b):
+            return round(a/b, 2) if b > 0 else 0.0
+            
+        stats['avg_points'] = safe_divide(stats['total_points'], stats['total_matches'])
+        stats['home']['avg_points'] = safe_divide(stats['home']['points'], stats['home']['matches'])
+        stats['away']['avg_points'] = safe_divide(stats['away']['points'], stats['away']['matches'])
+        stats['vs_top6']['avg_points'] = safe_divide(stats['vs_top6']['points'], stats['vs_top6']['matches'])
+        stats['vs_middle']['avg_points'] = safe_divide(stats['vs_middle']['points'], stats['vs_middle']['matches'])
+        stats['vs_bottom6']['avg_points'] = safe_divide(stats['vs_bottom6']['points'], stats['vs_bottom6']['matches'])
+        
+        return stats
+    
+    except Exception as e:
+        return {"error": f"Failed to fetch team stats: {str(e)}"}
+
+# # Add the tool to your agent
+# agent = CodeAgent(
+#     model=model,
+#     tools=[final_answer, image_generation_tool, get_pl_team_stats],  # Added our new tool
+#     max_steps=6,
+#     verbosity_level=1,
+#     grammar=None,
+#     planning_interval=None,
+#     name=None,
+#     description=None,
+#     prompt_templates=prompt_templates
+# )
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +160,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_generation_tool, get_pl_team_stats], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,