Update app.py

app.py

@@ -347,6 +347,50 @@ def update_quantity():
     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')
+    user_id = session.get('user_id')
+    if not email or not user_id:
+        return jsonify({"success": False, "message": "User not logged in"})
+
+    try:
+        # Fetch cart items for the user
+        result = sf.query(f"""
+            SELECT Id, Name, Price__c, Add_Ons_Price__c, Quantity__c, Add_Ons__c
+            FROM Cart_Item__c
+            WHERE Customer_Email__c = '{email}'
+        """)
+        cart_items = result.get("records", [])
+        if not cart_items:
+            return jsonify({"success": False, "message": "Cart is empty"})
+
+        # Calculate the total price of the order
+        total_price = sum(item['Price__c'] for item in cart_items)
+
+        # Create the order in Salesforce
+        order_data = {
+            "Customer_Name__c": user_id,
+            "Customer_Email__c": email,
+            "Total_Amount__c": total_price,
+            "Order_Status__c": "Pending",
+            "Order_Items__c": "\n".join(
+                [f"{item['Name']} (Qty: {item['Quantity__c']})" for item in cart_items]
+            ),
+            "Add_Ons__c": "\n".join(
+                [item['Add_Ons__c'] if item.get('Add_Ons__c') else "None" for item in cart_items]
+            ),
+        }
+        sf.Order__c.create(order_data)
+
+        # Clear the cart after placing the order
+        for item in cart_items:
+            sf.Cart_Item__c.delete(item["Id"])
+
+        return jsonify({"success": True, "message": "Order placed successfully!"})
+    except Exception as e:
+        print(f"Error during checkout: {str(e)}")
+        return jsonify({"success": False, "error": str(e)})
 
 
 @app.route("/order", methods=["GET"])