"""Run the preregistered P2-E1/P2-E2 one-second latent inversion."""
from __future__ import annotations

import argparse
import json
from pathlib import Path

from music3lab.inversion import run_inversion_session


def parser() -> argparse.ArgumentParser:
    result = argparse.ArgumentParser(description=__doc__)
    result.add_argument("--config", type=Path, required=True)
    result.add_argument("--snapshot", type=Path, required=True)
    result.add_argument("--base-manifest", type=Path, required=True)
    result.add_argument("--diffusers-root", type=Path, required=True)
    result.add_argument("--phase0-artifacts", type=Path, required=True)
    result.add_argument("--output-root", type=Path, required=True)
    return result


def main() -> int:
    args = parser().parse_args()
    verified = run_inversion_session(
        config_path=args.config,
        snapshot=args.snapshot,
        base_manifest=args.base_manifest,
        diffusers_root=args.diffusers_root,
        phase0_artifacts=args.phase0_artifacts,
        output_root=args.output_root,
    )
    print(json.dumps({
        "status": (
            "FEASIBILITY_PASS"
            if verified.manifest.all_feasibility_pass
            else "FAIL"
        ),
        "high_fidelity_status": (
            "HIGH_FIDELITY_PASS"
            if verified.manifest.all_high_fidelity_pass
            else "NOT_HIGH_FIDELITY"
        ),
        "quality_claim": verified.manifest.quality_claim,
        "session_semantic_digest": verified.manifest.semantic_digest,
        "session_file_sha256": verified.manifest_file_sha256,
        "output_root": str(args.output_root),
        "experiments": [{
            "experiment_id": item.manifest.experiment_id,
            "status": item.manifest.status,
            "high_fidelity_status": item.manifest.high_fidelity_status,
            "quality_claim": item.manifest.quality_claim,
            "manifest_semantic_digest": item.manifest.semantic_digest,
            "manifest_file_sha256": item.manifest_file_sha256,
            "selected_restart_seed": item.manifest.selected_restart_seed,
            "metrics": item.metrics.model_dump(mode="json"),
            "telemetry": item.manifest.telemetry.model_dump(mode="json"),
        } for item in verified.experiments],
    }, sort_keys=True))
    return 0


if __name__ == "__main__":
    raise SystemExit(main())