Update app.py

app.py

@@ -27,20 +27,13 @@ except Exception as e:
     sf = None
 
 # Route for login page
-@app.route('/')
-def login():
-    print("Login route accessed")  # Debug statement
-    return render_template('login.html')
-
-# Route to process login
 @app.route('/auth', methods=['POST'])
 def auth():
-    print("Auth route accessed")  # Debug statement
     email = request.form['email']
     password = request.form['password']
 
     if not sf:
-        return "Salesforce connection failed. Please check credentials and try again."
+        return render_template('login.html', error_message="Salesforce connection failed. Please try again.")
 
     try:
         # Query Salesforce for user authentication
@@ -48,15 +41,17 @@ def auth():
         result = sf.query(query)
 
         if result['totalSize'] == 0:
-            return "Invalid Login Details"
+            return render_template('login.html', error_message="Invalid login details. Please check your email and password.")
 
         customer = result['records'][0]
         reward_points = customer['Reward_Points__c']
 
-        # Redirect to rewards page
+        # Redirect to rewards page or any other page
         return redirect(url_for('rewards', customer_id=customer['Id'], points=reward_points))
+
     except Exception as e:
-        return f"Error during authentication: {e}"
+        return render_template('login.html', error_message=f"Error during authentication: {e}")
+
 
 # Route to display rewards page
 @app.route('/rewards/<customer_id>')