73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from wov_sdk.models import NodeManifest, WorkflowDefinition, WorkflowEdge, WorkflowNode
|
|
|
|
from app.db import Database
|
|
|
|
|
|
def seed_nodes(db: Database, workspace_root: Path) -> None:
|
|
for node_dir in sorted(workspace_root.glob("wov-node-*")):
|
|
manifest_path = node_dir / "node.manifest.json"
|
|
if not manifest_path.is_file():
|
|
continue
|
|
manifest = NodeManifest.load(str(manifest_path))
|
|
manifest.repo_dir = node_dir.name
|
|
db.upsert_node(manifest)
|
|
|
|
|
|
def seed_demo_workflow(db: Database) -> None:
|
|
if db.get_workflow("demo") is not None:
|
|
return
|
|
definition = WorkflowDefinition(
|
|
name="视频字幕生成",
|
|
version=1,
|
|
nodes=[
|
|
WorkflowNode(
|
|
id="extract",
|
|
node_type="ffmpeg-extract",
|
|
params={"sample_rate": 16000, "channels": 1},
|
|
inputs={"video_uri": "input.video_uri"},
|
|
),
|
|
WorkflowNode(
|
|
id="asr",
|
|
node_type="faster-whisper",
|
|
params={"language": "ja"},
|
|
inputs={"audio_uri": "extract.audio_uri"},
|
|
),
|
|
WorkflowNode(
|
|
id="translate",
|
|
node_type="llm-translate",
|
|
params={"target_language": "zh-CN"},
|
|
inputs={"srt_uri": "asr.srt_uri"},
|
|
),
|
|
WorkflowNode(
|
|
id="ass",
|
|
node_type="srt-to-dual-eye-ass",
|
|
params={"resolution": "3840x1920"},
|
|
inputs={"cn_srt_uri": "translate.cn_srt_uri"},
|
|
),
|
|
],
|
|
edges=[
|
|
WorkflowEdge(from_node="extract", to_node="asr"),
|
|
WorkflowEdge(from_node="asr", to_node="translate"),
|
|
WorkflowEdge(from_node="translate", to_node="ass"),
|
|
],
|
|
entry_inputs={"video_uri": "file"},
|
|
final_outputs={
|
|
"cn_srt": "translate.cn_srt_uri",
|
|
"ass": "ass.ass_uri",
|
|
},
|
|
)
|
|
db.upsert_workflow(
|
|
{
|
|
"id": "demo",
|
|
"name": definition.name,
|
|
"description": "上传视频，自动生成中文字幕和 VR 双眼 ASS。",
|
|
"published": 1,
|
|
"latest_version": 1,
|
|
}
|
|
)
|
|
db.create_workflow_version("demo", 1, definition.to_dict())
|