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

from app.core.config import settings
from app.core.runtime_constants import misc_defaults
from app.features.memoir.constants import memoir
from app.features.story.constants import story

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


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
    story_image_min_body_chars: int = 400

    def __init__(
        self,
        *,
        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,
        story_image_min_body_chars: int = 400,
    ) -> None:
        self.enabled = enabled
        self.provider = provider
        self.default_style = default_style
        self.default_size = default_size
        self.poll_interval_seconds = poll_interval_seconds
        self.max_attempts = max_attempts
        self.liblib_template_uuid = liblib_template_uuid
        self.story_image_min_body_chars = story_image_min_body_chars

    @classmethod
    def from_env(cls) -> "MemoirImageSettings":
        return cls(
            enabled=bool(settings.memoir_image_enabled),
            provider=memoir.image_provider or DEFAULT_IMAGE_PROVIDER,
            default_style=memoir.image_style_default or DEFAULT_IMAGE_STYLE,
            default_size=memoir.image_size_default or DEFAULT_IMAGE_SIZE,
            poll_interval_seconds=memoir.image_poll_interval,
            max_attempts=memoir.image_max_attempts,
            liblib_template_uuid=settings.liblib_template_uuid
            or DEFAULT_LIBLIB_TEMPLATE_UUID,
            story_image_min_body_chars=int(story.image_min_body_chars or 0),
        )