Update app.py

app.py

@@ -92,9 +92,16 @@ def register():
         email = data.get('email')
         phone = data.get('phone')
 
+        # Ensure phone number is in correct format (remove any unwanted characters)
+        phone = ''.join(e for e in phone if e.isalnum())  # Remove non-alphanumeric characters
+
         if not name or not email or not phone:
             return jsonify({"success": False, "message": "All fields are required."}), 400
 
+        # Check if the phone number is exactly 10 digits
+        if len(phone) != 10:
+            return jsonify({"success": False, "message": "Phone number must be 10 digits."}), 400
+
         try:
             # Create a new record in Salesforce's Customer_Login__c object
             customer_login = sf.Customer_Login__c.create({
@@ -102,20 +109,24 @@ def register():
                 'Email__c': email,
                 'Phone_Number__c': phone
             })
-            return jsonify({"success": True, "message": "Registration successful!"}), 200
+            return redirect(url_for('login'))  # Redirect to login page after successful registration
         except Exception as e:
             return jsonify({"success": False, "message": f"Registration failed: {str(e)}"}), 500
 
     return render_template("register.html")  # Registration page
 
-# Route for validating login
+# Route for Login Validation
 @app.route('/login', methods=['POST'])
 def login():
     try:
-        data = request.get_json()
+        data = request.form
         login_email = data.get("email")
         login_mobile = data.get("mobile")
 
+        # Ensure phone number is in correct format (remove unwanted characters)
+        login_mobile = ''.join(e for e in login_mobile if e.isalnum())
+
+        # Validate if email and mobile are present
         if not login_email or not login_mobile:
             return jsonify({"success": False, "message": "Missing email or mobile number"}), 400
 
@@ -124,7 +135,8 @@ def login():
 
         if result['records']:
             user_name = result['records'][0]['Name']
-            return jsonify({"success": True, "message": "Login successful", "name": user_name})
+            session['user_email'] = login_email  # Store user email in session
+            return redirect(url_for('menu_page'))  # Redirect to menu page after successful login
         else:
             return jsonify({"success": False, "message": "Invalid email or mobile number"}), 401