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"
    V2 = "hot_opening_step_io_v2"


class HotOpeningLogEventEnum(StrEnum):
    """爆款开头复刻模块业务日志事件。"""

    API_REQUEST_RECEIVED = "HOT_OPENING_API_REQUEST_RECEIVED"
    API_REQUEST_SUBMITTED = "HOT_OPENING_API_REQUEST_SUBMITTED"
    API_REQUEST_REJECTED = "HOT_OPENING_API_REQUEST_REJECTED"
    API_REQUEST_FAILED = "HOT_OPENING_API_REQUEST_FAILED"
    CELERY_DISABLED = "HOT_OPENING_CELERY_DISABLED"
    CELERY_DISPATCH_FAILED = "HOT_OPENING_CELERY_DISPATCH_FAILED"
    CELERY_DISPATCH_MARK_FAILED = "HOT_OPENING_CELERY_DISPATCH_MARK_FAILED"

    IMAGE_PROMPT_SUBMITTED = "HOT_OPENING_IMAGE_PROMPT_SUBMITTED"
    IMAGE_GENERATE_SUBMITTED = "HOT_OPENING_IMAGE_GENERATE_SUBMITTED"
    VIDEO_PROMPT_SUBMITTED = "HOT_OPENING_VIDEO_PROMPT_SUBMITTED"
    VIDEO_GENERATE_SUBMITTED = "HOT_OPENING_VIDEO_GENERATE_SUBMITTED"

    VIDEO_PROMPT_REMOTE_API_STARTED = "HOT_OPENING_VIDEO_PROMPT_REMOTE_API_STARTED"
    VIDEO_PROMPT_REMOTE_API_SUCCESS = "HOT_OPENING_VIDEO_PROMPT_REMOTE_API_SUCCESS"
    VIDEO_PROMPT_REMOTE_API_FAILED = "HOT_OPENING_VIDEO_PROMPT_REMOTE_API_FAILED"
    VIDEO_PROMPT_RESPONSE_PARSE_FAILED = "HOT_OPENING_VIDEO_PROMPT_RESPONSE_PARSE_FAILED"
    VIDEO_PROMPT_RESPONSE_EMPTY = "HOT_OPENING_VIDEO_PROMPT_RESPONSE_EMPTY"


class HotOpeningRemoteActionEnum(StrEnum):
    """爆款开头复刻远程模型动作。"""

    VIDEO_PROMPT_OPTIMIZE = "video_prompt_optimize"