Update app.py

app.py

@@ -2,10 +2,12 @@ from smolagents import CodeAgent, tool
 import datetime
 import pytz
 import yaml
+import requests
+import json
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Define a simple tool that doesn't rely on any external libraries
+# Text Analyzer Tool
 @tool
 def text_analyzer(text: str) -> str:
     """Analyzes text and returns statistics about it.
@@ -46,6 +48,7 @@ def text_analyzer(text: str) -> str:
     except Exception as e:
         return f"Error analyzing text: {str(e)}"
 
+# Timezone Tool
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -62,7 +65,69 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Set up the agent with minimal tools
+# Weather Forecast Tool
+@tool
+def weather_forecast(location: str) -> str:
+    """Fetches weather forecast for a specified location.
+    
+    Args:
+        location: The location to get weather forecast for (city name or coordinates).
+    """
+    try:
+        # Connect to a public weather API
+        api_url = f"https://wttr.in/{location}?format=j1"
+        
+        # Make the API request
+        response = requests.get(api_url, timeout=10)
+        response.raise_for_status()  # Raise an exception for HTTP errors
+        
+        # Parse the JSON response
+        weather_data = response.json()
+        
+        # Extract relevant information
+        current_condition = weather_data.get("current_condition", [{}])[0]
+        weather_desc = current_condition.get("weatherDesc", [{}])[0].get("value", "Unknown")
+        temp_c = current_condition.get("temp_C", "Unknown")
+        temp_f = current_condition.get("temp_F", "Unknown")
+        feels_like_c = current_condition.get("FeelsLikeC", "Unknown")
+        humidity = current_condition.get("humidity", "Unknown")
+        wind_speed = current_condition.get("windspeedKmph", "Unknown")
+        wind_dir = current_condition.get("winddir16Point", "Unknown")
+        
+        # Get forecast for upcoming days
+        forecast = weather_data.get("weather", [])
+        forecast_info = ""
+        
+        if forecast:
+            forecast_info = "\n\nForecast for the next few days:\n"
+            for day in forecast[:3]:  # Limit to 3 days
+                date = day.get("date", "Unknown")
+                max_temp_c = day.get("maxtempC", "Unknown")
+                min_temp_c = day.get("mintempC", "Unknown")
+                desc = day.get("hourly", [{}])[0].get("weatherDesc", [{}])[0].get("value", "Unknown")
+                
+                forecast_info += f"- {date}: {desc}, Max: {max_temp_c}°C, Min: {min_temp_c}°C\n"
+        
+        # Format the response
+        weather_report = f"""
+Weather for {location}:
+Current Conditions: {weather_desc}
+Temperature: {temp_c}°C / {temp_f}°F (Feels like: {feels_like_c}°C)
+Humidity: {humidity}%
+Wind: {wind_speed} km/h, Direction: {wind_dir}
+{forecast_info}
+"""
+        
+        return weather_report.strip()
+        
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching weather for {location}: Connection error - {str(e)}"
+    except json.JSONDecodeError:
+        return f"Error fetching weather for {location}: Invalid response from weather service"
+    except Exception as e:
+        return f"Error fetching weather for {location}: {str(e)}"
+
+# Set up the agent with our tools
 final_answer = FinalAnswerTool()
 
 with open("prompts.yaml", 'r') as stream:
@@ -77,10 +142,10 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Create agent with minimal tools
+# Create agent with our tools (now 4 including final_answer)
 agent = CodeAgent(
     model=model,
-    tools=[text_analyzer, get_current_time_in_timezone, final_answer],
+    tools=[text_analyzer, get_current_time_in_timezone, weather_forecast, final_answer],
     max_steps=6,
     verbosity_level=1,
     grammar=None,