Update app.py

app.py

@@ -48,6 +48,14 @@ def signup():
         password = request.form.get("password")
         referral = request.form.get("referral")
         try:
+            # Generate a unique referral code
+            while True:
+                referral_code = generate_referral_code()
+                query = f"SELECT Id FROM Customer_Login__c WHERE Customer_Referral_Code__c = '{referral_code}'"
+                result = sf.query(query)
+                if not result["records"]:  # If no existing record with this code, break the loop
+                    break
+                    
             sf.Customer_Login__c.create({
                 "Name": name,
                 "Phone_Number__c": phone,