Накоплено постов: {len(samples)}. Дополнений профиля: {len(notes)}.
" + f"{html.escape(profile)}
" + "Присылай новые посты или сведения в режиме «Обучение» — профиль будет дополняться.
" + ) + await send_rich_message(message, rich_html, "обновления профиля", reply_markup=mode_keyboard("training")) + + +async def process_training_text(message: Message, text: str) -> None: + if message.from_user is None or message.bot is None: + await message.answer("Не удалось определить пользователя или бота.") + return + with diagnostic_operation(): + try: + async with typing_status(message.bot, message.chat.id): + await _rebuild_account_profile(message.from_user.id, message, new_note=text) + except Exception as exc: + await report_error(message, "обучение профиля аккаунта", exc) + + +def training_confirmation_keyboard(user_id: int) -> InlineKeyboardMarkup: return InlineKeyboardMarkup( inline_keyboard=[[ InlineKeyboardButton(text="✅ Подтвердить", callback_data=f"training:reset:confirm:{user_id}"), @@ -483,7 +524,7 @@ async def clear_command(message: Message, state: FSMContext) -> None: "Будут удалены стиль, примеры постов, история публикаций и идей, а также настройки.\n" "Подписка, триал и служебные API-логи останутся.", parse_mode="HTML", - reply_markup=training_reset_confirmation_keyboard(message.from_user.id), + reply_markup=training_confirmation_keyboard(message.from_user.id), ) @@ -694,26 +735,11 @@ async def _process_image_group(messages: list[Message], bot: Bot) -> None: images.append(image_payload) if not images: return - previous_tone = await asyncio.to_thread(get_tone_of_voice, first.from_user.id) new_samples = await extract_posts_from_images(images, user_id=first.from_user.id) if not new_samples: - await first.answer("Не нашла читаемых текстов постов на скриншотах. Стиль не меняла.") + await first.answer("Не нашла читаемых текстов на скриншотах. Обучение не меняла.") return - all_samples = await asyncio.to_thread(add_post_samples, first.from_user.id, new_samples) - tone = await create_tone_profile(all_samples, previous_tone, user_id=first.from_user.id) - await asyncio.to_thread(save_tone_of_voice, first.from_user.id, tone) - rich_html = ( - "Образцов учтено: {len(all_samples)}.
" - f"{html.escape(tone)}
" - "Скриншоты сохранены как образцы; тексты постов не переписывались.
" - ) - await send_rich_message( - first, - rich_html, - "обновления стиля", - reply_markup=mode_keyboard("training"), - ) + await _rebuild_account_profile(first.from_user.id, first, new_posts=new_samples) async def _collect_media_group(message: Message, bot: Bot) -> list[Message]: @@ -800,12 +826,23 @@ async def _process_voice(message: Message, bot: Bot, state: FSMContext) -> None: if not transcript: await message.answer("Не удалось распознать речь в аудио.") return + if await current_mode(state) == "training": + if message.from_user is None: + await message.answer("Не удалось определить пользователя.") + return + await _rebuild_account_profile(message.from_user.id, message, new_note=transcript) + return await _classify_and_respond(message, transcript, show_typing=False, state=state) @router.message(F.voice | F.audio) async def transcribe_voice(message: Message, bot: Bot, state: FSMContext) -> None: - if not await require_mode(message, state, "generation"): + mode = await current_mode(state) + if mode not in {"training", "generation"}: + await message.answer( + "Голосовые принимаются в режимах «Обучение» и «Генерация».", + reply_markup=keyboard_for_mode(mode), + ) return await notify_development_deploy(bot) with diagnostic_operation(): @@ -864,10 +901,13 @@ async def handle_text(message: Message, state: FSMContext) -> None: return mode = await current_mode(state) + if mode == "training": + await process_training_text(message, text) + return if mode != "generation": prompts = { "lobby": "Выбери режим работы кнопкой ниже.", - "training": "Пришли скриншоты своих постов или используй /style, чтобы посмотреть стиль.", + "training": "Пришли пост текстом, скриншотом, голосовым или аудиофайлом — профиль будет накапливаться.", "subscription": "Статус уже показан выше. Для продления нажми кнопку «Продлить подписку».", "help": "Открой нужный раздел справки выше или выбери другой режим.", } @@ -885,7 +925,7 @@ async def handle_text(message: Message, state: FSMContext) -> None: async def handle_other_message(message: Message, state: FSMContext) -> None: mode = await current_mode(state) if mode == "training": - text = "В режиме «Обучение» пришли скриншоты постов." + text = "Пришли пост текстом, скриншотом, голосовым или аудиофайлом. Профиль будет постепенно дополняться." elif mode == "generation": text = "Для генерации отправь текстовый запрос или голосовое сообщение." elif mode == "subscription": diff --git a/help_message.py b/help_message.py index 9dcfa2f..7e9bcd2 100644 --- a/help_message.py +++ b/help_message.py @@ -28,8 +28,8 @@ HELP_SECTIONS: dict[str, tuple[str, str]] = { ), "tone": ( "🎨 Tone of Voice", - """Пришли скриншот или галерею скриншотов своих опубликованных постов. Я распознаю текст, сохраню его как образцы и обновлю только правила Tone of Voice. Сами публикации не переписываю.
-Чтобы удалить стиль и историю обучения, нажми «Сбросить данные» и подтверди действие.
""", + """Пришли скриншоты постов, текстовые примеры или просто расскажи об аккаунте. Можно отправить голосовое или аудиофайл. Я буду постепенно накапливать и обновлять профиль аккаунта и Tone of Voice; новые сведения дополняют прежние.
+Чтобы начать обучение заново и удалить накопленное, нажми «Сбросить данные» и подтверди действие.
""", ), "edit": ( "✏️ Правка поста", diff --git a/tests/test_keyboards.py b/tests/test_keyboards.py index 39f1939..17992b5 100644 --- a/tests/test_keyboards.py +++ b/tests/test_keyboards.py @@ -100,16 +100,24 @@ class _StubMessage: class StyleCommandTests(unittest.IsolatedAsyncioTestCase): async def test_style_without_tone_explains_how_to_set_it(self): message = _StubMessage() - with patch.object(bot.asyncio, "to_thread", new=AsyncMock(side_effect=["", []])): + with patch.object(bot.asyncio, "to_thread", new=AsyncMock(side_effect=["", [], []])): await bot.style_command(message, _StubState("training")) self.assertIn("скриншот", message.answers[0][0]) async def test_style_shows_saved_tone(self): message = _StubMessage() - with patch.object(bot.asyncio, "to_thread", new=AsyncMock(side_effect=["Короткие фразы", ["a", "b"]])): + with patch.object(bot.asyncio, "to_thread", new=AsyncMock(side_effect=["Короткие фразы", ["a", "b"], []])): await bot.style_command(message, _StubState("training")) self.assertIn("Короткие фразы", message.answers[0][0]) - self.assertIn("Образцов сохранено: 2", message.answers[1][0]) + self.assertIn("Накоплено постов: 2", message.answers[1][0]) + + async def test_style_shows_saved_account_profile(self): + message = _StubMessage("/style") + state = _StubState("training") + with patch.object(bot.asyncio, "to_thread", new=AsyncMock(side_effect=["Профиль аккаунта", ["Пост"], ["Заметка"]])): + await bot.style_command(message, state) + self.assertIn("Профиль аккаунта", message.answers[0][0]) + self.assertIn("Накоплено постов: 1", message.answers[1][0]) async def test_style_is_rejected_outside_training(self): message = _StubMessage() @@ -220,6 +228,12 @@ class ModeRoutingTests(unittest.IsolatedAsyncioTestCase): await bot.handle_command_text(message, state) classify.assert_awaited_once_with(message, "Напиши пост", state=state) + async def test_training_text_is_saved_as_note_without_entering_generation(self): + message = _StubMessage("Кофейня в Калининграде для семей с детьми") + with patch.object(bot, "_rebuild_account_profile", new=AsyncMock()) as rebuild: + await bot.handle_text(message, _StubState("training")) + rebuild.assert_awaited_once_with(42, message, new_note="Кофейня в Калининграде для семей с детьми") + async def test_plain_text_is_only_sent_to_generator_in_generation_mode(self): message = _StubMessage("Напиши пост про запуск продукта") state = _StubState("generation") @@ -271,16 +285,40 @@ class ModeRoutingTests(unittest.IsolatedAsyncioTestCase): process_image.assert_not_awaited() self.assertIn("только в режиме «Обучение»", message.answers[0][0]) - async def test_voice_is_rejected_outside_generation(self): + async def test_voice_is_rejected_outside_learning_and_generation(self): message = _StubMessage() with ( patch.object(bot, "notify_development_deploy", new=AsyncMock()) as notify, patch.object(bot, "_process_voice", new=AsyncMock()) as process_voice, ): - await bot.transcribe_voice(message, AsyncMock(), _StubState("training")) + await bot.transcribe_voice(message, AsyncMock(), _StubState("subscription")) notify.assert_not_awaited() process_voice.assert_not_awaited() - self.assertIn("только в режиме «Генерация»", message.answers[0][0]) + self.assertIn("«Обучение» и «Генерация»", message.answers[0][0]) + + async def test_voice_in_training_transcribes_then_updates_profile_not_generation(self): + message = SimpleNamespace( + from_user=SimpleNamespace(id=42), + voice=SimpleNamespace(file_size=4, mime_type="audio/ogg"), + audio=None, + answer=AsyncMock(), + chat=SimpleNamespace(id=555), + ) + bot_client = SimpleNamespace(download=AsyncMock(side_effect=lambda file, destination: destination.write(b"ogg"))) + state = _StubState("training") + with ( + patch.object(bot, "notify_development_deploy", new=AsyncMock()), + patch.dict("os.environ", {"APINEX_API_KEY": "not-logged"}, clear=False), + patch.object(bot, "transcribe_audio", new=AsyncMock(return_value="Мы — кофейня в Калининграде")) as stt, + patch.object(bot, "_rebuild_account_profile", new=AsyncMock()) as rebuild, + patch.object(bot, "_classify_and_respond", new=AsyncMock()) as classify, + patch.object(bot, "typing_status", return_value=AsyncMock()), + ): + await bot._process_voice(message, bot_client, state) + stt.assert_awaited_once() + rebuild.assert_awaited_once_with(42, message, new_note="Мы — кофейня в Калининграде") + classify.assert_not_awaited() + class MenuPublishTests(unittest.IsolatedAsyncioTestCase): diff --git a/tests/test_writer_features.py b/tests/test_writer_features.py index 7447c1f..b9599e1 100644 --- a/tests/test_writer_features.py +++ b/tests/test_writer_features.py @@ -431,6 +431,7 @@ class ScreenshotToneWorkflowTests(unittest.IsolatedAsyncioTestCase): with ( patch.object(bot, "_download_message_image", new=AsyncMock(return_value=(b"image", "image/png"))), patch.object(bot, "get_tone_of_voice", return_value="Старый профиль"), + patch.object(bot, "get_training_notes", return_value=[]), patch.object(bot, "extract_posts_from_images", new=AsyncMock(return_value=["Новый пост"])), patch.object(bot, "add_post_samples", return_value=["Старый пост", "Новый пост"]) as add_samples, patch.object(bot, "create_tone_profile", new=AsyncMock(return_value="Обновлённый стиль")) as create_profile, @@ -442,12 +443,11 @@ class ScreenshotToneWorkflowTests(unittest.IsolatedAsyncioTestCase): add_samples.assert_called_once_with(7, ["Новый пост"]) create_profile.assert_awaited_once_with( - ["Старый пост", "Новый пост"], "Старый профиль", user_id=7 + ["Старый пост", "Новый пост"], "Старый профиль", user_id=7, account_notes=[] ) save_tone.assert_called_once_with(7, "Обновлённый стиль") - self.assertIn("Стиль обновлён", str(send_rich.await_args)) - self.assertIn("не переписывались", str(send_rich.await_args)) - self.assertIn("обновления стиля", str(send_rich.await_args)) + self.assertIn("Профиль аккаунта обновлён", str(send_rich.await_args)) + self.assertIn("обновления профиля", str(send_rich.await_args)) async def test_clear_command_resets_only_calling_user_profile_and_samples(self): @@ -486,7 +486,40 @@ class ScreenshotToneWorkflowTests(unittest.IsolatedAsyncioTestCase): clear_data.assert_not_called() -class StorageTests(unittest.TestCase): + async def test_training_notes_and_reset_preserve_source_separation(self): + import tempfile + from pathlib import Path + with tempfile.TemporaryDirectory() as temp_dir: + db = str(Path(temp_dir) / "training.sqlite3") + user_store.init_db(db) + self.assertEqual(user_store.add_training_note(23, " Кофейня в Калининграде ", db), ["Кофейня в Калининграде"]) + user_store.add_post_samples(23, ["Пост об открытии"], db) + self.assertEqual(user_store.get_training_notes(23, db), ["Кофейня в Калининграде"]) + self.assertEqual(user_store.get_post_samples(23, db), ["Пост об открытии"]) + user_store.clear_user_data(23, db) + self.assertEqual(user_store.get_training_notes(23, db), []) + self.assertEqual(user_store.get_post_samples(23, db), []) + + async def test_create_account_profile_includes_previous_profile_posts_and_notes(self): + captured = {} + async def fake_chat(model_setting, system_prompt, user_content, **kwargs): + captured.update(model_setting=model_setting, system_prompt=system_prompt, user_content=user_content, **kwargs) + return "Профиль аккаунта: кофейня; ToV: дружелюбный" + + with patch.object(apinex_client, "chat_text", side_effect=fake_chat): + profile = await apinex_client.create_tone_profile( + ["Открыли летнюю террасу"], "Старый профиль", user_id=54, + account_notes=["Кофейня в Калининграде", "Целевая аудитория — жители района"], + ) + data = json.loads(captured["user_content"]) + self.assertEqual(data["previous_account_profile"], "Старый профиль") + self.assertEqual(data["accumulated_post_samples"], ["Открыли летнюю террасу"]) + self.assertEqual(data["accumulated_user_notes"], ["Кофейня в Калининграде", "Целевая аудитория — жители района"]) + profile_prompt = captured["system_prompt"].lower() + self.assertIn("накопленные примеры публикаций", profile_prompt) + self.assertIn("чем занимается аккаунт", profile_prompt) + self.assertIn("Профиль аккаунта", profile) + def test_api_log_schema_migrates_existing_database(self): import sqlite3 import tempfile diff --git a/user_store.py b/user_store.py index 5a53468..1b5d083 100644 --- a/user_store.py +++ b/user_store.py @@ -44,6 +44,17 @@ def init_db(db_path: str | None = None) -> None: ) """ ) + connection.execute( + """ + CREATE TABLE IF NOT EXISTS user_training_notes ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + telegram_user_id INTEGER NOT NULL, + note_text TEXT NOT NULL, + created_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP, + UNIQUE (telegram_user_id, note_text) + ) + """ + ) connection.execute( """ CREATE TABLE IF NOT EXISTS generated_posts ( @@ -225,6 +236,30 @@ def get_post_samples(user_id: int, db_path: str | None = None) -> list[str]: (user_id,), ).fetchall() return [str(row["post_text"]) for row in rows] +def add_training_note(user_id: int, note: str, db_path: str | None = None) -> list[str]: + text = note.strip() + if not text: + return get_training_notes(user_id, db_path) + with closing(connect(db_path)) as connection, connection: + connection.execute("INSERT OR IGNORE INTO users (telegram_user_id) VALUES (?)", (user_id,)) + connection.execute( + "INSERT OR IGNORE INTO user_training_notes (telegram_user_id, note_text) VALUES (?, ?)", + (user_id, text), + ) + rows = connection.execute( + "SELECT note_text FROM user_training_notes WHERE telegram_user_id = ? ORDER BY id", + (user_id,), + ).fetchall() + return [str(row["note_text"]) for row in rows] + + +def get_training_notes(user_id: int, db_path: str | None = None) -> list[str]: + with closing(connect(db_path)) as connection: + rows = connection.execute( + "SELECT note_text FROM user_training_notes WHERE telegram_user_id = ? ORDER BY id", + (user_id,), + ).fetchall() + return [str(row["note_text"]) for row in rows] def _bounded_json(value: str, max_bytes: int = API_LOG_MAX_FIELD_BYTES) -> str: @@ -354,6 +389,7 @@ def get_idea_history(user_id: int, limit: int = 30, db_path: str | None = None) def clear_user_data(user_id: int, db_path: str | None = None) -> None: with closing(connect(db_path)) as connection, connection: + connection.execute("DELETE FROM user_training_notes WHERE telegram_user_id = ?", (user_id,)) connection.execute("DELETE FROM user_post_samples WHERE telegram_user_id = ?", (user_id,)) connection.execute("DELETE FROM generated_posts WHERE telegram_user_id = ?", (user_id,)) connection.execute("DELETE FROM generated_ideas WHERE telegram_user_id = ?", (user_id,))