from enum import StrEnum


class GenerationProviderResultType(StrEnum):
    IMAGE = "image"
    VIDEO = "video"


class GenerationProviderTaskPhase(StrEnum):
    SUBMITTED = "submitted"
    POLLING = "polling"
    RESULT_READY = "result_ready"
    DOWNLOAD_PENDING = "download_pending"
    COMPLETED = "completed"
    FAILED = "failed"


class ImageProviderErrorType(StrEnum):
    TIMEOUT = "timeout"
    NETWORK = "network"
    RATE_LIMIT = "rate_limit"
    AUTH = "auth"
    INVALID_REQUEST = "invalid_request"
    CAPABILITY_MISMATCH = "capability_mismatch"
    CONTENT_REJECTED = "content_rejected"
    PROVIDER_INTERNAL = "provider_internal"
    INVALID_RESPONSE = "invalid_response"
    UNKNOWN = "unknown"


IMAGE_MULTI_OUTPUT_MIN = 1
IMAGE_MULTI_OUTPUT_MAX = 15
IMAGE_MULTI_REFERENCE_MAX = 14
IMAGE_PROVIDER_CLAIM_LEASE_SECONDS = 10 * 60

MULTI_IMAGE_PROMPT_TEMPLATE = (
    "请严格生成恰好{count}张内容相关但画面具有明显差异的图片。"
    "每张图片必须作为独立图片分别输出，不要把多个画面拼接到同一张图片中，"
    "不要生成九宫格、分镜图、组合图或包含多张子图的单张图片。"
)