Function main

scripts/plan_figures.py:477–503

()

def main() -> None:
    from common import emit

    args = parser().parse_args()
    primary = maybe_load_json_record(args.input) if args.input else None
    evidence = maybe_load_json_record(args.evidence) if args.evidence else None
    assets = maybe_load_json_record(args.assets) if args.assets else None
    data = merge_inputs(primary, evidence, assets)
    if not data:
        raise SystemExit("plan_figures.py requires at least one JSON input.")

    evidence_pack = data.get("evidence_pack", {}) if isinstance(data.get("evidence_pack"), dict) else {}
    page_assets = data.get("page_assets", []) if isinstance(data.get("page_assets"), list) else []
    image_assets = data.get("image_assets", []) if isinstance(data.get("image_assets"), list) else []
    figure_assets = data.get("figure_assets", []) if isinstance(data.get("figure_assets"), list) else []
    items = build_figure_items(evidence_pack, limit=args.max_items)
    items = attach_candidate_images(items, page_assets, image_assets, figure_assets)
    payload = {
        "status": "ok",
        "script": "plan_figures.py",
        "paper_id": args.paper_id or data.get("paper_id", ""),
        "figure_plan": {
            "paper_id": args.paper_id or data.get("paper_id", ""),
            "figures": items,
        },
    }
    emit(payload, args.output)


if __name__ == "__main__":