Update app.py

app.py

@@ -6,6 +6,8 @@ from smolagents import GradioUI, CodeAgent, HfApiModel
 from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
 from retriever import load_guest_dataset
 
+from tools import CountryLanguageTool
+
 # Initialize the Hugging Face model
 model = HfApiModel()
 
@@ -21,9 +23,11 @@ hub_stats_tool = HubStatsTool()
 # Load the guest dataset and initialize the guest info tool
 guest_info_tool = load_guest_dataset()
 
+country_language_tool = CountryLanguageTool()
+
 # Create Alfred with all the tools
 alfred = CodeAgent(
-    tools=[guest_info_tool, weather_info_tool, hub_stats_tool, search_tool], 
+    tools=[guest_info_tool, weather_info_tool, hub_stats_tool, search_tool, country_language_tool], 
     model=model,
     add_base_tools=True,  # Add any additional base tools
     planning_interval=3   # Enable planning every 3 steps