Adding code to support the expected arival date as an input.

app.py

@@ -11,9 +11,11 @@ A deployment is available at https://huggingface.co/spaces/sambanovasystems/trip
 
 import json
 import logging
+import datetime
 from typing import List, Tuple
 
 import gradio as gr
+from gradio_calendar import Calendar
 import plotly.graph_objects as go
 from crew import AddressSummaryCrew, TravelCrew
 
@@ -46,6 +48,7 @@ def filter_map(text_list: List[str], lat: List[str], lon: List[str]) -> go.Figur
 def run(
     origin: str,
     destination: str,
+    arrival_date: datetime,
     age: int,
     trip_duration: int,
     interests: List[str],
@@ -59,6 +62,7 @@ def run(
     Args:
         origin: Origin city of the traveller
         destination: Destination that traveller is going to
+        arrival_date: Approximate date when the trip will begin
         age: Age profile of traveller
         interests: Specific interests of the traveller
         cuisine_preferences: Specific cuisine preferences of the traveller
@@ -69,12 +73,13 @@ def run(
         Returns a tuple containing the itinerary and map
     """
     logger.info(
-        f'Origin: {origin}, Destination: {destination}, Age: {age}, Duration: {trip_duration},'
+        f'Origin: {origin}, Destination: {destination}, Arrival Date: {arrival_date}, Age: {age}, Duration: {trip_duration},'
         f' Interests: {interests}, Cuisines: {cuisine_preferences}, Children: {children}, Daily Budget: {budget}'
     )
     inputs = {
         'origin': origin,
         'destination': destination,
+        'arrival_date': arrival_date.strftime("%m-%d-%Y"),
         'age': age,
         'trip_duration': trip_duration,
         'interests': interests,
@@ -119,6 +124,7 @@ demo = gr.Interface(
     inputs=[
         gr.Textbox(label='Where are you travelling from?'),
         gr.Textbox(label='Where are you going?'),
+        Calendar(type="datetime", label="Approximate arrival date"),
         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.CheckboxGroup(
@@ -147,7 +153,12 @@ demo = gr.Interface(
         ),
     ],
     outputs=[
-        gr.Textbox(label='Complete Personalized Itinerary of your Trip', show_label=True, show_copy_button=True, autoscroll=False),
+        gr.Textbox(
+            label='Complete Personalized Itinerary of your Trip',
+            show_label=True,
+            show_copy_button=True,
+            autoscroll=False,
+        ),
         gr.Plot(label='Venues on a Map. Please verify with a Navigation System before traveling.'),
     ],
 )

config/agents.yaml

@@ -2,7 +2,7 @@ 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.
+    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. Pay importance to the planned arrival date of the traveller, which will indicate which will indicate the season.
   backstory: >
     You are skilled at creating personalized itineraries that cater to the specific preferences and demographics of travelers.
 

config/tasks.yaml

@@ -1,6 +1,8 @@
 personalized_activity_planning_task:
   description: >
     Research and find cool things to do at {destination}.
+    Check the arrival_date and schedule activities that are seasonal around the arrival date. As an example, if the arrival data in in winter, and there are ski resorts around the destination, recommend a ski resort.
+    Also map the date to the day of the week and check if the activity you are recommending in open on that day of the week.
     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.
 
@@ -9,6 +11,8 @@ personalized_activity_planning_task:
     - origin: {origin}
 
     - destination: {destination}
+ 
+    - Approximate arrival data in month-day-year format: {arrival_date}
 
     - Approximate age of the adult travelers: {age}
 
@@ -56,6 +60,7 @@ restaurant_scenic_location_scout_task:
 interest_scout_task:
   description: >
     Run specifically for the interests of the traveller provided {interests}. As an example if the traveller has interest in museums, plan activities related to museums at the destination.
+    Also consider the approximate arrival date, and plan activities specific to the month and the season {arrival_date}
     Check if children are accompanying in the trip. {children}
     If children are accompanying, find activities that might be more appealing to children.
     Use internet search tools, review sites, feedback from users, and travel guides.
@@ -67,6 +72,8 @@ interest_scout_task:
 
     - destination: {destination}
 
+    - Approximate arrival data in month-day-year format: {arrival_date}
+
     - age of the traveler: {age}
 
     - List of interests that the traveller may have: {interests}

crew.py

@@ -8,12 +8,16 @@ from crewai_tools import ScrapeWebsiteTool, SerperDevTool
 # Currently, we use the Llama 3.1 70B model because it seems the most versatile
 llm = LLM(model='sambanova/Meta-Llama-3.1-70B-Instruct')
 
+
 @CrewBase
 class TravelCrew:
     """Crew to do travel planning"""
 
-    agents_config = 'config/agents.yaml'
-    tasks_config = 'config/tasks.yaml'
+    def __init__(self) -> None:
+        """Initialize the crew."""
+        super().__init__()
+        self.agents_config = 'config/agents.yaml'
+        self.tasks_config = 'config/tasks.yaml'
 
     @agent
     def personalized_activity_planner(self) -> Agent:
@@ -97,7 +101,6 @@ class TravelCrew:
         """
         return Task(config=self.tasks_config['interest_scout_task'], llm=llm, max_iter=1, agent=self.interest_scout())
 
-
     @task
     def restaurant_scenic_location_scout_task(self) -> Task:
         """

requirements.txt

@@ -1,3 +1,4 @@
 crewai
 crewai-tools
 plotly
+gradio-calendar