Update app.py

app.py

@@ -57,12 +57,8 @@ def signup():
         referral_code = request.form.get("referral_code")
         try:
             # Generate a unique referral code
-            while True:
-                customer_referral_code = generate_referral_code()
-                query = f"SELECT Id FROM Customer_Login__c WHERE Customer_Referral_Code__c = '{customer_referral_code}'"
-                result = sf.query(query)
-                if not result["records"]:  # If no existing record with this code, break the loop
-                    break
+            customer_referral_code = generate_referral_code()
+                
                     
             sf.Customer_Login__c.create({
                 "Name": name,
@@ -90,12 +86,19 @@ def signup():
 
 
                     # Create a Referral Coupon record
-                    sf.Referral_Coupon__c.create({
+                    referrer_data = {
                         "Referral_Name__c": referrer_name,
                         "Referral_Email__c": referrer_email,
                         "Coupon_Code__c": coupon_code,
                         "Coupon_Status__c": "Active"
-                    })
+                    }
+                    try:
+                        sf.Referral_Coupon__c.create(coupon_data)
+                        
+                    except Exception as coupon_error:
+                        print(f"Failed to create referral coupon: {str(coupon_error)}")
+                        return render_template("signup.html", error="Failed to create referral coupon.")
+
                 else:
                     print("Referral code is invalid or does not exist.")
                     return render_template("signup.html", error="Invalid referral code provided.")