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><ul style='list-style-type: none; padding: 0;'>"
    
    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_price = float(item['price'].strip('$'))
        item_total = (item_price + extras_cost) * item['quantity']
        total_bill += item_total

        cart_html += f"<li style='margin-bottom: 20px; border: 1px solid #ddd; padding: 10px; border-radius: 8px;'>"
        cart_html += f"<strong>Item:</strong> {item['name']} - ${item_price:.2f}<br>"
        cart_html += f"<strong>Quantity x Price:</strong> {item['quantity']} x ${item_price:.2f} = ${item_price * item['quantity']:.2f}<br>"
        cart_html += f"<strong>Spice Level:</strong> {item['spiceLevel']}<br>"
        cart_html += f"<strong>Extras:</strong> {extras} - ${extras_cost:.2f}<br>"
        cart_html += f"<strong>Instructions:</strong> {item['instructions']}<br>"
        cart_html += f"<strong>Item Total:</strong> ${item_total:.2f}"
        cart_html += "</li>"

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

# Function to submit the cart and display on a new page
def submit_cart():
    if len(cart_items) == 0:
        return "<h3>Your Order</h3><p>Your cart is empty.</p>"

    total_bill = 0
    order_html = "<h3>Your Order</h3><ul style='list-style-type: none; padding: 0;'>"
    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_price = float(item['price'].strip('$'))
        item_total = (item_price + extras_cost) * item['quantity']
        total_bill += item_total

        order_html += f"<li style='margin-bottom: 20px; border: 1px solid #ddd; padding: 10px; border-radius: 8px;'>"
        order_html += f"<strong>Item:</strong> {item['name']} - ${item_price:.2f}<br>"
        order_html += f"<strong>Quantity x Price:</strong> {item['quantity']} x ${item_price:.2f} = ${item_price * item['quantity']:.2f}<br>"
        order_html += f"<strong>Spice Level:</strong> {item['spiceLevel']}<br>"
        order_html += f"<strong>Extras:</strong> {extras} - ${extras_cost:.2f}<br>"
        order_html += f"<strong>Instructions:</strong> {item['instructions']}<br>"
        order_html += f"<strong>Item Total:</strong> ${item_total:.2f}"
        order_html += "</li>"

    order_html += f"</ul><p><strong>Total Bill: ${total_bill:.2f}</strong></p>"
    return order_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"))

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

        # Submit button for the cart
        submit_button = gr.Button("Submit Order")

        # Output for the new page (order details)
        order_output = gr.HTML()

        # Submit button action to display the order details on a new page
        submit_button.click(fn=submit_cart, inputs=[], outputs=order_output)

        # Layout
        gr.Row([selected_preference])
        gr.Row(menu_output)
        gr.Row(cart_output)
        gr.Row([submit_button])
        gr.Row(order_output)

    return demo

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