Update app.py

app.py

@@ -11,6 +11,21 @@ from Gradio_UI import GradioUI
 # LOCATION + TIMEZONE TOOLS
 # -----------------------------
 
+@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').
+    """
+    try:
+        # Create timezone object
+        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}"
+    except Exception as e:
+        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+
 @tool
 def get_location() -> dict:
     """Get user's approximate latitude and longitude."""
@@ -120,54 +135,112 @@ def weather_icon(code):
 # -----------------------------
 # PLOT TOOL (ICONS + BANDS)
 # -----------------------------
-
 @tool
 def plot_weather(weather_data: list) -> str:
-    """Generate stylized weather visualization with icons and uncertainty bands.
+    """Generate an SVG weather visualization with icons and uncertainty bands.
 
     Args:
-        weather_data: A list of dictionaries containing hourly weather data.
+        weather_data: A list of dictionaries containing hourly weather data. Each dict
+                      should have 'time', 'temperature', 'temp_min', 'temp_max',
+                      'precipitation', and 'weathercode'.
 
     Returns:
-        A string file path to the generated image.
+        A string file path to the generated SVG image.
     """
-
-    times = [entry["time"][11:16] for entry in weather_data]
+    import datetime
+    from xml.etree.ElementTree import Element, SubElement, ElementTree, tostring
+
+    # Helper: map weather code to 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 "🌈"
+
+    # SVG canvas size
+    width = 1200
+    height = 400
+    margin = 50
+
+    # Compute min/max for scaling
     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()
+    temp_min_vals = [entry["temp_min"] for entry in weather_data]
+    temp_max_vals = [entry["temp_max"] for entry in weather_data]
+    overall_min = min(temp_min_vals) - 2
+    overall_max = max(temp_max_vals) + 2
+
+    n = len(weather_data)
+    step_x = (width - 2*margin) / (n-1) if n > 1 else 1
+
+    def y_scale(temp):
+        # invert y axis (SVG 0,0 is top-left)
+        return height - margin - (temp - overall_min)/(overall_max - overall_min)*(height - 2*margin)
+
+    # Create root SVG
+    svg = Element('svg', width=str(width), height=str(height), xmlns="http://www.w3.org/2000/svg")
+
+    # Draw uncertainty band
+    points = []
+    for i, entry in enumerate(weather_data):
+        x = margin + i*step_x
+        y = y_scale(entry["temp_max"])
+        points.append(f"{x},{y}")
+    for i, entry in reversed(list(enumerate(weather_data))):
+        x = margin + i*step_x
+        y = y_scale(entry["temp_min"])
+        points.append(f"{x},{y}")
+    band = SubElement(svg, "polygon", points=" ".join(points), fill="#a0c4ff", opacity="0.3")
+
+    # Draw temperature line
+    temp_line_points = []
+    for i, entry in enumerate(weather_data):
+        x = margin + i*step_x
+        y = y_scale(entry["temperature"])
+        temp_line_points.append(f"{x},{y}")
+    SubElement(svg, "polyline", points=" ".join(temp_line_points),
+               fill="none", stroke="#0077b6", stroke_width="2")
+
+    # Draw precipitation bars
+    max_precip = max([entry["precipitation"] for entry in weather_data] + [1])
+    for i, entry in enumerate(weather_data):
+        x = margin + i*step_x - step_x*0.2
+        y = y_scale(overall_min) - (entry["precipitation"]/max_precip)*(height - 2*margin)
+        bar_height = (entry["precipitation"]/max_precip)*(height - 2*margin)
+        SubElement(svg, "rect", x=str(x), y=str(y), width=str(step_x*0.4),
+                   height=str(bar_height), fill="#00b4d8", opacity="0.4")
+
+    # Draw icons and x-axis labels
+    for i, entry in enumerate(weather_data):
+        x = margin + i*step_x
+        y_temp = y_scale(entry["temperature"])
+        # icon above temp
+        text = SubElement(svg, "text", x=str(x), y=str(y_temp - 10),
+                          font_size="20", text_anchor="middle")
+        text.text = weather_icon(entry["weathercode"])
+        # time label below
+        t_label = SubElement(svg, "text", x=str(x), y=str(height - margin + 20),
+                             font_size="12", text_anchor="middle")
+        t_label.text = entry["time"][11:16]
+
+    # Save SVG
+    file_path = "/mnt/data/weather_visual.svg"
+    tree = ElementTree(svg)
+    tree.write(file_path)
 
     return file_path
 
 
+
 # -----------------------------
 # AGENT SETUP
 # -----------------------------