Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
@@ -7,124 +7,6 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-@tool
-def get_weather_api_key() -> str:
-    """Retrieves the OpenWeatherMap API key from an environment variable.
-
-    Returns:
-        The OpenWeatherMap API key, or an error message if not found.
-    """
-    api_key = '978b53bb6550f5b81b031a823a63f5dc'
-    #api_key = os.getenv("OPENWEATHERMAP_API_KEY")
-
-    return api_key
-'''
-@tool
-def get_weather_forecast(city: str, date: str, api_key: str) -> str:
-    """Retrieves the weather forecast for a specified city and date using OpenWeatherMap API.
-
-    Args:
-        city: The name of the city.
-        date: The date in YYY-MM-DD format.
-        api_key: Your OpenWeatherMap API key.
-
-    Returns:
-        A string containing the weather forecast, or an error message.
-    """
-    try:
-        date_obj = datetime.datetime.strptime(date, "%Y-%m-%d").date()
-        today = datetime.date.today()
-        if date_obj < today:
-          return "Cannot get weather from past dates."
-
-        # Convert date to Unix timestamp for OpenWeatherMap API
-        dt = int(datetime.datetime.combine(date_obj, datetime.datetime.min.time()).timestamp())
-
-        # Get latitude and longitude of the city using geocoding API
-        geocoding_url = f"http://api.openweathermap.org/geo/1.0/direct?q={city}&limit=1&appid={api_key}"
-        geocoding_response = requests.get(geocoding_url)
-        geocoding_data = geocoding_response.json()
-
-        if not geocoding_data:
-            return f"Could not find coordinates for {city}."
-
-        lat = geocoding_data[0]["lat"]
-        lon = geocoding_data[0]["lon"]
-
-        # Get historical weather data using One Call API
-        weather_url = f"https://api.openweathermap.org/data/3.0/onecall/day_summary?lat={lat}&lon={lon}&date={dt}&appid={api_key}"
-        weather_response = requests.get(weather_url)
-        weather_data = weather_response.json()
-
-        if "weather" in weather_data:
-            weather_description = weather_data["weather"][0]["description"]
-            temp_max = weather_data["temp"]["max"]
-            return f"{weather_description}, with a high of {temp_max} degrees Celsius."
-        else:
-            return "Weather data not available for the specified date."
-
-    except requests.exceptions.RequestException as e:
-        return f"Error connecting to OpenWeatherMap API: {e}"
-    except ValueError:
-        return "Invalid date format. Please use YYYY-MM-DD."
-    except KeyError:
-        return "Weather data not available or API key is invalid"
-'''
-
-@tool 
-def get_weather_forecast(city: str, date: str, api_key: str) -> str:
-    """Retrieves the weather forecast for a specified city and date using OpenWeatherMap API.
-
-    Args:
-        city: The name of the city.
-        date: The date in YYYY-MM-DD format.
-        api_key: Your OpenWeatherMap API key.
-
-    Returns:
-        A string containing the weather forecast, or an error message.
-    """
-    try:
-        date_obj = datetime.datetime.strptime(date, "%Y-%m-%d").date()
-        today = datetime.date.today()
-        if date_obj < today:
-            return "Cannot get weather from past dates."
-
-        # Convert date to Unix timestamp for OpenWeatherMap API
-        dt = int(datetime.datetime.combine(date_obj, datetime.datetime.min.time()).timestamp())
-
-        # Get latitude and longitude of the city using geocoding API
-        geocoding_url = f"http://api.openweathermap.org/geo/1.0/direct?q={city}&limit=1&appid={api_key}"
-        geocoding_response = requests.get(geocoding_url)
-        geocoding_data = geocoding_response.json()
-
-        if not geocoding_data:
-            return f"Could not find coordinates for {city}."
-
-        lat = geocoding_data[0]["lat"]
-        lon = geocoding_data[0]["lon"]
-
-        # Get weather data using One Call API day_summary endpoint
-        weather_url = f"https://api.openweathermap.org/data/3.0/onecall/day_summary?lat={lat}&lon={lon}&date={dt}&appid={api_key}"
-        weather_response = requests.get(weather_url)
-        weather_data = weather_response.json()
-
-        if "weather" in weather_data and weather_data["weather"]:
-            weather_description = weather_data["weather"][0]["description"]
-            temp_max = weather_data["temp"]["max"]
-            return f"{weather_description}, with a high of {temp_max} degrees Celsius."
-        elif "message" in weather_data:
-            return f"OpenWeatherMap API Error: {weather_data['message']}"
-        else:
-            return "Weather data not available for the specified date."
-
-    except requests.exceptions.RequestException as e:
-        return f"Error connecting to OpenWeatherMap API: {e}"
-    except ValueError:
-        return "Invalid date format. Please use YYYY-MM-DD."
-    except KeyError:
-        return "Weather data not available or API key is invalid"
-
-
 @tool
 def calculate_days_until_date(future_date: str) -> int:
     """Calculates the number of days until a specified future date.
@@ -156,18 +38,17 @@ def generate_travel_plan(weather_forecast: str, days_until_date: int, city: str)
     elif "sunny" in weather_forecast.lower():
         return f"Enjoy the sunshine in {city}! You are travelling in {days_until_date} days."
     else:
-        return f"Travel plan for {city} in {days_until_date} days. Weather is unknown."
-    
+        return f"Travel plan for {city} in {days_until_date} days. Weather is {weather_forecast}."
+
 @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:
@@ -176,34 +57,33 @@ def get_current_time_in_timezone(timezone: str) -> str:
 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='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
-#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='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
+    # 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)
+search_tool = DuckDuckGoSearchTool()
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-    
+
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,
-          image_generation_tool,
-          get_weather_api_key,
-          get_weather_forecast,
-          calculate_days_until_date,
-          generate_travel_plan,
-          get_current_time_in_timezone,
-
- ], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer,
+        image_generation_tool,
+        search_tool, #replace get_weather_forecast and get_weather_api_key
+        calculate_days_until_date,
+        generate_travel_plan,
+        get_current_time_in_timezone,
+    ],  ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -213,5 +93,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
 GradioUI(agent).launch()