Update app.py

app.py

@@ -1,26 +1,80 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
 import yaml
 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 !
+# Fully functional weather forecast tool
 @tool
-def get_weather_forecast(weather:int, feels_like: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 
+def get_weather_forecast(city: str, country_code: str = "") -> str:
+    """A tool that fetches the current weather and forecast for a specified location.
+    
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        city: The name of the city to get weather for (e.g., 'London', 'New York')
+        country_code: Optional two-letter country code (e.g., 'US', 'UK', 'FR')
     """
-    return "What magic will you build ?"
+    try:
+        # OpenWeatherMap API endpoint and parameters
+        # Note: In a real application, you should store API keys securely
+        API_KEY = "YOUR_OPENWEATHERMAP_API_KEY"  # Replace with your actual API key
+        base_url = "https://api.openweathermap.org/data/2.5/weather"
+        
+        # Build location string based on inputs
+        location = city
+        if country_code:
+            location = f"{city},{country_code}"
+            
+        # Create request parameters
+        params = {
+            'q': location,
+            'appid': API_KEY,
+            'units': 'metric'  # Use metric units (Celsius)
+        }
+        
+        # Make API request
+        response = requests.get(base_url, params=params)
+        response.raise_for_status()  # Raise exception for HTTP errors
+        
+        # Parse response
+        weather_data = response.json()
+        
+        # Extract relevant information
+        current_temp = weather_data['main']['temp']
+        feels_like = weather_data['main']['feels_like']
+        humidity = weather_data['main']['humidity']
+        conditions = weather_data['weather'][0]['description']
+        wind_speed = weather_data['wind']['speed']
+        
+        # Format and return results
+        result = (
+            f"Current Weather in {weather_data['name']}, {weather_data['sys']['country']}:\n"
+            f"• Temperature: {current_temp}°C (feels like {feels_like}°C)\n"
+            f"• Conditions: {conditions.capitalize()}\n"
+            f"• Humidity: {humidity}%\n"
+            f"• Wind Speed: {wind_speed} m/s"
+        )
+        
+        return result
+    
+    except requests.exceptions.HTTPError as http_err:
+        return f"HTTP Error occurred: {http_err}"
+    except requests.exceptions.ConnectionError:
+        return f"Connection Error: Unable to connect to weather service"
+    except requests.exceptions.Timeout:
+        return f"Timeout Error: Request timed out"
+    except requests.exceptions.RequestException as err:
+        return f"Error fetching weather data: {err}"
+    except KeyError as key_err:
+        return f"Error parsing weather data: {key_err}"
+    except Exception as e:
+        return f"Unexpected error: {str(e)}"
 
 @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').
     """
@@ -33,29 +87,26 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-
 final_answer = FinalAnswerTool()
 
 # 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' 
-
+# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
 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',  # it is possible that this model may be overloaded
+    custom_role_conversions=None,
 )
 
-
 # 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_weather_forecast, get_current_time_in_timezone, image_generation_tool],  # Added all tools here
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -65,5 +116,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
 GradioUI(agent).launch()