File size: 1,457 Bytes
90884df | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | #!/usr/bin/env python3
from __future__ import annotations
import argparse
import json
from pathlib import Path
from music3lab.editing.waveform_right_context_prepend_runner import (
load_waveform_right_context_prepend_config,
run_waveform_right_context_prepend,
)
def main() -> int:
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers(dest="command", required=True)
check = subparsers.add_parser("check")
check.add_argument("--config", type=Path, required=True)
run = subparsers.add_parser("run")
run.add_argument("--config", type=Path, required=True)
run.add_argument("--corpus-root", type=Path, required=True)
run.add_argument("--output-root", type=Path, required=True)
run.add_argument("--user-audio", type=Path)
args = parser.parse_args()
if args.command == "check":
config = load_waveform_right_context_prepend_config(args.config)
result = {"status": "READY", "schema_version": config["schema_version"]}
code = 0
else:
result = run_waveform_right_context_prepend(
config_path=args.config,
corpus_root=args.corpus_root,
output_root=args.output_root,
user_audio=args.user_audio,
)
code = 0 if result["status"] == "FUNCTIONAL_PASS" else 1
print(json.dumps(result, sort_keys=True, separators=(",", ":")))
return code
if __name__ == "__main__":
raise SystemExit(main())
|