life-echo/api/services/memoir_images/settings.py

import os
from dataclasses import dataclass

DEFAULT_LIBLIB_TEMPLATE_UUID = "5d7e67009b344550bc1aa6ccbfa1d7f4"


@dataclass(frozen=True)
class MemoirImageSettings:
    enabled: bool
    max_per_chapter: int
    provider: str
    default_style: str
    default_size: str
    poll_interval_seconds: int
    max_attempts: int
    liblib_template_uuid: str

    @classmethod
    def from_env(cls) -> "MemoirImageSettings":
        return cls(
            enabled=os.getenv("MEMOIR_IMAGE_ENABLED", "").lower() in {"1", "true", "yes"},
            max_per_chapter=_get_int_env("MEMOIR_IMAGE_MAX_PER_CHAPTER", 2),
            provider=os.getenv("MEMOIR_IMAGE_PROVIDER", "liblib"),
            default_style=os.getenv("MEMOIR_IMAGE_STYLE_DEFAULT", "watercolor"),
            default_size=os.getenv("MEMOIR_IMAGE_SIZE_DEFAULT", "1280x720"),
            poll_interval_seconds=_get_int_env("MEMOIR_IMAGE_POLL_INTERVAL", 3),
            max_attempts=_get_int_env("MEMOIR_IMAGE_MAX_ATTEMPTS", 60),
            liblib_template_uuid=os.getenv("LIBLIB_TEMPLATE_UUID") or DEFAULT_LIBLIB_TEMPLATE_UUID,
        )


def _get_int_env(name: str, default: int) -> int:
    value = os.getenv(name, str(default))
    try:
        return int(value)
    except (TypeError, ValueError):
        return default