Update app.py

app.py

@@ -768,30 +768,40 @@ def add_suggestion_to_cart():
 
 @app.route('/cart/add', methods=['POST'])
 def add_to_cart():
-    data = request.json
-    item_name = data.get('itemName').strip()
-    item_price = data.get('itemPrice')
-    item_image = data.get('itemImage')
-    addons = data.get('addons', [])
-    instructions = data.get('instructions', '')
-    category = data.get('category')
-    section = data.get('section')
-    customer_email = session.get('user_email')
-
-    if not item_name or not item_price:
-        return jsonify({"success": False, "error": "Item name and price are required."})
-
     try:
+        # Get data from request
+        data = request.json
+        item_name = data.get('itemName', '').strip()
+        item_price = data.get('itemPrice')
+        item_image = data.get('itemImage')
+        addons = data.get('addons', [])
+        instructions = data.get('instructions', '')
+        category = data.get('category')
+        section = data.get('section')
+        quantity = data.get('quantity', 1)  # Get the quantity field from the request
+        customer_email = session.get('user_email')
+
+        # Basic validation for required fields
+        if not item_name or not item_price:
+            return jsonify({"success": False, "error": "Item name and price are required."}), 400
+
+        if not customer_email:
+            return jsonify({"success": False, "error": "User email is required."}), 400
+
+        # Query to check if the item is already in the cart
         query = f"""
-            SELECT Id, Quantity__c, Add_Ons__c, Add_Ons_Price__c, Instructions__c FROM Cart_Item__c
+            SELECT Id, Quantity__c, Add_Ons__c, Add_Ons_Price__c, Instructions__c 
+            FROM Cart_Item__c
             WHERE Customer_Email__c = '{customer_email}' AND Name = '{item_name}'
         """
         result = sf.query(query)
         cart_items = result.get("records", [])
 
+        # Calculate the total price for the addons
         addons_price = sum(addon['price'] for addon in addons)
         new_addons = "; ".join([f"{addon['name']} (${addon['price']})" for addon in addons])
 
+        # If the item is already in the cart, update it
         if cart_items:
             cart_item_id = cart_items[0]['Id']
             existing_quantity = cart_items[0]['Quantity__c']
@@ -799,40 +809,46 @@ def add_to_cart():
             existing_addons_price = cart_items[0].get('Add_Ons_Price__c', 0)
             existing_instructions = cart_items[0].get('Instructions__c', "")
 
+            # Combine the new addons with the existing ones
             combined_addons = existing_addons if existing_addons != "None" else ""
             if new_addons:
                 combined_addons = f"{combined_addons}; {new_addons}".strip("; ")
 
+            # Combine existing instructions with new instructions
             combined_instructions = existing_instructions
             if instructions:
                 combined_instructions = f"{combined_instructions} | {instructions}".strip(" | ")
 
+            # Calculate total addons price
             combined_addons_list = combined_addons.split("; ")
             combined_addons_price = sum(
                 float(addon.split("($")[1][:-1]) for addon in combined_addons_list if "($" in addon
             )
 
+            # Update the cart item in Salesforce (updating quantity)
             sf.Cart_Item__c.update(cart_item_id, {
-                "Quantity__c": existing_quantity + 1,
+                "Quantity__c": existing_quantity + quantity,  # Add the selected quantity
                 "Add_Ons__c": combined_addons,
                 "Add_Ons_Price__c": combined_addons_price,
                 "Instructions__c": combined_instructions,
-                "Price__c": (existing_quantity + 1) * item_price + combined_addons_price,
+                "Price__c": (existing_quantity + quantity) * item_price + combined_addons_price,
                 "Category__c": category,
                 "Section__c": section
             })
         else:
+            # If the item is not already in the cart, create a new entry
             addons_string = "None"
             if addons:
                 addons_string = new_addons
 
-            total_price = item_price + addons_price
+            total_price = item_price * quantity + addons_price  # Multiply by the quantity
 
+            # Create new cart item in Salesforce
             sf.Cart_Item__c.create({
                 "Name": item_name,
                 "Price__c": total_price,
                 "Base_Price__c": item_price,
-                "Quantity__c": 1,
+                "Quantity__c": quantity,  # Use the selected quantity
                 "Add_Ons_Price__c": addons_price,
                 "Add_Ons__c": addons_string,
                 "Image1__c": item_image,
@@ -843,13 +859,15 @@ def add_to_cart():
             })
 
         return jsonify({"success": True, "message": "Item added to cart successfully."})
-    except Exception as e:
-        print(f"Error adding item to cart: {str(e)}")
-        return jsonify({"success": False, "error": str(e)})
-
-
 
+    except KeyError as e:
+        # Handle missing expected keys in request data
+        return jsonify({"success": False, "error": f"Missing required field: {str(e)}"}), 400
 
+    except Exception as e:
+        # Log the error for debugging and return a general error message
+        print(f"Error adding item to cart: {str(e)}")
+        return jsonify({"success": False, "error": "An error occurred while adding the item to the cart."}), 500
 
 
 @app.route("/cart/add_item", methods=["POST"])