life-echo/api/app/features/memoir/memoir_images/settings.py

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)