Update app.py

app.py

@@ -53,11 +53,13 @@ def signup():
         generated_referral_code = generate_referral_code()
 
         try:
+            ref=0
             # Check if a referral code is entered
             if referral_code:
                 # Query Salesforce to check if the referral code exists
                 referral_query = f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Referral__c = '{referral_code}'"
                 referral_result = sf.query(referral_query)
+                ref=50
 
                 if not referral_result['records']:
                     # If referral code does not exist, show an error message
@@ -66,20 +68,21 @@ def signup():
                 # If referral exists, get the first matching record
                 referrer = referral_result['records'][0]
                 referrer_id = referrer['Id']
-                existing_reward_points = referrer.get('Reward_Points__c', 0)
-
-                # Add 100 points to the referrer's Reward_Points__c field
+                # Fetch the existing reward points, defaulting to 0 if None
+                existing_reward_points = referrer.get('Reward_Points__c') or 0  # Default to 0 if the field is None
                 updated_reward_points = existing_reward_points + 100
+                # Update the referrer's Reward_Points__c field in Salesforce
                 sf.Customer_Login__c.update(referrer_id, {
                     "Reward_Points__c": updated_reward_points
                 })
-
+            
             # Create the new customer record in Salesforce
             sf.Customer_Login__c.create({
                 "Name": name,
                 "Phone_Number__c": phone,
                 "Email__c": email,
                 "Password__c": password,
+                "Reward_Points__c": ref,
                 "Referral__c": generated_referral_code
             })
             return redirect(url_for("login"))