mirror of
https://github.com/harry0703/MoneyPrinterTurbo.git
synced 2026-02-21 16:37:21 +08:00
120 lines
3.1 KiB
Python
120 lines
3.1 KiB
Python
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
|