from enum import Enum
from typing import Any, Optional

from pydantic import BaseModel
import warnings

# 忽略 Pydantic 的特定警告
warnings.filterwarnings("ignore", category=UserWarning, message="Field name.*shadows an attribute in parent.*")


class VideoAspect(str, Enum):
    landscape = "16:9"
    portrait = "9:16"
    square = "1:1"

    def to_resolution(self):
        if self == VideoAspect.landscape.value:
            return 1920, 1080
        elif self == VideoAspect.portrait.value:
            return 1080, 1920
        elif self == VideoAspect.square.value:
            return 1080, 1080
        return 1080, 1920


VoiceNames = [
    # zh-CN
    "female-zh-CN-XiaoxiaoNeural",
    "female-zh-CN-XiaoyiNeural",
    "female-zh-CN-liaoning-XiaobeiNeural",
    "female-zh-CN-shaanxi-XiaoniNeural",

    "male-zh-CN-YunjianNeural",
    "male-zh-CN-YunxiNeural",
    "male-zh-CN-YunxiaNeural",
    "male-zh-CN-YunyangNeural",

    # "female-zh-HK-HiuGaaiNeural",
    # "female-zh-HK-HiuMaanNeural",
    # "male-zh-HK-WanLungNeural",
    #
    # "female-zh-TW-HsiaoChenNeural",
    # "female-zh-TW-HsiaoYuNeural",
    # "male-zh-TW-YunJheNeural",

    # en-US

    "female-en-US-AnaNeural",
    "female-en-US-AriaNeural",
    "female-en-US-AvaNeural",
    "female-en-US-EmmaNeural",
    "female-en-US-JennyNeural",
    "female-en-US-MichelleNeural",

    "male-en-US-AndrewNeural",
    "male-en-US-BrianNeural",
    "male-en-US-ChristopherNeural",
    "male-en-US-EricNeural",
    "male-en-US-GuyNeural",
    "male-en-US-RogerNeural",
    "male-en-US-SteffanNeural",
]


class VideoParams:
    """
    {
      "video_subject": "",
      "video_aspect": "横屏 16:9（西瓜视频）",
      "voice_name": "女生-晓晓",
      "bgm_name": "random",
      "font_name": "STHeitiMedium 黑体-中",
      "text_color": "#FFFFFF",
      "font_size": 60,
      "stroke_color": "#000000",
      "stroke_width": 1.5
    }
    """
    video_subject: str
    video_aspect: Optional[VideoAspect] = VideoAspect.portrait.value
    voice_name: Optional[str] = VoiceNames[0]
    bgm_name: Optional[str] = "random"
    font_name: Optional[str] = "STHeitiMedium.ttc"
    text_fore_color: Optional[str] = "#FFFFFF"
    font_size: int = 60
    stroke_color: Optional[str] = "#000000"
    stroke_width: float = 1.5
    n_threads: Optional[int] = 2
    paragraph_number: Optional[int] = 1


class BaseResponse(BaseModel):
    status: int = 200
    message: Optional[str] = 'success'
    data: Any = None


class TaskVideoRequest(VideoParams, BaseModel):
    pass


class TaskQueryRequest(BaseModel):
    pass


######################################################################################################
######################################################################################################
######################################################################################################
######################################################################################################
class TaskResponse(BaseResponse):
    class TaskResponseData(BaseModel):
        task_id: str
        task_type: str = ""

    data: TaskResponseData


class TaskQueryResponse(BaseResponse):
    pass