#!/usr/bin/env python3
"""OpusBWE.AXERA demo: 16 kHz wav -> 48 kHz wav.

The neural feature net runs on the AX650 NPU; the DSP synthesis is the bundled
C++ backend. This wrapper starts `bin/bwe_ax650` with the right environment.
"""
import argparse
import os
import subprocess
import sys

ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))


def main():
    ap = argparse.ArgumentParser(description="OpusBWE.AXERA 16k -> 48k bandwidth extension")
    ap.add_argument("input", help="input wav (16 kHz)")
    ap.add_argument("output", help="output wav (48 kHz)")
    ap.add_argument("--model", default=os.path.join(ROOT, "models", "bwe_feature_net.axmodel"))
    ap.add_argument("--bin", default=os.path.join(ROOT, "bin", "bwe_ax650"))
    args = ap.parse_args()

    if not os.path.isfile(args.bin):
        sys.exit(f"backend not found: {args.bin}")
    env = dict(os.environ)
    env["LD_LIBRARY_PATH"] = "/soc/lib:" + env.get("LD_LIBRARY_PATH", "")
    cmd = [args.bin, args.model, args.input, args.output]
    print("+", " ".join(cmd))
    sys.exit(subprocess.run(cmd, env=env).returncode)


if __name__ == "__main__":
    main()