mirror of
https://github.com/linyqh/NarratoAI.git
synced 2025-12-12 11:17:37 +00:00
82823297f2
- 在 GenerateScriptRequest 和 ScriptGenerator 中新增 frame_interval_input 参数，默认为 5 - 更新 VideoProcessor 的 process_video_pipeline 方法，调整帧提取间隔逻辑 - 修改前端界面，增加帧间隔输入框并更新相关翻译文本 - 优化 generate_script_docu 函数，使用新的帧间隔参数进行视频处理
64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
from typing import Optional, List
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class GenerateScriptRequest(BaseModel):
|
|
video_path: str
|
|
video_theme: Optional[str] = ""
|
|
custom_prompt: Optional[str] = ""
|
|
frame_interval_input: Optional[int] = 5
|
|
skip_seconds: Optional[int] = 0
|
|
threshold: Optional[int] = 30
|
|
vision_batch_size: Optional[int] = 5
|
|
vision_llm_provider: Optional[str] = "gemini"
|
|
|
|
|
|
class GenerateScriptResponse(BaseModel):
|
|
task_id: str
|
|
script: List[dict]
|
|
|
|
|
|
class CropVideoRequest(BaseModel):
|
|
video_origin_path: str
|
|
video_script: List[dict]
|
|
|
|
|
|
class CropVideoResponse(BaseModel):
|
|
task_id: str
|
|
subclip_videos: dict
|
|
|
|
|
|
class DownloadVideoRequest(BaseModel):
|
|
url: str
|
|
resolution: str
|
|
output_format: Optional[str] = "mp4"
|
|
rename: Optional[str] = None
|
|
|
|
|
|
class DownloadVideoResponse(BaseModel):
|
|
task_id: str
|
|
output_path: str
|
|
resolution: str
|
|
format: str
|
|
filename: str
|
|
|
|
|
|
class StartSubclipRequest(BaseModel):
|
|
task_id: str
|
|
video_origin_path: str
|
|
video_clip_json_path: str
|
|
voice_name: Optional[str] = None
|
|
voice_rate: Optional[int] = 0
|
|
voice_pitch: Optional[int] = 0
|
|
subtitle_enabled: Optional[bool] = True
|
|
video_aspect: Optional[str] = "16:9"
|
|
n_threads: Optional[int] = 4
|
|
subclip_videos: list # 从裁剪视频接口获取的视频片段字典
|
|
|
|
|
|
class StartSubclipResponse(BaseModel):
|
|
task_id: str
|
|
state: str
|
|
videos: Optional[List[str]] = None
|
|
combined_videos: Optional[List[str]] = None
|