#!/usr/bin/env python3
"""OCR an image via the local PaddleOCR-VL server."""
from __future__ import annotations

import argparse
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parent))
from ocr_engine import DEFAULT_BASE_URL, DEFAULT_MODEL, OcrEngine


def main() -> int:
    p = argparse.ArgumentParser()
    p.add_argument("image", type=Path)
    p.add_argument("--base-url", default=DEFAULT_BASE_URL)
    p.add_argument("--model", default=DEFAULT_MODEL)
    args = p.parse_args()
    text, dt, _ = OcrEngine(args.base_url, args.model).ocr_image(args.image)
    print(text)
    print(f"\n# {dt:.2f}s", file=sys.stderr)
    return 0


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