import gradio as gr
import pandas as pd

# Function to load the menu data
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):
    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

    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 style=\"background-color: #28a745; color: white; border: none; padding: 8px 15px; font-size: 14px; border-radius: 5px; cursor: pointer;\" onclick=\"openModal('{item['Dish Name']}', '{item['Image URL']}', '{item['Description']}', '{item['Price ($)']}')\">Add</button>
            </div>
        </div>
        """
    return html_content

# JavaScript for modal and cart behavior
modal_and_cart_js = """
<script>
    let cart = [];
    const extrasPrices = {
        "Extra Raitha 4oz": 1,
        "Extra Raitha 8oz": 2,
        "Extra Salan 4oz": 1,
        "Extra Salan 8oz": 2,
        "Extra Onion": 1,
        "Extra Onion & Lemon": 2,
        "Extra Fried Onion 4oz": 2
    };
    let finalized = false;

    function openModal(name, image, description, price) {
        if (finalized) {
            alert("You cannot add more items after finalizing your order.");
            return;
        }
        document.getElementById('modal').style.display = 'block';
        document.getElementById('modal-image').src = image;
        document.getElementById('modal-name').innerText = name;
        document.getElementById('modal-description').innerText = description;
        document.getElementById('modal-price').innerText = price;
        // Reset spice levels and extras
        const spiceLevelInputs = document.querySelectorAll('input[name="spice-level"]');
        spiceLevelInputs.forEach(input => input.checked = false);
        const extrasInputs = document.querySelectorAll('input[name="biryani-extra"]');
        extrasInputs.forEach(input => input.checked = false);
        document.getElementById('quantity').value = 1;
        document.getElementById('special-instructions').value = '';
    }

    function closeModal() {
        document.getElementById('modal').style.display = 'none';
    }

    function addToCart() {
        if (finalized) {
            alert("You cannot add more items after finalizing your order.");
            return;
        }
        const name = document.getElementById('modal-name').innerText;
        const price = parseFloat(document.getElementById('modal-price').innerText.replace('$', ''));
        const spiceLevel = document.querySelector('input[name="spice-level"]:checked')?.value || "Not Selected";
        const quantity = parseInt(document.getElementById('quantity').value) || 1;
        const instructions = document.getElementById('special-instructions').value;
        const extras = Array.from(document.querySelectorAll('input[name="biryani-extra"]:checked')).map(extra => extra.value);
        const extrasCost = extras.reduce((sum, extra) => sum + (extrasPrices[extra] || 0), 0);
        const itemTotal = (price + extrasCost) * quantity;
        const cartItem = { name, price, spiceLevel, quantity, instructions, extras, itemTotal };
        cart.push(cartItem);
        alert(`${name} added to cart!`);
        updateCartDisplay();
        closeModal();
    }

    function updateCartDisplay() {
        let totalBill = 0;
        let cartHTML = "<h3>Your Cart:</h3><ul>";
        cart.forEach((item, index) => {
            totalBill += item.itemTotal;
            const extras = item.extras.join(', ');
            cartHTML += `<li><input type='checkbox' id='cart-item-${index}' value='${index}' checked onclick='removeItem(${index})'> ${item.name} (x${item.quantity}, Spice: ${item.spiceLevel}, Extras: ${extras}, Instructions: ${item.instructions}) - $${item.itemTotal.toFixed(2)}</li>`;
        });
        cartHTML += `</ul><p><strong>Total Bill: $${totalBill.toFixed(2)}</strong></p>`;
        document.getElementById('floating-cart').innerHTML = cartHTML;
    }

    function removeItem(index) {
        cart.splice(index, 1);
        updateCartDisplay();
    }

    function showSummary() {
        if (cart.length === 0) {
            alert("Your cart is empty. Please add items before viewing the summary.");
            return;
        }
        finalized = true;
        let summaryHTML = "<h3>Order Summary:</h3><ul>";
        let totalBill = 0;
        cart.forEach(item => {
            totalBill += item.itemTotal;
            const extras = item.extras.join(', ');
            summaryHTML += `<li>${item.name} (x${item.quantity}, Spice: ${item.spiceLevel}, Extras: ${extras}, Instructions: ${item.instructions}) - $${item.itemTotal.toFixed(2)}</li>`;
        });
        summaryHTML += `</ul><p><strong>Total Bill: $${totalBill.toFixed(2)}</strong></p>`;
        document.getElementById('summary-container').innerHTML = summaryHTML;
        document.getElementById('menu-container').style.display = 'none';
        document.getElementById('cart-container').style.display = 'none';
        document.getElementById('summary-container').style.display = 'block';
    }
</script>
"""

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

        # Menu container
        menu_container = gr.Group(visible=True, elem_id="menu-container")
        with menu_container:
            # 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"))

        # Cart container
        cart_container = gr.Group(visible=True, elem_id="cart-container")
        with cart_container:
            # Floating cart display
            cart_output = gr.HTML(value="Your cart is empty.", elem_id="floating-cart")

        # Summary container
        summary_container = gr.Group(visible=False, elem_id="summary-container")
        with summary_container:
            summary_output = gr.HTML(value="", elem_id="summary-container")

        # Finalize Order Button
        finalize_button = gr.Button("View Summary")

        # Finalize order button click event
        finalize_button.click(lambda: "", inputs=[], outputs=[], _js="showSummary")

        # Update menu dynamically based on preference
        selected_preference.change(filter_menu, inputs=[selected_preference], outputs=[menu_output])

        # Layout
        gr.Row([menu_container])
        gr.Row([cart_container])
        gr.Row([finalize_button])
        gr.Row([summary_container])
        gr.HTML(modal_and_cart_js)

    return demo

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