"""JSONL inference CLI; compiles but never executes tools."""

import argparse
import json
import sys

from core import compile_js
from model import Model


def main():
    ap = argparse.ArgumentParser(description=__doc__)
    ap.add_argument(
        "--checkpoint", default=None, help="NPZ weights (default: packaged CNN)"
    )
    ap.add_argument("--js", action="store_true")
    args = ap.parse_args()
    model = Model(args.checkpoint)
    for line in sys.stdin:
        try:
            item = json.loads(line)
            plan = model.predict(item)
            print(
                json.dumps(
                    {"plan": plan, "js": compile_js(plan, item["tools"])}
                    if args.js
                    else {"plan": plan}
                )
            )
        except (ValueError, KeyError, IndexError) as e:
            print(json.dumps({"diagnostic": str(e)}))


if __name__ == "__main__":
    main()