Update app.py

app.py

@@ -162,98 +162,90 @@ def generate_custom_dish():
     except Exception as e:
         return jsonify({"success": False, "error": str(e)}), 500
 
-import re
-@app.route("/edit_profile", methods=["GET", "POST"])
-def edit_profile():
+@app.route("/customer_details", methods=["GET"])
+def customer_details():
     email = session.get('user_email')  # Get logged-in user's email
     if not email:
         return redirect(url_for("login"))
 
     try:
-        # Fetch user details from Salesforce
-        result = sf.query(f"""
-            SELECT Id, Name, Email__c, Phone_Number__c, Password__c
+        # Fetch customer details from Salesforce based on the email
+        customer_record = sf.query(f"""
+            SELECT Id, Name, Email__c, Phone_Number__c, Referral__c, Reward_Points__c
             FROM Customer_Login__c
             WHERE Email__c = '{email}'
+            LIMIT 1
         """)
 
-        if not result['records']:
+        if not customer_record.get("records"):
+            flash("Customer not found", "danger")
             return redirect(url_for("login"))
 
-        user = result['records'][0]
-        user_id = user.get("Id")
-        user_name = user.get("Name")
-        user_phone = user.get("Phone_Number__c")
-        user_email = user.get("Email__c")
-
-    except Exception as e:
-        print(f"Error fetching user data: {str(e)}")
-        return jsonify({"success": False, "message": "Error fetching user data"})
-
-    try:
-        # Process user profile update
-        new_name = request.form.get('name')
-        new_email = request.form.get('email')
-        new_phone = request.form.get('phone')
-        new_password = request.form.get('password')
+        customer = customer_record["records"][0]
 
-        update_data = {
-            'Name': new_name,
-            'Email__c': new_email,
-            'Phone_Number__c': new_phone
+        # 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)
         }
 
-        if new_password:
-            update_data['Password__c'] = new_password
-
-        # Update Salesforce record
-        sf.Customer_Login__c.update(user_id, update_data)
-
-        return redirect(url_for('customer_details'))
+        return render_template("customer_details.html", customer=customer_data)
 
     except Exception as e:
-        return render_template("edit_profile.html", user_name=user_name, user_phone=user_phone, user_email=user_email, error=str(e))
+        flash(f"Error fetching customer details: {str(e)}", "danger")
+        return redirect(url_for("login"))
 
-import re
-@app.route("/customer_details", methods=["GET"])
-def customer_details():
+@app.route("/update_profile", methods=["POST"])
+def update_profile():
     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
+        return jsonify({'status': 'error', 'message': 'User not logged in'})
 
     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
+        # Fetch user details from Salesforce
+        result = sf.query(f"""
+            SELECT Id, Name, Email__c, Phone_Number__c, Referral__c, Reward_Points__c
             FROM Customer_Login__c
             WHERE Email__c = '{email}'
-            LIMIT 1
         """)
 
-        # If no customer record found, handle it
-        if not customer_record.get("records"):
-            return jsonify({"success": False, "message": "Customer not found in Salesforce"})
+        if not result['records']:
+            return jsonify({'status': 'error', 'message': 'User not found'})
 
-        # Get the customer details
-        customer = customer_record["records"][0]
+        user = result['records'][0]
+        user_id = user.get("Id")
 
-        # 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)
+        # Get updated profile data from the form
+        new_name = request.form.get('customerName')
+        new_email = request.form.get('email')
+        new_phone = request.form.get('phone')
+        new_referral_code = request.form.get('referralCode')
+        new_reward_points = request.form.get('rewardPoints')
+
+        # Prepare data for Salesforce update
+        update_data = {
+            'Name': new_name,
+            'Email__c': new_email,
+            'Phone_Number__c': new_phone,
+            'Referral__c': new_referral_code,
+            'Reward_Points__c': new_reward_points
         }
 
-        # Return the customer details as JSON response
-        return render_template("customer_details.html", customer=customer_data)
+        # Update Salesforce record
+        sf.Customer_Login__c.update(user_id, update_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 jsonify({
+            'status': 'success',
+            'message': 'Profile updated successfully!',
+            'data': update_data
+        })
 
+    except Exception as e:
+        return jsonify({'status': 'error', 'message': str(e)})
+        
 @app.route("/order-history", methods=["GET"])
 def order_history():
     email = session.get('user_email')  # Get logged-in user's email