Update app.py

app.py

@@ -4,6 +4,7 @@ import datetime
 import pytz
 import math
 import os
+import requests
 from deep_translator import GoogleTranslator
 
 # ============ TOOLS ============
@@ -21,6 +22,43 @@ def translator_tool(text: str, target_language: str) -> str:
     except Exception as e:
         return f"Translation error: {str(e)}. Make sure the language name is correct."
 
+@tool
+def weather_tool(location: str) -> str:
+    """Get the current weather for any location worldwide.
+    
+    Args:
+        location: The name of the city or place (e.g., 'London' or 'Tokyo').
+    """
+    try:
+        # 1. Geocoding: Convert city name to coordinates using a free service
+        geo_url = f"https://geocoding-api.open-meteo.com/v1/search?name={location}&count=1&language=en&format=json"
+        geo_res = requests.get(geo_url).json()
+        
+        if not geo_res.get('results'):
+            return f"I couldn't find coordinates for '{location}'. Please check the spelling."
+        
+        data = geo_res['results'][0]
+        lat, lon = data['latitude'], data['longitude']
+        city_full = f"{data.get('name')}, {data.get('country')}"
+
+        # 2. Weather: Get current data using coordinates
+        weather_url = f"https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={lon}&current=temperature_2m,relative_humidity_2m,apparent_temperature,weather_code,wind_speed_10m"
+        w_res = requests.get(weather_url).json()
+        
+        curr = w_res['current']
+        temp = curr['temperature_2m']
+        feels = curr['apparent_temperature']
+        hum = curr['relative_humidity_2m']
+        wind = curr['wind_speed_10m']
+
+        return (f"🌤️ Weather for {city_full}:\n"
+                f"• Temperature: {temp}°C (Feels like {feels}°C)\n"
+                f"• Humidity: {hum}%\n"
+                f"• Wind Speed: {wind} km/h")
+    except Exception as e:
+        return f"Weather Error: {str(e)}"
+
+
 @tool
 def web_search_tool(query: str) -> str:
     """Search the web for current information using DuckDuckGo.
@@ -130,9 +168,9 @@ model = InferenceClientModel(
 
 agent = CodeAgent(
     model=model,
-    tools=[web_search_tool, time_tool, calculator_tool, translator_tool],
+    tools=[web_search_tool, time_tool, calculator_tool, translator_tool, weather_tool],
     max_steps=5,
-    additional_authorized_imports=['math', 'datetime', 'pytz', 'deep_translator']
+    additional_authorized_imports=['math', 'datetime', 'pytz', 'deep_translator', 'requests']
 )
 
 # ============ GRADIO UI ============
@@ -156,7 +194,8 @@ demo = gr.ChatInterface(
         "Convert 150 pounds to kg",
         "Calculate the square root of 144 plus 50",
         "Translate 'Hello, how are you today?' into Japanese",
-        "Search for the latest news on AI agents"
+        "Search for the latest news on AI agents",
+        "What's the weather in Paris right now?", "How's the weather in New York compared to Tokyo?"
     ],
 )