Spaces:
Running on Zero
Running on Zero
| """Workspace paths, manifests, and segment file ordering.""" | |
| from __future__ import annotations | |
| import json | |
| from datetime import datetime, timezone | |
| from pathlib import Path | |
| from typing import List, Tuple | |
| from .constants import _SEGMENT_RE | |
| def _utc_now() -> str: | |
| return datetime.now(timezone.utc).isoformat() | |
| def workspace_paths(work_dir: Path) -> dict[str, Path]: | |
| return { | |
| "root": work_dir, | |
| "input": work_dir / "00_input", | |
| "segments": work_dir / "01_segments", | |
| "transcripts": work_dir / "02_transcripts.json", | |
| "aligned": work_dir / "03_aligned.json", | |
| "spans": work_dir / "04_spans", | |
| "inpainted": work_dir / "05_inpainted", | |
| "stitched": work_dir / "06_stitched.wav", | |
| "manifest": work_dir / "manifest.json", | |
| } | |
| def load_manifest(path: Path) -> dict: | |
| if path.is_file(): | |
| return json.loads(path.read_text(encoding="utf-8")) | |
| return {} | |
| def save_manifest(path: Path, manifest: dict) -> None: | |
| path.parent.mkdir(parents=True, exist_ok=True) | |
| path.write_text(json.dumps(manifest, indent=2, ensure_ascii=False), encoding="utf-8") | |
| def mark_stage(manifest: dict, stage: str) -> None: | |
| manifest.setdefault("stages", {})[stage] = {"done": True, "at": _utc_now()} | |
| def stage_done(manifest: dict, stage: str) -> bool: | |
| return bool(manifest.get("stages", {}).get(stage, {}).get("done")) | |
| def sorted_segment_wavs(segments_dir: Path) -> List[Path]: | |
| wavs = list(segments_dir.glob("*.wav")) | |
| if not wavs: | |
| return [] | |
| def _key(p: Path) -> Tuple[int, str]: | |
| m = _SEGMENT_RE.search(p.name) | |
| return (int(m.group(1)) if m else 10**9, p.name) | |
| return sorted(wavs, key=_key) | |