main.py

from __future__ import annotations

import asyncio
import logging

from aiogram import Bot, Dispatcher
from aiogram.client.default import DefaultBotProperties
from aiogram.enums import ParseMode

from app.bot import setup_routers
from app.config import get_settings
from app.db.session import init_engine
from app.logging import configure_logging
from app.dependencies import set_generation_service, set_model_registry, set_storage_service
from app.services.generation import GenerationService
from app.services.model_registry import ModelRegistry
from app.services.storage import StorageService


async def _on_startup() -> None:
    configure_logging()
    settings = get_settings()
    await init_engine()

    model_registry = ModelRegistry()
    if not model_registry.list_models():
        logging.warning("No diffusion models were discovered. /draw will be unavailable until models are added.")

    generation_service = GenerationService(model_registry)
    storage_service = StorageService()

    set_model_registry(model_registry)
    set_generation_service(generation_service)
    set_storage_service(storage_service)

    bot = Bot(token=settings.bot_token, default=DefaultBotProperties(parse_mode=ParseMode.HTML))
    dispatcher = Dispatcher()

    setup_routers(dispatcher)

    await bot.delete_webhook(drop_pending_updates=True)
    await dispatcher.start_polling(bot)


def main() -> None:
    try:
        asyncio.run(_on_startup())
    except (KeyboardInterrupt, SystemExit):
        logging.info("Bot stopped.")


if __name__ == "__main__":
    main()