diff --git a/bot.py b/bot.py
index 60eeeca..7e56d9e 100644
--- a/bot.py
+++ b/bot.py
@@ -781,6 +781,7 @@ class MenuBuilder:
         # Для главного меню — динамически меняем кнопку ИИ
         if menu_name == "main" and user_id:
             state = state_manager.get(user_id)
+            logger.info(f"get_keyboard: user_id={user_id}, ai_chat_mode={state.ai_chat_mode}")
             
             for item in items:
                 if item.callback == "toggle_ai_chat":
@@ -791,6 +792,7 @@ class MenuBuilder:
                     else:
                         label = f"❌ Включить чат с ИИ"
                         callback = "toggle_ai_chat_on"
+                    logger.info(f"get_keyboard: label={label}, callback={callback}")
                     button = InlineKeyboardButton(label, callback_data=callback)
                 else:
                     button = InlineKeyboardButton(item.label, callback_data=item.callback)
@@ -1445,30 +1447,14 @@ async def menu_callback(update: Update, context: ContextTypes.DEFAULT_TYPE):
         )
         state.current_menu = "main"
     
-    elif callback == "toggle_ai_chat":
-        # Переключаем режим
-        state.ai_chat_mode = not state.ai_chat_mode
-        
-        ai_status = "✅ ВКЛЮЧЕН" if state.ai_chat_mode else "❌ ВЫКЛЮЧЕН"
-        action = "включён" if state.ai_chat_mode else "выключен"
-        
-        await query.edit_message_text(
-            f"🏠 *Главное меню*\n\n"
-            f"💬 *Чат с ИИ:* {ai_status}\n\n"
-            f"Режим чата с агентом {action}.\n"
-            f"Теперь все сообщения будут отправляться в Qwen Code.",
-            parse_mode="Markdown",
-            reply_markup=menu_builder.get_keyboard("main", user_id=query.from_user.id)
-        )
-        state.current_menu = "main"
-    
     elif callback in ["toggle_ai_chat_on", "toggle_ai_chat_off"]:
-        # Переключаем режим (новые callback_data для правильной работы кнопки)
+        # Переключаем режим чата с ИИ
         state.ai_chat_mode = not state.ai_chat_mode
-        
+        logger.info(f"toggle_ai_chat: user_id={user_id}, new_mode={state.ai_chat_mode}")
+
         ai_status = "✅ ВКЛЮЧЕН" if state.ai_chat_mode else "❌ ВЫКЛЮЧЕН"
         action = "включён" if state.ai_chat_mode else "выключен"
-        
+
         await query.edit_message_text(
             f"🏠 *Главное меню*\n\n"
             f"💬 *Чат с ИИ:* {ai_status}\n\n"