Update app.py

app.py

@@ -118,6 +118,60 @@ def login():
             return render_template("login.html", error=f"Error: {str(e)}")
 
     return render_template("login.html")
+@app.route('/reward_status')
+def reward_status():
+    # Get the user email from session (assuming the session is already set during checkout)
+    email = session.get('user_email')
+    
+    if not email:
+        return "User not logged in", 400
+
+    # Query Salesforce to fetch the user's reward points
+    query = f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}'"
+    user_data = query_salesforce(query)
+
+    # Check if the user record was found
+    if not user_data.get("records"):
+        return "User not found", 400
+    
+    # Get the user's reward points from Salesforce
+    user_points = user_data["records"][0].get("Reward_Points__c", 0)
+    
+    # Define point ranges for each tier
+    tiers = {
+        "Bronze": 1,
+        "Silver": 100,
+        "Gold": 300,
+        "Platinum": 500
+    }
+    
+    # Get the user's current tier and the points required for the next tier
+    current_tier = ""
+    points_needed_for_next_tier = 0
+    
+    for tier, threshold in tiers.items():
+        if user_points < threshold:
+            current_tier = tier
+            points_needed_for_next_tier = threshold - user_points
+            break
+        else:
+            current_tier = tier
+            points_needed_for_next_tier = 0
+
+    # Progress bar logic (percentage for current tier progress)
+    if current_tier != "Platinum":
+        next_tier_threshold = tiers[current_tier] if current_tier else 0
+        progress_percentage = (user_points - next_tier_threshold) / (tiers.get(current_tier, 500) - next_tier_threshold) * 100
+    else:
+        progress_percentage = 100
+
+    return render_template(
+        'reward_status.html',
+        user_points=user_points,
+        current_tier=current_tier,
+        points_needed_for_next_tier=points_needed_for_next_tier,
+        progress_percentage=progress_percentage
+    )
 @app.route("/logout")
 def logout():
     # Retrieve table number before clearing session