"""从 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("全部配音生成完成")