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}")

# Initialize cart globally
cart_items = []

# Pricing for extras
EXTRAS_PRICES = {
    "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 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

# Function to update the cart display
def update_cart():
    if len(cart_items) == 0:
        return "Your cart is empty."

    total_bill = 0
    cart_html = "<h3>Your Cart:</h3><table style='border: 1px solid black; border-collapse: collapse; width: 100%;'>"
    cart_html += "<tr><th style='border: 1px solid black; padding: 8px;'>Selected Item</th><th style='border: 1px solid black; padding: 8px;'>Actual Price</th><th style='border: 1px solid black; padding: 8px;'>Quantity x Price</th><th style='border: 1px solid black; padding: 8px;'>Spice Level</th><th style='border: 1px solid black; padding: 8px;'>Extras</th><th style='border: 1px solid black; padding: 8px;'>Instructions</th></tr>"
    
    for item in cart_items:
        extras = ", ".join(item.get("extras", []))
        extras_cost = sum(EXTRAS_PRICES.get(extra, 0) for extra in item.get("extras", []))
        item_total = (float(item['price'].strip('$')) + extras_cost) * item['quantity']
        total_bill += item_total
        cart_html += f"<tr>"
        cart_html += f"<td style='border: 1px solid black; padding: 8px;'>{item['name']}</td>"
        cart_html += f"<td style='border: 1px solid black; padding: 8px;'>${float(item['price'].strip('$')):.2f}</td>"
        cart_html += f"<td style='border: 1px solid black; padding: 8px;'>${item_total:.2f}</td>"
        cart_html += f"<td style='border: 1px solid black; padding: 8px;'>{item['spiceLevel']}</td>"
        cart_html += f"<td style='border: 1px solid black; padding: 8px;'>{extras}</td>"
        cart_html += f"<td style='border: 1px solid black; padding: 8px;'>{item['instructions']}</td>"
        cart_html += f"</tr>"

    cart_html += "</table>"
    cart_html += f"<p><strong>Total Bill: ${total_bill:.2f}</strong></p>"
    return cart_html

# 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"))

        # Redirect to cart page
        def redirect_to_cart():
            return update_cart()

        cart_page = gr.HTML("<h2>Your Cart</h2>")

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

        # 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, image, description, price) {
                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;
            }

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

            function addToCart() {
                const name = document.getElementById('modal-name').innerText;
                const price = document.getElementById('modal-price').innerText;
                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 = (parseFloat(price.replace('$', '')) + 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 => {
                    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;
            }
        </script>
        """

        # Modal window
        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>
        """)

        # Interactivity
        selected_preference.change(filter_menu, inputs=[selected_preference], outputs=[menu_output])
        gr.Button("View Cart").click(redirect_to_cart, outputs=[cart_page])

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

    return demo

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