32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
"""从 narration.md 解析 SEG 分段，逐段调用 TTS 生成 MP3。"""
|
||
import re
|
||
import subprocess
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
WS = Path(r"D:\ItemSpace\14.工业智核\BH-PPT\aps-agent\video")
|
||
PLUGIN = Path(r"D:\KimiData\daimon-share\daimon\runtime\kimi-code\home\plugins\managed\audio_generation")
|
||
VOICE = "05Cdh2gw2NMzDvykn1nm"
|
||
|
||
text = (WS / "narration.md").read_text(encoding="utf-8")
|
||
segments = re.findall(r"SEG (\d+)（[^）]*）\s*\n(.*?)(?=\n\nSEG |\Z)", text, re.S)
|
||
print(f"共解析 {len(segments)} 段")
|
||
|
||
for num, body in segments:
|
||
out = WS / "audio" / f"seg{num}.mp3"
|
||
if out.exists() and out.stat().st_size > 10_000:
|
||
print(f"seg{num} 已存在，跳过")
|
||
continue
|
||
body = " ".join(body.split())
|
||
result = subprocess.run(
|
||
[sys.executable, str(PLUGIN / "scripts" / "audio_generation_tool.py"), "speech",
|
||
"--text", body, "--voice-id", VOICE, "--output", str(out)],
|
||
cwd=PLUGIN, capture_output=True, text=True, timeout=600,
|
||
)
|
||
ok = out.exists() and out.stat().st_size > 10_000
|
||
print(f"seg{num}: {'OK' if ok else 'FAIL'} ({len(body)} 字)")
|
||
if not ok:
|
||
print(result.stdout[-500:], result.stderr[-500:])
|
||
sys.exit(1)
|
||
print("全部配音生成完成")
|