Add fetching of coordinates by location name

app.py

@@ -3,21 +3,27 @@ import datetime
 import requests
 import pytz
 import yaml
+from geopy.geocoders import Nominatim
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
 
 from Gradio_UI import GradioUI
 
+def get_city_coordinates(city: str) ->tuple[str, str]:
+    geolocator = Nominatim(user_agent="test_agent")
+    location = geolocator.geocode(city)
+    return (str(location.latitude), str(location.longitude))
+    
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def sunrise_sunset_tool(lat:str, lng:str)-> str: #it's import to specify the return type
+def sunrise_sunset_tool(city:str)-> 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 returns sunrise and sunset times for location specified by latitude and longitude
+    """A tool that returns sunrise and sunset times for location specified by name
     Args:
-        lat: A string representing location latitude
-        lng: A string representing location longitude
+        city: A string representing location name
     """
     try:
+        lat, lng = get_city_coordinates()
         api_url = "https://api.sunrise-sunset.org/json"
         payload = {'lat': lat, 'lng': lng}
         ret = requests.get(api_url, params=payload)
@@ -29,7 +35,7 @@ def sunrise_sunset_tool(lat:str, lng:str)-> str: #it's import to specify the ret
             solar_noon = js["results"]["solar_noon"]
             time_zone = js["tzid"]
         elif ret.status_code == 200:
-            return f"Error fetching sunrise/sunset data for location lat={lat} long={lng}. Error code = {js["status"]}"
+            return f"Error fetching sunrise/sunset data for location lat={lat} long={lng}. Error code = {js['status']}"
         else:
                 return f"Error fetching sunrise/sunset data for location lat={lat} long={lng}. Error code = {ret.status_code}, response={js}"