Update app.py

app.py

@@ -203,61 +203,43 @@ def login():
         print(f"Login attempt with email: {email}")  # Debug log
 
         try:
-            # Fetch user details, including Name and Reward_Points__c
+            # Fetch user details from Salesforce
             query = f"SELECT Id, Name, Email__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c='{email}' AND Password__c='{password}'"
             result = sf.query(query)
 
             if result["records"]:
                 user = result["records"][0]
                 session['user_id'] = user['Id']
-                session['user_email'] = email
-                print(f"Session variables set: user_id={session['user_id']}, user_email={session['user_email']}")
-
-                user_name = user.get("Name", "")  # Get the user's name
-                reward_points = user.get("Reward_Points__c") or 0  # Ensures reward_points is always an integer
 
+                # ✅ Always store or update session email
+                if 'user_email' not in session or session['user_email'] != email:
+                    session['user_email'] = email
+                    session['user_name'] = user.get("Name", "")
+                    print(f"✅ Session email updated: {session['user_email']}")
 
+                reward_points = user.get("Reward_Points__c") or 0
 
+                # Coupon generation logic (if reward points >= 500)
                 if reward_points >= 500:
-                    print(f"User {email} has {reward_points} reward points. Generating coupon...")
-
-                    # Generate a new coupon code
                     new_coupon_code = generate_coupon_code()
-
-                    # Check if user already has a record in Referral_Coupon__c
-                    coupon_query = sf.query(f"""
-                        SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'
-                    """)
+                    coupon_query = sf.query(f"SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'")
 
                     if coupon_query["records"]:
-                        # If record exists, append the new coupon
                         coupon_record = coupon_query["records"][0]
                         referral_coupon_id = coupon_record["Id"]
                         existing_coupons = coupon_record.get("Coupon_Code__c", "")
 
-                        # Append the new coupon on the next line
                         updated_coupons = f"{existing_coupons}\n{new_coupon_code}".strip()
-
-                        # Update the Referral_Coupon__c record
-                        sf.Referral_Coupon__c.update(referral_coupon_id, {
-                            "Coupon_Code__c": updated_coupons
-                        })
-                        print(f"Updated existing coupon record for {email}. New Coupon: {new_coupon_code}")
+                        sf.Referral_Coupon__c.update(referral_coupon_id, {"Coupon_Code__c": updated_coupons})
                     else:
-                        # If no record exists, create a new one with Referral_Name__c
                         sf.Referral_Coupon__c.create({
                             "Referral_Email__c": email,
-                            "Name": user_name,  # Store user's name in Referral_Coupon__c
+                            "Name": user.get("Name", ""),
                             "Coupon_Code__c": new_coupon_code
                         })
-                        print(f"Created new coupon record for {email} with name {user_name}. Coupon: {new_coupon_code}")
 
-                    # Subtract 500 reward points from user's account
                     new_reward_points = reward_points - 500
-                    sf.Customer_Login__c.update(user['Id'], {
-                        "Reward_Points__c": new_reward_points
-                    })
-                    print(f"Coupon {new_coupon_code} generated and 500 points deducted. New balance: {new_reward_points}")
+                    sf.Customer_Login__c.update(user['Id'], {"Reward_Points__c": new_reward_points})
 
                 return redirect(url_for("menu"))