app.py

import bcrypt
import gradio as gr
from simple_salesforce import Salesforce

# Salesforce Connection
sf = Salesforce(username='diggavalli98@gmail.com', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')

# Function to Hash Password
def hash_password(password):
    return bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')

# Function to Verify Password
def verify_password(plain_password, hashed_password):
    return bcrypt.checkpw(plain_password.encode('utf-8'), hashed_password.encode('utf-8'))

# Signup function
def signup(name, email, phone, password):
    try:
        email = email.strip()
        query = f"SELECT Id FROM Customer_Login__c WHERE Email__c = '{email}'"
        result = sf.query(query)

        if len(result['records']) > 0:
            return "Email already exists! Please use a different email."

        hashed_password = hash_password(password)

        sf.Customer_Login__c.create({
            'Name': name.strip(),
            'Email__c': email,
            'Phone_Number__c': phone.strip(),
            'Password__c': hashed_password
        })
        return "Signup successful! You can now login."
    except Exception as e:
        return f"Error during signup: {str(e)}"

# Login function
def login(email, password):
    try:
        email = email.strip()
        query = f"SELECT Name, Password__c FROM Customer_Login__c WHERE Email__c = '{email}'"
        result = sf.query(query)

        if len(result['records']) == 0:
            return "Invalid email or password.", None

        user = result['records'][0]
        stored_password = user['Password__c']

        if verify_password(password.strip(), stored_password):
            return "Login successful!", user['Name']
        else:
            return "Invalid email or password.", None
    except Exception as e:
        return f"Error during login: {str(e)}", None

# Function to load menu data
def load_menu_from_salesforce():
    try:
        query = "SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
        result = sf.query(query)
        return result['records']
    except Exception as e:
        return []

# Function to load add-ons data
def load_add_ons_from_salesforce():
    try:
        query = "SELECT Name, Price__c FROM Add_Ons__c"
        result = sf.query(query)
        return result['records']
    except Exception as e:
        return []

# Function to filter menu items
def filter_menu(preference):
    menu_data = load_menu_from_salesforce()

    filtered_data = {}
    for item in menu_data:
        if "Section__c" not in item or "Veg_NonVeg__c" not in item:
            continue

        if item["Section__c"] not in filtered_data:
            filtered_data[item["Section__c"]] = []

        if preference == "All" or (preference == "Veg" and item["Veg_NonVeg__c"] in ["Veg", "Both"]) or (preference == "Non-Veg" and item["Veg_NonVeg__c"] in ["Non veg", "Both"]):
            filtered_data[item["Section__c"].strip()].append(item)

    html_content = '<div style="padding: 0 10px; max-width: 1200px; margin: auto;">'
    for section, items in filtered_data.items():
        html_content += f"<h2 style='text-align: center; margin-top: 5px;'>{section}</h2>"
        html_content += '<div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(250px, 1fr)); gap: 15px; justify-content: center; margin-top: 10px;">'
        for item in items:
            html_content += f"""
            <div style="border: 1px solid #ddd; border-radius: 10px; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1); overflow: hidden; height: 350px;">
                <img src="{item.get('Image1__c', '')}" style="width: 100%; height: 200px; object-fit: cover;">
                <div style="padding: 10px;">
                    <h3 style='font-size: 1.2em; text-align: center;'>{item['Name']}</h3>
                    <p style='font-size: 1.1em; color: green; text-align: center;'>${item['Price__c']}</p>
                    <p style='font-size: 0.9em; text-align: justify; margin: 5px;'>{item['Description__c']}</p>
                </div>
            </div>
            """
        html_content += '</div>'
    html_content += '</div>'

    if not any(filtered_data.values()):
        return "<p>No items match your filter.</p>"

    return html_content

# Generate final order summary
def generate_order_summary(cart):
    if not cart:
        return "<p>Your cart is empty.</p>"

    summary_html = "<h2>Order Summary</h2>"
    total_cost = 0

    for item in cart:
        item_cost = item['price'] * item['quantity']
        add_ons_summary = "<ul>" + "".join([f"<li>{addon['name']} x {addon['quantity']} (+${addon['price'] * addon['quantity']:.2f})</li>" for addon in item['extras']]) + "</ul>"
        total_item_cost = item_cost + sum(addon['price'] * addon['quantity'] for addon in item['extras'])
        summary_html += f"""
        <div style='margin-bottom: 20px;'>
            <h3>{item['name']}</h3>
            <p>Quantity: {item['quantity']}</p>
            <p>Special Instructions: {item['instructions'] or 'None'}</p>
            <p>Add-ons:</p>
            {add_ons_summary}
            <p>Total Cost for this Item: ${total_item_cost:.2f}</p>
        </div>
        """
        total_cost += total_item_cost

    summary_html += f"<h3>Total Cart Cost: ${total_cost:.2f}</h3>"
    return summary_html

# Gradio App
cart = []

with gr.Blocks() as app:
    with gr.Row():
        gr.HTML("<h1 style='text-align: center;'>Welcome to Biryani Hub</h1>")

    with gr.Row(visible=True) as login_page:
        with gr.Column():
            login_email = gr.Textbox(label="Email")
            login_password = gr.Textbox(label="Password", type="password")
            login_button = gr.Button("Login")
            signup_button = gr.Button("Go to Signup")
            login_output = gr.Textbox(label="Status")

    with gr.Row(visible=False) as signup_page:
        with gr.Column():
            signup_name = gr.Textbox(label="Name")
            signup_email = gr.Textbox(label="Email")
            signup_phone = gr.Textbox(label="Phone")
            signup_password = gr.Textbox(label="Password", type="password")
            submit_signup = gr.Button("Signup")
            login_redirect = gr.Button("Go to Login")
            signup_output = gr.Textbox(label="Status")

    with gr.Row(visible=False) as menu_page:
        with gr.Column():
            preference = gr.Radio(choices=["All", "Veg", "Non-Veg"], label="Filter Preference", value="All")
            menu_output = gr.HTML()
            gr.HTML("<div id='cart-button' style='position: fixed; top: 20px; right: 20px; background: #28a745; color: white; padding: 10px 20px; border-radius: 30px; cursor: pointer; z-index: 1000;' onclick='openCartModal()'>View Cart</div>")
            gr.HTML("<div id='cart-modal' style='display: none; position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: white; z-index: 1000; overflow-y: auto;'><div style='padding: 20px;'><div style='text-align: right;'><button onclick='closeCartModal()' style='background: none; border: none; font-size: 24px; cursor: pointer;'>&times;</button></div><h1>Your Cart</h1><div id='cart-items'></div><p id='cart-total-cost' style='font-size: 1.2em; font-weight: bold;'>Total Cart Cost: $0.00</p><button style='background: #ff5722; color: white; padding: 10px 20px; border-radius: 5px; border: none; cursor: pointer;' onclick='proceedToCheckout()'>Proceed to Checkout</button></div></div>")
            gr.HTML(create_modal_window())

    login_button.click(
        lambda email, password: (
            gr.update(visible=False),
            gr.update(visible=True),
            gr.update(value=filter_menu("All")),
            "Login successful!"
        ) if login(email, password)[0] == "Login successful!" else (
            gr.update(),
            gr.update(),
            gr.update(),
            "Invalid email or password."
        ),
        [login_email, login_password],
        [login_page, menu_page, menu_output, login_output]
    )

    signup_button.click(
        lambda: (gr.update(visible=False), gr.update(visible=True)),
        inputs=[],
        outputs=[login_page, signup_page]
    )

    login_redirect.click(
        lambda: (gr.update(visible=True), gr.update(visible=False)),
        inputs=[],
        outputs=[login_page, signup_page]
    )

    submit_signup.click(
        lambda name, email, phone, password: signup(name, email, phone, password),
        [signup_name, signup_email, signup_phone, signup_password],
        signup_output
    )

    preference.change(
        lambda pref: filter_menu(pref),
        [preference],
        menu_output
    )

    submit_order = gr.Button("Submit Order")
    order_summary = gr.HTML(visible=False)

    submit_order.click(
        lambda: gr.update(visible=True, value=generate_order_summary(cart)),
        inputs=[],
        outputs=[order_summary]
    )

    with gr.Row():
        gr.HTML(value="", visible=False).style(name="order-summary")

app.launch()