Update app.py

app.py

@@ -9,14 +9,31 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(location_name:str)-> str: #it's import to specify the return type
+def my_custom_tool(city:str, state:str, country="United States")-> 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 fetches the weather for a provided location_name
+    """A tool that fetches the weather for a provided city and state. This tool only works for the united states
     Args:
         location_name: weather location
     """
     api_key = "3MfEYSjmLMlWRZCbw7VSLhaGyptxpUYP"
-    location_key = "35341"
+
+    url_city = f"http://dataservice.accuweather.com/locations/v1/cities/{country}/{state}/search"
+    params = {
+        "apikey": api_key,
+        "q": city,
+    }
+    
+    response = responses.get(url_city, params=params)
+
+    if response.status_code == 200:
+        data = response.json()
+        if data:
+            location_key = data[0]["Key"]
+        else:
+            return "No results found"
+
+    
+    #location_key = "35341"
     url = f"http://dataservice.accuweather.com/currentconditions/v1/{location_key}?apikey={api_key}"
 
     try: