pi-demo-mode 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 dutifuldev
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,53 @@
+# pi-demo-mode
+
+pi-demo-mode is a [Pi](https://github.com/earendil-works/pi) extension package
+that puts Pi into a self-driving demo mode. It sends a story prompt when the
+session starts, keeps asking the model to continue after every turn (compacting
+the session when the context fills up), and strips the TUI down to the chat
+plus a single status line — no input bar, no directory line, no context meter.
+The result is a terminal that endlessly writes a story on its own, which is
+what you want for screen recordings and demo walls.
+
+## Install
+
+Into a plain Pi setup:
+
+```bash
+pi install github:osolmaz/pi-demo-mode
+```
+
+Or as an npm git dependency of a Pi launcher, pointing Pi at
+`extensions/demo-mode.ts` inside the installed package.
+
+## Usage
+
+The extension is inert until it is enabled, so it can stay installed for
+interactive sessions. Configuration is entirely through environment
+variables:
+
+| Variable                       | Meaning                                                         |
+| ------------------------------ | --------------------------------------------------------------- |
+| `PI_DEMO_MODE=1`               | enable demo mode                                                |
+| `PI_DEMO_INITIAL_PROMPT`       | first prompt text                                               |
+| `PI_DEMO_INITIAL_PROMPT_FILE`  | UTF-8 file with the first prompt (wins over the inline text)    |
+| `PI_DEMO_FOLLOWUP_PROMPT`      | repeated prompt after each turn                                 |
+| `PI_DEMO_FOLLOWUP_PROMPT_FILE` | UTF-8 file with the repeated prompt (wins over the inline text) |
+
+Without prompt overrides, a built-in never-ending sci-fi story prompt is
+used. Run Pi with tools disabled so nothing interrupts the story:
+
+```bash
+PI_DEMO_MODE=1 pi --no-tools --no-approve
+```
+
+The demo stops on its own when a turn is aborted or errors, and treats any
+message you type as a live director note for the story.
+
+## Used by
+
+- [localpi](https://github.com/osolmaz/localpi) (`localpi --demo`)
+- [diffusionpi](https://github.com/osolmaz/diffusionpi) (`diffusionpi demo`)
+
+## License
+
+[MIT](LICENSE)

package/extensions/demo-mode.ts

@@ -0,0 +1,233 @@
+// Self-driving demo mode for Pi: sends a story prompt on startup, keeps
+// asking the model to continue after every turn, and strips the TUI down to
+// the chat plus a single status line. Everything is gated on PI_DEMO_MODE=1
+// so the extension can stay installed in interactive sessions without doing
+// anything.
+//
+// Configuration (all optional):
+//   PI_DEMO_MODE=1                   enable demo mode
+//   PI_DEMO_INITIAL_PROMPT           first prompt text
+//   PI_DEMO_INITIAL_PROMPT_FILE      UTF-8 file with the first prompt
+//   PI_DEMO_FOLLOWUP_PROMPT          repeated prompt text after each turn
+//   PI_DEMO_FOLLOWUP_PROMPT_FILE     UTF-8 file with the repeated prompt
+//
+// A *_FILE variable wins over its inline counterpart; without either, the
+// built-in never-ending sci-fi story prompts are used.
+import { readFile } from "node:fs/promises";
+
+import type { ExtensionAPI, ExtensionContext, TurnEndEvent } from "@earendil-works/pi-coding-agent";
+
+const demoEnabled = process.env["PI_DEMO_MODE"] === "1";
+
+const fallbackInitialPrompt =
+  "You are narrating a never-ending sci-fi adventure. Continue in short paragraphs. Whenever the user sends a message, treat it as a live director note and incorporate it immediately. Never end the story.";
+const fallbackFollowupPrompt = "Continue. Try to write as long as possible.";
+
+const compactAtContextPercent = 70;
+const demoCompactionInstructions = [
+  "Preserve the demo narrative state, named entities, current setting,",
+  "unresolved plot threads, and latest user direction.",
+  "Keep the summary concise so the story can continue after compaction."
+].join(" ");
+
+async function promptText(inlineVar: string, fileVar: string, fallback: string): Promise<string> {
+  const file = process.env[fileVar];
+  if (file !== undefined && file !== "") {
+    try {
+      const text = (await readFile(file, "utf8")).trim();
+      if (text !== "") {
+        return text;
+      }
+    } catch {
+      // Fall through to the inline value or fallback below.
+    }
+  }
+  const inline = process.env[inlineVar];
+  return inline === undefined || inline === "" ? fallback : inline;
+}
+
+// Demo chrome: recordings only need the chat and the perf status line, so the
+// footer is replaced with a single dim line (extension statuses + model name)
+// and the input editor is hidden. The built-in footer's cwd and context lines
+// are intentionally dropped.
+type DemoFooterTheme = { fg(color: string, text: string): string };
+type DemoFooterData = { getExtensionStatuses(): ReadonlyMap<string, string> };
+type DemoFooterComponent = { render(width: number): string[] };
+type DemoChromeUi = {
+  setFooter(
+    factory:
+      | ((tui: unknown, theme: DemoFooterTheme, footerData: DemoFooterData) => DemoFooterComponent)
+      | undefined
+  ): void;
+  setEditorComponent(
+    factory: ((tui: unknown, theme: unknown, keybindings: unknown) => unknown) | undefined
+  ): void;
+};
+
+const ansiPattern = new RegExp(String.fromCharCode(27) + "\\[[0-9;]*m", "gu");
+
+function visibleLength(text: string): number {
+  return text.replace(ansiPattern, "").length;
+}
+
+function demoFooterLines(left: string, right: string, width: number): string[] {
+  const leftWidth = visibleLength(left);
+  const rightWidth = visibleLength(right);
+  if (left === "") {
+    return [" ".repeat(Math.max(0, width - rightWidth)) + right];
+  }
+  if (leftWidth + 2 + rightWidth <= width) {
+    return [left + " ".repeat(width - leftWidth - rightWidth) + right];
+  }
+  return [left, " ".repeat(Math.max(0, width - rightWidth)) + right];
+}
+
+function applyDemoChrome(ctx: ExtensionContext): void {
+  const ui = ctx.ui as unknown as Partial<DemoChromeUi>;
+  if (typeof ui.setFooter !== "function" || typeof ui.setEditorComponent !== "function") {
+    return;
+  }
+  const modelName = ctx.model?.id ?? "";
+  ui.setFooter((_tui, theme, footerData) => ({
+    render(width: number): string[] {
+      const left = Array.from(footerData.getExtensionStatuses().entries())
+        .sort(([a], [b]) => a.localeCompare(b))
+        .map(([, text]) => text.replace(/[\r\n\t]+/gu, " ").trim())
+        .filter((text) => text !== "")
+        .join(" ");
+      return demoFooterLines(left, theme.fg("dim", modelName), width);
+    }
+  }));
+  void hideEditor(ui as DemoChromeUi);
+}
+
+async function hideEditor(ui: DemoChromeUi): Promise<void> {
+  try {
+    const pkg = (await import("@earendil-works/pi-coding-agent")) as unknown as {
+      CustomEditor: new (
+        tui: unknown,
+        theme: unknown,
+        keybindings: unknown
+      ) => { render(width: number): string[] };
+    };
+    class HiddenEditor extends pkg.CustomEditor {
+      override render(): string[] {
+        return [];
+      }
+    }
+    ui.setEditorComponent((tui, theme, keybindings) => new HiddenEditor(tui, theme, keybindings));
+  } catch {
+    // Keep the default editor when the package cannot be imported (e.g. the
+    // extension is loaded outside Pi's module loader in tests).
+  }
+}
+
+export default function piDemoMode(pi: ExtensionAPI): void {
+  if (!demoEnabled) {
+    return;
+  }
+  let started = false;
+  let stopped = false;
+  let compacting = false;
+
+  function queueInitialPrompt(): void {
+    void promptText(
+      "PI_DEMO_INITIAL_PROMPT",
+      "PI_DEMO_INITIAL_PROMPT_FILE",
+      fallbackInitialPrompt
+    ).then((prompt) => {
+      if (!stopped) {
+        pi.sendUserMessage(prompt);
+      }
+    });
+  }
+
+  function queueFollowup(): void {
+    void promptText(
+      "PI_DEMO_FOLLOWUP_PROMPT",
+      "PI_DEMO_FOLLOWUP_PROMPT_FILE",
+      fallbackFollowupPrompt
+    ).then((prompt) => {
+      if (!stopped && !compacting) {
+        pi.sendUserMessage(prompt, { deliverAs: "followUp" });
+      }
+    });
+  }
+
+  function compactThenFollowup(ctx: ExtensionContext): void {
+    if (compacting) {
+      return;
+    }
+    compacting = true;
+    ctx.compact({
+      customInstructions: demoCompactionInstructions,
+      onComplete: () => {
+        compacting = false;
+        queueFollowup();
+      },
+      onError: (error) => {
+        compacting = false;
+        stopped = true;
+        ctx.ui.notify("Demo compaction failed: " + error.message, "error");
+      }
+    });
+  }
+
+  pi.on("session_start", (event, ctx) => {
+    if (ctx.mode !== "tui") {
+      return;
+    }
+    applyDemoChrome(ctx);
+    if (started || stopped || event.reason !== "startup") {
+      return;
+    }
+    started = true;
+    queueInitialPrompt();
+  });
+
+  pi.on("turn_end", (event, ctx) => {
+    if (!started || stopped || compacting || ctx.mode !== "tui") {
+      return;
+    }
+    if (event.message.role !== "assistant") {
+      return;
+    }
+    switch (event.message.stopReason) {
+      case "aborted":
+      case "error":
+        stopped = true;
+        return;
+      case "toolUse":
+        return;
+    }
+    if (shouldCompactBeforeFollowup(event, ctx)) {
+      compactThenFollowup(ctx);
+      return;
+    }
+    queueFollowup();
+  });
+
+  pi.on("session_shutdown", () => {
+    stopped = true;
+  });
+}
+
+function shouldCompactBeforeFollowup(event: TurnEndEvent, ctx: ExtensionContext): boolean {
+  const contextPercent = currentContextPercent(event, ctx);
+  return contextPercent !== undefined && contextPercent >= compactAtContextPercent;
+}
+
+function currentContextPercent(event: TurnEndEvent, ctx: ExtensionContext): number | undefined {
+  const usage = ctx.getContextUsage();
+  if (usage?.percent !== undefined && usage.percent !== null) {
+    return usage.percent;
+  }
+  if (event.message.role !== "assistant") {
+    return undefined;
+  }
+  const contextWindow = ctx.model?.contextWindow;
+  if (contextWindow === undefined || contextWindow <= 0) {
+    return undefined;
+  }
+  return (event.message.usage.totalTokens / contextWindow) * 100;
+}

package/package.json

@@ -0,0 +1,46 @@
+{
+  "name": "pi-demo-mode",
+  "version": "0.1.0",
+  "description": "Self-driving demo mode for Pi: continuous story generation with a stripped-down TUI",
+  "keywords": [
+    "pi-package",
+    "demo",
+    "extension"
+  ],
+  "license": "MIT",
+  "type": "module",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/osolmaz/pi-demo-mode.git"
+  },
+  "bugs": {
+    "url": "https://github.com/osolmaz/pi-demo-mode/issues"
+  },
+  "homepage": "https://github.com/osolmaz/pi-demo-mode#readme",
+  "files": [
+    "extensions/**",
+    "README.md",
+    "LICENSE"
+  ],
+  "pi": {
+    "extensions": [
+      "./extensions"
+    ]
+  },
+  "scripts": {
+    "format": "prettier --check .",
+    "typecheck": "tsc --noEmit",
+    "test": "vitest run",
+    "check": "npm run format && npm run typecheck && npm test"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*"
+  },
+  "devDependencies": {
+    "@earendil-works/pi-coding-agent": "^0.82.1",
+    "@types/node": "^22.0.0",
+    "prettier": "^3.0.0",
+    "typescript": "^5.0.0",
+    "vitest": "^3.0.0"
+  }
+}