xiaoxia-saas/apps/api/app/schemas/lipsync.py

"""对口型 API Schema 定义 — #1796, #1809 参数调整."""

from __future__ import annotations

from datetime import datetime
from typing import Optional

from pydantic import BaseModel, Field, field_validator


class LipsyncJobResponse(BaseModel):
    """对口型任务响应."""

    id: str
    user_id: str
    project_id: str
    video_url: str
    audio_url: str
    enable_video_loop: bool
    mediakit_task_id: str
    status: str
    output_video_url: str
    output_duration: float
    error_message: str
    error_code: str
    submitted_at: Optional[datetime] = None
    completed_at: Optional[datetime] = None
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True


class CreateLipsyncJobRequest(BaseModel):
    """创建对口型任务请求 — #1809.

    前端传 {voice_id, script_text, video_url}，
    后端内部调 TTS 生成 audio_url 再提交 MediaKit。
    """

    video_url: str = Field(..., description="人物视频 URL（MP4，≤30min，单人真人）")
    voice_id: str = Field(..., description="音色 ID（预设音色或克隆音色 profile ID）")
    script_text: str = Field(..., description="要合成的脚本文本")
    enable_video_loop: bool = Field(False, description="音频长于视频时是否循环画面")
    project_id: str = Field("", description="项目 ID（可选）")

    @field_validator("video_url")
    @classmethod
    def validate_video_url(cls, v: str) -> str:
        v = v.strip()
        if not v:
            raise ValueError("video_url 不能为空")
        if not v.startswith(("http://", "https://")):
            raise ValueError("video_url 必须是 HTTP/HTTPS URL")
        lower = v.lower().split("?")[0]
        if not lower.endswith(".mp4"):
            raise ValueError("video_url 仅支持 MP4 格式")
        return v

    @field_validator("voice_id")
    @classmethod
    def validate_voice_id(cls, v: str) -> str:
        v = v.strip()
        if not v:
            raise ValueError("voice_id 不能为空")
        return v

    @field_validator("script_text")
    @classmethod
    def validate_script_text(cls, v: str) -> str:
        v = v.strip()
        if not v:
            raise ValueError("script_text 不能为空")
        if len(v) > 5000:
            raise ValueError("script_text 最长 5000 字符")
        return v