from enum import Enum


class GenerationStatus(str, Enum):
    """生成状态。"""
    prompt_optimized = "prompt_optimized"
    generating = "generating"
    completed = "completed"
    failed = "failed"


class GenerationRecordPipelineStage(str, Enum):
    """GenerationRecord 图片/视频生成、下载与超分流水线阶段。"""

    QUEUED = "queued"
    PREPARING = "preparing"
    CREATING_PROVIDER_TASK = "creating_provider_task"
    PROVIDER_RESULT_STAGED = "provider_result_staged"
    WAITING_REMOTE = "waiting_remote"
    POLLING = "polling"
    RESULT_READY = "result_ready"
    DOWNLOAD_QUEUED = "download_queued"
    DOWNLOADING = "downloading"
    RETRY_WAITING = "retry_waiting"
    UPSCALE_QUEUED = "upscale_queued"
    UPSCALE_PROCESSING = "upscale_processing"
    UPSCALE_POLLING = "upscale_polling"
    UPSCALE_DOWNLOADING = "upscale_downloading"
    UPSCALE_FINALIZING = "upscale_finalizing"
    UPSCALE_RETRY_WAITING = "upscale_retry_waiting"
    UPSCALE_FAILED = "upscale_failed"
    DONE = "done"
    FAILED = "failed"
    TIMEOUT = "timeout"
    DOWNLOAD_FAILED = "download_failed"


class GenerationType(str, Enum):
    """生成类型。"""
    video = "video"
    image = "image"


# 生成配置常量
DURATIONS = [4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
ASPECT_RATIOS = ["16:9", "9:16", "1:1", "4:3", "3:4", "21:9"]
RESOLUTIONS = ["480p", "720p", "1080p"]
IMAGE_SIZES = ["1K", "2K", "4K"]