"""CLI: аудио файл → текст.

Пример:
    python -m audio_audit.cli examples/jfk.wav
    python -m audio_audit.cli examples/jfk.wav --language ru --json
"""

from __future__ import annotations

import argparse
import json
import sys

from .engine import AuditEngine


def main(argv=None) -> int:
    ap = argparse.ArgumentParser(
        prog="audit", description="Audit: аудио → текст (Whisper base)"
    )
    ap.add_argument("audio", help="путь к аудио: wav, mp3, m4a, ogg, flac")
    ap.add_argument(
        "--language",
        "-l",
        default=None,
        help="язык, напр. ru или en; по умолчанию автоопределение",
    )
    ap.add_argument("--beam-size", type=int, default=5, help="размер beam search")
    ap.add_argument(
        "--model-dir",
        default=None,
        help="каталог с весами (по умолчанию ./model)",
    )
    ap.add_argument("--json", action="store_true", help="вывести полный JSON")
    args = ap.parse_args(argv)

    engine = AuditEngine(model_dir=args.model_dir)
    result = engine.transcribe(args.audio, language=args.language, beam_size=args.beam_size)

    if args.json:
        print(json.dumps(result, ensure_ascii=False, indent=2))
    else:
        print(f"[{result['language']} p={result['language_probability']}]")
        print(result["text"])
    return 0


if __name__ == "__main__":
    sys.exit(main())