| 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() | |