Галингер Р.С.
d6c029f708
- Add WHISPER_MODEL_SIZE and WHISPER_MIN_FREE_SPACE_BYTES config options - voice.py now preloads/downloads the Whisper model at bot startup - Detailed logging for model presence, disk space, download progress and readiness - If model is already present locally, preload is skipped - If disk space is insufficient, bot fails fast with a clear error - main.py calls preload_model() after proxy setup and before polling - Document voice recognition model download behavior in README
99 lines
3.2 KiB
Python
99 lines
3.2 KiB
Python
import asyncio
|
|
import logging
|
|
|
|
from aiogram.exceptions import TelegramNetworkError
|
|
from aiohttp import ClientConnectionError
|
|
|
|
from aiogram import F
|
|
from aiogram.types import Message
|
|
from aiogram.dispatcher.event.bases import SkipHandler
|
|
|
|
from bot.bot import dp, bot, setup_proxy
|
|
from bot.routers import moderation, layout, weather, ai, voice, yadisk, dialogue
|
|
from bot.utils.database import init_db, save_chat_user
|
|
from bot.utils.ai_client import start_model_health_check
|
|
from bot.utils.logging_config import setup_logging
|
|
from bot.utils.voice import preload_model
|
|
from bot.setup_commands import setup_bot_commands
|
|
from config import PROXY_ENABLED
|
|
|
|
setup_logging()
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Router order matters: moderation first so it can delete messages before others process them.
|
|
dp.include_router(moderation.router)
|
|
dp.include_router(layout.router)
|
|
dp.include_router(weather.router)
|
|
dp.include_router(ai.router)
|
|
dp.include_router(voice.router)
|
|
dp.include_router(yadisk.router)
|
|
dp.include_router(dialogue.router)
|
|
|
|
|
|
@dp.message(F.text)
|
|
async def save_user_info(message: Message):
|
|
try:
|
|
await save_chat_user(
|
|
message.from_user.id,
|
|
message.chat.id,
|
|
message.from_user.username,
|
|
message.from_user.full_name or "",
|
|
)
|
|
except Exception as exc:
|
|
logger.warning("Failed to save chat user: %s", exc)
|
|
raise SkipHandler
|
|
|
|
|
|
async def main():
|
|
await init_db()
|
|
logging.info("База данных инициализирована")
|
|
await setup_bot_commands()
|
|
|
|
health_check_task = asyncio.create_task(start_model_health_check())
|
|
|
|
try:
|
|
if PROXY_ENABLED:
|
|
await setup_proxy()
|
|
else:
|
|
me = await bot.me()
|
|
logging.info("Бот запущен без прокси: @%s", me.username)
|
|
except TelegramNetworkError:
|
|
if PROXY_ENABLED:
|
|
logging.error("Не могу соединиться с прокси сервером. Попробуй другой прокси.")
|
|
else:
|
|
logging.error("Не могу соединиться с Telegram. Проверь интернет-соединение.")
|
|
health_check_task.cancel()
|
|
return
|
|
except ClientConnectionError:
|
|
if PROXY_ENABLED:
|
|
logging.error("Не могу соединиться с прокси сервером. Попробуй другой прокси.")
|
|
else:
|
|
logging.error("Не могу соединиться с Telegram. Проверь интернет-соединение.")
|
|
health_check_task.cancel()
|
|
return
|
|
except Exception as exc:
|
|
logging.error("Ошибка при запуске бота: %s", exc)
|
|
health_check_task.cancel()
|
|
return
|
|
|
|
try:
|
|
await preload_model()
|
|
logging.info("Модель Whisper готова к работе")
|
|
except Exception as exc:
|
|
logging.error("Не удалось загрузить модель Whisper: %s", exc)
|
|
health_check_task.cancel()
|
|
return
|
|
|
|
try:
|
|
await dp.start_polling(bot)
|
|
finally:
|
|
health_check_task.cancel()
|
|
try:
|
|
await health_check_task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|