#!/usr/bin/env python3
"""Print average metrics from eval.json for quick tracking."""

import json
from datetime import datetime
from pathlib import Path


def mean(values):
    return sum(values) / len(values) if values else None


def main():
    base_dir = Path(__file__).resolve().parent
    eval_path = base_dir / "results" / "eval.json"
    if not eval_path.exists():
        raise SystemExit(f"missing eval.json: {eval_path}")

    obj = json.loads(eval_path.read_text(encoding="utf-8"))
    ks = list(obj.get("continuous_ks", {}).values())
    jsd = list(obj.get("discrete_jsd", {}).values())
    lag = list(obj.get("continuous_lag1_diff", {}).values())

    avg_ks = mean(ks)
    avg_jsd = mean(jsd)
    avg_lag1 = mean(lag)

    print("avg_ks", avg_ks)
    print("avg_jsd", avg_jsd)
    print("avg_lag1_diff", avg_lag1)

    history_path = base_dir / "results" / "metrics_history.csv"
    if not history_path.exists():
        history_path.write_text("timestamp,avg_ks,avg_jsd,avg_lag1_diff\n", encoding="utf-8")
    with history_path.open("a", encoding="utf-8") as f:
        f.write(f"{datetime.utcnow().isoformat()},{avg_ks},{avg_jsd},{avg_lag1}\n")


if __name__ == "__main__":
    main()