import gradio as gr
import pandas as pd

# Load menu data from Excel
def load_menu():
    menu_file = "menu.xlsx"  # Ensure this file exists in the same directory
    try:
        return pd.read_excel(menu_file)
    except Exception as e:
        raise ValueError(f"Error loading menu file: {e}")

# Main menu view
def render_menu(preference):
    menu_data = load_menu()

    if preference == "Halal/Non-Veg":
        filtered_data = menu_data[menu_data["Ingredients"].str.contains("Chicken|Mutton|Fish|Prawns|Goat", case=False, na=False)]
    elif preference == "Vegetarian":
        filtered_data = menu_data[~menu_data["Ingredients"].str.contains("Chicken|Mutton|Fish|Prawns|Goat", case=False, na=False)]
    elif preference == "Guilt-Free":
        filtered_data = menu_data[menu_data["Description"].str.contains(r"Fat: ([0-9]|10)g", case=False, na=False)]
    else:
        filtered_data = menu_data

    menu_html = ""
    for _, item in filtered_data.iterrows():
        menu_html += f"""
        <div style="display: flex; align-items: center; border: 1px solid #ddd; border-radius: 8px; padding: 15px; margin-bottom: 10px;">
            <div style="flex: 1; margin-right: 15px;">
                <h3 style="margin: 0;">{item['Dish Name']}</h3>
                <p style="margin: 5px 0;">${item['Price ($)']}</p>
                <p>{item['Description']}</p>
            </div>
            <div>
                <img src="{item['Image URL']}" alt="{item['Dish Name']}" style="width: 100px; height: 100px; border-radius: 8px; object-fit: cover;">
                <button style="background-color: #28a745; color: white; padding: 8px 15px; border: none; cursor: pointer;" 
                        onclick="return '{item['Dish Name']}'">View Details</button>
            </div>
        </div>
        """
    return menu_html

# Dish details view
def render_dish_details(dish_name):
    menu_data = load_menu()
    dish = menu_data[menu_data["Dish Name"] == dish_name].iloc[0]
    return {
        "image": dish["Image URL"],
        "name": dish_name,
        "description": dish["Description"],
        "price": dish["Price ($)"]
    }

# Add to cart
def add_to_cart(dish_name, spice_level, extras, quantity, special_instructions, cart):
    cart.append({
        "name": dish_name,
        "spice_level": spice_level,
        "extras": extras,
        "quantity": quantity,
        "instructions": special_instructions
    })
    return cart, f"Added {dish_name} to cart!"

# Gradio app
def app():
    with gr.Blocks() as demo:
        cart_state = gr.State([])

        # Menu page
        menu_html = gr.HTML(render_menu("All"))
        detailed_view = gr.Column(visible=False)
        cart_view = gr.Column(visible=False)

        # Detailed view inputs
        spice_level = gr.Dropdown(choices=["Mild", "Medium", "Spicy"], label="Spice Level")
        extras = gr.CheckboxGroup(choices=["Extra Raita", "Extra Salan", "Extra Onion"], label="Extras")
        quantity = gr.Number(value=1, label="Quantity")
        special_instructions = gr.Textbox(placeholder="Add instructions", label="Special Instructions")
        add_button = gr.Button("Add to Cart")
        back_button = gr.Button("Back to Menu")

        # Cart view
        cart_html = gr.HTML(value="Your cart is empty.")

        # Switch to detailed view
        def show_dish_details(dish_name):
            details = render_dish_details(dish_name)
            return gr.update(visible=False), gr.update(visible=True), details["image"], details["name"], details["description"], details["price"]

        # Add to cart
        def handle_add_to_cart(dish_name, spice_level, extras, quantity, instructions, cart):
            return add_to_cart(dish_name, spice_level, extras, quantity, instructions, cart)

        # Navigation
        menu_html.change(show_dish_details, inputs=["dish_name"], outputs=[menu_html, detailed_view])

        add_button.click(handle_add_to_cart, inputs=[spice_level, extras, quantity, special_instructions, cart_state], outputs=[cart_state, cart_html])

        back_button.click(lambda: (gr.update(visible=True), gr.update(visible=False)), outputs=[menu_html, detailed_view])

        return demo

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