support local videos

This commit is contained in:
harry
2024-04-27 08:33:44 +08:00
parent c161ab3124
commit 5d06530a39
10 changed files with 201 additions and 66 deletions

View File

@@ -1,6 +1,7 @@
from enum import Enum
from typing import Any, Optional
from typing import Any, Optional, List
import pydantic
from pydantic import BaseModel
import warnings
@@ -28,6 +29,11 @@ class VideoAspect(str, Enum):
return 1080, 1920
class _Config:
arbitrary_types_allowed = True
@pydantic.dataclasses.dataclass(config=_Config)
class MaterialInfo:
provider: str = "pexels"
url: str = ""
@@ -95,6 +101,9 @@ class VideoParams(BaseModel):
video_clip_duration: Optional[int] = 5
video_count: Optional[int] = 1
video_source: Optional[str] = "pexels"
video_materials: Optional[List[MaterialInfo]] = None # 用于生成视频的素材
video_language: Optional[str] = "" # auto detect
voice_name: Optional[str] = ""