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 load menu data
+# Function to fetch menu items from Salesforce
 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"
+        query = "SELECT Name, Price__c, Description__c, Image1__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
+# Function to fetch add-ons from Salesforce
 def load_add_ons_from_salesforce():
     try:
         query = "SELECT Name, Price__c FROM Add_Ons__c"
@@ -73,211 +73,61 @@ def load_add_ons_from_salesforce():
     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;" 
-                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
-
-# Function to save cart details to Salesforce
-def save_cart_to_salesforce(cart):
+# Function to save cart data to Salesforce
+def save_cart_to_salesforce(email, name, cart_items, total_bill):
     try:
-        # Loop through each cart item and create a new Order__c record
-        for item in cart:
-            order_data = {
-                'Name2__c': item['name'],  # Custom name for the order (can be the item name or order ID)
-                'Item_Name__c': item['name'],  # Name of the item in the order
-                'Quantity__c': item['quantity'],  # Quantity of the item
-                'Price__c': item['price'],  # Price of the item
-                'Total_Cost__c': item['totalCost'],  # Total cost for the item
-                'Special_Instructions__c': item['instructions'],  # Special instructions (if any)
-            }
-
-            # Handle add-ons if available
-            if item['extras']:
-                add_ons = ", ".join([add_on['name'] for add_on in item['extras']])
-                order_data['Add_Ons__c'] = add_ons  # Add-ons associated with the item
-
-            # Create the order in Salesforce
-            sf.Order__c.create(order_data)
-        
-        return "Cart saved successfully to Salesforce!"
+        # 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 while saving cart to Salesforce: {str(e)}"
+        return f"Error saving order to Salesforce: {str(e)}"
 
-# Function to create the modal window with add-ons
-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>
-        """
+# 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)
 
-    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}
+# 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>
-        <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>
-    """
-    return modal_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')),
-            }));
-            const extrasCost = extras.reduce((total, extra) => total + (extra.price * 1), 0); // Default quantity for add-ons set to 1
-            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} (+$${extra.price})`).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);
-            }
-            const item = cart[index];
-            const price = item.price;
-            const extrasCost = item.extras.reduce((total, extra) => total + (extra.price * 1), 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() {
-            // Save the cart to Salesforce when the user checks out
-            save_cart_to_salesforce(cart);
-            alert("Proceeding to checkout...");
-        }
-        // 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 by default
-        }
-    </script>
-    """
-    return modal_script
+    # 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>"
+
+    return html
 
 # Gradio App
 with gr.Blocks() as app:
@@ -306,16 +156,24 @@ 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()
-            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())
-            gr.HTML(modal_js())
+            checkout_button = gr.Button("Proceed to Checkout")
+            checkout_output = gr.Textbox(label="Checkout Status")
 
     login_button.click(
-        lambda email, password: (gr.update(visible=False), gr.update(visible=True), gr.update(value=filter_menu("All")), "Login successful!")
+        lambda email, password: (gr.update(visible=False), gr.update(visible=True), gr.update(value=generate_menu_html("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)
 
-app.launch()
+    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()