Update app.py

app.py

@@ -12,14 +12,6 @@ import requests
 import pytz
 import yaml
 
-@tool
-def my_custom_tool(arg1: str, arg2: int) -> str:
-    """A tool that does nothing yet
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -34,6 +26,57 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+
+@tool
+def get_current_weather(location: str) -> str:
+    """A tool that fetches the current weather for a given city/location.
+    Args:
+        location: The name of the city or location (e.g., 'Ho Chi Minh City').
+    """
+    try:
+        # Geocode
+        geo_response = requests.get(
+            f"https://geocoding-api.open-meteo.com/v1/search?name={location}&count=1"
+        ).json()
+
+        if not geo_response.get("results"):
+            return f"Could not find location: {location}"
+
+        result = geo_response["results"][0]
+        lat, lon = result["latitude"], result["longitude"]
+        city_name = result["name"]
+        country = result.get("country", "")
+
+        # Fetch weather
+        weather_response = requests.get(
+            f"https://api.open-meteo.com/v1/forecast?"
+            f"latitude={lat}&longitude={lon}"
+            f"&current_weather=true"
+        ).json()
+        current = weather_response["current_weather"]
+
+        weather_codes = {
+            0: "Clear sky", 1: "Mainly clear", 2: "Partly cloudy", 3: "Overcast",
+            45: "Foggy", 48: "Icy fog", 51: "Light drizzle", 53: "Moderate drizzle",
+            61: "Slight rain", 63: "Moderate rain", 65: "Heavy rain",
+            71: "Slight snow", 73: "Moderate snow", 75: "Heavy snow",
+            80: "Slight showers", 81: "Moderate showers", 82: "Violent showers",
+            95: "Thunderstorm", 96: "Thunderstorm with hail",
+        }
+        description = weather_codes.get(current["weathercode"], "Unknown")
+
+        return (
+            f"Weather in {city_name}, {country}:\n"
+            f"  🌡️  Temperature     : {current['temperature']}°C\n"
+            f"  💨  Wind Speed      : {current['windspeed']} km/h\n"
+            f"  🧭  Wind Direction  : {current['winddirection']}°\n"
+            f"  🌤️  Condition       : {description}\n"
+        )
+
+    except Exception as e:
+        return f"Error fetching weather for '{location}': {str(e)}"
+
+
 final_answer = FinalAnswerTool()
 
 model = HfApiModel(
@@ -43,7 +86,6 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Load image generation tool correctly (only once)
 image_generation_tool = load_tool("m-ric/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
@@ -53,9 +95,10 @@ agent = CodeAgent(
     model=model,
     tools=[
         final_answer,
-        image_generation_tool,        # ✅ image tool
-        get_current_time_in_timezone, # ✅ custom tools
-        my_custom_tool,
+        image_generation_tool,
+        get_current_time_in_timezone,
+        get_current_weather,          # ✅ new weather tool
+        DuckDuckGoSearchTool(),
     ],
     max_steps=6,
     verbosity_level=1,