from pathlib import Path

root = Path(r"C:\Users\jiwei\Documents\GitHub\sim-proj\dev-docs\comsol-hbm-hbn-repro")
model_dir = root / "model"
input_dir = root / "input"
model_dir.mkdir(parents=True, exist_ok=True)

case_slug = "arxiv_2510_11461_hbm_hbn_repro"
model_path = model_dir / f"{case_slug}_00_identity.mph"

model.label(case_slug)
try:
    model.title("arXiv 2510.11461 GPU-HBM h-BN thermal reproduction")
except Exception:
    pass
model.modelPath(str(model_dir))

for stale in ["paper_id", "work_root", "assumption_level"]:
    try:
        model.param().remove(stale)
    except Exception:
        pass

model.param().set("paper_rev", "2510.11461")
model.param().descr("paper_rev", "Source paper identifier: arXiv:2510.11461")
model.param().set("work_root_id", "1")
model.param().descr("work_root_id", f"Codex reproduction work folder: {root}")
model.param().set("assumption_level_id", "1")
model.param().descr(
    "assumption_level_id",
    "Paper does not specify full dimensions; geometry is a documented representative reproduction.",
)

model.save(str(model_path))

_result = {
    "step": "identity",
    "case_slug": case_slug,
    "model_path": str(model_path),
    "input_dir": str(input_dir),
    "status": "ok",
}