Update app.py

app.py

@@ -9,17 +9,30 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_chess_leaderboard() -> str:
+def get_chess_leaderboard(prompt: str) -> str:
     """A tool that fetches the current chess leaderboard from Chess.com API.
     """
     url = "https://api.chess.com/pub/leaderboards"
-    headers = {"User-Agent": "MarcoM003/First_agent_template"}
+    headers = {"User-Agent": "MiAgenteIA/1.0 (ejecutando desde VSCode)"}
     respuesta = requests.get(url, headers=headers)
     if respuesta.status_code == 200:
         datos = respuesta.json()
-        leaderboard_str = ""
-        for jugador in datos["live_blitz"][:10]:
-            leaderboard_str += f"{jugador['username']} - Rating: {jugador['score']}\n"
+        if "blitz" in prompt.lower():
+            leaderboard_str = "Leaderboard de Blitz:\n"
+            for jugador in datos["live_blitz"][:10]:
+                leaderboard_str += f"{jugador['username']} - Rating: {jugador['score']}\n"
+        elif "rapid" in prompt.lower():
+            leaderboard_str = "Leaderboard de Rapid:\n"
+            for jugador in datos["live_rapid"][:10]:
+                leaderboard_str += f"{jugador['username']} - Rating: {jugador['score']}\n"
+        elif "daily" in prompt.lower():
+            leaderboard_str = "Leaderboard diario:\n"
+            for jugador in datos["daily"][:10]:
+                leaderboard_str += f"{jugador['username']} - Rating: {jugador['score']}\n"
+        else:
+            leaderboard_str = "Leaderboard de Blitz:\n"
+            for jugador in datos["live_blitz"][:10]:
+                leaderboard_str += f"{jugador['username']} - Rating: {jugador['score']}\n"
         return leaderboard_str
     else:
         return f"Error al obtener datos: {respuesta.status_code}"