import gradio as gr
import pandas as pd

# Function to load the 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}")

# Function to filter menu items based on preference
def filter_menu(preference):
    # Load menu data
    menu_data = load_menu()

    # Define filter conditions
    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:  # Default to "All"
        filtered_data = menu_data

    # Generate HTML for the menu
    html_content = ""
    for _, item in filtered_data.iterrows():
        html_content += f"""
        <div style="display: flex; align-items: center; border: 1px solid #ddd; border-radius: 8px; padding: 15px; margin-bottom: 10px; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);">
            <div style="flex: 1; margin-right: 15px;">
                <h3 style="margin: 0; font-size: 18px;">{item['Dish Name']}</h3>
                <p style="margin: 5px 0; font-size: 16px; color: #888;">${item['Price ($)']}</p>
                <p style="margin: 5px 0; font-size: 14px; color: #555;">{item['Description']}</p>
            </div>
            <div style="flex-shrink: 0; text-align: center;">
                <img src="{item['Image URL']}" alt="{item['Dish Name']}" style="width: 100px; height: 100px; border-radius: 8px; object-fit: cover; margin-bottom: 10px;">
                <button value="{item['Dish Name']}" style="background-color: #28a745; color: white; border: none; padding: 8px 15px; font-size: 14px; border-radius: 5px; cursor: pointer;" 
                        onclick="return '{item['Dish Name']}'">Add</button>
            </div>
        </div>
        """
    return html_content

# Function to render detailed view of a dish
def render_dish_details(dish_name):
    menu_data = load_menu()
    dish = menu_data[menu_data["Dish Name"] == dish_name].iloc[0]
    return (
        dish["Image URL"],
        f"## {dish_name}",
        dish["Description"],
        f"**Price:** ${dish['Price ($)']}"
    )

# Gradio app definition
def app():
    with gr.Blocks(title="Dynamic Menu with Filters") as demo:
        gr.Markdown("## Dynamic Menu with Preferences")

        # Radio button for selecting preference
        selected_preference = gr.Radio(
            choices=["All", "Vegetarian", "Halal/Non-Veg", "Guilt-Free"],
            value="All",
            label="Choose a Preference",
        )

        # Output area for menu items
        menu_output = gr.HTML(value=filter_menu("All"))

        # Detailed view
        detailed_view = gr.Column(visible=False)
        dish_image = gr.Image()
        dish_name = gr.Markdown()
        dish_description = gr.Markdown()
        dish_price = gr.Markdown()
        spice_level = gr.Radio(
            choices=["Mild", "Medium", "Spicy"],
            label="Choose Spice Level",
        )
        extras = gr.CheckboxGroup(
            choices=["Extra Raita", "Extra Salan", "Extra Fried Onion"],
            label="Choose Extras",
        )
        quantity = gr.Number(label="Quantity", value=1, interactive=True)
        special_instructions = gr.Textbox(
            label="Special Instructions",
            placeholder="Add any requests here",
        )
        add_to_bag_button = gr.Button("Add to Bag")
        back_button = gr.Button("Back to Menu")

        # Cart
        cart_state = gr.State([])
        cart_output = gr.HTML(value="Your cart is empty.")

        # Define interactivity
        def update_menu(preference):
            return filter_menu(preference)

        def show_dish_details(dish_name):
            img, name, desc, price = render_dish_details(dish_name)
            return gr.update(visible=False), gr.update(visible=True), img, name, desc, price

        def add_to_cart(dish_name, spice_level, extras, quantity, instructions, cart):
            cart.append({
                "name": dish_name,
                "spice_level": spice_level,
                "extras": extras,
                "quantity": quantity,
                "instructions": instructions
            })
            cart_html = "<br>".join(
                [f"{item['quantity']}x {item['name']} - {item['spice_level']} (Extras: {', '.join(item['extras'])})" for item in cart]
            )
            return cart, cart_html

        # Event handlers
        selected_preference.change(
            update_menu,
            inputs=[selected_preference],
            outputs=[menu_output],
        )

        menu_output.change(
            show_dish_details,
            inputs=[menu_output],
            outputs=[menu_output, detailed_view, dish_image, dish_name, dish_description, dish_price],
        )

        add_to_bag_button.click(
            add_to_cart,
            inputs=[dish_name, spice_level, extras, quantity, special_instructions, cart_state],
            outputs=[cart_state, cart_output],
        )

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

        # Layout
        with gr.Row():
            menu_output

        with detailed_view:
            dish_image
            dish_name
            dish_description
            dish_price
            spice_level
            extras
            quantity
            special_instructions
            add_to_bag_button
            back_button

        with gr.Row():
            cart_output

    return demo


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