import gradio as gr
from agent import get_itinerary

with gr.Blocks(fill_height=True) as demo:
    gr.Markdown("""
    # Travel Itinerary Planner Agent

    Welcome to the Travel Itinerary Planner Agent! This application helps you plan your travel itineraries effortlessly. 
    Simply provide your preferences and details, and the agent will generate a personalized travel plan for you.
    """)
    gr.ChatInterface(
        get_itinerary,
        type="messages",
        cache_mode="lazy"
    )


if __name__ == "__main__":
    demo.queue(max_size=3)
    demo.launch()