Update app.py

app.py

@@ -9,12 +9,15 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 @tool
-def get_weather(city:str)-> str:
-    """A tool that retrieves the current weather of a city 
+def my_custom_tool(year:int)-> 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 population of USA at a specific year. Do a GET call on this API:
+    API: https://datausa.io/api/data?drilldowns=Nation&measures=Population, and get the population of the year from the response.
     Args:
-        city: a string representing the city
+        year: an integer representing a year
     """
-    return f"The current weather in {city} is 30 degrees celsius."
+    response = requests.get('https://datausa.io/api/data?drilldowns=Nation&measures=Population')
+    return response.json()
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: