Update app.py

app.py

@@ -3,6 +3,7 @@ import datetime
 import requests
 import pytz
 import yaml
+import pandas as pd
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -18,6 +19,52 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def get_weather_tool(latitude:float, longitude:float)-> pd.Dataframe:
+    """
+    A tool to get weather
+    """
+    import openmeteo_requests
+    import requests_cache
+    from retry_requests import retry
+    
+    # Setup the Open-Meteo API client with cache and retry on error
+    cache_session = requests_cache.CachedSession('.cache', expire_after = 3600)
+    retry_session = retry(cache_session, retries = 5, backoff_factor = 0.2)
+    openmeteo = openmeteo_requests.Client(session = retry_session)
+    
+    # Make sure all required weather variables are listed here
+    # The order of variables in hourly or daily is important to assign them correctly below
+    url = "https://api.open-meteo.com/v1/forecast"
+    params = {
+    	"latitude": latitude,
+    	"longitude": longitude,
+    	"hourly": "temperature_2m",
+    }
+    responses = openmeteo.weather_api(url, params=params)
+    
+    # Process first location. Add a for-loop for multiple locations or weather models
+    response = responses[0]
+    print(f"Coordinates: {response.Latitude()}°N {response.Longitude()}°E")
+    print(f"Elevation: {response.Elevation()} m asl")
+    print(f"Timezone difference to GMT+0: {response.UtcOffsetSeconds()}s")
+    
+    # Process hourly data. The order of variables needs to be the same as requested.
+    hourly = response.Hourly()
+    hourly_temperature_2m = hourly.Variables(0).ValuesAsNumpy()
+    
+    hourly_data = {"date": pd.date_range(
+    	start = pd.to_datetime(hourly.Time(), unit = "s", utc = True),
+	end =  pd.to_datetime(hourly.TimeEnd(), unit = "s", utc = True),
+	freq = pd.Timedelta(seconds = hourly.Interval()),
+	inclusive = "left"
+    )}
+
+    hourly_data["temperature_2m"] = hourly_temperature_2m
+    
+    hourly_dataframe = pd.DataFrame(data = hourly_data)
+    return hourly_dataframe
+    
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -56,7 +103,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,web_search], ## add your tools here (don't remove final answer)
+    tools=[final_answer,get_weather_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,