Update app.py

app.py

@@ -64,33 +64,59 @@ def load_menu_from_salesforce():
     except Exception as e:
         return []
 
+# Function to load add-ons data from Salesforce
+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 []
+
 # Cart Management Functions
 cart = []
 total_cart_cost = 0
+cart_id = None  # Store the Cart__c record ID
 
 # Function to add item to the cart and Salesforce
-def add_to_cart(item_name, item_price, quantity, special_instructions):
-    global total_cart_cost
+def add_to_cart(item_name, item_price, quantity, special_instructions, selected_addons):
+    global total_cart_cost, cart_id
     total_cost = item_price * quantity
+    addons_cost = sum([addon['Price__c'] for addon in selected_addons])
 
-    # Store the cart item in Salesforce
+    # If this is the first item, create a new Cart__c record in Salesforce
+    if not cart_id:
+        cart_record = sf.Cart__c.create({
+            'Total_Cost__c': total_cost + addons_cost
+        })
+        cart_id = cart_record['Id']
+    else:
+        # Update the total cost of the cart
+        sf.Cart__c.update(cart_id, {
+            'Total_Cost__c': total_cart_cost + total_cost + addons_cost
+        })
+
+    # Create the Cart_Item__c record and link it to Cart__c
     cart_item = sf.Cart_Item__c.create({
         'Name': item_name,
         'Quantity__c': quantity,
         'Price__c': item_price,
         'Special_Instructions__c': special_instructions,
+        'Cart__c': cart_id,  # Link this item to the Cart__c
+        'Add_ons__c': [addon['Id'] for addon in selected_addons]  # Link selected add-ons
     })
     
-    # Update the total cart cost
+    # Add the item to the cart list (for local tracking)
     cart.append({
         'name': item_name,
         'price': item_price,
         'quantity': quantity,
         'instructions': special_instructions,
-        'total_cost': total_cost
+        'total_cost': total_cost + addons_cost,
+        'addons': selected_addons
     })
     
-    total_cart_cost += total_cost
+    total_cart_cost += total_cost + addons_cost
     return cart, total_cart_cost
 
 # Function to remove item from the cart
@@ -98,6 +124,12 @@ def remove_from_cart(index):
     global total_cart_cost
     total_cart_cost -= cart[index]['total_cost']
     cart.pop(index)
+    
+    # Update the Cart__c record in Salesforce to reflect the new total cost
+    sf.Cart__c.update(cart_id, {
+        'Total_Cost__c': total_cart_cost
+    })
+    
     return cart, total_cart_cost
 
 # Checkout Summary
@@ -107,6 +139,8 @@ def proceed_to_checkout():
         summary += f"{item['name']} (x{item['quantity']}) - ${item['total_cost']:.2f}\n"
         if item['instructions']:
             summary += f"Instructions: {item['instructions']}\n"
+        if item['addons']:
+            summary += f"Add-ons: {', '.join([addon['Name'] for addon in item['addons']])}\n"
     summary += f"Total Bill: ${total_cart_cost:.2f}"
     return summary
 
@@ -133,7 +167,7 @@ def filter_menu(preference):
             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="openItemModal('{item['Name']}', '{item.get('Image2__c', '')}', '{item['Description__c']}', '${item['Price__c']}')">
+                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>
@@ -146,11 +180,20 @@ def filter_menu(preference):
 
     if not any(filtered_data.values()):
         return "<p>No items match your filter.</p>"
-
     return html_content
 
 # Modal Simulation Function (handled in Python)
 def open_modal(item_name, item_image, item_description, item_price):
+    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="add-on" value="{add_on['Name']}" data-price="{add_on['Price__c']}" />
+            {add_on['Name']} + ${add_on['Price__c']}
+        </label><br>
+        """
+
     modal_content = f"""
     <div style="background: white; padding: 20px; width: 60%; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.2); position: absolute; top: 20%; left: 20%; z-index: 1000;">
         <img src="{item_image}" style="width: 100%; height: 250px; object-fit: cover; border-radius: 8px;">
@@ -160,8 +203,10 @@ def open_modal(item_name, item_image, item_description, item_price):
         <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 onclick="addToCart()" style="background-color: #28a745; color: white; padding: 10px 20px; border: none; font-size: 14px; border-radius: 5px; cursor: pointer;" >Add to Cart</button>
-        <button onclick="closeModal()" style="background-color: #ff5722; color: white; padding: 10px 20px; border: none; font-size: 14px; border-radius: 5px; cursor: pointer;" >Close</button>
+        <h4>Select Add-ons:</h4>
+        {add_ons_html}
+        <button onclick="addToCart()" style="background-color: #28a745; color: white; padding: 10px 20px; border: none; font-size: 14px; border-radius: 5px; cursor: pointer;">Add to Cart</button>
+        <button onclick="closeModal()" style="background-color: #ff5722; color: white; padding: 10px 20px; border: none; font-size: 14px; border-radius: 5px; cursor: pointer;">Close</button>
     </div>
     """
     return gr.HTML(modal_content)