Update routes.py

routes.py

@@ -60,20 +60,33 @@ def register():
 
 @app.route('/chat')
 def chat():
-    if 'user_id' not in session:
-        return redirect(url_for('landing'))
-    
-    user = User.query.get(session['user_id'])
-    if not user:
+    try:
+        user_id = session.get('user_id')
+        if not user_id:
+            app.logger.warning("No user_id in session, redirecting to landing")
+            return redirect(url_for('landing'))
+        
+        user = User.query.get(user_id)
+        if not user:
+            app.logger.warning(f"User {user_id} not found in database, clearing session")
+            session.clear()
+            return redirect(url_for('landing'))
+        
+        # Update user online status
+        user.online = True
+        user.last_seen = datetime.utcnow()
+        db.session.commit()
+        
+        # Ensure session is still valid after database operations
+        session.modified = True
+        
+        app.logger.info(f"Rendering chat for user {user_id}")
+        return render_template('chat.html', user=user)
+        
+    except Exception as e:
+        app.logger.error(f"Error in chat route: {e}")
         session.clear()
         return redirect(url_for('landing'))
-    
-    # Update user online status
-    user.online = True
-    user.last_seen = datetime.utcnow()
-    db.session.commit()
-    
-    return render_template('chat.html', user=user)
 
 @app.route('/settings')
 def settings():
@@ -101,6 +114,12 @@ def logout():
 
 # API Routes
 
+@app.route('/api/check_session')
+def check_session():
+    if 'user_id' in session:
+        return jsonify({'success': True, 'user_id': session['user_id']})
+    return jsonify({'success': False, 'message': 'No active session'}), 401
+
 @app.route('/api/register', methods=['POST'])
 def api_register():
     try:
@@ -124,14 +143,25 @@ def api_register():
         session.clear()  # Clear any existing session
         session['user_id'] = user.user_id
         session.permanent = True  # Make the session permanent
+        session.modified = True  # Ensure the session is saved
         
-        app.logger.info(f"User registered successfully: {user.user_id}")
+        # Force a database commit to ensure user is saved
+        db.session.commit()
         
-        return jsonify({
+        app.logger.info(f"User registered successfully: {user.user_id}, Session: {session}")
+        
+        # Create response
+        response = jsonify({
             'success': True, 
             'message': 'Account created successfully',
-            'redirect': url_for('chat')
+            'redirect': url_for('chat'),
+            'user_id': user.user_id
         })
+        
+        # Set a cookie to help with session persistence
+        response.set_cookie('user_registered', 'true', max_age=60)
+        
+        return response
     
     except Exception as e:
         app.logger.error(f"Registration error: {e}")