Update app.py

app.py

@@ -7,6 +7,10 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 from smolagents.agent_types import AgentImage
 import uuid
+import requests
+import os
+
+CRICKET_API_KEY = os.getenv("CRICKET_API_KEY")
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -53,6 +57,73 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+@tool
+def get_live_cricket_score(team: str) -> str:
+    """
+    Fetches live cricket score for a given team.
+    
+    Args:
+        team: Name of the cricket team (e.g., India, Australia).
+    
+    Returns:
+        Live match details as formatted string.
+    """
+    
+    url = "https://api.cricapi.com/v1/currentMatches"
+    
+    params = {
+        "apikey": CRICKET_API_KEY,
+        "offset": 0
+    }
+    
+    response = requests.get(url, params=params)
+    data = response.json()
+    
+    if data.get("status") != "success":
+        return "Unable to fetch live matches at the moment."
+    
+    matches = data.get("data", [])
+    
+    for match in matches:
+        if team.lower() in match["name"].lower():
+            return f"""
+Match: {match['name']}
+Status: {match['status']}
+Score: {match.get('score', 'Score not available')}
+"""
+    
+    return f"No live match found for {team} right now."
+
+
+@tool
+def get_today_matches() -> str:
+    """
+    Fetches all current live cricket matches.
+    """
+    
+    url = "https://api.cricapi.com/v1/currentMatches"
+    
+    params = {
+        "apikey": CRICKET_API_KEY,
+        "offset": 0
+    }
+    
+    response = requests.get(url, params=params)
+    data = response.json()
+    
+    if data.get("status") != "success":
+        return "Unable to fetch matches."
+    
+    matches = data.get("data", [])
+    
+    result = "Today's Matches:\n\n"
+    
+    for match in matches:
+        result += f"{match['name']} — {match['status']}\n"
+    
+    return result
+
+
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -72,7 +143,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, generate_image, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, generate_image, get_current_time_in_timezone, get_live_cricket_score, get_today_matches], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,