python/inflect_tts_sdk/__main__.py

"""CLI: python -m inflect_tts_sdk --text "Hello" --output out.wav [--chip ax650]"""
import argparse
from pathlib import Path
from .tts_engine import InflectTTSEngine

def main():
    ap = argparse.ArgumentParser(description="Inflect-Micro-v2 TTS (AX NPU)")
    ap.add_argument("--text", required=True)
    ap.add_argument("--output", default="output.wav")
    ap.add_argument("--speed", type=float, default=1.0)
    ap.add_argument("--variation", type=float, default=0.667)
    ap.add_argument("--seed", type=int, default=0)
    ap.add_argument("--model-dir", default=None)
    ap.add_argument("--chip", default="ax650", choices=["ax650", "ax620e", "ax637"])
    args = ap.parse_args()

    engine = InflectTTSEngine(model_dir=args.model_dir, chip=args.chip)
    engine.save(args.text, args.output, speed=args.speed,
                variation=args.variation, seed=args.seed)
    print(f"🔊 {args.output}")

if __name__ == "__main__":
    main()