Update app.py

app.py

@@ -9,21 +9,23 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_weather(city:str)-> str: 
-    """A tool that returns the weather of a city
+def get_weather(city:str,lat: float, lng:float)-> str: 
+    """A tool that returns the weather of a city given its lattitude and longitude
     Args:
-        arg1: city name
+        city: city name
+        lat: city lattitude
+        lng: city longitude
     """
-    city_coords = {
-        "New York": (40.7128, -74.0060),
-        "London": (51.5074, -0.1278),
-        "Tokyo": (35.6895, 139.6917),
-    }
+    # city_coords = {
+    #     "New York": (40.7128, -74.0060),
+    #     "London": (51.5074, -0.1278),
+    #     "Tokyo": (35.6895, 139.6917),
+    # }
     
-    if city not in city_coords:
-        return "City not found. Try New York, London, or Tokyo."
+    # if city not in city_coords:
+    #     return "City not found. Try New York, London, or Tokyo."
 
-    lat, lon = city_coords[city]
+    # lat, lon = city_coords[city]
     url = f"https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={lon}&current_weather=true"
     
     response = requests.get(url)
@@ -72,7 +74,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, get_weather], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, search_tool, get_weather], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,