Update app.py

app.py

@@ -120,76 +120,69 @@ def signup():
 
 
 
-@app.route("/login", methods=["GET", "POST"])
-def login():
-    if request.method == "POST":
-        email = request.form.get("email")
-        password = request.form.get("password")
-        print(f"Login attempt with email: {email}")  # Debug log
+@app.route("/redeem_coupon", methods=["POST"])
+def redeem_coupon():
+    email = session.get('user_email')
 
-        try:
-            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']}")
-
-                reward_points = user.get("Reward_Points__c", 0)
-
-                if reward_points >= 500:
-                    print(f"User {email} has {reward_points} reward points. Generating coupon...")
-
-                    # Generate 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}'
-                    """)
-
-                    if coupon_query["records"]:
-                        # Append the new coupon to existing ones
-                        coupon_record = coupon_query["records"][0]
-                        referral_coupon_id = coupon_record["Id"]
-                        existing_coupons = coupon_record.get("Coupon_Code__c", "")
-
-                        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
-                        })
-                    else:
-                        # Create a new record for the user
-                        sf.Referral_Coupon__c.create({
-                            "Referral_Email__c": email,
-                            "Coupon_Code__c": 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}")
+    if not email:
+        return jsonify({"success": False, "message": "User not logged in"})
 
-                return redirect(url_for("menu"))
+    try:
+        # Fetch reward points
+        customer_record = sf.query(f"""
+            SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}'
+        """)
+        customer = customer_record.get("records", [])[0] if customer_record else None
 
-            else:
-                print("Invalid credentials!")
-                return render_template("login.html", error="Invalid credentials!")
+        if not customer:
+            return jsonify({"success": False, "message": "User not found"})
 
-        except Exception as e:
-            print(f"Error during login: {str(e)}")
-            return render_template("login.html", error=f"Error: {str(e)}")
+        reward_points = customer.get("Reward_Points__c", 0)
 
-    return render_template("login.html")
+        if reward_points < 500:
+            return jsonify({"success": False, "message": "Not enough reward points to redeem a 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}'
+        """)
+
+        if coupon_query["records"]:
+            # ✅ Case 1: User already has a referral coupon record → Append new coupon
+            coupon_record = coupon_query["records"][0]
+            referral_coupon_id = coupon_record["Id"]
+            existing_coupons = coupon_record.get("Coupon_Code__c", "")
+
+            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"✅ Coupon appended for existing user: {email}")
+        else:
+            # ❌ Case 2: No record found → Create a new record in Referral_Coupon__c
+            sf.Referral_Coupon__c.create({
+                "Referral_Email__c": email,
+                "Coupon_Code__c": new_coupon_code
+            })
+            print(f"✅ New Referral_Coupon__c record created for {email}")
+
+        # ✅ Deduct 500 reward points
+        new_reward_points = reward_points - 500
+        sf.Customer_Login__c.update(customer['Id'], {
+            "Reward_Points__c": new_reward_points
+        })
+        print(f"✅ Reward points updated: New balance = {new_reward_points}")
+
+        return jsonify({"success": True, "message": f"Coupon {new_coupon_code} generated!", "new_reward_points": new_reward_points})
+
+    except Exception as e:
+        return jsonify({"success": False, "error": str(e)})
 
-    return render_template("login.html")
 @app.route("/menu", methods=["GET", "POST"])
 def menu():
     selected_category = request.args.get("category", "All")