app.py

import streamlit as st
from datetime import datetime, timedelta, date
from config import load_secrets
from agents.agent import Agent  

# Load the API secrets at the start of your app
secrets = load_secrets()

 # Assuming you only need the OPENAI_API_KEY for this test
agent = Agent(openai_api_key=secrets["OPENAI_API_KEY"], debug=True)

# Then use the secrets as needed, for example
openai_api_key = secrets["OPENAI_API_KEY"]
google_maps_api_key = secrets["GOOGLE_MAPS_API_KEY"]
google_palm_api_key = secrets["GOOGLE_PALM_API_KEY"]

# Title of the application
st.title('Smart Travel Itinerary Suggester')

# Update this function to be used as an on_change callback for the start_date
def update_end_date():
    # If end_date is before start_date, set it to one day after start_date
    if 'end_date' in st.session_state and st.session_state['end_date'] <= st.session_state['start_date']:
        st.session_state['end_date'] = st.session_state['start_date'] + timedelta(days=1)

# Sidebar for input
with st.sidebar:
    st.header('Enter Your Travel Details')

    # Start location input
    start_location = st.text_input('From', help='Enter the starting location for your trip.')

    # End location input
    end_location = st.text_input('To', help='Enter the final location for your trip.')

    # Create the start date input widget
    start_date = st.date_input(
        'Start Date',
        min_value=datetime.today().date(),
        key='start_date',
        on_change=update_end_date   # This will call update_end_date whenever start_date changes
    )

    # Create the end date input widget
    end_date = st.date_input(
        'End Date',
        min_value=st.session_state.get('start_date', datetime.today().date()) + timedelta(days=1),
        value=st.session_state.get('end_date', st.session_state.get('start_date', datetime.today().date()) + timedelta(days=1)),
        key='end_date',
        help='End Date must be after the Start Date.'
    )

    # Preferences multiselect
    attractions = st.multiselect(
        'What attractions are you interested in?',
        ['Museums', 'Parks', 'Beaches', 'Historical Sites', 'Art Galleries', 'Famous', 'Carnivals'],
        help='Select multiple options by holding down the Ctrl (Cmd on Mac) button while clicking.'
    )

    # Budget slider
    budget = st.slider(
        'Select your travel budget range',
        min_value=100, max_value=5000, value=(500, 1500),
        help='Drag the slider to indicate your minimum and maximum travel budget.'
    )

    # Transportation selectbox
    transportation = st.selectbox(
        'Preferred mode of transportation',
        ['Public Transport', 'Rental Car', 'Bike', 'Walk'],
        help='Select how you plan to get around at your destination.'
    )

    # Accommodation radio buttons
    accommodation = st.radio(
        'Choose accommodation type',
        ['Hotel', 'Hostel', 'Airbnb', 'Apartment'],
        help='Select where you prefer to stay during your trip.'
    )

    # Daily itinerary planner
    schedule = st.selectbox(
        'How packed do you want your daily schedule to be?',
        ['Packed', 'Balanced', 'Relaxed', 'Very Relaxed'],
        help='Choose how many activities you would like to do each day.'
    )

    # Submit button
    submit = st.button('Generate Itinerary')

# Main page layout
st.header('Your Itinerary')

# Create a loading spinner
loading_spinner = st.spinner("Generating itinerary...")

# Read custom CSS from the external file
with open("styles.css", "r") as css_file:
    custom_css = css_file.read()


if submit:
    missing_fields = []

    if not start_location:
        missing_fields.append("Start Location")

    if not end_location:
        missing_fields.append("End Location")

    if not attractions:
        missing_fields.append("Attractions")

    if not start_date:
        missing_fields.append("Start Date")

    if not end_date:
        missing_fields.append("End Date")

    if not accommodation:
        missing_fields.append("Accommodation")

    if not schedule:
        missing_fields.append("Schedule")

    if not transportation:
        missing_fields.append("Transportation")

    if not missing_fields:
        user_details = {
            'start_location': start_location,
            'end_location': end_location,
            'start_date': start_date,
            'end_date': end_date,
            'attractions': attractions,
            'budget': budget,
            'transportation': transportation,
            'accommodation': accommodation,
            'schedule': schedule
        }

        # Display loading spinner while generating the itinerary
        with loading_spinner:

            # Apply custom CSS to the spinner only
            st.markdown(f"<style>{custom_css}</style>", unsafe_allow_html=True)
            
            # Call the generate_itinerary method (without artificial delay)
            itinerary_result = agent.generate_itinerary(user_details)

            if itinerary_result and itinerary_result["itinerary_suggestion"]:

                # Display the itinerary and other information (same as before)
                st.subheader("Itinerary Suggestion")
                st.write(itinerary_result["itinerary_suggestion"])
                st.write("List of Places:", itinerary_result["list_of_places"])
                st.write("Validation Dict:", itinerary_result["validation_dict"])

            elif itinerary_result and itinerary_result["itinerary_suggestion"] is None:

                # Display a message for unrealistic edge cases (same as before)
                st.markdown(
                    """<div style='background-color:#ffebeb; padding:10px; border-radius:5px;'>
                    <p style='color:#ff0000; font-weight:bold;'>Error: The travel plan is not valid.</p>
                    <p>Please review your input and try again.</p>
                    </div>""",
                    unsafe_allow_html=True
                )
            else:

                st.markdown(
                    """<div style='background-color:#ffebeb; padding:10px; border-radius:5px;'>
                    <p style='color:#ff0000; font-weight:bold;'>An error occurred while generating the itinerary.</p>
                    <p>Please try again later or contact support.</p>
                    </div>""",
                    unsafe_allow_html=True
                )

        note = """Here is your personalized travel itinerary covering all the major locations you want to visit. 
        This map gives you a general overview of your travel route from start to finish. 
        For daily plans, please review the sequence of waypoints to ensure the best experience, 
        as the route may vary based on daily activities and traffic conditions."""
    else:
        st.error(f'Please fill out the following required fields: {", ".join(missing_fields)}')