from __future__ import annotations

import argparse
import json
from pathlib import Path

from inferscale.reports import generate_research_report


def main() -> None:
    parser = argparse.ArgumentParser(description="Generate an InferScale Markdown research report from exported study artifacts.")
    parser.add_argument("robust_study", type=Path)
    parser.add_argument("--calibration", type=Path)
    parser.add_argument("--output", type=Path, default=Path("inferscale-research-consolidation.md"))
    args = parser.parse_args()

    robust = json.loads(args.robust_study.read_text())
    calibration = json.loads(args.calibration.read_text()) if args.calibration else None
    args.output.write_text(generate_research_report(robust, calibration))
    print(f"Wrote {args.output}")


if __name__ == "__main__":
    main()