@getmegabrain/cli 0.1.8 → 0.1.9

package/dist/science/kernel_driver.py

@@ -13,9 +13,14 @@ cell are visible in the next.
 import contextlib
 import io
 import json
+import os
 import sys
 import traceback
 
+# Render matplotlib to a non-interactive backend so plt.show() never blocks; we capture
+# figures ourselves after each cell (see emit_figures). Set before any user import.
+os.environ.setdefault("MPLBACKEND", "Agg")
+
 # Persistent namespace shared across all cells in this session.
 NAMESPACE: dict = {"__name__": "__mb_science__"}
 
@@ -25,6 +30,30 @@ def emit(frame: dict) -> None:
     sys.stdout.flush()
 
 
+def emit_figures(cell_id: str) -> int:
+    """Emit any open matplotlib figures as base64 PNG image frames, then close them.
+
+    This is what makes plots appear inline in the transcript (a science-notebook staple).
+    No-op if matplotlib hasn't been used in this session."""
+    pyplot = sys.modules.get("matplotlib.pyplot")
+    if pyplot is None:
+        return 0
+    import base64
+
+    count = 0
+    try:
+        for num in pyplot.get_fignums():
+            buf = io.BytesIO()
+            pyplot.figure(num).savefig(buf, format="png", bbox_inches="tight", dpi=110)
+            data = base64.b64encode(buf.getvalue()).decode("ascii")
+            emit({"type": "image", "cellId": cell_id, "mime": "image/png", "data": data})
+            count += 1
+        pyplot.close("all")
+    except Exception:  # noqa: BLE001 - figure capture is best-effort
+        pass
+    return count
+
+
 def run_cell(cell_id: str, code: str) -> None:
     emit({"type": "status", "cellId": cell_id, "state": "busy"})
     stdout, stderr = io.StringIO(), io.StringIO()
@@ -41,6 +70,11 @@ def run_cell(cell_id: str, code: str) -> None:
         emit({"type": "stream", "cellId": cell_id, "kind": "stdout", "data": out})
     if err:
         emit({"type": "stream", "cellId": cell_id, "kind": "stderr", "data": err})
+    figures = emit_figures(cell_id)
+    if figures:
+        # A note in stdout so the (text-only) model knows a figure was produced.
+        emit({"type": "stream", "cellId": cell_id, "kind": "stdout",
+              "data": f"[{figures} figure(s) displayed]\n"})
     emit({"type": "status", "cellId": cell_id, "state": "idle"})
 
 

package/dist/science/pages.js

@@ -307,6 +307,8 @@ export function sessionPage(projectId, sessionId) {
         white-space:pre-wrap; overflow-x:auto; border-top:1px solid var(--border); }
       .step .out { color:var(--muted); background:var(--bg); }
       .step .out.err { color:#c0392b; }
+      img.figure { max-width:calc(100% - 24px); border-radius:8px; margin:10px 12px; display:block;
+        background:#fff; border:1px solid var(--border); }
       /* Composer */
       .composer { flex-shrink:0; border-top:1px solid var(--border); padding:14px 20px; }
       .composer-inner { max-width:760px; margin:0 auto; display:flex; gap:10px; align-items:flex-end;

package/dist/science/session-client.js

@@ -64,6 +64,18 @@
     if (stderr) out.classList.add('err');
     scroll();
   }
+  // Inline a rendered figure (matplotlib PNG) under the current step, or on its own if none.
+  function addFigure(mime, data) {
+    var img = el('img', 'figure');
+    img.src = 'data:' + mime + ';base64,' + data;
+    if (lastCard) lastCard.appendChild(img);
+    else {
+      var row = el('div', 'msg agent');
+      row.appendChild(img);
+      transcript.appendChild(row);
+    }
+    scroll();
+  }
 
   function setRunning(on) {
     sendBtn.disabled = on;
@@ -112,6 +124,7 @@
       if (m.channel === 'kernel') {
         var f = m.payload;
         if (f.type === 'status') kstate.textContent = f.state === 'busy' ? 'running' : f.state;
+        else if (f.type === 'image') addFigure(f.mime, f.data);
         return;
       }
       var p = m.payload;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@getmegabrain/cli",
-  "version": "0.1.8",
+  "version": "0.1.9",
   "description": "MegaBrain CLI — sign in once, then code from your terminal via OpenCode wired to the MegaBrain Gateway, or run `megabrain science` for the local research workbench.",
   "license": "MIT",
   "type": "module",