Update app.py

app.py

@@ -147,20 +147,32 @@ def admin_page():
 @app.route('/register')
 def register_page():
     """User registration page - redirect if already logged in"""
-    if 'user_id' in session:
-        role = session.get('role')
+    current_user = get_current_user()
+    if current_user:
+        role = current_user.role
         if role == 'admin': return redirect(url_for('admin_page'))
         return redirect(url_for('student_page' if role == 'student' else 'lecturer_page'))
+    
+    # Clear stale session if user doesn't exist
+    if 'user_id' in session:
+        session.clear()
+        
     return render_template('register.html')
 
 
 @app.route('/login')
 def login_page():
     """Login page - redirect if already logged in"""
-    if 'user_id' in session:
-        role = session.get('role')
+    current_user = get_current_user()
+    if current_user:
+        role = current_user.role
         if role == 'admin': return redirect(url_for('admin_page'))
         return redirect(url_for('student_page' if role == 'student' else 'lecturer_page'))
+    
+    # Clear stale session if user doesn't exist
+    if 'user_id' in session:
+        session.clear()
+
     return render_template('login.html')