feat: ai letter + disabled fl

main.py

@@ -4,14 +4,16 @@ import asyncio
 import logging
 
 from aiogram import Bot, Dispatcher, Router
-from aiogram.types import CallbackQuery
+from aiogram.types import CallbackQuery, Message
 from aiogram.client.session.aiohttp import AiohttpSession
 from apscheduler.schedulers.asyncio import AsyncIOScheduler
 
+from app.ai import OfferReplyGenerator
 from app.config import load_settings
 from app.service import OfferProcessingService
-from app.sources import FlSource, KworkSource
-from app.state import SentOffersStore
+from app.sources import KworkSource
+# from app.sources import FlSource
+from app.state import PublishedOffersStore, SentOffersStore
 from app.telegram import TelegramPublisher
 
 
@@ -22,10 +24,75 @@ logging.basicConfig(
 
 logger = logging.getLogger(__name__)
 
+def _build_direct_message_text(message: Message, offer_url: str, reply_text: str) -> str:
+    message_link = _build_message_link(message)
+    if message_link:
+        return f"Пост: {message_link}\nОффер: {offer_url}\n\n{reply_text}"
+    return f"Оффер: {offer_url}\n\n{reply_text}"
 
-def build_dispatcher() -> Dispatcher:
+
+def _build_message_link(message: Message) -> str | None:
+    if message.chat.username:
+        return f"https://t.me/{message.chat.username}/{message.message_id}"
+
+    chat_id = str(message.chat.id)
+    if not chat_id.startswith("-100"):
+        return None
+
+    internal_chat_id = chat_id[4:]
+    if not internal_chat_id:
+        return None
+
+    return f"https://t.me/c/{internal_chat_id}/{message.message_id}"
+
+
+def build_dispatcher(
+    reply_generator: OfferReplyGenerator,
+    published_offers_store: PublishedOffersStore,
+) -> Dispatcher:
     router = Router()
 
+    @router.callback_query(lambda callback: callback.data == "accept_offer")
+    async def accept_offer_callback(callback: CallbackQuery, bot: Bot) -> None:
+        if callback.message is None:
+            await callback.answer()
+            return
+
+        offer = published_offers_store.get(callback.message.chat.id, callback.message.message_id)
+        if offer is None:
+            await callback.answer("Не удалось найти оффер", show_alert=True)
+            return
+
+        await callback.answer("Готовлю сообщение")
+
+        try:
+            reply_text = await reply_generator.generate_reply(offer)
+            await bot.send_message(
+                chat_id=callback.from_user.id,
+                text=_build_direct_message_text(callback.message, offer.url, reply_text),
+                disable_web_page_preview=True,
+            )
+            logger.info(
+                "Generated DM for user_id=%s from chat_id=%s message_id=%s",
+                callback.from_user.id,
+                callback.message.chat.id,
+                callback.message.message_id,
+            )
+        except Exception:
+            logger.exception(
+                "Failed to generate or send DM for user_id=%s chat_id=%s message_id=%s",
+                callback.from_user.id,
+                callback.message.chat.id,
+                callback.message.message_id,
+            )
+            try:
+                await bot.send_message(
+                    chat_id=callback.from_user.id,
+                    text="Не удалось подготовить сообщение. Убедитесь, что вы начали диалог с ботом, и попробуйте снова.",
+                )
+            except Exception:
+                logger.exception("Failed to send fallback DM to user_id=%s", callback.from_user.id)
+
     @router.callback_query(lambda callback: callback.data == "delete_post")
     async def delete_post_callback(callback: CallbackQuery) -> None:
         if callback.message is None:
@@ -34,6 +101,7 @@ def build_dispatcher() -> Dispatcher:
 
         try:
             await callback.message.delete()
+            published_offers_store.remove(callback.message.chat.id, callback.message.message_id)
             logger.info(
                 "Deleted Telegram message chat_id=%s message_id=%s by user_id=%s",
                 callback.message.chat.id,
@@ -56,12 +124,20 @@ def build_dispatcher() -> Dispatcher:
 
 async def main() -> None:
     settings = load_settings()
-    dispatcher = build_dispatcher()
 
     session = AiohttpSession(proxy=settings.telegram_proxy_url) if settings.telegram_proxy_url else None
     bot = Bot(token=settings.telegram_bot_token, session=session)
     publisher = TelegramPublisher(bot=bot, channel_id=settings.telegram_channel_id)
     store = SentOffersStore(settings.state_file)
+    published_offers_store = PublishedOffersStore(settings.published_offers_file)
+    reply_generator = OfferReplyGenerator(
+        api_base_url=settings.openai_api_base_url,
+        api_key=settings.openai_api_key,
+        model=settings.openai_model,
+        system_prompt_file=settings.openai_system_prompt_file,
+        timeout_seconds=settings.requests_timeout_seconds,
+    )
+    dispatcher = build_dispatcher(reply_generator, published_offers_store)
 
     sources = [
         KworkSource(
@@ -69,18 +145,19 @@ async def main() -> None:
             timeout_seconds=settings.requests_timeout_seconds,
         )
     ]
-    if settings.fl_rss_url:
-        sources.append(
-            FlSource(
-                rss_url=settings.fl_rss_url,
-                timeout_seconds=settings.requests_timeout_seconds,
-            )
-        )
+    # if settings.fl_rss_url:
+    #     sources.append(
+    #         FlSource(
+    #             rss_url=settings.fl_rss_url,
+    #             timeout_seconds=settings.requests_timeout_seconds,
+    #         )
+    #     )
 
     service = OfferProcessingService(
         sources=sources,
         publisher=publisher,
         sent_offers_store=store,
+        published_offers_store=published_offers_store,
         include_keywords=settings.keywords_include,
         exclude_keywords=settings.keywords_exclude,
         min_price_amount=settings.min_price_amount,
@@ -100,4 +177,4 @@ async def main() -> None:
 
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    asyncio.run(main())