@audiofool/moondog 0.2.2 → 0.3.0-beta.2

package/src/surfaces/cli/tui.mjs

@@ -1,4 +1,5 @@
-import { LOCALES, N_, liveScreenTranslator, normalizeLocale } from "../../i18n/index.mjs";
+import { homedir } from "node:os";
+import { LOCALES, N_, intlLocale, liveScreenTranslator, normalizeLocale, translator } from "../../i18n/index.mjs";
 import {
   CombinedAutocompleteProvider,
   Container,
@@ -53,6 +54,10 @@ import {
   setBrandText,
 } from "./brand-components.mjs";
 
+// What making a track says in the footer, stage by stage (src/integrations/stable-audio).
+const generationStages = { starting: N_("warming up"), reading: N_("reading your taste"), composing: N_("composing"),
+  decoding: N_("pressing the record"), writing: N_("saving it") };
+
 // The lines Moondog shows, as a radio host, while the model thinks; the window has the same ones.
 export const hostLines = (tr) => [tr("Reading the room"), tr("Checking the request line"), tr("Finding the groove"), tr("Working the decks"),
   tr("Planning the segue"), tr("Mic check"), tr("Riding the faders"), tr("Finding the cue")];
@@ -120,6 +125,8 @@ const slashCommands = [
   { name: "endpoint", description: N_("Add a model service by its address") },
   { name: "web", description: N_("Look things up on public music sites") },
   { name: "service", description: N_("Choose where Moondog plays music") },
+  { name: "generation", description: N_("Let Moondog make music on this Mac") },
+  { name: "onair", description: N_("Go on air: Moondog keeps making music from your taste") },
   { name: "spotify", description: N_("Connect and control Spotify") },
   { name: "skip", description: N_("Skip to the next song preview") },
   { name: "stop", description: N_("Stop the song previews") },
@@ -223,6 +230,8 @@ export async function runMoondogTui({
   // The terminal's preview player. The room shows what it plays (subscribe) and
   // gives the listener /skip and /stop.
   previewPlayer,
+  // Making music on this Mac with Stable Audio 3: { status, check, save }, or nothing where the room can't play it.
+  generation,
   environment = process.env,
 }) {
   if (providers === listPiProviders && models === listPiModels) {
@@ -238,7 +247,10 @@ export async function runMoondogTui({
   // /skip and /stop exist only where the room controls the previews itself.
   const previewControls = typeof previewPlayer?.subscribe === "function" &&
     typeof previewPlayer.skip === "function" && typeof previewPlayer.stop === "function";
-  const commands = slashCommands.filter(({ name }) => previewControls || !["skip", "stop"].includes(name));
+  // /onair needs both: the room's own player and music making here.
+  const onAirControls = previewControls && typeof generation?.save === "function";
+  const commands = slashCommands.filter(({ name }) => (previewControls || !["skip", "stop"].includes(name)) &&
+    (name !== "generation" || typeof generation?.save === "function") && (name !== "onair" || onAirControls));
   const localizedCommands = commands.map(({ name, description }) => ({ name, get description() { return tr.marked(description); } }));
   let activeRuntime = runtime;
   application.startNewSession?.("tui_start");
@@ -282,6 +294,8 @@ export async function runMoondogTui({
   let firstRunNote = null;
   let firstRunSetup = null;
   let firstRunTicket = 0;
+  // The skip being saved, which the room waits for before it closes.
+  let firstRunSaving = null;
   let stopWatchingFirstRun = null;
   let importPage = "start";
   let importReturnHome = false;
@@ -315,7 +329,20 @@ export async function runMoondogTui({
     : Math.min(messageQueue.length, 4) + (messageQueue.length > 4 ? 1 : 0);
   // The preview now playing sits on its own line above the draft.
   let nowPlaying = null;
-  const nowPlayingRows = () => (nowPlaying && !profileView && !importView && !firstRunView ? 1 : 0);
+  // Moondog's stream as the core shows it: null, or { phase, started_at, ends_at, tracks_made }.
+  let onAir = application.streamView?.() ?? null;
+  // Between two of the stream's tracks nothing plays, and the line says the next one is being made.
+  const dryGap = () => !nowPlaying && onAir?.phase === "waiting";
+  const nowPlayingRows = () => ((nowPlaying || dryGap()) && !profileView && !importView && !firstRunView ? 1 : 0);
+  // When the stream ends at the latest, as a clock time: an upper bound, so never a countdown. The clock
+  // follows the screen's language, as the words around it do.
+  const onAirUntil = () => {
+    const ends = new Date(onAir?.ends_at);
+    let time;
+    try { time = new Intl.DateTimeFormat(intlLocale(tr.locale), { hour: "numeric", minute: "2-digit" }).format(ends); }
+    catch { time = ends.toTimeString().slice(0, 5); }
+    return tr("on air until {time}", { time });
+  };
   let cancelWork = null;
   let cancellationRequested = false;
   let workStartedAt = 0;
@@ -470,14 +497,21 @@ export async function runMoondogTui({
     render(width) {
       if (profileView || importView || firstRunView) return [];
       const lines = [];
-      if (nowPlaying) {
-        const { track, position, count, type } = nowPlaying;
+      if (nowPlaying || dryGap()) {
+        const { track, position, count, type } = nowPlaying ?? {};
         const prefix = `  ${theme.plain ? ">" : "♪"} `;
         // Where the terminal can't open links, the note in the conversation shows the address.
-        const apple = getCapabilities().hyperlinks ? hyperlink("Apple Music", track.catalog_url) : "Apple Music";
+        const apple = nowPlaying && getCapabilities().hyperlinks ? hyperlink("Apple Music", track.catalog_url) : "Apple Music";
         const state = type === "loading" ? tr("loading") : type === "paused" ? tr("paused") : "";
-        const controls = `  ${position}/${count}${state ? ` · ${state}` : ""} · ${apple} · /skip · /stop`;
-        const song = sanitizeTerminalText(`${track.title} · ${track.artist_credit}`).replace(/\s+/gu, " ").trim();
+        // Once its first track plays, a made track is the stream's, which goes on to the next with /skip.
+        const onAirTrack = track?.generated && onAir && onAir.phase !== "starting";
+        // A track Moondog made is its own record, kept on this Mac, with nothing to open elsewhere.
+        const controls = !nowPlaying ? `  · ${onAirUntil()} · /stop`
+          : onAirTrack ? `${state ? `  ${state}` : ""}  · ${onAirUntil()} · /skip · /stop`
+          : track.generated ? `${state ? `  ${state}` : ""}  · /stop`
+          : `  ${position}/${count}${state ? ` · ${state}` : ""} · ${apple} · /skip · /stop`;
+        const song = !nowPlaying ? tr("Cutting the next record")
+          : sanitizeTerminalText(`${track.title} · ${track.generated ? tr("Made by Moondog") : track.artist_credit}`).replace(/\s+/gu, " ").trim();
         const room = width - visibleWidth(prefix) - visibleWidth(controls);
         lines.push(paintBrandLine(room >= 12
           ? theme.accent(prefix) + theme.text(truncateToWidth(song, room, "…")) + theme.faint(controls)
@@ -702,18 +736,82 @@ export async function runMoondogTui({
   };
   const stopWatchingPreviews = previewControls ? previewPlayer.subscribe((event) => {
     if (cleanedUp) return;
-    if (event?.type === "queue") addPreviewNote(event);
+    // A track Moondog made needs no preview note: the reply names it, and the host's line says where it was made.
+    if (event?.type === "queue") { if (!event.tracks?.some((track) => track.generated)) addPreviewNote(event); }
     else if (event?.type === "failed") markPreviewFailed(event);
     else if (["loading", "playing", "paused"].includes(event?.type)) nowPlaying = event;
     else if (event?.type === "stopped") nowPlaying = null;
     tui.requestRender();
   }) : null;
+  // The stream redraws the line above the draft as it moves on. Ending by itself, it goes dark
+  // without a word; only a track that couldn't be made says so.
+  const stopWatchingStream = application.subscribeStream?.(({ view, ended } = {}) => {
+    if (cleanedUp) return;
+    onAir = view ?? null;
+    if (ended === "failed") setFooter(tr("Off air: the next record couldn't be made."), failure);
+    tui.requestRender();
+  }) ?? null;
   // These act at once, even while Moondog is answering.
   const controlPreviews = (command) => {
-    const acted = command === "stop" ? previewPlayer.stop() : previewPlayer.skip();
-    if (!busy) setFooter(!acted ? tr("No previews are playing.") : command === "stop" ? tr("Previews stopped.") : tr("Skipped."));
+    // /stop also ends Moondog's stream while its next track is still being made and nothing plays.
+    const streamStopped = command === "stop" && application.stopStream?.()?.state === "stopped";
+    const acted = command === "stop" ? previewPlayer.stop() || streamStopped : previewPlayer.skip();
+    const waiting = !acted && command === "skip" && application.streamView?.()?.phase === "waiting";
+    // Off air shows even during a reply: a stream stopped while its first track is made has no row to lose.
+    if (!busy || streamStopped) {
+      setFooter(waiting ? tr("The next record is still being cut.") : !acted ? tr("No previews are playing.")
+        : streamStopped ? tr("Off air.") : command === "stop" ? tr("Previews stopped.") : tr("Skipped."));
+    }
     tui.requestRender();
   };
+  // /onair goes off air at once while a stream runs, even while Moondog is answering, and says so
+  // when music making is off. Otherwise it returns the request to send, in the reply's language.
+  const onAirRequest = () => {
+    if (application.streamView?.()) {
+      application.stopStream();
+      // Said even during a reply: a stream stopped while its first track is made has no line to clear.
+      setFooter(tr("Off air."));
+      return null;
+    }
+    // Music making can be turned off, here or in the window, after the room opened.
+    if (!application.streamReady?.()) {
+      setFooter(tr("Turn on music making first: /generation on."), warning);
+      return null;
+    }
+    return translator(application.locale)("onAir.request");
+  };
+
+  // Making music on this Mac. /generation says where it stands; /generation on [folder] turns it on,
+  // with the folder of Stability AI's MLX build when it is not the usual one; /generation off turns it off.
+  const home = homedir();
+  const shownFolder = (folder) => sanitizeTerminalText(String(folder ?? "")).replace(home, "~");
+  const generationMessage = (status) => {
+    const folder = shownFolder(status?.path);
+    if (status?.install === "unsupported") return tr("Stable Audio 3 needs a Mac with Apple silicon, so Moondog can't make music here.");
+    if (status?.state === "ready") return tr("Moondog makes music on this Mac with Stable Audio 3, powered by Stability AI. Ask for a track. Stable Audio 3 runs offline, and the track stays on this Mac.");
+    if (status?.state === "off") {
+      if (status.install === "ready") return tr("Making music on this Mac is off. Stable Audio 3 is ready at {folder}; turn it on with /generation on.", { folder });
+      if (status.install === "incomplete") return tr("Making music on this Mac is off, and Stable Audio 3 at {folder} isn't fully installed. Run ./install.sh --download medium there, then use /generation on.", { folder });
+      return tr("Making music on this Mac is off, and Stable Audio 3 isn't at {folder}. Install Stability AI's MLX build from github.com/Stability-AI/stable-audio-3, then use /generation on with its folder.", { folder });
+    }
+    if (status?.state === "incomplete") {
+      return status.missing === "weights"
+        ? tr("Stable Audio 3 at {folder} doesn't have the Medium model yet. Run ./install.sh --download medium there, then use /generation again.", { folder })
+        : tr("Stable Audio 3 at {folder} isn't installed yet. Run ./install.sh --download medium there, then use /generation again.", { folder });
+    }
+    return tr("Stable Audio 3 isn't at {folder}. Use /generation on with the folder that holds its sa3 script.", { folder });
+  };
+  const manageGeneration = async (args) => {
+    const [action, ...rest] = args;
+    if ((action && !["on", "off"].includes(action)) || (action === "off" && rest.length)) {
+      throw new Error(tr("Use /generation, /generation on [folder], or /generation off."));
+    }
+    const status = action ? await generation.save({ enabled: action === "on", ...(rest.length ? { path: rest.join(" ") } : {}) })
+      : await generation.check();
+    addMoondogMessage(generationMessage(status));
+    setFooter(status.state === "ready" ? tr("Music making on.") : status.state === "off" ? tr("Music making off.") : tr("Music making isn't ready."),
+      status.state === "ready" || status.state === "off" ? dim : warning);
+  };
 
   const removeLifecycleListeners = () => {
     signalTarget.removeListener("SIGTERM", handleSigterm);
@@ -732,6 +830,7 @@ export async function runMoondogTui({
     stopWatchingModel?.();
     stopWatchingFirstRun?.();
     stopWatchingPreviews?.();
+    stopWatchingStream?.();
     clearInterval(indicator);
     indicator = null;
     clearInterval(homeMotion);
@@ -760,6 +859,8 @@ export async function runMoondogTui({
   };
 
   const shutdown = () => {
+    // A reply still running stops with the room, and so does a track it is making.
+    try { activeRuntime.abort(); } catch { /* Nothing was running. */ }
     cleanup();
     // The spoken line that closes Eclipse, the last thing on The Dark Side of the Moon.
     try {
@@ -1804,10 +1905,11 @@ export async function runMoondogTui({
       ? tr("All set. /spotify help shows how to connect Spotify.")
       : tr("All set. Say anything, or press Tab to look around."), success);
   };
-  // The view closes at once; the saved skip then reaches the window too.
+  // The view closes at once; the saved skip then reaches the window too, and a room closed right
+  // after Esc still saves it.
   const skipFirstRun = () => {
     closeFirstRun(tr("Skipped. /service and /model are here when you want them."));
-    void firstRun.finish({ skipped: true }).catch(() => {});
+    firstRunSaving = firstRun.finish({ skipped: true }).catch(() => {});
   };
   const handleFirstRunAction = async ({ type, value } = {}) => {
     if (busy || cleanedUp || !firstRunView) return;
@@ -2144,6 +2246,10 @@ export async function runMoondogTui({
       await pickService(args);
       return;
     }
+    if (command === "generation") {
+      await manageGeneration(args);
+      return;
+    }
     if (command === "spotify") {
       await runSpotifyCommand(args);
       return;
@@ -2303,6 +2409,12 @@ export async function runMoondogTui({
         await openCommandPalette();
         return;
       }
+      // Going on air is a request like any other, which the agent answers by starting the stream.
+      if (command === "onair") {
+        const request = onAirRequest();
+        if (request) await submitListeningTurn(request);
+        return;
+      }
       try {
         if (!["home", "theme", "art", "motion", "model", "service", "resume", "rewind", "branch", "new", "taste", "profile", "import"].includes(command)) enterConversation();
         if (command !== "auth") editor.addToHistory(value);
@@ -2312,7 +2424,7 @@ export async function runMoondogTui({
         setFooter(`/${command}...`);
         const argumentText = value.slice(1).trim().slice(rawCommand.length).trim();
         const parsedArgs = command === "import" ? argumentText ? [argumentText] : []
-          : ["profile", "spotify", "web"].includes(command)
+          : ["profile", "spotify", "web", "generation"].includes(command)
           ? parseTuiArguments(argumentText)
           : args;
         await runLocalCommand(command, parsedArgs);
@@ -2408,6 +2520,12 @@ export async function runMoondogTui({
           if (!cancellationRequested) setFooter(tr("Can't reach the model. Trying again ({attempt}/{max})...", { attempt, max: maxRetries }), warning);
         },
         onToolStart: (tool) => updateTool(tool, "running"),
+        // Making a track says where it is, from warming up to saving it.
+        onToolUpdate: (tool) => {
+          const stage = generationStages[tool?.stage];
+          if (cleanedUp || cancellationRequested || !stage) return;
+          setFooter(`${sanitizeTerminalText(tr.marked(tool.label ?? "Music tool"))} · ${tr.marked(stage)}...`);
+        },
         onToolEnd: (tool) => updateTool(tool, tool?.isError ? "failed" : "completed"),
         // The model names a new conversation after its first reply; the header shows that name from then on.
         onConversationNamed: ({ session_id, title }) => {
@@ -2484,9 +2602,11 @@ export async function runMoondogTui({
         controlPreviews(pending.slice(1));
         return;
       }
-      if (pending) {
-        if (messageQueue.length >= 30) editor.setText(pending);
-        else messageQueue.push(pending);
+      // /onair stops a stream at once; the request to go on air waits in the queue like any message.
+      const queued = onAirControls && pending === "/onair" ? onAirRequest() : pending;
+      if (queued) {
+        if (messageQueue.length >= 30) editor.setText(queued);
+        else messageQueue.push(queued);
         if (rewindPending) queuePaused = true;
       }
       if (!cleanedUp) tui.requestRender();
@@ -2739,6 +2859,7 @@ export async function runMoondogTui({
   } finally {
     cleanup();
     await lyricTask;
+    await firstRunSaving;
   }
   return exitRequest;
 }