Update routes.py

routes.py

@@ -120,9 +120,18 @@ def api_register():
         db.session.add(user)
         db.session.commit()
         
+        # Set session data
+        session.clear()  # Clear any existing session
         session['user_id'] = user.user_id
+        session.permanent = True  # Make the session permanent
         
-        return jsonify({'success': True, 'message': 'Account created successfully'})
+        app.logger.info(f"User registered successfully: {user.user_id}")
+        
+        return jsonify({
+            'success': True, 
+            'message': 'Account created successfully',
+            'redirect': url_for('chat')
+        })
     
     except Exception as e:
         app.logger.error(f"Registration error: {e}")