Update routes.py

routes.py

@@ -122,10 +122,15 @@ def api_register():
         
         session['user_id'] = user.user_id
         
-        return jsonify({'success': True, 'message': 'Account created successfully'})
+        # Verify session was set properly
+        if 'user_id' not in session or session['user_id'] != user.user_id:
+            return jsonify({'success': False, 'message': 'Session creation failed'})
+        
+        return jsonify({'success': True, 'message': 'Account created successfully', 'redirect': url_for('chat')})
     
     except Exception as e:
         app.logger.error(f"Registration error: {e}")
+        db.session.rollback()  # Rollback on error
         return jsonify({'success': False, 'message': 'Registration failed'})
 
 @app.route('/api/conversations')