Commit c5f1f11
Changed files (1)
src
src/ai/chat_summary.py
@@ -6,7 +6,6 @@ import re
from datetime import datetime, timedelta
from zoneinfo import ZoneInfo
-from glom import Coalesce, glom
from loguru import logger
from pyrogram.client import Client
from pyrogram.types import Chat, Message
@@ -14,14 +13,12 @@ from pyrogram.types.messages_and_media.message import Str
from ai.main import ai_text_generation
from ai.utils import BOT_TIPS
-from config import AI, MAX_MESSAGE_SUMMARY, PREFIX, TID, TZ, cache
+from config import AI, MAX_MESSAGE_SUMMARY, PREFIX, TID, TZ
from messages.chat_history import get_history_info_list
from messages.parser import parse_msg
from messages.progress import modify_progress
from messages.sender import send2tg
-from messages.utils import equal_prefix, remove_prefix, startswith_prefix, to_int
-from networking import match_social_media_link
-from subtitles.subtitle import get_subtitle
+from messages.utils import equal_prefix, startswith_prefix, to_int
from utils import nowdt, rand_number, strings_list
# ruff: noqa: RUF001
@@ -128,21 +125,6 @@ async def ai_chat_summary(
return
if not startswith_prefix(message.content, prefix=[PREFIX.CHAT_SUMMARY, PREFIX.COMBINATION]):
return
- # summary Youtube & Bilibili video (skip for summaring chat history)
- if startswith_prefix(message.text, prefix=PREFIX.CHAT_SUMMARY) and not remove_prefix(message.text, prefix=PREFIX.CHAT_SUMMARY).strip().startswith("#"):
- # Youtube & Bilibili links in message's content or reply_to_message's content or reply_to_message's entity_urls
- links_to_check = [message.content, glom(message, Coalesce("reply_to_message.content"), default="")]
- if message.reply_to_message:
- reply_info = parse_msg(message.reply_to_message, use_cache=False, silent=True)
- links_to_check.extend(reply_info["entity_urls"])
- for link in links_to_check:
- matched = await match_social_media_link(link)
- if matched["platform"] in ["youtube", "bilibili"]:
- cache.delete(f"parse_msg-{message.chat.id}-{message.id}")
- msg = Message(id=glom(message, Coalesce("reply_to_message.id", "id")), chat=message.chat, text=Str(f"{PREFIX.SUBTITLE} {matched['url']}"))
- kwargs |= {"ai_summary": True, "send_subtitle_as": "none"}
- await get_subtitle(client, msg, **kwargs)
- return
info = parse_msg(message, silent=True)
need_summay = startswith_prefix(info["text"], prefix=PREFIX.CHAT_SUMMARY)