add a tour guide agent

agents/tour_guide_agent.py

@@ -0,0 +1,58 @@
+# agents/tour_guide_agent.py
+from smolagents import CodeAgent
+from typing import Dict, List, Optional, Any
+
+class TourGuideAgent(CodeAgent):
+    """
+    The Tour Guide Agent specializes in finding popular tourist destinations
+    for a given location using web search.
+    """
+    
+    def __init__(
+        self,
+        model,
+        tools=None,
+        managed_agents=None,
+        prompt_templates=None,
+        planning_interval=None,
+        max_steps=3,
+        verbosity_level=1,
+        name="Tour Guide Agent",
+        description="Recommends popular tourist destinations in a given location",
+        **kwargs
+    ):
+        super().__init__(
+            model=model,
+            tools=tools,
+            managed_agents=managed_agents,
+            prompt_templates=prompt_templates,
+            planning_interval=planning_interval,
+            max_steps=max_steps,
+            verbosity_level=verbosity_level,
+            name=name,
+            description=description,
+            **kwargs
+        )
+        
+        # Add specialized agent prompt
+        self.system_prompt_extension = """
+        You are the Tour Guide Agent for Journi, an AI travel companion system.
+        Your expertise is in finding popular tourist attractions and destinations.
+        
+        When given a task from the Coordinator Agent, you should:
+        1. Focus on finding the most popular tourist destinations for the location
+        2. Use web_search to get up-to-date information about tourist spots
+        3. Return a concise list of the top 5-7 attractions with 1-2 sentence descriptions
+        4. Format your response as a bulleted list for easy reading
+        
+        Only focus on actual tourist attractions and sights - not hotels, restaurants, or general travel advice.
+        Provide a clear, organized list of the must-see places for travelers.
+        """
+        
+        if prompt_templates and "system_prompt" in prompt_templates:
+            prompt_templates["system_prompt"] += self.system_prompt_extension
+        else:
+            if not prompt_templates:
+                prompt_templates = {}
+            prompt_templates["system_prompt"] = self.system_prompt_extension
+            self.prompt_templates = prompt_templates

app.py

@@ -4,11 +4,14 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from tools.web_search import DuckDuckGoSearchTool
-from tools.visit_webpage import VisitWebpageTool
 from Gradio_UI import GradioUI
 from smolagents.agent_types import AgentImage
 
+# Import tools
+from tools.final_answer import FinalAnswerTool
+from tools.web_search import DuckDuckGoSearchTool
+from agents.tour_guide_agent import TourGuideAgent
+
 # 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
@@ -58,28 +61,42 @@ image_generation_tool = Tool.from_space(
     description="Generate an image from a text prompt. The prompt should be detailed to create high-quality images."
 )
 
+# Create the web search tool for the tour guide agent
+web_search_tool = DuckDuckGoSearchTool(max_results=5)
+
+# Initialize the tour guide agent
+tour_guide_agent = TourGuideAgent(
+    model=model,
+    tools=[web_search_tool],
+    max_steps=3,
+    verbosity_level=1
+)
 
 # Final answer tool
 final_answer = FinalAnswerTool()
 
-with open("prompts_journi.ymal", 'r') as stream:
+with open("prompts_mas.ymal", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
-agent = CodeAgent(
+# Create the coordinator agent
+coordinator_agent = CodeAgent(
     model=model,
     tools=[
         final_answer,
-        image_generation_tool,
-        get_current_time_in_timezone
+        image_generation_tool
     ],
+    managed_agents={
+        'tour_guide_agent': tour_guide_agent
+    },
     max_steps=6,
     verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
+    name="Journi",
+    description="Your AI travel companion",
     prompt_templates=prompt_templates
 )
 
-
-GradioUI(agent).launch()
+# Launch the UI
+if __name__ == "__main__":
+    print("✈️ Launching Journi - Your AI Travel Companion")
+    print("Ask about any destination to see images and get tourist recommendations!")
+    GradioUI(coordinator_agent).launch()

prompts_mas.ymal

@@ -0,0 +1,38 @@
+"system_prompt": |-
+  You are Journi, a coordinator agent that helps travelers learn about destinations.
+  
+  When a user asks about a destination, ALWAYS follow these exact steps in order:
+  
+  STEP 1: Generate an image of the destination
+  ```python
+  # Generate destination image first
+  destination = "Paris"  # Extract the exact destination from user query
+  destination_image = image_generator(prompt=f"Beautiful travel photo of {destination}, showing iconic landmarks and scenery, high-quality professional photography")
+  print(f"Generated image of {destination}")
+  ```
+  
+  STEP 2: Get tourist recommendations from the tour guide agent
+  ```python
+  # Get tourist recommendations
+  tour_info = tour_guide_agent(task=f"What are the most popular tourist attractions in {destination}?")
+  print(f"Found tourist information for {destination}")
+  ```
+  
+  STEP 3: Combine everything into a final answer
+  ```python
+  # Create comprehensive response with both image and information
+  comprehensive_answer = f'''
+  {destination_image}
+  
+  # Welcome to {destination}!
+  
+  ## Top Tourist Attractions:
+  {tour_info}
+  
+  Enjoy your trip to {destination}!
+  '''
+  
+  final_answer(comprehensive_answer)
+  ```
+  
+  This sequence must be followed for all destination queries. Always generate the image first, then get tourist information, then create the final answer.