Update app.py

app.py

@@ -95,11 +95,27 @@ def login():
         return jsonify({'error': 'Missing required fields'}), 400
 
     try:
-        customer_login = create_salesforce_record(sf, name, email, phone_number)
-        return redirect("/menu")  # Redirect to the menu page after successful login
+        # Check if the user already exists in Salesforce using email or phone number
+        query = f"SELECT Id, Name, Email__c, Phone_Number__c FROM Customer_Login__c WHERE Email__c = '{email}' AND Phone_Number__c = '{phone_number}' LIMIT 1"
+        result = sf.query(query)
+
+        if result['totalSize'] > 0:
+            # User exists and credentials match, proceed to dashboard
+            return redirect("/menu")  # Redirect to the menu page after successful login
+        else:
+            # User not found or credentials don't match
+            return jsonify({'error': 'Invalid email or phone number'}), 401  # Unauthorized
+
     except Exception as e:
-        return jsonify({'error': f'Failed to create record in Salesforce: {str(e)}'}), 500
+        return jsonify({'error': f'Error during login or Salesforce query: {str(e)}'}), 500
+
 
+@app.route('/logout')
+def logout():
+    # Clear the session to log out the user
+    session.pop('user', None)  # Removes the 'user' from session
+    return redirect(url_for('login'))  # Redirect to login page
+    
 @app.route("/submit", methods=["POST"])
 def submit():
     data = request.json