Update app.py

app.py

@@ -9,14 +9,82 @@ 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_weather(location: str) -> str:
+    """Get the current weather for a specified location.
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        location: A city, region, or country name, such as 'Beijing', 'Tokyo', or 'New York'.
     """
-    return "What magic will you build ?"
+    try:
+        url = f"https://wttr.in/{location}?format=j1"
+        response = requests.get(url, timeout=10)
+        response.raise_for_status()
+        data = response.json()
+
+        current = data["current_condition"][0]
+
+        temp_c = current.get("temp_C", "Unknown")
+        feels_like_c = current.get("FeelsLikeC", "Unknown")
+        humidity = current.get("humidity", "Unknown")
+        weather_desc = current.get("weatherDesc", [{}])[0].get("value", "Unknown")
+        wind_kph = current.get("windspeedKmph", "Unknown")
+
+        return (
+            f"Current weather in {location}:\n"
+            f"- Condition: {weather_desc}\n"
+            f"- Temperature: {temp_c}°C\n"
+            f"- Feels like: {feels_like_c}°C\n"
+            f"- Humidity: {humidity}%\n"
+            f"- Wind speed: {wind_kph} km/h"
+        )
+    except Exception as e:
+        return f"Error fetching weather for '{location}': {str(e)}"
+
+
+@tool
+def get_nba_games(date: str) -> str:
+    """Get NBA game results for a specific date.
+    Args:
+        date: A date in YYYY-MM-DD format, for example '2026-03-08'.
+    """
+    try:
+        url = "https://www.balldontlie.io/api/v1/games"
+        params = {
+            "dates[]": date,
+            "per_page": 100
+        }
+
+        response = requests.get(url, params=params, timeout=10)
+        response.raise_for_status()
+        data = response.json()
+
+        games = data.get("data", [])
+        if not games:
+            return f"No NBA games found for {date}."
+
+        results = [f"NBA games on {date}:"]
+        for game in games:
+            home_team = game["home_team"]["full_name"]
+            visitor_team = game["visitor_team"]["full_name"]
+            home_score = game.get("home_team_score", 0)
+            visitor_score = game.get("visitor_team_score", 0)
+            status = game.get("status", "Unknown")
+
+            if home_score > visitor_score:
+                winner = home_team
+            elif visitor_score > home_score:
+                winner = visitor_team
+            else:
+                winner = "Tie / Not finished"
+
+            results.append(
+                f"- {visitor_team} {visitor_score} : {home_score} {home_team} "
+                f"(Status: {status}, Winner: {winner})"
+            )
+
+        return "\n".join(results)
+
+    except Exception as e:
+        return f"Error fetching NBA games for '{date}': {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -46,6 +114,8 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
 custom_role_conversions=None,
 )
 
+# Search tool
+search_tool = DuckDuckGoSearchTool()
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
@@ -55,12 +125,12 @@ 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, my_custom_tool, get_nba_games, get_weather, get_current_time_in_timezone, search_tool, image_generation_tool]
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
-    name=None,
+    name="Qiang Chen",
     description=None,
     prompt_templates=prompt_templates
 )