import gradio as gr
from agent import get_itinerary

# async def respond(message, history):
#     partial_message = ""
#     async for token in get_itinerary(message, history):
#         partial_message += str(token)
#         # Use yield to stream each update
#         yield partial_message

demo = gr.ChatInterface(
    get_itinerary,
    type="messages"
)


if __name__ == "__main__":
    demo.launch()