from dataclasses import dataclass from typing import TYPE_CHECKING if TYPE_CHECKING: from app.core.config import Settings DEFAULT_LIBLIB_TEMPLATE_UUID = "5d7e67009b344550bc1aa6ccbfa1d7f4" DEFAULT_IMAGE_PROVIDER = "liblib" DEFAULT_IMAGE_STYLE = "watercolor" DEFAULT_IMAGE_SIZE = "1280x720" DEFAULT_POLL_INTERVAL_SECONDS = 5 DEFAULT_MAX_ATTEMPTS = 60 @dataclass(frozen=True) class MemoirImageSettings: enabled: bool = False provider: str = DEFAULT_IMAGE_PROVIDER default_style: str = DEFAULT_IMAGE_STYLE default_size: str = DEFAULT_IMAGE_SIZE poll_interval_seconds: int = DEFAULT_POLL_INTERVAL_SECONDS max_attempts: int = DEFAULT_MAX_ATTEMPTS liblib_template_uuid: str = DEFAULT_LIBLIB_TEMPLATE_UUID @classmethod def from_settings(cls, settings: "Settings") -> "MemoirImageSettings": s = settings return cls( enabled=bool(s.memoir_image_enabled), provider=s.memoir_image_provider or DEFAULT_IMAGE_PROVIDER, default_style=s.memoir_image_style_default or DEFAULT_IMAGE_STYLE, default_size=s.memoir_image_size_default or DEFAULT_IMAGE_SIZE, poll_interval_seconds=s.memoir_image_poll_interval, max_attempts=s.memoir_image_max_attempts, liblib_template_uuid=s.liblib_template_uuid or DEFAULT_LIBLIB_TEMPLATE_UUID, ) @classmethod def from_env(cls) -> "MemoirImageSettings": from app.core.config import settings as _s return cls.from_settings(_s)