create app.py

app.py

@@ -0,0 +1,199 @@
+import streamlit as st
+from crewai import Crew, Process
+from textwrap import dedent
+from trip_agents import TripAgents
+from trip_tasks import TripTasks
+from dotenv import load_dotenv
+
+load_dotenv()
+
+class TripCrew:
+    def __init__(self, origin, cities, date_range, interests):
+        self.cities = cities
+        self.origin = origin
+        self.interests = interests
+        self.date_range = date_range
+
+    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()
+        return result
+
+# Function to initialize chat history in session state
+def initialize_chat_history():
+    if 'chat_history' not in st.session_state:
+        st.session_state['chat_history'] = []
+
+# Function to handle user input and AI response
+def handle_user_input():
+    user_input = st.session_state['user_input']
+    if user_input:
+        # Append user message to chat history
+        st.session_state['chat_history'].append(f"<div style='text-align: right;'>You: {user_input}</div>")
+        
+        # Get user input for trip planning
+        location = user_input.split(",")[0].strip()
+        cities = user_input.split(",")[1].strip()
+        date_range = user_input.split(",")[2].strip()
+        interests = user_input.split(",")[3].strip()
+        
+        # Run the trip planning agent
+        trip_crew = TripCrew(location, cities, date_range, interests)
+        result = trip_crew.run()
+        
+        # Append agent response to chat history
+        st.session_state['chat_history'].append(f"<div style='text-align: right;'>{result}</div>")
+        
+        # Clear the input field
+        st.session_state['user_input'] = ""
+
+# Initialize chat history
+initialize_chat_history()
+
+# Main page layout
+st.title("Welcome to Paris Travel Advisor!")
+
+# Sidebar
+st.sidebar.title("Chat History")
+st.sidebar.write("Recent Conversations")
+# Dummy chat history for UI demonstration
+for i in range(5):
+    st.sidebar.write(f"Conversation {i + 1}")
+st.sidebar.button("New Chat")
+st.sidebar.button("Clear Conversations")
+st.sidebar.write("My Account")
+st.sidebar.write("Logout")
+
+# Main content
+col1, col2, col3 = st.columns(3)
+
+# Custom CSS for box with shadow effect and equal heights
+box_shadow = """
+<style>
+.custom-box {
+    box-shadow: 5px 5px 10px rgba(0, 0, 0, 0.2);
+    padding: 15px;
+    border-radius: 5px;
+    background-color: #fff;
+    height: 100%;
+    display: flex;
+    flex-direction: column;
+}
+
+.custom-box h3 {
+    margin-top: 0;
+    font-size: 1.2rem;
+}
+
+.custom-box p {
+    flex-grow: 1;
+    font-size: 0.8rem;
+}
+</style>
+"""
+st.markdown(box_shadow, unsafe_allow_html=True)
+
+with col1:
+    # Plan Your Trip section with box
+    st.markdown(
+        """
+        <div class="custom-box">
+            <h3>Plan Your Trip</h3>
+            <p>Plan your itinerary with ease. Search for flights, hotels, and activities, all in one place. Book your selections and create a personalized itinerary.</p>
+        </div>
+        """,
+        unsafe_allow_html=True,
+    )
+
+with col2:
+    # Trip Inspiration section with box
+    st.markdown(
+        """
+        <div class="custom-box">
+            <h3>Trip Inspiration</h3>
+            <p>Find your dream vacation. Explore destinations, discover things to do, and get ideas for your next trip.</p>
+        </div>
+        """,
+        unsafe_allow_html=True,
+    )
+
+with col3:
+    # Travel Restrictions and Advisories section with box
+    st.markdown(
+        """
+        <div class="custom-box">
+            <h3>Travel Restrictions and Advisories</h3>
+            <p>Stay informed, travel safe! Get the latest updates on travel advisories, visa requirements, and COVID-19 restrictions for your destinations.</p>
+        </div>
+        """,
+        unsafe_allow_html=True,
+    )
+
+# Display chat history
+for message in st.session_state['chat_history']:
+    st.markdown(message, unsafe_allow_html=True)
+
+# CSS styles for input container with absolute positioning and flexbox for centering
+st.markdown(
+    """
+    <style>
+    .input-container {
+        position: absolute;
+        bottom: 0;
+        width: 100%;
+        background-color: #f1f1f1;
+        padding: 10px;
+        box-sizing: border-box;
+        display: flex; /* Added flexbox for centering */
+        justify-content: center; /* Horizontally center elements */
+    }
+    .input-container .column {
+        margin: 0 5px;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True,
+)
+
+# Input field and send button icon inside a container
+input_container = st.container()
+with input_container:
+    col1, col2 = st.columns([10, 1])
+    with col1:
+        user_input = st.text_input("", key="user_input", placeholder="Enter your trip details (location, cities, date range, interests)", help="Enter your trip details separated by commas")
+    with col2:
+        submit_button = st.button("⬆", on_click=handle_user_input)