93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
"""FFmpeg 提音节点入口。
|
|
|
|
通过标准节点 HTTP 服务对外提供音频提取能力。ffmpeg 解析顺序为：
|
|
FFMPEG_BIN 环境变量 > PATH 中的 ffmpeg > imageio-ffmpeg 内置二进制。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
from wov_sdk.models import InvokeRequest, InvokeResponse, NodeManifest
|
|
from wov_sdk.server import run_node
|
|
|
|
|
|
def _bundled_ffmpeg() -> str | None:
|
|
"""尝试获取 imageio-ffmpeg 内置的 ffmpeg 可执行文件路径。"""
|
|
try:
|
|
import imageio_ffmpeg
|
|
|
|
return imageio_ffmpeg.get_ffmpeg_exe()
|
|
except Exception: # noqa: BLE001
|
|
# 未安装 imageio-ffmpeg 或获取失败时返回 None，交由上层回退。
|
|
return None
|
|
|
|
|
|
def _ffmpeg_bin() -> str:
|
|
"""按优先级解析 ffmpeg 可执行文件，返回最终命令路径。"""
|
|
# 显式配置优先，便于部署环境指定自定义二进制。
|
|
configured = os.getenv("FFMPEG_BIN")
|
|
if configured:
|
|
return configured
|
|
# 其次查找 PATH 中的系统 ffmpeg。
|
|
found = shutil.which("ffmpeg")
|
|
if found:
|
|
return found
|
|
# 最后回退到 imageio-ffmpeg 内置二进制；都没有时保留 "ffmpeg" 交给调用失败处理。
|
|
return _bundled_ffmpeg() or "ffmpeg"
|
|
|
|
|
|
def invoke(request: InvokeRequest) -> InvokeResponse:
|
|
"""提取输入视频/音频的标准化音频，产物为 audio.wav。"""
|
|
video_uri = request.inputs.get("video_uri")
|
|
if not video_uri:
|
|
return InvokeResponse(status="failed", error="video_uri is required")
|
|
|
|
# 找不到可用 ffmpeg 时直接返回失败，避免子进程报晦涩错误。
|
|
ffmpeg = _ffmpeg_bin()
|
|
if shutil.which(ffmpeg) is None and not Path(ffmpeg).is_file():
|
|
return InvokeResponse(status="failed", error="ffmpeg not found")
|
|
|
|
output_dir = Path(request.output_dir)
|
|
output_dir.mkdir(parents=True, exist_ok=True)
|
|
output_path = output_dir / "audio.wav"
|
|
# ASR 节点默认期望 16kHz 单声道；参数可覆盖。
|
|
channels = str(request.params.get("channels", 1))
|
|
sample_rate = str(request.params.get("sample_rate", 16000))
|
|
command = [
|
|
ffmpeg,
|
|
"-y", # 覆盖可能存在的同名输出文件。
|
|
"-i",
|
|
str(video_uri),
|
|
"-vn", # 丢弃视频流，只保留音频。
|
|
"-ac",
|
|
channels,
|
|
"-ar",
|
|
sample_rate,
|
|
str(output_path),
|
|
]
|
|
result = subprocess.run(command, capture_output=True, text=True)
|
|
if result.returncode != 0:
|
|
# 返回 stderr 尾部，保留最有诊断价值的错误信息。
|
|
return InvokeResponse(
|
|
status="failed",
|
|
error=result.stderr[-2000:] or "ffmpeg failed",
|
|
)
|
|
return InvokeResponse(status="completed", outputs={"audio_uri": str(output_path)})
|
|
|
|
|
|
def main() -> None:
|
|
"""加载节点清单并以本模块的 invoke 处理器启动服务。"""
|
|
manifest_path = Path(__file__).resolve().parent.parent / "node.manifest.json"
|
|
with open(manifest_path, "r", encoding="utf-8") as f:
|
|
manifest = NodeManifest.from_dict(json.load(f))
|
|
run_node(manifest, invoke)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|