"""Run one oruk file request and print the complete JSON response.

Python 3.10+, oruk SDK 0.2.5+. Set ORUK_API_KEY, then:
  python analyze-file.py recording.wav --task analysis
  python analyze-file.py recording.wav --task analysis --diarize
  python analyze-file.py speaking.wav --task proficiency

This calls the metered file API, not the separate realtime WebSocket API.
For local fixtures/private deployments only, set ORUK_API_BASE_URL.
"""
from __future__ import annotations

import argparse
import json
import os
from pathlib import Path
import sys

from oruk import Oruk, OrukAPIError


METHODS = {
    "analysis": "analyze",
    "transcriptions": "transcribe",
    "emotions": "emotions",
    "styles": "styles",
    "affect": "affect",
    "proficiency": "proficiency",
}


def main() -> int:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("file", type=Path, help="local recorded English audio")
    parser.add_argument("--task", choices=METHODS, default="analysis")
    parser.add_argument("--model", choices=["oruk-resonance", "oruk-fourier", "oruk-proficiency-1"])
    parser.add_argument("--diarize", action="store_true", help="Resonance file endpoints only")
    parser.add_argument("--num-speakers", type=int, help="known speaker count, 1–32; requires --diarize")
    parser.add_argument("--transcript-file", type=Path, help="UTF-8 transcript for proficiency only")
    parser.add_argument("--request-id", help="optional trace ID, reused across SDK retries")
    args = parser.parse_args()

    model = args.model or ("oruk-proficiency-1" if args.task == "proficiency" else "oruk-resonance")
    if (args.task == "proficiency") != (model == "oruk-proficiency-1"):
        parser.error("proficiency requires oruk-proficiency-1; other tasks use Resonance or Fourier")
    if args.diarize and model != "oruk-resonance":
        parser.error("--diarize requires a Resonance file endpoint")
    if args.num_speakers is not None and (not args.diarize or not 1 <= args.num_speakers <= 32):
        parser.error("--num-speakers requires --diarize and a count from 1 to 32")
    if args.transcript_file and args.task != "proficiency":
        parser.error("--transcript-file is supported only for proficiency")
    if not args.file.is_file():
        parser.error(f"audio file does not exist: {args.file}")
    api_key = os.environ.get("ORUK_API_KEY")
    if not api_key:
        parser.error("set ORUK_API_KEY in the environment; do not put keys in source")

    try:
        options = {"model": model}
        if args.request_id:
            options["request_id"] = args.request_id
        if args.diarize:
            options["diarize"] = True
        if args.num_speakers is not None:
            options["num_speakers"] = args.num_speakers
        if args.transcript_file:
            transcript = args.transcript_file.read_text(encoding="utf-8")
            if not transcript.strip():
                parser.error("--transcript-file must contain the recording's transcript")
            options["transcript"] = transcript
        with Oruk(
            api_key=api_key,
            base_url=os.environ.get("ORUK_API_BASE_URL", "https://speech-api.oruk.ai"),
        ) as client:
            result = getattr(client, METHODS[args.task])(args.file, **options)
        # Preserve empty labels, reference usage fields, and proficiency check.status.
        # Reference estimated_cost_usd is not the subscription invoice.
        print(json.dumps(result, indent=2, ensure_ascii=False))
        return 0
    except OrukAPIError as error:
        print(json.dumps({"status": error.status, "code": error.code,
                          "request_id": error.request_id, "message": str(error)}), file=sys.stderr)
        return 1
    except Exception as error:
        print(f"Request failed: {error}", file=sys.stderr)
        return 1


if __name__ == "__main__":
    raise SystemExit(main())