Initial v0.1 version of the app

app.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python
+import sys
+import gradio as gr
+from crew import SurpriseTravelCrew
+
+
+def run(origin, destination, age, trip_duration):
+    # Replace with your inputs, it will automatically interpolate any tasks and agents information
+    inputs = {
+        'origin': origin,
+        'destination': destination,
+        'age': age,
+        'trip_duration': trip_duration 
+    }
+    result = SurpriseTravelCrew().crew().kickoff(inputs=inputs)
+    return (result)
+
+
+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",
+    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)],
+    outputs=["text"],
+)
+demo.launch(share=True)

config/agents.yaml

@@ -0,0 +1,23 @@
+personalized_activity_planner:
+  role: >
+    Activity Planner
+  goal: >
+    Research and find cool things to do at the destination, including activities and events that match the traveler's interests and age group. Create detailed notes on the activity or event.
+  backstory: >
+    You are skilled at creating personalized itineraries that cater to the specific preferences and demographics of travelers.
+
+restaurant_scout:
+  role: >
+    Restaurant Scout
+  goal: >
+    Find highly-rated restaurants and dining experiences including off-the-beat options at the destination, and recommend scenic locations and fun activities
+  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.
+
+itinerary_compiler:
+  role: >
+    Itinerary Compiler
+  goal: >
+    Compile all researched information into a comprehensive day-by-day itinerary, ensuring the integration of flights and hotel information
+  backstory: >
+    With an eye for detail, you organize all the information into a coherent and enjoyable travel plan.

config/tasks.yaml

@@ -0,0 +1,51 @@
+personalized_activity_planning_task:
+  description: >
+    Research and find cool things to do at {destination}.
+    Focus on activities and events that match the traveler's interests and age group.
+    Utilize internet search tools and recommendation engines to gather the information.
+
+
+    Traveler's information:
+
+
+    - origin: {origin}
+
+    - destination: {destination}
+
+    - age of the traveler: {age}
+
+    - how long is the trip: {trip_duration}
+  expected_output: >
+    A list of recommended activities and events for each day of the trip.
+    Each entry should include the activity name, location, a brief description, and why it's suitable for the traveler.
+    And potential reviews and ratings of the activities.
+
+restaurant_scenic_location_scout_task:
+  description: >
+    Find highly-rated restaurants and dining experiences at {destination}.
+    Recommend scenic locations and fun activities that align with the traveler's preferences.
+    Use internet search tools, restaurant review sites, and travel guides.
+    Make sure to find a variety of options to suit different tastes and budgets, and ratings for them.
+
+    Traveler's information:
+
+
+    - origin: {origin}
+
+    - destination: {destination}
+
+    - age of the traveler: {age}
+
+    - how long is the trip: {trip_duration}
+  expected_output: >
+    A list of recommended restaurants, scenic locations, and fun activities for each day of the trip.
+    Each entry should include the name, location (address), type of cuisine or activity, and a brief description and ratings.
+
+itinerary_compilation_task:
+  description: >
+    Compile all researched information into a comprehensive day-by-day itinerary for the trip to {destination}.
+    Ensure the itinerary integrates flights, hotel information, and all planned activities and dining experiences.
+    Use text formatting and document creation tools to organize the information.
+  expected_output: >
+    A detailed itinerary document, the itinerary should include a day-by-day
+    plan with activities, restaurants, and scenic locations. A day can contain multiple activities. In such a case a recommened mode of transport should be provided to help move from one activity location to another.

crew.py

@@ -0,0 +1,118 @@
+from crewai import Agent, Crew, Process, Task, LLM
+from crewai.project import CrewBase, agent, crew, task
+
+# Uncomment the following line to use an example of a custom tool
+# from surprise_travel.tools.custom_tool import MyCustomTool
+
+# Check our tools documentation for more information on how to use them
+from crewai_tools import SerperDevTool, ScrapeWebsiteTool
+#from pydantic import BaseModel, Field
+from typing import List, Optional
+import os
+
+os.environ["OPENAI_API_KEY"] = "aGlhbWl0YWJoYV9fZ21haWwuY29tOnhvbnhGMmh2MEU="
+os.environ["OPENAI_API_BASE"] = "https://api.sambanova.ai/v1"
+os.environ["OPENAI_MODEL_NAME"] = "sambanova/Meta-Llama-3.1-70B-Instruct"
+
+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")
+"""
+
+@CrewBase
+class SurpriseTravelCrew():
+    """SurpriseTravel crew"""
+    agents_config = 'config/agents.yaml'
+    tasks_config = 'config/tasks.yaml'
+
+    @agent
+    def personalized_activity_planner(self) -> Agent:
+        return Agent(
+            config=self.agents_config['personalized_activity_planner'],
+            llm=llm,
+            max_iter=1,
+            tools=[SerperDevTool(), ScrapeWebsiteTool()], # Example of custom tool, loaded at the beginning of file
+            verbose=True,
+            allow_delegation=False,
+        )
+
+    @agent
+    def restaurant_scout(self) -> Agent:
+        return Agent(
+            config=self.agents_config['restaurant_scout'],
+            llm=llm,
+            max_iter=1,
+            tools=[SerperDevTool(), ScrapeWebsiteTool()],
+            verbose=True,
+            allow_delegation=False,
+        )
+
+    @agent
+    def itinerary_compiler(self) -> Agent:
+        return Agent(
+            config=self.agents_config['itinerary_compiler'],
+            llm=llm,
+            max_iter=1,
+            tools=[SerperDevTool()],
+            verbose=True,
+            allow_delegation=False,
+        )
+
+    @task
+    def personalized_activity_planning_task(self) -> Task:
+        return Task(
+            config=self.tasks_config['personalized_activity_planning_task'],
+            llm=llm,
+            max_iter=1,
+            agent=self.personalized_activity_planner()
+        )
+
+    @task
+    def restaurant_scenic_location_scout_task(self) -> Task:
+        return Task(
+            config=self.tasks_config['restaurant_scenic_location_scout_task'],
+            llm=llm,
+            max_iter=1,
+            agent=self.restaurant_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
+        )
+
+    @crew
+    def crew(self) -> Crew:
+        """Creates the SurpriseTravel crew"""
+        return Crew(
+            agents=self.agents, # Automatically created by the @agent decorator
+            tasks=self.tasks, # Automatically created by the @task decorator
+            process=Process.sequential,
+            verbose=True,
+            # process=Process.hierarchical, # In case you want to use that instead https://docs.crewai.com/how-to/Hierarchical/
+        )

requirements.txt

@@ -0,0 +1,2 @@
+crewai
+crewai-tools