Upload app.py

This my edited version of the original app.py, I have added some tools to my agent such as get a GDP of a country, web search tool or image generation tool

app.py

@@ -9,14 +9,25 @@ 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(arg1:str, arg2: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 does nothing yet 
+def get_gdp(country_iso_code: str, year: int) -> str:
+    """Retrieves the GDP of a given country for a specified year in USD.
+    
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        country_iso_code: The ISO 3166-1 alpha-3 code of the country (e.g., "FRA" for France, "USA" for United States).
+        year: The year for which to retrieve GDP data.
+
+    Returns:
+        The GDP of the country in the given year (in US dollars) or an error message.
     """
-    return "What magic will you build ?"
+    base_url = "https://api.worldbank.org/v2/country/{}/indicator/NY.GDP.MKTP.CD"
+    response = requests.get(base_url.format(country_iso_code), params={"date": year, "format": "json"})
+
+    if response.status_code == 200:
+        data = response.json()
+        if data and len(data) > 1 and "value" in data[1][0]:
+            return f"GDP of {country_iso_code} in {year}: ${data[1][0]['value']:,} USD"
+    
+    return "GDP data not available."
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -35,6 +46,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 
 final_answer = FinalAnswerTool()
+web_search_tool = DuckDuckGoSearchTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
@@ -55,7 +67,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_gdp, image_generation_tool, web_search_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,