import gradio as gr
import order_assistant  # The module you renamed to 'order_assistant'
import asyncio

# Gradio interface function
async def food_order_interface(audio_input, food_type):
    audio, order_details = await order_assistant.food_order_assistant(audio_input, food_type)
    return audio, order_details

# Create Gradio application
async def create_demo():
    demo = gr.Interface(fn=food_order_interface, 
                        inputs=[gr.Audio(type="filepath", label="Order Food"),  # Microphone input for audio file
                                gr.Dropdown(choices=["Starters", "Main Course", "Desserts"], label="Food Type")],
                        outputs=[gr.Audio(label="Order Confirmation Audio"), 
                                 gr.Textbox(label="Confirmed Order Details", interactive=False)],
                        title="AI Restaurant Voice Ordering",
                        description="Order food easily with the AI assistant's voice ordering system.")
    return demo

# Run the application
if __name__ == "__main__":
    demo = asyncio.run(create_demo())
    demo.launch()