from __future__ import annotations

from enum import StrEnum


class ModuleCodeEnum(StrEnum):
    """可复用模块编码。"""

    HOT_OPENING_REPLICATE = "hot_opening_replicate"


class HotOpeningStepCodeEnum(StrEnum):
    """爆款开头复刻子任务步骤编码。"""

    MATERIAL_INPUT = "material_input"
    IMAGE_PROMPT_OPTIMIZE = "image_prompt_optimize"
    IMAGE_GENERATE = "image_generate"
    VIDEO_PROMPT_OPTIMIZE = "video_prompt_optimize"
    VIDEO_GENERATE = "video_generate"


class HotOpeningGenerationModeEnum(StrEnum):
    """复用 ChatGenerationTask 时使用的 generation_mode。"""

    HOT_OPENING_REPLICATE = "hot_opening_replicate"



class HotOpeningStepIOSchemaVersionEnum(StrEnum):
    """爆款开头复刻子任务 input_json/output_json 结构版本。"""

    V1 = "hot_opening_step_io_v1"
