Update app.py

app.py

@@ -7,16 +7,44 @@ 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 compare_tennis_players(player1_name: str, player2_name: str) -> str:
+    """A tool that compares statistics between two tennis players.
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        player1_name: The name of the first tennis player.
+        player2_name: The name of the second tennis player.
     """
-    return "What magic will you build ?"
+    base_url = "https://api.openligadb.de"  # API gratuita para datos de deportes
+    
+    try:
+        # Obtener datos de partidos (esto es un ejemplo, ajusta según la API)
+        matches_response = requests.get(f"{base_url}/getmatchdata/tennis")
+        matches_data = matches_response.json()
+        
+        # Filtrar partidos de los jugadores
+        player1_matches = [match for match in matches_data if player1_name in match["team1"]["teamName"] or player1_name in match["team2"]["teamName"]]
+        player2_matches = [match for match in matches_data if player2_name in match["team1"]["teamName"] or player2_name in match["team2"]["teamName"]]
+        
+        # Calcular estadísticas básicas (victorias, derrotas, etc.)
+        player1_wins = sum(1 for match in player1_matches if player1_name in match["winner"]["teamName"])
+        player1_losses = len(player1_matches) - player1_wins
+        
+        player2_wins = sum(1 for match in player2_matches if player2_name in match["winner"]["teamName"])
+        player2_losses = len(player2_matches) - player2_wins
+        
+        # Formatear la respuesta
+        result = f"Comparación entre {player1_name} y {player2_name}:\n"
+        result += f"{player1_name} - Victorias: {player1_wins}, Derrotas: {player1_losses}\n"
+        result += f"{player2_name} - Victorias: {player2_wins}, Derrotas: {player2_losses}\n"
+        
+        return result
+    
+    except Exception as e:
+        return f"Error al comparar jugadores: {str(e)}"
+
+
+###############
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +83,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, compare_tennis_players], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,