Update app.py

app.py

@@ -321,23 +321,30 @@ def update_quantity():
         base_price = cart_items[0]['Base_Price__c']
         addons_price = cart_items[0].get('Add_Ons_Price__c', 0)
 
-        # Calculate the new price
-        new_price = base_price * quantity
+        # Calculate the new item price
+        new_item_price = (base_price * quantity) + addons_price
 
         # Update the record in Salesforce
         sf.Cart_Item__c.update(cart_item_id, {
             "Quantity__c": quantity,
-            "Price__c": new_price,  # Update base price
-            "Add_Ons_Price__c": addons_price  # Add-ons price remains constant
+            "Price__c": new_item_price,  # Update base price
         })
 
-        return jsonify({"success": True, "new_quantity": quantity, "new_price": new_price})
+        # Recalculate the subtotal for all items in the cart
+        cart_items = sf.query(f"""
+            SELECT Price__c, Add_Ons_Price__c 
+            FROM Cart_Item__c 
+            WHERE Customer_Email__c = '{email}'
+        """)['records']
+        new_subtotal = sum(item['Price__c'] for item in cart_items) + sum(item['Add_Ons_Price__c'] for item in cart_items)
+
+        # Return updated item price and subtotal
+        return jsonify({"success": True, "new_item_price": new_item_price, "subtotal": new_subtotal})
     except Exception as e:
         print(f"Error updating quantity: {str(e)}")
         return jsonify({"success": False, "error": str(e)}), 500
 
 
-
 @app.route("/checkout", methods=["POST"])
 def checkout():
     email = session.get('user_email')