serve.py

#!/usr/bin/env python3
"""Run the HARRIER FastAPI service.

    HARRIER_WEIGHTS=/path/to/best.pt HARRIER_BACKEND=cuda \\
        python scripts/serve.py --host 0.0.0.0 --port 8010
"""

from __future__ import annotations

import argparse
import sys
from pathlib import Path

_REPO_ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(_REPO_ROOT / "src"))


def main(argv: list[str] | None = None) -> int:  # pragma: no cover - runtime only
    parser = argparse.ArgumentParser(prog="anima-harrier-serve")
    parser.add_argument("--host", default="0.0.0.0")
    parser.add_argument("--port", type=int, default=8010)
    parser.add_argument("--workers", type=int, default=1)
    args = parser.parse_args(argv)

    import uvicorn  # type: ignore

    uvicorn.run(
        "anima_harrier.serving.app:build_app",
        host=args.host,
        port=args.port,
        workers=args.workers,
        factory=True,
    )
    return 0


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