Update routes.py

routes.py

@@ -120,18 +120,30 @@ def api_register():
         db.session.add(user)
         db.session.commit()
         
+        # Set session
+        session.clear()  # Clear any existing session
         session['user_id'] = user.user_id
+        session.modified = True  # Ensure session is saved
+        
+        app.logger.info(f"Setting session for user {user.user_id}")
         
         # Verify session was set properly
         if 'user_id' not in session or session['user_id'] != user.user_id:
+            app.logger.error(f"Session creation failed for user {user.user_id}")
             return jsonify({'success': False, 'message': 'Session creation failed'})
         
-        return jsonify({'success': True, 'message': 'Account created successfully', 'redirect': url_for('chat')})
+        app.logger.info(f"Session created successfully for user {user.user_id}")
+        return jsonify({
+            'success': True, 
+            'message': 'Account created successfully',
+            'redirect': url_for('chat'),
+            'user_id': user.user_id  # Send user_id back to client
+        })
     
     except Exception as e:
         app.logger.error(f"Registration error: {e}")
         db.session.rollback()  # Rollback on error
-        return jsonify({'success': False, 'message': 'Registration failed'})
+        return jsonify({'success': False, 'message': str(e)})
 
 @app.route('/api/conversations')
 def api_conversations():