Updated to get weather forecast for a zip code

app.py

@@ -3,7 +3,11 @@ import datetime
 import requests
 import pytz
 import yaml
+from smartystreets_python_sdk import StaticCredentials, exceptions, ClientBuilder
+from smartystreets_python_sdk.us_street import Lookup as StreetLookup
+from smartystreets_python_sdk.us_zipcode import Lookup as ZipcodeLookup
 from tools.final_answer import FinalAnswerTool
+import os
 
 from Gradio_UI import GradioUI
 
@@ -33,6 +37,44 @@ 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_weather_by_zipcode(zipcode: str) -> str:
+    """Get weather forecast for a given US zip code
+    Args:
+        zipcode: A 5-digit US zip code
+    """
+    try:
+        # Validate zipcode
+        if not (zipcode.isdigit() and len(zipcode) == 5):
+            return "Please provide a valid 5-digit US zip code"
+
+        # Get forecast data using WeatherAPI.com
+        weather_api_key = os.getenv('WEATHERAPI_KEY')
+        url = f"http://api.weatherapi.com/v1/forecast.json?key={weather_api_key}&q={zipcode}&days=3&aqi=no"
+        
+        response = requests.get(url)
+        if response.status_code != 200:
+            return "Error fetching weather data"
+            
+        weather_data = response.json()
+        location = weather_data['location']['name']
+        forecast = weather_data['forecast']['forecastday']
+        
+        result = f"3-Day Forecast for {location} ({zipcode}):\n\n"
+        
+        for day in forecast:
+            date = day['date']
+            max_temp = day['day']['maxtemp_f']
+            min_temp = day['day']['mintemp_f']
+            condition = day['day']['condition']['text']
+            result += f"{date}:\n"
+            result += f"High: {max_temp}°F, Low: {min_temp}°F\n"
+            result += f"Conditions: {condition}\n\n"
+        
+        return result
+
+    except Exception as e:
+        return f"Error getting weather: {str(e)}"
 
 final_answer = FinalAnswerTool()
 
@@ -55,7 +97,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_weather_by_zipcode, final_answer],  # Added weather tool
     max_steps=6,
     verbosity_level=1,
     grammar=None,