"""Transcribe one file locally, or download weights with an explicit --repo."""
import argparse
from huggingface_hub import snapshot_download
from haas_cpu.pipeline import Transcriber
from haas_cpu.external.audio_io import read_audio

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument('audio')
    parser.add_argument('--repo', help='Published Hugging Face model ID (download only)')
    parser.add_argument('--format', choices=('int4-onecomp', 'int8', 'fp32'), default='int4-onecomp')
    args = parser.parse_args()
    models = snapshot_download(repo_id=args.repo) if args.repo else None
    model = Transcriber(models, fmt=args.format, threads=4)
    print(model.transcribe_long(read_audio(args.audio)).text)