import gradio as gr
import pandas as pd
import json

# 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['Price ($)']}', '{item['Description']}')\">Add</button>
            </div>
        </div>
        """
    return html_content

# Function to finalize the order and display the final page
def finalize_order(cart_json):
    cart = json.loads(cart_json)
    if not cart:
        return "<p style='color: red; font-size: 18px;'>Your cart is empty. Please add items before finalizing.</p>"

    total_bill = 0
    final_order_html = "<h3>Your Final Order:</h3><ul>"
    for item in cart:
        total_bill += item['itemTotal']
        extras = ', '.join(item['extras']) if item['extras'] else 'None'
        final_order_html += f"<li>{item['name']} (x{item['quantity']}, Spice: {item['spiceLevel']}, Extras: {extras}, Instructions: {item['instructions']}) - ${item['itemTotal']:.2f}</li>"
    final_order_html += f"</ul><p><strong>Total Bill: ${total_bill:.2f}</strong></p>"
    final_order_html += "<p style='color: green; font-size: 18px;'>Your final order has been placed. Thank you!</p>"
    return final_order_html

# 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
    };
    function openModal(name, price, description) {
        document.getElementById('modal').style.display = 'block';
        document.getElementById('modal-name').innerText = name;
        document.getElementById('modal-price').innerText = price;
        document.getElementById('modal-description').innerText = description;
        document.getElementById('modal-image').src = "";

        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() {
        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);
        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>${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;
        document.getElementById('cart-data').value = JSON.stringify(cart);
    }
    function finalizeCart() {
        document.getElementById('cart-data').value = JSON.stringify(cart);
    }
</script>
"""

# Gradio app definition
def app():
    with gr.Blocks() 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"))

        # Floating cart display
        cart_output = gr.HTML(value="Your cart is empty.", elem_id="floating-cart")

        # Hidden input to store cart data
        cart_data = gr.Textbox(visible=False, elem_id="cart-data")

        # Final order display
        final_order_output = gr.HTML(value="", elem_id="final-page")

        # Modal window for item customization
        modal_window = gr.HTML("""
        <div id="modal" style="display: none; position: fixed; top: 40%; left: 50%; transform: translate(-50%, -40%); width: 50%; background: white; border-radius: 8px; box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2); padding: 20px; z-index: 1000;">
            <div style="text-align: right;">
                <button onclick="closeModal()" style="background: none; border: none; font-size: 18px; cursor: pointer;">&times;</button>
            </div>
            <img id="modal-image" style="width: 100%; height: auto; border-radius: 8px; margin-bottom: 20px;" />
            <h2 id="modal-name"></h2>
            <p id="modal-description"></p>
            <p id="modal-price"></p>
            <!-- Spice Levels -->
            <label for="spice-level">Choose a Spice Level (Required):</label>
            <div id="spice-level-options" style="display: flex; flex-wrap: wrap; gap: 10px; margin: 10px 0;">
                <label><input type="radio" name="spice-level" value="American Mild" /> American Mild</label>
                <label><input type="radio" name="spice-level" value="American Medium" /> American Medium</label>
                <label><input type="radio" name="spice-level" value="American Spicy" /> American Spicy</label>
                <label><input type="radio" name="spice-level" value="Indian Mild" /> Indian Mild</label>
                <label><input type="radio" name="spice-level" value="Indian Medium" /> Indian Medium</label>
                <label><input type="radio" name="spice-level" value="Indian Very Spicy" /> Indian Very Spicy</label>
            </div>
            <!-- Biryani Extras -->
            <label for="biryani-extras">Biryani Extras (Optional - Choose as many as you like):</label>
            <div id="biryani-extras-options" style="display: flex; flex-wrap: wrap; gap: 10px; margin: 10px 0;">
                <label><input type="checkbox" name="biryani-extra" value="Extra Raitha 4oz" /> Extra Raitha 4oz + $1.00</label>
                <label><input type="checkbox" name="biryani-extra" value="Extra Raitha 8oz" /> Extra Raitha 8oz + $2.00</label>
                <label><input type="checkbox" name="biryani-extra" value="Extra Salan 4oz" /> Extra Salan 4oz + $1.00</label>
                <label><input type="checkbox" name="biryani-extra" value="Extra Salan 8oz" /> Extra Salan 8oz + $2.00</label>
                <label><input type="checkbox" name="biryani-extra" value="Extra Onion" /> Extra Onion + $1.00</label>
                <label><input type="checkbox" name="biryani-extra" value="Extra Onion & Lemon" /> Extra Onion & Lemon + $2.00</label>
                <label><input type="checkbox" name="biryani-extra" value="Extra Fried Onion 4oz" /> Extra Fried Onion 4oz + $2.00</label>
            </div>
            <!-- Quantity and Special Instructions -->
            <label for="quantity">Quantity:</label>
            <input type="number" id="quantity" value="1" min="1" style="width: 50px;" />
            <br><br>
            <textarea id="special-instructions" placeholder="Add special instructions here..." style="width: 100%; height: 60px;"></textarea>
            <br><br>
            <!-- Add to Cart Button -->
            <button style="background-color: #28a745; color: white; border: none; padding: 10px 20px; font-size: 14px; border-radius: 5px; cursor: pointer;" onclick="addToCart()">Add to Cart</button>
        </div>
        """)

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

        # Finalize order button click event
        finalize_button.click(finalize_order, inputs=[cart_data], outputs=[final_order_output])

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

        # Layout
        gr.Row([selected_preference])
        gr.Row(menu_output)
        gr.Row(cart_output)
        gr.Row(modal_window)
        gr.Row(finalize_button)
        gr.Row(final_order_output)
        gr.HTML(modal_and_cart_js)

    return demo

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