You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.9 KiB
78 lines
2.9 KiB
#!/usr/bin/env python3
|
|
"""Phase 3 Step 3.4: Move preview audio files from project root to voice_samples/samples/.
|
|
|
|
Usage: python3 scripts/organize_preview_files.py
|
|
|
|
Reads preview_plan.json for filenames, moves .mp3 and _subtitle.json from project root
|
|
to voice_samples/samples/, and updates assets.json path field.
|
|
Idempotent: skips files already moved or not found at root.
|
|
"""
|
|
import json, os, shutil
|
|
from pathlib import Path
|
|
|
|
PROJECT_DIR = Path(os.getcwd())
|
|
AUDIOBOOK_ROOT = PROJECT_DIR / ".audiobook"
|
|
BOOK_DIR = next(d for d in sorted(AUDIOBOOK_ROOT.iterdir()) if d.is_dir() and (d / ".audiobook-state.json").exists())
|
|
|
|
plan_path = BOOK_DIR / "voice_samples" / "preview_plan.json"
|
|
assets_path = PROJECT_DIR / ".hilo" / "assets.json"
|
|
target_dir = BOOK_DIR / "voice_samples" / "samples"
|
|
target_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
plan = json.loads(plan_path.read_text(encoding="utf-8"))
|
|
assets = json.loads(assets_path.read_text(encoding="utf-8"))
|
|
|
|
# Collect all preview filenames from plan
|
|
filenames = []
|
|
for role in plan:
|
|
for c in role["candidates"]:
|
|
if "filename" not in c:
|
|
print(f"ERROR: candidate {c['voice_id']} in role '{role['role']}' has no 'filename' field.")
|
|
print(" preview_plan.json must include filename for each candidate (see Phase 3 Step 3.1).")
|
|
exit(1)
|
|
filenames.append(c["filename"])
|
|
|
|
# Build assets lookup: filename (without ext) -> list index
|
|
name_to_idx = {}
|
|
for i, a in enumerate(assets):
|
|
if a.get("type") == "audio" and a.get("name"):
|
|
name_key = a["name"].rsplit(".", 1)[0]
|
|
name_to_idx[name_key] = i
|
|
|
|
moved = 0
|
|
skipped = 0
|
|
for fn in filenames:
|
|
# Move .mp3
|
|
src_mp3 = PROJECT_DIR / f"{fn}.mp3"
|
|
dst_mp3 = target_dir / f"{fn}.mp3"
|
|
if src_mp3.exists() and not dst_mp3.exists():
|
|
shutil.move(str(src_mp3), str(dst_mp3))
|
|
moved += 1
|
|
print(f" moved {fn}.mp3")
|
|
elif dst_mp3.exists():
|
|
skipped += 1
|
|
else:
|
|
print(f" skip {fn}.mp3 (not found at root)")
|
|
|
|
# Move subtitle json if present
|
|
src_sub = PROJECT_DIR / f"{fn}_subtitle.json"
|
|
dst_sub = target_dir / f"{fn}_subtitle.json"
|
|
if src_sub.exists() and not dst_sub.exists():
|
|
shutil.move(str(src_sub), str(dst_sub))
|
|
print(f" moved {fn}_subtitle.json")
|
|
|
|
# Update assets.json path field
|
|
if fn in name_to_idx:
|
|
idx = name_to_idx[fn]
|
|
rel_path = str(dst_mp3.relative_to(PROJECT_DIR))
|
|
if assets[idx]["path"] != rel_path:
|
|
assets[idx]["path"] = rel_path
|
|
# Update subtitle_path in metadata if present
|
|
meta = assets[idx].get("metadata", {})
|
|
if meta.get("subtitle_path"):
|
|
meta["subtitle_path"] = str(dst_sub.relative_to(PROJECT_DIR))
|
|
|
|
with open(assets_path, "w", encoding="utf-8") as f:
|
|
json.dump(assets, f, ensure_ascii=False, indent=2)
|
|
|
|
print(f"\nDone: {moved} moved, {skipped} already in place, assets.json updated.")
|
|
|