Update app.py

app.py

@@ -79,26 +79,19 @@ def signup():
                 query = f"SELECT Id, Name, Email__c FROM Customer_Login__c WHERE Customer_Referral_Code__c = '{referral_code}'"
                 referral_result = sf.query(query)
 
-                if len(result['records']) == 0:
-                    return "Referral not found. No coupon generated."
-                    # Fetch referral details
-                referrer = referral_result["records"][0]
-                    
-
-                    # Generate a unique coupon code
-                coupon_code = generate_coupon_code()
-
-                sf.Referral_Coupon__c.create({
-                    "Referral_Name__c": referrer["Name"],
-                    "Referral_Email__c": referrer["Referral_Email__c"],
-                    "Coupon_Code__c": coupon_code,
-                    "Coupon_Status__c": "Active"
-                })
-                    
+                if len(referral_result['records']) == 0:
+                        return "Referral not found. No coupon generated."
                 else:
-                    print("Referral code is invalid or does not exist.")
-                    return render_template("signup.html", error="Invalid referral code provided.")
-
+                    referrer = referral_result["records"][0]
+                        # Generate a unique coupon code
+                    coupon_code = generate_coupon_code()
+                    
+                    sf.Referral_Coupon__c.create({
+                        "Referral_Name__c": referrer["Name"],
+                        "Referral_Email__c": referrer["Email__c"],
+                        "Coupon_Code__c": coupon_code,
+                        "Coupon_Status__c": "Active"
+                    })
             return redirect(url_for("login"))
         except Exception as e:
             return render_template("signup.html", error=f"Error: {str(e)}")