Update app.py

app.py

@@ -1,69 +1,186 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
 import yaml
+import matplotlib.pyplot as plt
 from tools.final_answer import FinalAnswerTool
-
 from Gradio_UI import GradioUI
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
+# -----------------------------
+# LOCATION + TIMEZONE TOOLS
+# -----------------------------
+
+@tool
+def get_location() -> dict:
+    """Get user's approximate latitude and longitude."""
+    try:
+        res = requests.get("http://ip-api.com/json/")
+        data = res.json()
+        return {
+            "latitude": data["lat"],
+            "longitude": data["lon"]
+        }
+    except Exception as e:
+        return {"error": str(e)}
+
 @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 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
+def get_local_timezone() -> str:
+    """Detects the user's local timezone based on IP."""
+    try:
+        res = requests.get("http://ip-api.com/json/")
+        data = res.json()
+        return data.get("timezone", "UTC")
+    except Exception as e:
+        return f"Error detecting timezone: {str(e)}"
+
+
+# -----------------------------
+# WEATHER TOOL (24H + UNCERTAINTY)
+# -----------------------------
 
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
+def get_hourly_weather(latitude: float, longitude: float, timezone: str) -> list:
+    """Fetch hourly weather forecast for next 24 hours including uncertainty bands."""
     try:
-        # Create timezone object
+        url = "https://api.open-meteo.com/v1/forecast"
+
+        params = {
+            "latitude": latitude,
+            "longitude": longitude,
+            "hourly": "temperature_2m,precipitation,weathercode",
+            "forecast_days": 2,
+            "timezone": timezone
+        }
+
+        res = requests.get(url, params=params)
+        data = res.json()
+
+        hourly = data["hourly"]
         tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
+        now = datetime.datetime.now(tz)
+
+        result = []
+
+        for t, temp, precip, code in zip(
+            hourly["time"],
+            hourly["temperature_2m"],
+            hourly["precipitation"],
+            hourly["weathercode"]
+        ):
+            dt = datetime.datetime.fromisoformat(t)
+
+            if dt >= now and len(result) < 24:
+                # simple synthetic uncertainty band (±2°C)
+                result.append({
+                    "time": t,
+                    "temperature": temp,
+                    "temp_min": temp - 2,
+                    "temp_max": temp + 2,
+                    "precipitation": precip,
+                    "weathercode": code
+                })
+
+        return result
+
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        return [{"error": str(e)}]
 
 
-final_answer = FinalAnswerTool()
+# -----------------------------
+# WEATHER CODE → ICON
+# -----------------------------
+
+def weather_icon(code):
+    if code == 0:
+        return "☀️"
+    elif code in [1, 2, 3]:
+        return "⛅"
+    elif code in [45, 48]:
+        return "🌫"
+    elif code in [51, 53, 55, 61, 63, 65]:
+        return "🌧"
+    elif code in [71, 73, 75]:
+        return "❄️"
+    elif code in [95]:
+        return "⛈"
+    else:
+        return "🌈"
 
-# 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:
-# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
+
+# -----------------------------
+# PLOT TOOL (ICONS + BANDS)
+# -----------------------------
+
+@tool
+def plot_weather(weather_data: list) -> str:
+    """Generate stylized weather visualization with icons + uncertainty bands."""
+
+    times = [entry["time"][11:16] for entry in weather_data]
+    temps = [entry["temperature"] for entry in weather_data]
+    temp_min = [entry["temp_min"] for entry in weather_data]
+    temp_max = [entry["temp_max"] for entry in weather_data]
+    precip = [entry["precipitation"] for entry in weather_data]
+    codes = [entry["weathercode"] for entry in weather_data]
+
+    icons = [weather_icon(c) for c in codes]
+
+    plt.figure(figsize=(12, 6))
+
+    # Temperature line
+    plt.plot(times, temps, label="Temperature")
+
+    # Uncertainty band
+    plt.fill_between(times, temp_min, temp_max, alpha=0.2)
+
+    # Precip bars
+    plt.bar(times, precip, alpha=0.3, label="Precipitation")
+
+    # Add icons above line
+    for i, icon in enumerate(icons):
+        plt.text(i, temps[i] + 1.5, icon, ha='center')
+
+    plt.xticks(rotation=45)
+    plt.title("24-Hour Weather Forecast")
+    plt.legend()
+    plt.tight_layout()
+
+    file_path = "/mnt/data/weather_visual.png"
+    plt.savefig(file_path)
+    plt.close()
+
+    return file_path
+
+
+# -----------------------------
+# AGENT SETUP
+# -----------------------------
+
+final_answer = FinalAnswerTool()
 
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 )
 
-
-# Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-    
+
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer,
+        get_location,
+        get_local_timezone,
+        get_hourly_weather,
+        plot_weather
+    ],
     max_steps=6,
     verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
     prompt_templates=prompt_templates
 )
 
-
-GradioUI(agent).launch()
+GradioUI(agent).launch