Update app.py

app.py

@@ -1,24 +1,21 @@
-# app.py
 import gradio as gr
+from models.hotel_booking_model import HotelBookingAgent
 
-def chatbot_response(message):
-    # Simple response logic
-    responses = {
-        "book a hotel": "Sure! What date and location?",
-        "check availability": "Let me check the availability...",
-        "hotel pricing": "Here are the pricing options...",
-    }
-    return responses.get(message.lower(), "Sorry, I didn't understand that.")
+# Create an instance of the chatbot model
+model = HotelBookingAgent()
 
-with gr.Blocks() as chatbot:
-    gr.Markdown("# Hotel Booking Chatbot")
-    with gr.Row():
-        message_input = gr.Textbox(label="Your Message")
-    with gr.Row():
-        submit_button = gr.Button("Send")
-    with gr.Row():
-        output = gr.Textbox(label="Chatbot Response")
-    
-    submit_button.click(chatbot_response, inputs=message_input, outputs=output)
+def chatbot_function(user_input):
+    return model.generate_response(user_input)
 
-chatbot.launch()
+# 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()