Update app.py

app.py

@@ -17,7 +17,41 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
         arg2: the second argument
     """
     return "What magic will you build ?"
+    
+@tool
+def get_country_economics(country_code: str) -> str:
+    """Fetches basic economic data for a country using World Bank's API.
+    Args:
+        country_code: The ISO 3166-1 alpha-3 country code (e.g., 'USA' for United States, 'IND' for India).
+    """
+    try:
+        # Fetch GDP (current US$)
+        gdp_url = f"http://api.worldbank.org/v2/country/{country_code}/indicator/NY.GDP.MKTP.CD?format=json&per_page=1"
+        gdp_response = requests.get(gdp_url).json()
+        gdp_value = gdp_response[1][0]['value']
+        gdp_year = gdp_response[1][0]['date']
+
+        # Fetch population
+        pop_url = f"http://api.worldbank.org/v2/country/{country_code}/indicator/SP.POP.TOTL?format=json&per_page=1"
+        pop_response = requests.get(pop_url).json()
+        pop_value = pop_response[1][0]['value']
+        pop_year = pop_response[1][0]['date']
+
+        if gdp_value is None or pop_value is None:
+            return f"Data unavailable for country code '{country_code}'. Try a different one like 'USA', 'CHN', or 'IND'."
+
+        gdp_per_capita = gdp_value / pop_value
+        return (
+            f"Economic data for {country_code}:\n"
+            f"• Year: {gdp_year}\n"
+            f"• GDP: ${gdp_value:,.2f} USD\n"
+            f"• Population: {pop_value:,}\n"
+            f"• GDP per Capita: ${gdp_per_capita:,.2f}"
+        )
+    except Exception as e:
+        return f"Failed to fetch economic data for '{country_code}': {str(e)}"
 
+        
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.