Update app.py

app.py

@@ -127,16 +127,21 @@ def reward_status():
         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)
+    try:
+        # Query the Customer_Login__c object based on the user's email
+        query = f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}'"
+        user_data = sf.query(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)
+        # 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)
     
+    except Exception as e:
+        return f"Error querying Salesforce: {str(e)}", 500
+
     # Define point ranges for each tier
     tiers = {
         "Bronze": 1,
@@ -144,7 +149,7 @@ def reward_status():
         "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
@@ -172,6 +177,7 @@ def reward_status():
         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