Update app.py

app.py

@@ -31,8 +31,9 @@ def authenticate_user(email, password):
         if result['totalSize'] == 0:
             return "Invalid Login Details", None, None, gr.update(visible=False), gr.update(visible=False)
 
+        # Ensure we get the correct record
         customer = result['records'][0]
-        reward_points = customer['Reward_Points__c']
+        reward_points = customer.get('Reward_Points__c', 0)  # Default to 0 if not available
 
         return f"Welcome, you have {reward_points} points. Proceed to rewards.", customer['Id'], reward_points, gr.update(visible=True), gr.update(visible=True)
     except Exception as e:
@@ -41,8 +42,9 @@ def authenticate_user(email, password):
 # Function to handle reward points logic
 def handle_rewards(customer_id, bill_amount, apply_rewards):
     try:
+        # Ensure the customer exists in Salesforce
         customer = sf.Customer_Login__c.get(customer_id)
-        points = customer['Reward_Points__c']
+        points = customer.get('Reward_Points__c', 0)
         gst = 0.18 * bill_amount
 
         if points >= 500 and apply_rewards: