Update app.py

app.py

@@ -55,16 +55,16 @@ def login(email, password):
     except Exception as e:
         return f"Error during login: {str(e)}", None
 
-# Function to fetch menu items from Salesforce
+# Function to load menu data
 def load_menu_from_salesforce():
     try:
-        query = "SELECT Name, Price__c, Description__c, Image1__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
+        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 fetch add-ons from Salesforce
+# Function to load add-ons data
 def load_add_ons_from_salesforce():
     try:
         query = "SELECT Name, Price__c FROM Add_Ons__c"
@@ -73,61 +73,212 @@ def load_add_ons_from_salesforce():
     except Exception as e:
         return []
 
-# Function to save cart data to Salesforce
-def save_cart_to_salesforce(email, name, cart_items, total_bill):
-    try:
-        # Create the parent Cart__c record
-        parent_cart = sf.Cart__c.create({
-            'Customer_Email__c': email,
-            'Customer_Name__c': name,
-            'Total_Bill__c': total_bill
-        })
-        parent_cart_id = parent_cart['id']  # Retrieve the ID of the created Cart__c record
-
-        # Create child Cart_Item__c records for each item in the cart
-        for item in cart_items:
-            sf.Cart_Item__c.create({
-                'Cart__c': parent_cart_id,  # Associate the child with the parent
-                'Item_Name__c': item['name'],
-                'Quantity__c': item['quantity'],
-                'Price__c': item['price'],
-                'Total_Price__c': item['totalCost']
-            })
-
-        return "Order placed successfully!"
-    except Exception as e:
-        return f"Error saving order to Salesforce: {str(e)}"
+# Function to filter menu items
+def filter_menu(preference):
+    menu_data = load_menu_from_salesforce()
 
-# Process cart checkout
-def process_checkout(email, name, cart):
-    total_bill = sum(item['totalCost'] for item in cart)  # Calculate the total bill
-    return save_cart_to_salesforce(email, name, cart, total_bill)
+    filtered_data = {}
+    for item in menu_data:
+        if "Section__c" not in item or "Veg_NonVeg__c" not in item:
+            continue
 
-# Generate the menu HTML
-def generate_menu_html(preference):
-    menu_data = load_menu_from_salesforce()
-    add_ons_data = load_add_ons_from_salesforce()
-
-    filtered_menu = [item for item in menu_data if preference in ['All', item['Veg_NonVeg__c']]]
-    html = "<div>"
-
-    # Menu Items
-    for item in filtered_menu:
-        html += f"""
-        <div style="border: 1px solid #ddd; padding: 10px; margin: 10px;">
-            <h3>{item['Name']}</h3>
-            <p>Price: ${item['Price__c']}</p>
-            <p>Description: {item['Description__c']}</p>
-        </div>
+        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;" 
+                onclick="openModal('{item['Name']}', '{item.get('Image2__c', '')}', '{item['Description__c']}', '${item['Price__c']}')">
+                <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
+
+# Create Modal Window HTML
+def create_modal_window():
+    add_ons = load_add_ons_from_salesforce()
+    add_ons_html = ""
+    for add_on in add_ons:
+        add_ons_html += f"""
+        <label>
+            <input type="checkbox" name="biryani-extra" value="{add_on['Name']}" data-price="{add_on['Price__c']}" />
+            {add_on['Name']} + ${add_on['Price__c']}
+        </label>
+        <br>
         """
 
-    # Add-ons
-    html += "<h4>Add-ons:</h4><ul>"
-    for add_on in add_ons_data:
-        html += f"<li>{add_on['Name']} - ${add_on['Price__c']}</li>"
-    html += "</ul></div>"
+    modal_html = f"""
+    <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: 300px; border-radius: 8px; margin-bottom: 20px;" />
+        <h2 id="modal-name"></h2>
+        <p id="modal-description"></p>
+        <p id="modal-price"></p>
+        <label for="biryani-extras"><strong>Add-ons :</strong></label>
+        <div id="biryani-extras-options" style="display: flex; flex-wrap: wrap; gap: 10px; margin: 10px 0;">
+            {add_ons_html}
+        </div>
+        <label for="quantity">Quantity:</label>
+        <input type="number" id="quantity" value="1" min="1" style="width: 50px;" />
+        <textarea id="special-instructions" placeholder="Add your special instructions here..." style="width: 100%; height: 60px;"></textarea>
+        <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>
+    <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>
+            <div id="order-summary" style="margin-top: 20px;">
+                <h2>Final Order Summary:</h2>
+                <div id="final-order-summary"></div>
+                <p><strong>Total Bill: $<span id="total-bill"></span></strong></p>
+            </div>
+            <button style="background: #ff5722; color: white; padding: 10px 20px; border-radius: 5px; border: none; cursor: pointer;" onclick="proceedToCheckout()">Proceed to Checkout</button>
+        </div>
+    </div>
+    """
+    return modal_html
 
-    return html
+# JavaScript for Modal and Cart
+def modal_js():
+    modal_script = """
+    <script>
+        let cart = [];
+        let totalCartCost = 0;
+        function openModal(name, image2, description, price) {
+            const modal = document.getElementById('modal');
+            modal.style.display = 'block';
+            modal.style.position = 'fixed';
+            modal.style.width = window.innerWidth <= 768 ? '90%' : '30%';
+            modal.style.top = ${event.clientY}px;
+            modal.style.left = '50%';
+            modal.style.transform = 'translate(-50%, -50%)';
+            document.getElementById('modal-image').src = image2;
+            document.getElementById('modal-name').innerText = name;
+            document.getElementById('modal-description').innerText = description;
+            document.getElementById('modal-price').innerText = price;
+            document.getElementById('quantity').value = 1;
+            document.getElementById('special-instructions').value = '';
+            resetAddOns();  // Reset add-ons when opening the modal
+        }
+        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 quantity = parseInt(document.getElementById('quantity').value) || 1;
+            const instructions = document.getElementById('special-instructions').value;
+            const selectedAddOns = Array.from(document.querySelectorAll('input[name="biryani-extra"]:checked'));
+            const extras = selectedAddOns.map(extra => ({
+                name: extra.value,
+                price: parseFloat(extra.getAttribute('data-price')),
+                quantity: 1  // Default quantity for add-ons is 1
+            }));
+            const extrasCost = extras.reduce((total, extra) => total + (extra.price * extra.quantity), 0);
+            const totalCost = (price * quantity) + extrasCost;
+            // Add the item to the cart with its specific add-ons
+            cart.push({ name, price, quantity, extras, instructions, totalCost });
+            totalCartCost += totalCost;  // Update the total cost of the cart
+            updateCartButton();
+            updateCartTotalCost();  // Update total cost displayed
+            closeModal();
+        }
+        function updateCartButton() {
+            const cartButton = document.getElementById('cart-button');
+            cartButton.innerText = View Cart (${cart.length} items);
+        }
+        function openCartModal() {
+            const cartModal = document.getElementById('cart-modal');
+            const cartItemsContainer = document.getElementById('cart-items');
+            cartItemsContainer.innerHTML = "";
+            cart.forEach((item, index) => {
+                const extrasList = item.extras.map(extra => ${extra.name} x<input type="number" value="${extra.quantity}" min="1" style="width: 50px;" onchange="updateCartItem(${index}, 'extra', this.value)" /> (+$${(extra.price * extra.quantity).toFixed(2)})).join(', ');
+                cartItemsContainer.innerHTML += 
+                    <div style="border: 1px solid #ddd; padding: 10px; margin-bottom: 10px; border-radius: 8px;">
+                        <h3>${item.name}</h3>
+                        <p>Quantity: <input type="number" value="${item.quantity}" min="1" style="width: 50px;" onchange="updateCartItem(${index}, 'item', this.value)" /></p>
+                        <p>Extras: ${extrasList || 'None'}</p>
+                        <p>Special Instructions: ${item.instructions || 'None'}</p>
+                        <p>Total Cost: $<span id="item-${index}-total">${item.totalCost.toFixed(2)}</span></p>
+                        <button onclick="removeFromCart(${index})" style="color: red;">Remove</button>
+                    </div>
+                ;
+            });
+            cartModal.style.display = 'block';
+        }
+        function closeCartModal() {
+            document.getElementById('cart-modal').style.display = 'none';
+        }
+        function removeFromCart(index) {
+            totalCartCost -= cart[index].totalCost;  // Deduct the cost of the removed item from total cost
+            cart.splice(index, 1);
+            updateCartButton();
+            updateCartTotalCost();  // Update total cost displayed
+            openCartModal();
+        }
+        function updateCartItem(index, type, value) {
+            if (type === 'item') {
+                cart[index].quantity = parseInt(value);
+            } else if (type === 'extra') {
+                cart[index].extras[0].quantity = parseInt(value); // Assuming one add-on for simplicity
+            }
+            const item = cart[index];
+            const price = item.price;
+            const extrasCost = item.extras.reduce((total, extra) => total + (extra.price * extra.quantity), 0);
+            item.totalCost = (price * item.quantity) + extrasCost;
+            document.getElementById(item-${index}-total).innerText = item.totalCost.toFixed(2);
+            updateCartTotalCost();  // Update total cost displayed
+        }
+        function updateCartTotalCost() {
+            const totalCostElement = document.getElementById('cart-total-cost');
+            totalCartCost = cart.reduce((total, item) => total + item.totalCost, 0);
+            totalCostElement.innerText = Total Cart Cost: $${totalCartCost.toFixed(2)};
+        }
+        function proceedToCheckout() {
+            // Build the cart summary string with item details
+            const cartSummary = cart.map(item => 
+                ${item.name} (x${item.quantity}) - $${item.totalCost.toFixed(2)}
+                Extras: ${item.extras.map(extra => extra.name).join(', ') || 'None'}
+                Instructions: ${item.instructions || 'None'}
+            ).join('<br>');
+            // Insert the summary into the modal
+            document.getElementById('final-order-summary').innerHTML = cartSummary;
+            document.getElementById('total-bill').innerText = totalCartCost.toFixed(2);
+            // Show the cart modal with the summary
+            openCartModal();
+        }
+        // Reset all selected add-ons when opening a new item modal
+        function resetAddOns() {
+            const checkboxes = document.querySelectorAll('input[name="biryani-extra"]');
+            checkboxes.forEach(checkbox => checkbox.checked = false);  // Uncheck all add-ons
+        }
+    </script>
+    """
+    return modal_script
 
 # Gradio App
 with gr.Blocks() as app:
@@ -156,24 +307,16 @@ with gr.Blocks() as app:
         with gr.Column():
             preference = gr.Radio(choices=["All", "Veg", "Non-Veg"], label="Filter Preference", value="All")
             menu_output = gr.HTML()
-            checkout_button = gr.Button("Proceed to Checkout")
-            checkout_output = gr.Textbox(label="Checkout Status")
+            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><div id='order-summary'><h2>Final Order Summary:</h2><div id='final-order-summary'></div><p><strong>Total Bill: $<span id='total-bill'></span></strong></p></div><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())
+            gr.HTML(modal_js())
 
     login_button.click(
-        lambda email, password: (gr.update(visible=False), gr.update(visible=True), gr.update(value=generate_menu_html("All")), "Login successful!")
+        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]
     )
+    preference.change(lambda pref: filter_menu(pref), [preference], menu_output)
 
-    preference.change(lambda pref: generate_menu_html(pref), [preference], menu_output)
-
-    checkout_button.click(
-        lambda: process_checkout(login_email.value, "John Doe", [
-            {"name": "Biryani", "quantity": 2, "price": 10, "totalCost": 20},
-            {"name": "Naan", "quantity": 1, "price": 5, "totalCost": 5}
-        ]),
-        inputs=[],
-        outputs=checkout_output
-    )
-
-app.launch()
+app.launch()