Spaces:
Sleeping
Sleeping
| #!/usr/bin/env python3 | |
| """Create four forensic plots using only one NotePilot debug bundle.""" | |
| import json | |
| import sys | |
| from pathlib import Path | |
| def main(): | |
| if len(sys.argv) not in (2, 3): | |
| print('usage: plot_debug_bundle.py bundle.json [output-directory]', file=sys.stderr); return 2 | |
| bundle = json.loads(Path(sys.argv[1]).read_text()) | |
| out = Path(sys.argv[2]) if len(sys.argv) == 3 else Path(sys.argv[1]).with_suffix('') | |
| out.mkdir(parents=True, exist_ok=True) | |
| rows = bundle['diagnostics']['timeline'] | |
| t = [r['tMs'] / 1000 for r in rows] | |
| def vals(name): return [r.get(name) for r in rows] | |
| events = bundle['events'] | |
| def marks(kind): return [e['sessionTimeMs'] / 1000 for e in events if e.get('type') == kind] | |
| # Self-contained SVG avoids a plotting-package dependency in a fresh shell. | |
| def graph(name, title, series, vertical=()): | |
| width, height, pad = 1200, 460, 55 | |
| numeric = [v for _, values, _ in series for v in values if isinstance(v, (int, float))] | |
| lo, hi = (min(numeric), max(numeric)) if numeric else (0, 1) | |
| if lo == hi: lo -= 1; hi += 1 | |
| end = max(t) if t else 1 | |
| def xy(i, value): return (pad + (t[i] / end) * (width - 2 * pad), height - pad - ((value - lo) / (hi - lo)) * (height - 2 * pad)) | |
| paths = [] | |
| for label, values, color in series: | |
| points = [f'{x:.2f},{y:.2f}' for i, v in enumerate(values) if isinstance(v, (int, float)) for x, y in [xy(i, v)]] | |
| paths.append(f'<polyline fill="none" stroke="{color}" stroke-width="1.5" points="{" ".join(points)}"/><text x="{pad}" y="{22 + 16*len(paths)}" fill="{color}">{label}</text>') | |
| lines = ''.join(f'<line x1="{pad+(x/end)*(width-2*pad):.2f}" x2="{pad+(x/end)*(width-2*pad):.2f}" y1="{pad}" y2="{height-pad}" stroke="#c44" opacity=".25"/>' for x in vertical) | |
| svg = f'<svg xmlns="http://www.w3.org/2000/svg" width="{width}" height="{height}" viewBox="0 0 {width} {height}"><rect width="100%" height="100%" fill="white"/><text x="{pad}" y="20" font-size="16">{title}</text><line x1="{pad}" x2="{pad}" y1="{pad}" y2="{height-pad}" stroke="black"/><line x1="{pad}" x2="{width-pad}" y1="{height-pad}" y2="{height-pad}" stroke="black"/>{lines}{"".join(paths)}<text x="{pad}" y="{height-15}">session seconds (range 0–{end:.2f}); value range {lo:.3g}–{hi:.3g}</text></svg>' | |
| (out / name).write_text(svg) | |
| attempts = [e['sessionTimeMs']/1000 for e in events if e.get('physicalAttemptId') is not None] | |
| graph('01-mic-level.svg', 'Microphone level / noise floor / attempts', [('mic dB', vals('db'), '#1769aa'), ('noise floor dB', vals('noiseFloorDb'), '#777')], attempts + marks('mic.gate')) | |
| graph('02-pitch.svg', 'Blind YIN pitch and expected score', [('blind MIDI', vals('blindMidi'), '#1769aa'), ('expected MIDI', vals('expectedMidi'), '#e68613')], attempts) | |
| graph('03-onset.svg', 'Onset feature and attack flags', [('onset ratio', vals('onsetValue'), '#6a3d9a')], [r['tMs']/1000 for r in rows if r.get('attack')]) | |
| graph('04-follower.svg', 'Follower posterior / WRONG / evidence', [('posterior', vals('posterior'), '#1769aa'), ('committed posterior', vals('committedPosterior'), '#e68613'), ('WRONG', vals('wrongProbability'), '#c44'), ('evidence nats', vals('evidenceNats'), '#6a3d9a')]) | |
| print(f'Wrote four plots to {out}'); return 0 | |
| if __name__ == '__main__': raise SystemExit(main()) | |