import gradio as gr
from models.hotel_booking_model import HotelBookingAgent

# Create an instance of the chatbot model
model = HotelBookingAgent()

def chatbot_function(user_input):
    return model.generate_response(user_input)

# Gradio interface
iface = gr.Interface(
    fn=chatbot_function,
    inputs=["text", "audio"],
    outputs=["text"],
    live=True,
    title="Hotel Booking Chatbot",
    description="Ask about hotel bookings with voice or text input",
    theme="huggingface"
)

iface.launch()