try to fix response issue

app.py

@@ -4,14 +4,14 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from typing import List, Dict
 
 from Gradio_UI import GradioUI
 
 @tool
-def get_country_data(country_name):
+def get_country_data(country_name) -> List[Dict]:
     """
     Fetches and prints data for a country from the REST Countries API.
-    
     Args:
         country_name (str): The common name of the country (e.g., "France").
     """
@@ -35,7 +35,7 @@ def get_country_data(country_name):
         print(f"Capital: {country['capital'][0]}")
         print(f"Region: {country['region']}")
         print(f"Population: {country['population']:,}")
-        
+        return data
     except requests.exceptions.HTTPError as errh:
         print(f"HTTP Error: {errh}")
     except requests.exceptions.ConnectionError as errc:
@@ -49,7 +49,7 @@ def get_country_data(country_name):
     except Exception as e:
         print(f"An unexpected error occurred: {e}")
 
-
+    
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.