Update cart.py

cart.py

@@ -375,10 +375,9 @@ def checkout():
 def get_loyalty_data():
     email = session.get('user_email')
     if not email:
-        return redirect(url_for('login'))  # Redirect to login if not authenticated
+        return redirect(url_for('login'))
 
     try:
-        # Fetch customer loyalty data
         customer_record = sf.query(f"""
             SELECT Reward_Points__c, Total_Orders__c, Total_Spent__c, Badge__c
             FROM Customer_Login__c WHERE Email__c = '{email}'
@@ -387,18 +386,33 @@ def get_loyalty_data():
         if not customer:
             return "Customer not found", 404
 
-        # Define available rewards
+        # Clean up data
+        reward_points = round(customer.get("Reward_Points__c", 0) or 0, 2)  # Round to 2 decimals
+        total_orders = customer.get("Total_Orders__c", 0) or 0
+        total_spent = customer.get("Total_Spent__c", 0) or 0
+        badge = customer.get("Badge__c") or "Newbie"  # Default to "Newbie" if None
+
+        # Define rewards with image URLs (you can store these in Salesforce or locally)
         rewards = [
-            {"name": "$5 Off", "points": 50, "description": "Get $5 off your next order"},
-            {"name": "Free Drink", "points": 100, "description": "Redeem for a free drink"},
-            {"name": "20% Off", "points": 200, "description": "20% off your next order"}
+            {"name": "$5 Off", "points": 50, "description": "Get $5 off your next order", "image": "/static/images/reward_discount.png"},
+            {"name": "Free Drink", "points": 100, "description": "Redeem for a free drink", "image": "/static/images/reward_drink.png"},
+            {"name": "20% Off", "points": 200, "description": "20% off your next order", "image": "/static/images/reward_percentage.png"}
         ]
 
+        # Badge images (map badge names to image paths)
+        badge_images = {
+            "Newbie": "/static/images/badge_bronze.png",
+            "Regular": "/static/images/badge_silver.png",
+            "VIP": "/static/images/badge_gold.png",
+            "Gold Member": "/static/images/badge_platinum.png"
+        }
+
         loyalty_data = {
-            "reward_points": customer.get("Reward_Points__c", 0),
-            "total_orders": customer.get("Total_Orders__c", 0),
-            "total_spent": customer.get("Total_Spent__c", 0),
-            "badge": customer.get("Badge__c", "Newbie"),
+            "reward_points": reward_points,
+            "total_orders": total_orders,
+            "total_spent": total_spent,
+            "badge": badge,
+            "badge_image": badge_images.get(badge, "/static/images/badge_bronze.png"),  # Default to bronze
             "available_rewards": rewards
         }
         return render_template("loyalty.html", loyalty_data=loyalty_data)