Update app.py

app.py

@@ -56,8 +56,13 @@ def signup():
         password = request.form.get("password")
         referral_code = request.form.get("referral_code")
         try:
-            # Generate a unique referral code
-            customer_referral_code = generate_referral_code()
+            # Step 1: Generate a unique referral code for the new user
+            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"]:
+                    break
                 
                     
             sf.Customer_Login__c.create({
@@ -84,21 +89,13 @@ def signup():
                     coupon_code = generate_coupon_code()
                     print(f"Creating referral coupon: Name={referrer_name}, Email={referrer_email}, Code={coupon_code}")
 
-
-                    # Create a Referral Coupon record
-                    referrer_data = {
-                        "Referral_Name__c": "Test Name",
-                        "Referral_Email__c": "[email protected]",
-                        "Coupon_Code__c": "TESTCODE123",
+                    sf.Referral_Coupon__c.create({
+                        "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.")