fix: show toc in static export (marimo-team#1885)

manzt · Jul 25, 2024 · 461d6df · 461d6df
1 parent 6374404
commit 461d6df
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/frontend/src/core/cells/cells.ts b/frontend/src/core/cells/cells.ts
@@ -18,7 +18,7 @@ import { prepareCellForExecution, transitionCell } from "./cell";
 import { store } from "../state/jotai";
 import { createReducerAndAtoms } from "../../utils/createReducer";
 import { foldAllBulk, unfoldAllBulk } from "../codemirror/editing/commands";
-import { findCollapseRange, mergeOutlines } from "../dom/outline";
+import { findCollapseRange, mergeOutlines, parseOutline } from "../dom/outline";
 import type { CellHandle } from "@/components/editor/Cell";
 import { Logger } from "@/utils/Logger";
 import { Objects } from "@/utils/objects";
@@ -143,9 +143,13 @@ function initialNotebookState(): NotebookState {
         config: deserializeJson(deserializeBase64(config)),
         serializedEditorState: null,
       };
+      const outputMessage = output
+        ? deserializeJson(deserializeBase64(output))
+        : null;
       cellRuntime[cellId] = {
         ...createCellRuntimeState(),
-        output: output ? deserializeJson(deserializeBase64(output)) : null,
+        output: outputMessage,
+        outline: parseOutline(outputMessage),
         consoleOutputs: outputs.map((output) =>
           deserializeJson(deserializeBase64(output)),
         ),