真人/虚拟人像库app build

video-gen-api/app/schemas/private_portrait.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from typing import Any
 
-from pydantic import BaseModel, Field, field_validator
+from pydantic import BaseModel, Field, field_validator, model_validator
 
 from app.enums.private_portrait import (
     PRIVATE_PORTRAIT_ENABLED_ASSET_TYPES,
@@ -13,6 +13,9 @@ from app.enums.private_portrait import (
 )
 from app.schemas.common import NaiveDatetimeOptional
 
+PRIVATE_PORTRAIT_VIDEO_MIN_DURATION_SECONDS = 2
+PRIVATE_PORTRAIT_VIDEO_MAX_DURATION_SECONDS = 15
+
 
 class PrivatePortraitEnumItem(BaseModel):
     value: str
@@ -160,6 +163,18 @@ class PrivatePortraitAssetCreate(BaseModel):
             raise ValueError("Audio 暂未开放，当前仅支持 Image/Video")
         return value
 
+    @model_validator(mode="after")
+    def validate_video_duration(self) -> "PrivatePortraitAssetCreate":
+        if self.asset_type == PrivatePortraitAssetType.VIDEO.value:
+            duration = self.video_duration
+            if duration is None:
+                raise ValueError("Video 素材必须提供 video_duration")
+            if duration < PRIVATE_PORTRAIT_VIDEO_MIN_DURATION_SECONDS:
+                raise ValueError(f"视频素材最短不能少于 {PRIVATE_PORTRAIT_VIDEO_MIN_DURATION_SECONDS} 秒")
+            if duration > PRIVATE_PORTRAIT_VIDEO_MAX_DURATION_SECONDS:
+                raise ValueError(f"视频素材最长不能超过 {PRIVATE_PORTRAIT_VIDEO_MAX_DURATION_SECONDS} 秒")
+        return self
+
 
 class PrivatePortraitAssetOut(BaseModel):
     id: str