Update app.py

app.py

@@ -68,23 +68,45 @@ def generate_coupon_code(length=10):
 import re
 
 import re
-@app.route('/customer_details')
+@app.route("/customer_details", methods=["GET"])
 def customer_details():
-    email = session.get('user_email')  # Get logged-in user email from session
+    email = session.get('user_email')  # Get logged-in user's email
+
+    if not email:
+        return redirect(url_for("login"))  # If no email is found, redirect to login
+
+    try:
+        # Fetch customer details from Salesforce based on the email
+        customer_record = sf.query(f"""
+            SELECT Name, Email__c, Phone_Number__c, Referral__c, Reward_Points__c
+            FROM Customer_Login__c
+            WHERE Email__c = '{email}'
+            LIMIT 1
+        """)
 
-    # Fetch customer details using the email
-    customer = get_customer_details_from_db(email)  # Query Salesforce or your DB
+        # If no customer record found, handle it
+        if not customer_record.get("records"):
+            return jsonify({"success": False, "message": "Customer not found in Salesforce"})
 
-    # Create a dictionary with customer details, including referral code and reward points
-    customer_data = {
-        'name': customer.Name,
-        'email': customer.Email__c,
-        'phone': customer.Phone_Number__c,
-        'referral_code': customer.Referral__c,  # Referral code
-        'reward_points': customer.Reward_Points__c  # Reward points
-    }
+        # Get the customer details
+        customer = customer_record["records"][0]
+
+        # Prepare the data to return to the frontend
+        customer_data = {
+            "name": customer.get("Name", ""),
+            "email": customer.get("Email__c", ""),
+            "phone": customer.get("Phone_Number__c", ""),
+            "referral_code": customer.get("Referral__c", ""),
+            "reward_points": customer.get("Reward_Points__c", 0)
+        }
+
+        # Return the customer details as JSON response
+        return render_template("customer_details.html", customer=customer_data)
+
+    except Exception as e:
+        print(f"Error fetching customer details: {str(e)}")
+        return jsonify({"success": False, "message": f"Error fetching customer details: {str(e)}"})
 
-    return render_template('customer_details.html', customer=customer_data)
 @app.route("/order-history", methods=["GET"])
 def order_history():
     email = session.get('user_email')  # Get logged-in user's email