|
|
from crewai import Crew |
|
|
from trip_agents import TripAgents, StreamToExpander |
|
|
from trip_tasks import TripTasks |
|
|
import streamlit as st |
|
|
import datetime |
|
|
import sys |
|
|
|
|
|
st.set_page_config(page_icon="✈️", layout="wide") |
|
|
|
|
|
|
|
|
def icon(emoji: str): |
|
|
"""Shows an emoji as a Notion-style page icon.""" |
|
|
st.write( |
|
|
f'<span style="font-size: 78px; line-height: 1">{emoji}</span>', |
|
|
unsafe_allow_html=True, |
|
|
) |
|
|
|
|
|
|
|
|
class TripCrew: |
|
|
|
|
|
def __init__(self, origin, cities, date_range, interests): |
|
|
self.cities = cities |
|
|
self.origin = origin |
|
|
self.interests = interests |
|
|
self.date_range = date_range |
|
|
self.output_placeholder = st.empty() |
|
|
|
|
|
def run(self): |
|
|
agents = TripAgents() |
|
|
tasks = TripTasks() |
|
|
|
|
|
city_selector_agent = agents.city_selection_agent() |
|
|
local_expert_agent = agents.local_expert() |
|
|
travel_concierge_agent = agents.travel_concierge() |
|
|
|
|
|
identify_task = tasks.identify_task( |
|
|
city_selector_agent, |
|
|
self.origin, |
|
|
self.cities, |
|
|
self.interests, |
|
|
self.date_range |
|
|
) |
|
|
|
|
|
gather_task = tasks.gather_task( |
|
|
local_expert_agent, |
|
|
self.origin, |
|
|
self.interests, |
|
|
self.date_range |
|
|
) |
|
|
|
|
|
plan_task = tasks.plan_task( |
|
|
travel_concierge_agent, |
|
|
self.origin, |
|
|
self.interests, |
|
|
self.date_range |
|
|
) |
|
|
|
|
|
crew = Crew( |
|
|
agents=[ |
|
|
city_selector_agent, local_expert_agent, travel_concierge_agent |
|
|
], |
|
|
tasks=[identify_task, gather_task, plan_task], |
|
|
verbose=True |
|
|
) |
|
|
|
|
|
result = crew.kickoff() |
|
|
self.output_placeholder.markdown(result) |
|
|
|
|
|
return result |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
icon("🏖️ VacAIgent") |
|
|
|
|
|
st.subheader("Let AI agents plan your next vacation!", |
|
|
divider="rainbow", anchor=False) |
|
|
|
|
|
import datetime |
|
|
|
|
|
today = datetime.datetime.now().date() |
|
|
next_year = today.year + 1 |
|
|
jan_16_next_year = datetime.date(next_year, 1, 10) |
|
|
|
|
|
with st.sidebar: |
|
|
st.header("👇 Enter your trip details") |
|
|
with st.form("my_form"): |
|
|
location = st.text_input( |
|
|
"Where are you currently located?", placeholder="San Mateo, CA") |
|
|
cities = st.text_input( |
|
|
"City and country are you interested in vacationing at?", placeholder="Bali, Indonesia") |
|
|
date_range = st.date_input( |
|
|
"Date range you are interested in traveling?", |
|
|
min_value=today, |
|
|
value=(today, jan_16_next_year + datetime.timedelta(days=6)), |
|
|
format="MM/DD/YYYY", |
|
|
) |
|
|
interests = st.text_area("High level interests and hobbies or extra details about your trip?", |
|
|
placeholder="2 adults who love swimming, dancing, hiking, and eating") |
|
|
|
|
|
submitted = st.form_submit_button("Submit") |
|
|
|
|
|
st.divider() |
|
|
|
|
|
|
|
|
st.sidebar.markdown( |
|
|
""" |
|
|
Credits to [**@joaomdmoura**](https://twitter.com/joaomdmoura) |
|
|
for creating **crewAI** 🚀 |
|
|
""", |
|
|
unsafe_allow_html=True |
|
|
) |
|
|
|
|
|
st.sidebar.info("Click the logo to visit GitHub repo", icon="👇") |
|
|
st.sidebar.markdown( |
|
|
""" |
|
|
<a href="https://github.com/joaomdmoura/crewAI" target="_blank"> |
|
|
<img src="https://raw.githubusercontent.com/joaomdmoura/crewAI/main/docs/crewai_logo.png" alt="CrewAI Logo" style="width:100px;"/> |
|
|
</a> |
|
|
""", |
|
|
unsafe_allow_html=True |
|
|
) |
|
|
|
|
|
|
|
|
if submitted: |
|
|
with st.status("🤖 **Agents at work...**", state="running", expanded=True) as status: |
|
|
with st.container(height=500, border=False): |
|
|
sys.stdout = StreamToExpander(st) |
|
|
trip_crew = TripCrew(location, cities, date_range, interests) |
|
|
result = trip_crew.run() |
|
|
status.update(label="✅ Trip Plan Ready!", |
|
|
state="complete", expanded=False) |
|
|
|
|
|
st.subheader("Here is your Trip Plan", anchor=False, divider="rainbow") |
|
|
st.markdown(result) |
|
|
|
