Update app.py

app.py

@@ -281,16 +281,21 @@ def update_quantity():
             f"SELECT Id, Quantity__c FROM Cart_Item__c WHERE Customer_Email__c = '{email}' AND Name = '{item_name}'"
         )['records']
 
-        if not cart_items:
-            return jsonify({"success": False, "error": "Cart item not found."}), 404
+        if cart_items:
+            cart_item_id = cart_items[0]['Id']
+            base_price = cart_items[0]['Base_Price__c']
+            addons_price = sum([float(price) for price in cart_items[0].get('Add_Ons__c', "0").split("$")[1::2]])  # Example parsing logic
+
+            new_price = (base_price + addons_price) * quantity
 
-        # Get the first matching record ID (assuming the item name is unique per user)
-        cart_item_id = cart_items[0]['Id']
+            # Update Salesforce record
+            sf.Cart_Item__c.update(cart_item_id, {
+                "Quantity__c": quantity,
+                "Price__c": new_price
+            })
 
-        # Update the quantity in Salesforce
-        sf.Cart_Item__c.update(cart_item_id, {"Quantity__c": quantity})
+            return jsonify({"success": True, "new_quantity": quantity, "new_price": new_price})
 
-        return jsonify({"success": True, "new_quantity": quantity})
     except Exception as e:
         return jsonify({"success": False, "error": str(e)}), 500