Update app.py

app.py

@@ -66,6 +66,60 @@ def generate_coupon_code(length=10):
 
 
 import re
+@app.route("/edit_profile", methods=["GET", "POST"])
+def edit_profile():
+    user_email = session.get('user_email')
+
+    if not user_email:
+        return redirect(url_for('login'))  # If user is not logged in, redirect to login page
+
+    try:
+        # Fetch current user details from Salesforce
+        user_query = f"SELECT Name, Email__c, Phone_Number__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
+        user_result = sf.query(user_query)
+
+        if not user_result['records']:
+            return redirect(url_for('login'))  # If no user found, redirect to login
+
+        current_user = user_result['records'][0]
+        user_name = current_user.get('Name')
+        user_phone = current_user.get('Phone_Number__c')
+        user_email = current_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"})
+
+    # Handle form submission if POST request
+    if request.method == 'POST':
+        new_name = request.form.get('name')
+        new_email = request.form.get('email')
+        new_phone = request.form.get('phone')
+        new_password = request.form.get('password')  # Assuming password change is optional
+
+        # Update the customer details in Salesforce
+        try:
+            # Create the update dictionary
+            update_data = {
+                'Name': new_name,
+                'Email__c': new_email,
+                'Phone_Number__c': new_phone
+            }
+
+            # Only update password if provided
+            if new_password:
+                update_data['Password__c'] = new_password
+
+            # Update user details in Salesforce
+            sf.Customer_Login__c.update(current_user['Id'], update_data)
+            flash("Profile updated successfully!", "success")
+            return redirect(url_for('customer_details'))  # Redirect back to profile page
+
+        except Exception as e:
+            flash(f"Error updating profile: {str(e)}", "danger")
+            return render_template('edit_profile.html', user_name=user_name, user_phone=user_phone, user_email=user_email)
+
+    return render_template('edit_profile.html', user_name=user_name, user_phone=user_phone, user_email=user_email)
 
 import re
 @app.route("/customer_details", methods=["GET"])