Update app.py

app.py

@@ -149,7 +149,9 @@ def login():
                 print(f"Session variables set: user_id={session['user_id']}, user_email={session['user_email']}")
 
                 user_name = user.get("Name", "")  # Get the user's name
-                reward_points = user.get("Reward_Points__c", 0)
+                reward_points = user.get("Reward_Points__c") or 0  # Ensures reward_points is always an integer
+
+
 
                 if reward_points >= 500:
                     print(f"User {email} has {reward_points} reward points. Generating coupon...")
@@ -536,7 +538,7 @@ def checkout():
 
         # Fetch cart items
         result = sf.query(f"""
-            SELECT Id, Name, Price__c, Add_Ons_Price__c, Quantity__c, Add_Ons__c, Instructions__c
+            SELECT Id, Name, Price__c, Add_Ons_Price__c, Quantity__c, Add_Ons__c, Instructions__c, Image1__c
             FROM Cart_Item__c
             WHERE Customer_Email__c = '{email}'
         """)
@@ -556,15 +558,13 @@ def checkout():
         has_coupons = bool(coupon_query["records"])  # Check if user has any coupons
 
         if selected_coupon:
-            # Case 3: User selected a valid coupon → Apply discount & remove coupon
+            # Apply discount & remove coupon
             discount = total_price * 0.10  # 10% discount
             referral_coupon_id = coupon_query["records"][0]["Id"]
             existing_coupons = coupon_query["records"][0]["Coupon_Code__c"].split("\n")
 
             # Remove only the selected coupon
             updated_coupons = [coupon for coupon in existing_coupons if coupon.strip() != selected_coupon]
-
-            # Convert list back to a string with newlines
             updated_coupons_str = "\n".join(updated_coupons).strip()
 
             # Update the Referral_Coupon__c record with the remaining coupons
@@ -572,7 +572,7 @@ def checkout():
                 "Coupon_Code__c": updated_coupons_str
             })
         else:
-            # Case 1 & Case 2: User has no coupons or has coupons but didn’t select one → Add 10% to reward points
+            # Add 10% to reward points if no coupon is used
             reward_points_to_add = total_price * 0.10
 
             # Fetch current reward points
@@ -586,17 +586,17 @@ def checkout():
                 current_reward_points = customer.get("Reward_Points__c") or 0
                 new_reward_points = current_reward_points + reward_points_to_add
 
-                print(f"Updating reward points: Current = {current_reward_points}, Adding = {reward_points_to_add}, New = {new_reward_points}")
-
                 # Update reward points in Salesforce
                 sf.Customer_Login__c.update(customer["Id"], {
                     "Reward_Points__c": new_reward_points
                 })
-                print(f"Successfully updated reward points for {email}")
 
         total_bill = total_price - discount
 
-        # Store the order details
+        # ✅ Store Order Details (Including Images)
+        order_details = "\n".join(f"{item['Name']} x{item['Quantity__c']}|{item['Image1__c']}" for item in cart_items)
+
+        # Store the order details in Order__c
         order_data = {
             "Customer_Name__c": user_id,
             "Customer_Email__c": email,
@@ -604,18 +604,14 @@ def checkout():
             "Discount__c": discount,
             "Total_Bill__c": total_bill,
             "Order_Status__c": "Pending",
-            "Order_Details__c": "\n".join(f"{item['Name']} x{item['Quantity__c']}" for item in cart_items)
+            "Order_Details__c": order_details  # ✅ Updated to include images
         }
 
         sf.Order__c.create(order_data)
 
-        # ✅ Fix: Check if cart items exist before deleting
-        if cart_items:
-            for item in cart_items:
-                try:
-                    sf.Cart_Item__c.delete(item["Id"])
-                except Exception as e:
-                    print(f"Error deleting cart item {item['Id']}: {str(e)}")
+        # ✅ Delete cart items after order is placed
+        for item in cart_items:
+            sf.Cart_Item__c.delete(item["Id"])
 
         return jsonify({"success": True, "message": "Order placed successfully!"})