Adding a new museum scout

Adding good labels for easy readability
Improving agents and tasks

app.py

@@ -4,7 +4,7 @@ import json
 import gradio as gr
 import plotly.graph_objects as go
 import logging
-from crew import SurpriseTravelCrew, AddressSummaryCrew
+from crew import TravelCrew, AddressSummaryCrew
 
 def filter_map(text_list, lat, lon):
     fig = go.Figure(go.Scattermapbox(
@@ -43,7 +43,7 @@ def run(origin, destination, age, trip_duration, children, budget):
         'children': children,
         'budget': budget
     }
-    result = SurpriseTravelCrew().crew().kickoff(inputs=inputs)
+    result = TravelCrew().crew().kickoff(inputs=inputs)
     inputs_for_address = {
         'text': str(result)
     }
@@ -73,13 +73,14 @@ demo = gr.Interface(
     title="Plan your itinerary with the help of AI",
     description="Use this app to create a detailed itinerary on how to explore a new place. Itinerary is customized to your taste. Powered by Sambanova Cloud.",
     fn=run,
-    inputs=["text", "text", gr.Slider(value=30, minimum=15, maximum=90, step=5), 
-            gr.Slider(value=5, minimum=1, maximum=14, step=1),
-            gr.Checkbox(),
-            gr.Slider(value=100, minimum=20, maximum=1000, step=20)],
+    inputs=[gr.Textbox(label="Origin"), gr.Textbox(label="Destination"),
+            gr.Slider(label="Your age?", value=30, minimum=15, maximum=90, step=5),
+            gr.Slider(label="How many days are you travelling?", value=5, minimum=1, maximum=14, step=1),
+            gr.Checkbox(label="Check if children are travelling with you"),
+            gr.Slider(label="Total budget of trip in USD", show_label=True, value=1000, minimum=500, maximum=10000, step=500)],
     outputs=[
         gr.Textbox(label="Complete Itinerary", show_copy_button=True, autoscroll=False),
-        gr.Plot(label="Venues on a Map. Please verify with a Navigation System before going")
+        gr.Plot(label="Venues on a Map. Please verify with a Navigation System before traveling.")
     ]
 )
 demo.launch()

config/agents.yaml

@@ -14,6 +14,14 @@ restaurant_scout:
   backstory: >
     As a food lover, you know the best spots in town for a delightful culinary experience. You also have a knack for finding picturesque and entertaining locations.
 
+museum_scout:
+  role: >
+    Museum Scout
+  goal: >
+    Find well-known as well as off-the-beat museums matching the age profile. Also pay particular attention to if there are children. If children are accompanying, pick up museums which might be more interesting to children.
+  backstory: >
+    You know the best museums in town for different age groups. Older people may like quieter museums such as art museums. Young children like museums where they can get to touch, feel, and play.
+
 itinerary_compiler:
   role: >
     Itinerary Compiler

config/tasks.yaml

@@ -16,7 +16,7 @@ personalized_activity_planning_task:
 
     - how long is the trip: {trip_duration}
 
-    - budget per day in dollars: {budget}
+    - budget for entire trip in dollars: {budget}
   expected_output: >
     A list of recommended activities and events for each day of the trip.
     Each entry should include the activity name, location including a , a brief description, and why it's suitable for the traveler.
@@ -41,7 +41,31 @@ restaurant_scenic_location_scout_task:
 
     - how long is the trip: {trip_duration}
 
-    - budget per day in dollars: {budget}
+    - budget for entire trip in dollars: {budget}
+  expected_output: >
+    A list of recommended restaurants for each day of the trip.
+    Each entry should include the name, location including a detailed address, type of cuisine or activity, and a brief description and ratings.
+
+museum_scout_task:
+  description: >
+    Find popular and off-the-beat museums at {destination}. Consider the {age} of the traveller while recommending museums. Check if children are accompanying in the trip. {children}
+    If children are accompanying, find museums that might be more appealing to children.
+    Use internet search tools, museum review sites, feedback from users, and travel guides.
+    Extract the address of the museum so that the same can be displayed to the user.
+
+    Traveler's information:
+
+    - origin: {origin}
+
+    - destination: {destination}
+
+    - age of the traveler: {age}
+
+    - Are children accompanying the trip: {children}
+
+    - how long is the trip: {trip_duration}
+
+    - budget for entire trip in dollars: {budget}
   expected_output: >
     A list of recommended restaurants for each day of the trip.
     Each entry should include the name, location including a detailed address, type of cuisine or activity, and a brief description and ratings.

crew.py

@@ -10,34 +10,13 @@ from crewai_tools import SerperDevTool, ScrapeWebsiteTool
 from typing import List, Optional
 
 import os
-llm = LLM(model="sambanova/Meta-Llama-3.1-70B-Instruct")
 
-"""
-class Activity(BaseModel):
-    name: str = Field(..., description="Name of the activity")
-    location: str = Field(..., description="Location of the activity")
-    description: str = Field(..., description="Description of the activity")
-    date: str = Field(..., description="Date of the activity")
-    cousine: str = Field(..., description="Cousine of the restaurant")
-    why_its_suitable: str = Field(..., description="Why it's suitable for the traveler")
-    reviews: Optional[List[str]] = Field(..., description="List of reviews")
-    rating: Optional[float] = Field(..., description="Rating of the activity")
-
-class DayPlan(BaseModel):
-	date: str = Field(..., description="Date of the day")
-	activities: List[Activity] = Field(..., description="List of activities")
-	restaurants: List[str] = Field(..., description="List of restaurants")
-	flight: Optional[str] = Field(None, description="Flight information")
-
-class Itinerary(BaseModel):
-  name: str = Field(..., description="Name of the itinerary, something funny")
-  day_plans: List[DayPlan] = Field(..., description="List of day plans")
-  hotel: str = Field(..., description="Hotel information")
-"""
+# Change the model which you want to use below.
+llm = LLM(model="sambanova/Meta-Llama-3.1-70B-Instruct")
 
 @CrewBase
-class SurpriseTravelCrew():
-    """SurpriseTravel crew"""
+class TravelCrew():
+    """Travel crew"""
     agents_config = 'config/agents.yaml'
     tasks_config = 'config/tasks.yaml'
 
@@ -61,6 +40,16 @@ class SurpriseTravelCrew():
             allow_delegation=False,
         )
 
+    @agent
+    def museum_scout(self) -> Agent:
+        return Agent(
+            config=self.agents_config['museum_scout'],
+            llm=llm,
+            max_iter=1,
+            tools=[SerperDevTool(), ScrapeWebsiteTool()],
+            allow_delegation=False,
+        )
+
     @agent
     def itinerary_compiler(self) -> Agent:
         return Agent(
@@ -88,19 +77,27 @@ class SurpriseTravelCrew():
             agent=self.restaurant_scout()
         )
 
+    @task
+    def museum_scout_task(self) -> Task:
+        return Task(
+            config=self.tasks_config['museum_scout_task'],
+            llm=llm,
+            max_iter=1,
+            agent=self.museum_scout()
+        )
+
     @task
     def itinerary_compilation_task(self) -> Task:
         return Task(
             config=self.tasks_config['itinerary_compilation_task'],
             llm=llm,
             max_iter=1,
-            agent=self.itinerary_compiler(),
-            # output_json=Itinerary
+            agent=self.itinerary_compiler()
         )
 
     @crew
     def crew(self) -> Crew:
-        """Creates the SurpriseTravel crew"""
+        """Creates the Travel crew"""
         return Crew(
             agents=self.agents, # Automatically created by the @agent decorator
             tasks=self.tasks, # Automatically created by the @task decorator