1#!/usr/bin/env python3
2"""Human-facing status dashboard for the autonomous research loop."""
3 
4from __future__ import annotations
5 
6import argparse
7import json
8import sys
9from pathlib import Path
10from typing import Any
11 
12from loop_common import (
13    QUEUE_DIR,
14    REPORTS_DIR,
15    RESEARCHER,
16    ROOT,
17    categorize_runs,
18    load_run_state,
19    read_jsonl,
20    runs_created_today,
21    utc_now,
22)
23 
24 
25STATUS_FILE = REPORTS_DIR / "status.md"
26PARKED_REVIEW_FILE = REPORTS_DIR / "parked-review.md"
27 
28 
29def latest_benchmark() -> dict[str, Any] | None:
30    history = REPORTS_DIR / "benchmark-history.jsonl"
31    if not history.exists():
32        return None
33    last: dict[str, Any] | None = None
34    for line in history.read_text(encoding="utf-8").splitlines():
35        if not line.strip():
36            continue
37        try:
38            last = json.loads(line)
39        except json.JSONDecodeError:
40            continue
41    return last
42 
43 
44def latest_snapshot() -> Path | None:
45    snapshot_dir = REPORTS_DIR / "snapshots"
46    if not snapshot_dir.exists():
47        return None
48    candidates = sorted(snapshot_dir.glob("*.md"))
49    return candidates[-1] if candidates else None
50 
51 
52def render_status() -> dict[str, Any]:
53    buckets = categorize_runs()
54    parked = read_jsonl(QUEUE_DIR / "parked.jsonl")
55    inbox = read_jsonl(QUEUE_DIR / "inbox.jsonl")
56    quarantine = read_jsonl(QUEUE_DIR / "quarantine.jsonl")
57    done = read_jsonl(QUEUE_DIR / "done.jsonl")
58    bench = latest_benchmark()
59    snapshot = latest_snapshot()
60 
61    lines = [
62        f"# Researcher Loop Status",
63        f"_generated {utc_now()}_",
64        "",
65        "## Queue",
66        f"- inbox: {len(inbox)}",
67        f"- active runs: {len(buckets['active'])}",
68        f"- parked runs: {len(parked)}",
69        f"- closed runs (state): {len(buckets['closed'])}",
70        f"- done ledger: {len(done)}",
71        f"- quarantined: {len(quarantine)}",
72        f"- runs created today: {runs_created_today()}",
73        "",
74        "## Latest Benchmark",
75    ]
76    if bench:
77        summary = bench.get("summary", {})
78        lines.extend(
79            [
80                f"- timestamp: {bench.get('timestamp')}",
81                f"- ok: {bench.get('ok')}",
82                f"- checks: {summary.get('benchmarks')} failures: {summary.get('failures')} scenarios: {summary.get('scenarios')}",
83            ]
84        )
85    else:
86        lines.append("- no benchmark history yet")
87    lines.append("")
88    lines.append("## Latest Snapshot")
89    lines.append(f"- {snapshot.relative_to(ROOT)}" if snapshot else "- no snapshot yet")
90    lines.append("")
91    lines.append("## Active Runs")
92    if not buckets["active"]:
93        lines.append("- none")
94    for run_dir in buckets["active"]:
95        state = load_run_state(run_dir) or {}
96        lines.append(
97            f"- {run_dir.name}: state={state.get('current_state')} updated={state.get('updated_at')}"
98        )
99    lines.append("")
100    lines.append("## Parked Runs")
101    if not parked:
102        lines.append("- none")
103    for record in parked:
104        lines.append(f"- {record.get('run_id')}: {record.get('reason')} (parked {record.get('parked_at')})")
105    lines.append("")
106    lines.append("## Inbox Preview")
107    if not inbox:
108        lines.append("- empty")
109    for record in inbox[:5]:
110        lines.append(f"- {record.get('source_id')} {record.get('url')}")
111 
112    STATUS_FILE.parent.mkdir(parents=True, exist_ok=True)
113    STATUS_FILE.write_text("\n".join(lines) + "\n", encoding="utf-8")
114 
115    if parked:
116        parked_lines = [f"# Parked Runs Needing Review", f"_generated {utc_now()}_", ""]
117        for record in parked:
118            run_dir = RESEARCHER / "runs" / record.get("run_id", "")
119            state = load_run_state(run_dir) or {}
120            parked_lines.extend(
121                [
122                    f"## {record.get('run_id')}",
123                    f"- reason: {record.get('reason')}",
124                    f"- parked at: {record.get('parked_at')}",
125                    f"- current state: {state.get('current_state')}",
126                    f"- title: {state.get('title')}",
127                    f"- source: {state.get('source_url')}",
128                    f"- thread: {(run_dir / 'THREAD.md').relative_to(ROOT) if (run_dir / 'THREAD.md').exists() else 'missing'}",
129                    "",
130                ]
131            )
132        PARKED_REVIEW_FILE.write_text("\n".join(parked_lines), encoding="utf-8")
133    elif PARKED_REVIEW_FILE.exists():
134        PARKED_REVIEW_FILE.write_text("# Parked Runs Needing Review\n\n- none\n", encoding="utf-8")
135 
136    return {
137        "status_path": str(STATUS_FILE.relative_to(ROOT)),
138        "parked_review_path": str(PARKED_REVIEW_FILE.relative_to(ROOT)) if parked else None,
139        "queue": {
140            "inbox": len(inbox),
141            "active": len(buckets["active"]),
142            "parked": len(parked),
143            "closed": len(buckets["closed"]),
144            "done": len(done),
145            "quarantine": len(quarantine),
146        },
147        "runs_created_today": runs_created_today(),
148        "latest_benchmark_ok": bench.get("ok") if bench else None,
149    }
150 
151 
152def main() -> int:
153    parser = argparse.ArgumentParser(description="Write loop status dashboard")
154    parser.add_argument("--json", action="store_true")
155    args = parser.parse_args()
156    result = render_status()
157    if args.json:
158        print(json.dumps(result, indent=2))
159    else:
160        print(f"status written to {result['status_path']}")
161        if result["parked_review_path"]:
162            print(f"parked review at {result['parked_review_path']}")
163    return 0
164 
165 
166if __name__ == "__main__":
167    sys.exit(main())
168