Update app.py

app.py

@@ -8,6 +8,8 @@ from tools.final_answer import FinalAnswerTool
 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
@@ -18,6 +20,23 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+    
+
+@tool
+def travel_recommendations_for_city(city: str) -> str:
+    """A tool that provides tourist recommendations for a given city.
+    Args:
+        city: A string representing a specific city (e.g., 'Nairobi/Johannesburg').
+    """
+    search = DuckDuckGoSearchTool()
+    results = search(f"Top 5 things to do in {city}")
+    if not results:
+        return f"Couldn't find any recommendations for {city}."
+    recommendations = "\n".join([f"- {r['body']}" for r in results[:5]])
+    return f"Here are some things to do in {city}:\n{recommendations}"
+
+
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -34,6 +53,19 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+@tool
+def generate_travel_brochure_image(description: str) -> str:
+    """A tool that generates a travel brochure image based on a description.
+    Args:
+        description: A string representing text to be converted into a brochure.
+    """
+    return image_generation_tool(description)
+
+
+
 final_answer = FinalAnswerTool()
 
 # 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:
@@ -47,21 +79,23 @@ custom_role_conversions=None,
 )
 
 
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer,
+        get_current_time_in_timezone,
+        travel_recommendations_for_city,
+        generate_travel_brochure_image
+    ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
-    name=None,
-    description=None,
+    name="Travel Brochure Agent",
+    description="Provides local time, travel tips, and generates a brochure image.",
     prompt_templates=prompt_templates
 )