Update app.py

app.py

@@ -54,6 +54,71 @@ def get_weather(location: str) -> str:
     except Exception as e:
         return f"Error fetching weather data: {str(e)}"
 
+@tool
+def get_weather_forecast(location: str, days: int = 3) -> str:
+    """Get weather forecast for a specified location for the next few days.
+    Args:
+        location: City name and optionally country code (e.g., 'London' or 'London,UK')
+        days: Number of days for forecast (1-5, default 3)
+    """
+    try:
+        # Validate days input
+        days = max(1, min(5, int(days)))
+        
+        API_KEY = "de1f560f2d4bea4e02dcdbe2e6f197a6"
+        base_url = "http://api.openweathermap.org/data/2.5/forecast"
+        
+        params = {
+            'q': location,
+            'appid': API_KEY,
+            'units': 'metric',
+            'cnt': days * 8  # 8 forecasts per day (every 3 hours)
+        }
+        
+        response = requests.get(base_url, params=params)
+        data = response.json()
+        
+        if response.status_code == 200:
+            forecast = []
+            current_date = ""
+            
+            for item in data['list']:
+                date = item['dt_txt'].split()[0]
+                if date != current_date:
+                    if current_date:
+                        forecast.append(day_forecast)
+                    current_date = date
+                    day_forecast = {
+                        'date': date,
+                        'temps': [],
+                        'weathers': []
+                    }
+                
+                day_forecast['temps'].append(item['main']['temp'])
+                day_forecast['weathers'].append(item['weather'][0]['description'])
+            
+            # Add the last day
+            if current_date:
+                forecast.append(day_forecast)
+            
+            # Format the response
+            result = [f"Weather forecast for {location}:"]
+            for day in forecast[:days]:  # Limit to requested days
+                avg_temp = sum(day['temps']) / len(day['temps'])
+                main_weather = max(set(day['weathers']), key=day['weathers'].count)
+                
+                result.append(
+                    f"{day['date']}: {main_weather}, average temp {avg_temp:.1f}°C "
+                    f"(min {min(day['temps']):.1f}°C, max {max(day['temps']):.1f}°C)"
+                )
+            
+            return "\n".join(result)
+        else:
+            return f"Error getting forecast: {data.get('message', 'Unknown error')}"
+    
+    except Exception as e:
+        return f"Error fetching weather forecast: {str(e)}"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.