import gradio as gr
import pandas as pd
from bcrypt import hashpw, gensalt, checkpw

# Path to the Excel file for user storage
USER_FILE = "users.xlsx"

# Load users from Excel
def load_users():
    try:
        df = pd.read_excel(USER_FILE)
        return {row['Email']: row for _, row in df.iterrows()}
    except FileNotFoundError:
        df = pd.DataFrame(columns=["Name", "Phone", "Email", "Password"])
        df.to_excel(USER_FILE, index=False)
        return {}

# Save users to Excel
def save_users(users):
    df = pd.DataFrame(users.values())
    df.to_excel(USER_FILE, index=False)

# Signup user
def signup_user(name, phone, email, password):
    users = load_users()
    if email in users:
        return False, "Email already exists. Please use a different email."
    hashed_password = hashpw(password.encode(), gensalt()).decode()
    users[email] = {"Name": name, "Phone": phone, "Email": email, "Password": hashed_password}
    save_users(users)
    return True, "Signup successful! Redirecting to the login page..."

# Validate login
def validate_login(email, password):
    users = load_users()
    if email in users and checkpw(password.encode(), users[email]["Password"].encode()):
        return True, "Login successful! Redirecting to the menu page..."
    return False, "Invalid email or password."

# Session management
active_sessions = {}

def login_user(email, password):
    success, message = validate_login(email, password)
    if success:
        active_sessions[email] = True
        return True, message
    return False, message

def is_logged_in(email):
    return active_sessions.get(email, False)
    
# 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

# Preserving your JavaScript for modal and cart functionality (Your original JavaScript logic)
modal_and_cart_js = """
<style>
  .cart-container {
    border: 1px solid #ddd;
    border-radius: 8px;
    padding: 15px;
    margin-bottom: 15px;
    background-color: #f9f9f9;
    display: flex;
    flex-direction: column;
    gap: 10px;
  }
  .cart-item {
    display: flex;
    justify-content: space-between;
    align-items: center;
    border-bottom: 1px solid #ddd;
    padding: 10px 0;
  }
  .cart-item:last-child {
    border-bottom: none;
  }
  .cart-item span {
    font-size: 16px;
    margin-right: 10px;
  }
  .cart-item .quantity-container {
    display: flex;
    align-items: center;
    gap: 5px;
  }
  .cart-total {
    font-size: 18px;
    font-weight: bold;
    text-align: right;
    margin-top: 10px;
  }
</style>
<script>
    let cart = [];
    const extrasPrices = {
        "Thums up": 2,
        "Sprite": 2,
        "Extra Raitha": 1,
        "Extra Salan": 2,
        "Extra Onion & Lemon": 2,
        "Chilli Chicken": 14,
        "Veg Manchurian": 12
    };
    let finalized = false;
    function openModal(name, image, description, price) {
        if (finalized) {
            alert("You cannot add more items after finalizing your order.");
            return;
        }
        const modal = document.getElementById('modal');
        modal.style.display = 'block';
        modal.style.position = 'fixed';
        if (window.innerWidth <= 768) {
            modal.style.width = '90%';
        } else {
            modal.style.width = '30%';
        }
        modal.style.top = '15%';
        modal.style.left = '50%';
        modal.style.transform = 'translate(-50%, -50%)';
        document.getElementById('modal-image').src = image;
        document.getElementById('modal-name').innerText = name;
        document.getElementById('modal-description').innerText = description;
        document.getElementById('modal-price').innerText = price;
        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 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, quantity, instructions, extras, itemTotal, extrasQuantities: extras.map(() => 1) };
        cart.push(cartItem);
        alert(`${name} added to cart!`);
        updateCartDisplay();
        closeModal();
    }
    function updateCartDisplay() {
        let totalBill = 0;
        let cartHTML = "<div class='cart-container'>";
        cart.forEach((item, index) => {
            totalBill += item.itemTotal;
            const extras = item.extras.map((extra, i) => {
                const extraQuantity = item.extrasQuantities ? item.extrasQuantities[i] || 1 : 1;
                const extraTotal = extrasPrices[extra] * extraQuantity;
                totalBill += extraTotal;
                return `<div class='cart-item'>
                            <span>${extra}</span>
                            <span>Price: $${extrasPrices[extra].toFixed(2)}</span>
                            <div class='quantity-container'>
                                <label for='extra-quantity-${index}-${i}'>Quantity:</label>
                                <input type='number' id='extra-quantity-${index}-${i}' value='${extraQuantity}' min='1' style='width: 50px;' onchange='updateExtraQuantity(${index}, ${i}, this.value)'>
                            </div>
                            <span>Total: $${extraTotal.toFixed(2)}</span>
                            <input type='checkbox' id='extra-remove-${index}-${i}' onclick='removeExtra(${index}, ${i})'> Remove
                        </div>`;
            }).join('');
            cartHTML += `<div class='cart-item'>
                            <span>${item.name}</span>
                            <span>Item Price: $${item.price.toFixed(2)}</span>
                            <div class='quantity-container'>
                                <label for='item-quantity-${index}'>Quantity:</label>
                                <input type='number' id='item-quantity-${index}' value='${item.quantity}' min='1' style='width: 50px;' onchange='updateItemQuantity(${index}, this.value)'>
                            </div>
                            <span>Total: $${(item.price * item.quantity).toFixed(2)}</span>
                            <input type='checkbox' id='item-remove-${index}' onclick='removeItem(${index})'> Remove
                        </div>
                        ${extras}
                        <div class='cart-item'><strong>Instructions:</strong> ${item.instructions || "None"}</div>`;
        });
        cartHTML += `</div><p class='cart-total'>Total Bill: $${totalBill.toFixed(2)}</p>`;
        cartHTML += `<button style='margin-top: 10px; background-color: #007bff; color: white; border: none; padding: 10px; border-radius: 5px; width: 100%; cursor: pointer;' onclick='submitCart()'>Submit</button>`;
        document.getElementById('floating-cart').innerHTML = cartHTML;
    }
    function updateItemQuantity(index, newQuantity) {
        const quantity = parseInt(newQuantity) || 1;
        cart[index].quantity = quantity;
        cart[index].itemTotal = cart[index].price * quantity;
        updateCartDisplay();
    }
    function updateExtraQuantity(cartIndex, extraIndex, newQuantity) {
        const quantity = parseInt(newQuantity) || 1;
        cart[cartIndex].extrasQuantities = cart[cartIndex].extrasQuantities || [];
        cart[cartIndex].extrasQuantities[extraIndex] = quantity;
        updateCartDisplay();
    }
    function removeExtra(cartIndex, extraIndex) {
        cart[cartIndex].extras.splice(extraIndex, 1);
        if (cart[cartIndex].extrasQuantities) {
            cart[cartIndex].extrasQuantities.splice(extraIndex, 1);
        }
        updateCartDisplay();
    }
    function removeItem(index) {
        cart.splice(index, 1);
        updateCartDisplay();
    }
    function submitCart() {
        let finalOrderHTML = "<h3>Final Order:</h3><ul>";
        let totalBill = 0;
        cart.forEach(item => {
            totalBill += item.itemTotal;
            const extras = item.extras.map((extra, i) => {
                const extraQuantity = item.extrasQuantities ? item.extrasQuantities[i] || 1 : 1;
                const extraTotal = extrasPrices[extra] * extraQuantity;
                totalBill += extraTotal;
                return `${extra} (x${extraQuantity}) - $${extraTotal.toFixed(2)}`;
            }).join(', ');
            finalOrderHTML += `<li>
                ${item.name} (x${item.quantity}) - $${item.itemTotal.toFixed(2)}
                <br>Extras: ${extras}
                <br>Instructions: ${item.instructions || "None"}
            </li>`;
        });
        finalOrderHTML += `</ul><p><strong>Total Bill: $${totalBill.toFixed(2)}</strong></p>`;
        document.getElementById('final-order').innerHTML = finalOrderHTML;
        alert("Your final order has been submitted!");
    }
</script>
"""


# Gradio app
def app():
    with gr.Blocks() as demo:
        gr.Markdown("## Secure Food Ordering System")

        # State variable to track the current page
        current_page = gr.State("signup")  # Default page is Signup

        # Signup Page
        with gr.Row(visible=lambda state: state == "signup", state=current_page):
            gr.Markdown("### Signup Page")
            name = gr.Textbox(label="Name")
            phone = gr.Textbox(label="Phone Number")
            email = gr.Textbox(label="Email")
            password = gr.Textbox(label="Password", type="password")
            signup_btn = gr.Button("Signup")
            signup_message = gr.Label()

            # Signup callback
            def handle_signup(name, phone, email, password):
                message, next_page = signup_user(name, phone, email, password)
                return message, next_page

            signup_btn.click(
                handle_signup,
                inputs=[name, phone, email, password],
                outputs=[signup_message, current_page],
            )

        # Login Page
        with gr.Row(visible=lambda state: state == "login", state=current_page):
            gr.Markdown("### Login Page")
            login_email = gr.Textbox(label="Email")
            login_password = gr.Textbox(label="Password", type="password")
            login_btn = gr.Button("Login")
            login_message = gr.Label()

            # Login callback
            def handle_login(email, password):
                message, next_page = login_user(email, password)
                return message, next_page

            login_btn.click(
                handle_login,
                inputs=[login_email, login_password],
                outputs=[login_message, current_page],
            )

        # Menu Page
        with gr.Row(visible=lambda state: state == "menu", state=current_page):
            gr.Markdown("### Menu Page (Accessible Only After Login)")

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

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

            # Modal window
            modal_window = gr.HTML("""
            <div id="modal" style="display: none; position: fixed; 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>
                <!-- Biryani Extras -->
                <label for="biryani-extras">Biryani Extras :</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="Thums up" /> Thums up + $2.00</label>
                    <label><input type="checkbox" name="biryani-extra" value="Sprite" /> Sprite + $2.00</label>
                    <label><input type="checkbox" name="biryani-extra" value="Extra Raitha" /> Extra Raitha  + $1.00</label>
                    <label><input type="checkbox" name="biryani-extra" value="Extra Salan" /> Extra Salan  + $2.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="Chilli Chicken" /> Chilli Chicken + $14.00</label>
                    <label><input type="checkbox" name="biryani-extra" value="Veg Manchurian" /> Veg Manchurian + $12.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>
            """)

            # 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(final_order_output)
            gr.HTML(modal_and_cart_js)

        return demo


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