import json
from pathlib import Path

from sealcell.model import get_model

model = get_model()
source = json.loads(Path("data/fixtures/playground-probes.json").read_text())
out = Path("reports/local-probes.jsonl")
with out.open("w") as f:
    for case in source["cases"]:
        result = model.classify(case["text"], source["schema"])
        f.write(json.dumps({"fixture_id": case["id"], **result}) + "\n")
        f.flush()
        print(
            case["id"],
            result["error"] or {k: v["selected_label"] for k, v in result["heads"].items()},
            round(result["total_ms"]),
            flush=True,
        )
Path("reports/model-metadata.json").write_text(json.dumps(model.metadata, indent=2) + "\n")