codemaxxing 0.3.1 → 0.4.1

package/src/index.tsx

@@ -5,7 +5,7 @@ import { render, Box, Text, useInput, useApp, useStdout } from "ink";
 import { EventEmitter } from "events";
 import TextInput from "ink-text-input";
 import { CodingAgent } from "./agent.js";
-import { loadConfig, detectLocalProvider, parseCLIArgs, applyOverrides, listModels } from "./config.js";
+import { loadConfig, saveConfig, detectLocalProvider, parseCLIArgs, applyOverrides, listModels } from "./config.js";
 import { listSessions, getSession, loadMessages, deleteSession } from "./utils/sessions.js";
 import { execSync } from "child_process";
 import { isGitRepo, getBranch, getStatus, getDiff, undoLastCommit } from "./utils/git.js";
@@ -13,6 +13,9 @@ import { getTheme, listThemes, THEMES, DEFAULT_THEME, type Theme } from "./theme
 import { PROVIDERS, getCredentials, openRouterOAuth, anthropicSetupToken, importCodexToken, importQwenToken, copilotDeviceFlow, saveApiKey } from "./utils/auth.js";
 import { listInstalledSkills, installSkill, removeSkill, getRegistrySkills, searchRegistry, createSkillScaffold, getActiveSkills, getActiveSkillCount } from "./utils/skills.js";
 import { listServers, addServer, removeServer, getAllMCPTools, getConnectedServers } from "./utils/mcp.js";
+import { detectHardware, formatBytes, type HardwareInfo } from "./utils/hardware.js";
+import { getRecommendations, getRecommendationsWithLlmfit, getFitIcon, isLlmfitAvailable, type ScoredModel } from "./utils/models.js";
+import { isOllamaInstalled, isOllamaRunning, getOllamaInstallCommand, startOllama, stopOllama, pullModel, listInstalledModelsDetailed, deleteModel, getGPUMemoryUsage, type PullProgress } from "./utils/ollama.js";
 
 const VERSION = "0.1.9";
 
@@ -64,6 +67,12 @@ const SLASH_COMMANDS = [
   { cmd: "/mcp add", desc: "add MCP server" },
   { cmd: "/mcp remove", desc: "remove MCP server" },
   { cmd: "/mcp reconnect", desc: "reconnect MCP servers" },
+  { cmd: "/ollama", desc: "Ollama status & models" },
+  { cmd: "/ollama list", desc: "list installed models" },
+  { cmd: "/ollama start", desc: "start Ollama server" },
+  { cmd: "/ollama stop", desc: "stop Ollama server" },
+  { cmd: "/ollama pull", desc: "download a model" },
+  { cmd: "/ollama delete", desc: "delete a model" },
   { cmd: "/quit", desc: "exit" },
 ];
 
@@ -172,6 +181,21 @@ function App() {
     resolve: (decision: "yes" | "no" | "always") => void;
   } | null>(null);
 
+  // ── Ollama Management State ──
+  const [ollamaDeleteConfirm, setOllamaDeleteConfirm] = useState<{ model: string; size: number } | null>(null);
+  const [ollamaPulling, setOllamaPulling] = useState<{ model: string; progress: PullProgress } | null>(null);
+  const [ollamaExitPrompt, setOllamaExitPrompt] = useState(false);
+
+  // ── Setup Wizard State ──
+  type WizardScreen = "connection" | "models" | "install-ollama" | "pulling" | null;
+  const [wizardScreen, setWizardScreen] = useState<WizardScreen>(null);
+  const [wizardIndex, setWizardIndex] = useState(0);
+  const [wizardHardware, setWizardHardware] = useState<HardwareInfo | null>(null);
+  const [wizardModels, setWizardModels] = useState<ScoredModel[]>([]);
+  const [wizardPullProgress, setWizardPullProgress] = useState<PullProgress | null>(null);
+  const [wizardPullError, setWizardPullError] = useState<string | null>(null);
+  const [wizardSelectedModel, setWizardSelectedModel] = useState<ScoredModel | null>(null);
+
   // Listen for paste events from stdin interceptor
   useEffect(() => {
     const handler = ({ content, lines }: { content: string; lines: number }) => {
@@ -210,10 +234,11 @@ function App() {
         setConnectionInfo([...info]);
       } else {
         info.push("✗ No local LLM server found.");
-        info.push("  /connect  — retry after starting LM Studio or Ollama");
-        info.push("  /login    — authenticate with a cloud provider");
         setConnectionInfo([...info]);
         setReady(true);
+        // Show the setup wizard on first run
+        setWizardScreen("connection");
+        setWizardIndex(0);
         return;
       }
     } else {
@@ -369,7 +394,8 @@ function App() {
         // Commands that need args (like /commit, /model) — fill input instead of executing
         if (selected.cmd === "/commit" || selected.cmd === "/model" || selected.cmd === "/session delete" ||
             selected.cmd === "/skills install" || selected.cmd === "/skills remove" || selected.cmd === "/skills search" ||
-            selected.cmd === "/skills on" || selected.cmd === "/skills off" || selected.cmd === "/architect") {
+            selected.cmd === "/skills on" || selected.cmd === "/skills off" || selected.cmd === "/architect" ||
+            selected.cmd === "/ollama pull" || selected.cmd === "/ollama delete") {
           setInput(selected.cmd + " ");
           setCmdIndex(0);
           setInputKey((k) => k + 1);
@@ -397,7 +423,20 @@ function App() {
     addMsg("user", trimmed);
 
     if (trimmed === "/quit" || trimmed === "/exit") {
-      exit();
+      // Check if Ollama is running and offer to stop it
+      const ollamaUp = await isOllamaRunning();
+      if (ollamaUp) {
+        const config = loadConfig();
+        if (config.defaults.stopOllamaOnExit) {
+          addMsg("info", "Stopping Ollama...");
+          await stopOllama();
+          exit();
+        } else {
+          setOllamaExitPrompt(true);
+        }
+      } else {
+        exit();
+      }
       return;
     }
     if (trimmed === "/login" || trimmed === "/auth") {
@@ -440,6 +479,12 @@ function App() {
         "  /mcp add   — add MCP server to global config",
         "  /mcp remove — remove MCP server",
         "  /mcp reconnect — reconnect all MCP servers",
+        "  /ollama    — Ollama status, models & GPU usage",
+        "  /ollama list — list installed models with sizes",
+        "  /ollama start — start Ollama server",
+        "  /ollama stop — stop Ollama server (frees GPU RAM)",
+        "  /ollama pull <model> — download a model",
+        "  /ollama delete <model> — delete a model from disk",
         "  /quit      — exit",
       ].join("\n"));
       return;
@@ -600,6 +645,135 @@ function App() {
       return;
     }
 
+    // ── Ollama commands (work without agent) ──
+    if (trimmed === "/ollama" || trimmed === "/ollama status") {
+      const running = await isOllamaRunning();
+      const lines: string[] = [`Ollama: ${running ? "running" : "stopped"}`];
+      if (running) {
+        const models = await listInstalledModelsDetailed();
+        if (models.length > 0) {
+          lines.push(`Installed models (${models.length}):`);
+          for (const m of models) {
+            const sizeGB = (m.size / (1024 * 1024 * 1024)).toFixed(1);
+            lines.push(`  ${m.name}  (${sizeGB} GB)`);
+          }
+        } else {
+          lines.push("No models installed.");
+        }
+        const gpuMem = getGPUMemoryUsage();
+        if (gpuMem) lines.push(`GPU: ${gpuMem}`);
+      } else {
+        lines.push("Start with: /ollama start");
+      }
+      addMsg("info", lines.join("\n"));
+      return;
+    }
+    if (trimmed === "/ollama list") {
+      const running = await isOllamaRunning();
+      if (!running) {
+        addMsg("info", "Ollama is not running. Start with /ollama start");
+        return;
+      }
+      const models = await listInstalledModelsDetailed();
+      if (models.length === 0) {
+        addMsg("info", "No models installed. Pull one with /ollama pull <model>");
+      } else {
+        const lines = models.map((m) => {
+          const sizeGB = (m.size / (1024 * 1024 * 1024)).toFixed(1);
+          return `  ${m.name}  (${sizeGB} GB)`;
+        });
+        addMsg("info", `Installed models:\n${lines.join("\n")}`);
+      }
+      return;
+    }
+    if (trimmed === "/ollama start") {
+      const running = await isOllamaRunning();
+      if (running) {
+        addMsg("info", "Ollama is already running.");
+        return;
+      }
+      if (!isOllamaInstalled()) {
+        addMsg("error", `Ollama is not installed. Install with: ${getOllamaInstallCommand(process.platform === "darwin" ? "macos" : process.platform === "win32" ? "windows" : "linux")}`);
+        return;
+      }
+      startOllama();
+      addMsg("info", "Starting Ollama server...");
+      // Wait for it to come up
+      for (let i = 0; i < 10; i++) {
+        await new Promise(r => setTimeout(r, 1000));
+        if (await isOllamaRunning()) {
+          addMsg("info", "Ollama is running.");
+          return;
+        }
+      }
+      addMsg("error", "Ollama did not start in time. Try running 'ollama serve' manually.");
+      return;
+    }
+    if (trimmed === "/ollama stop") {
+      const running = await isOllamaRunning();
+      if (!running) {
+        addMsg("info", "Ollama is not running.");
+        return;
+      }
+      addMsg("info", "Stopping Ollama...");
+      const result = await stopOllama();
+      addMsg(result.ok ? "info" : "error", result.ok ? `\u2705 ${result.message}` : `\u274C ${result.message}`);
+      return;
+    }
+    if (trimmed.startsWith("/ollama pull ")) {
+      const modelId = trimmed.replace("/ollama pull ", "").trim();
+      if (!modelId) {
+        addMsg("info", "Usage: /ollama pull <model>\n  Example: /ollama pull qwen2.5-coder:7b");
+        return;
+      }
+      if (!isOllamaInstalled()) {
+        addMsg("error", "Ollama is not installed.");
+        return;
+      }
+      // Ensure ollama is running
+      let running = await isOllamaRunning();
+      if (!running) {
+        startOllama();
+        addMsg("info", "Starting Ollama server...");
+        for (let i = 0; i < 10; i++) {
+          await new Promise(r => setTimeout(r, 1000));
+          if (await isOllamaRunning()) { running = true; break; }
+        }
+        if (!running) {
+          addMsg("error", "Could not start Ollama. Run 'ollama serve' manually.");
+          return;
+        }
+      }
+      setOllamaPulling({ model: modelId, progress: { status: "starting", percent: 0 } });
+      try {
+        await pullModel(modelId, (p) => {
+          setOllamaPulling({ model: modelId, progress: p });
+        });
+        setOllamaPulling(null);
+        addMsg("info", `\u2705 Downloaded ${modelId}`);
+      } catch (err: any) {
+        setOllamaPulling(null);
+        addMsg("error", `Failed to pull ${modelId}: ${err.message}`);
+      }
+      return;
+    }
+    if (trimmed.startsWith("/ollama delete ")) {
+      const modelId = trimmed.replace("/ollama delete ", "").trim();
+      if (!modelId) {
+        addMsg("info", "Usage: /ollama delete <model>");
+        return;
+      }
+      // Look up size for confirmation
+      const models = await listInstalledModelsDetailed();
+      const found = models.find((m) => m.name === modelId || m.name.startsWith(modelId));
+      if (!found) {
+        addMsg("error", `Model "${modelId}" not found. Use /ollama list to see installed models.`);
+        return;
+      }
+      setOllamaDeleteConfirm({ model: found.name, size: found.size });
+      return;
+    }
+
     // ── MCP commands (partially work without agent) ──
     if (trimmed === "/mcp" || trimmed === "/mcp list") {
       const servers = listServers(process.cwd());
@@ -1141,6 +1315,265 @@ function App() {
       return;
     }
 
+    // ── Ollama delete confirmation ──
+    if (ollamaDeleteConfirm) {
+      if (inputChar === "y" || inputChar === "Y") {
+        const model = ollamaDeleteConfirm.model;
+        setOllamaDeleteConfirm(null);
+        const result = deleteModel(model);
+        addMsg(result.ok ? "info" : "error", result.ok ? `\u2705 ${result.message}` : `\u274C ${result.message}`);
+        return;
+      }
+      if (inputChar === "n" || inputChar === "N" || key.escape) {
+        setOllamaDeleteConfirm(null);
+        addMsg("info", "Delete cancelled.");
+        return;
+      }
+      return;
+    }
+
+    // ── Ollama exit prompt ──
+    if (ollamaExitPrompt) {
+      if (inputChar === "y" || inputChar === "Y") {
+        setOllamaExitPrompt(false);
+        stopOllama().then(() => exit());
+        return;
+      }
+      if (inputChar === "n" || inputChar === "N") {
+        setOllamaExitPrompt(false);
+        exit();
+        return;
+      }
+      if (inputChar === "a" || inputChar === "A") {
+        setOllamaExitPrompt(false);
+        saveConfig({ defaults: { ...loadConfig().defaults, stopOllamaOnExit: true } });
+        addMsg("info", "Saved preference: always stop Ollama on exit.");
+        stopOllama().then(() => exit());
+        return;
+      }
+      if (key.escape) {
+        setOllamaExitPrompt(false);
+        return;
+      }
+      return;
+    }
+
+    // ── Setup Wizard Navigation ──
+    if (wizardScreen) {
+      if (wizardScreen === "connection") {
+        const items = ["local", "openrouter", "apikey", "existing"];
+        if (key.upArrow) {
+          setWizardIndex((prev) => (prev - 1 + items.length) % items.length);
+          return;
+        }
+        if (key.downArrow) {
+          setWizardIndex((prev) => (prev + 1) % items.length);
+          return;
+        }
+        if (key.escape) {
+          setWizardScreen(null);
+          return;
+        }
+        if (key.return) {
+          const selected = items[wizardIndex];
+          if (selected === "local") {
+            // Scan hardware and show model picker (use llmfit if available)
+            const hw = detectHardware();
+            setWizardHardware(hw);
+            const { models: recs } = getRecommendationsWithLlmfit(hw);
+            setWizardModels(recs.filter(m => m.fit !== "skip"));
+            setWizardScreen("models");
+            setWizardIndex(0);
+          } else if (selected === "openrouter") {
+            setWizardScreen(null);
+            addMsg("info", "Starting OpenRouter OAuth — opening browser...");
+            setLoading(true);
+            setSpinnerMsg("Waiting for authorization...");
+            openRouterOAuth((msg: string) => addMsg("info", msg))
+              .then(() => {
+                addMsg("info", "✅ OpenRouter authenticated! Use /connect to connect.");
+                setLoading(false);
+              })
+              .catch((err: any) => { addMsg("error", `OAuth failed: ${err.message}`); setLoading(false); });
+          } else if (selected === "apikey") {
+            setWizardScreen(null);
+            setLoginPicker(true);
+            setLoginPickerIndex(0);
+          } else if (selected === "existing") {
+            setWizardScreen(null);
+            addMsg("info", "Start your LLM server, then type /connect to retry.");
+          }
+          return;
+        }
+        return;
+      }
+
+      if (wizardScreen === "models") {
+        const models = wizardModels;
+        if (key.upArrow) {
+          setWizardIndex((prev) => (prev - 1 + models.length) % models.length);
+          return;
+        }
+        if (key.downArrow) {
+          setWizardIndex((prev) => (prev + 1) % models.length);
+          return;
+        }
+        if (key.escape) {
+          setWizardScreen("connection");
+          setWizardIndex(0);
+          return;
+        }
+        if (key.return) {
+          const selected = models[wizardIndex];
+          if (selected) {
+            setWizardSelectedModel(selected);
+            // Check if Ollama is installed
+            if (!isOllamaInstalled()) {
+              setWizardScreen("install-ollama");
+            } else {
+              // Start pulling the model
+              setWizardScreen("pulling");
+              setWizardPullProgress({ status: "starting", percent: 0 });
+              setWizardPullError(null);
+
+              (async () => {
+                try {
+                  // Ensure ollama is running
+                  const running = await isOllamaRunning();
+                  if (!running) {
+                    setWizardPullProgress({ status: "Starting Ollama server...", percent: 0 });
+                    startOllama();
+                    // Wait for it to come up
+                    for (let i = 0; i < 15; i++) {
+                      await new Promise(r => setTimeout(r, 1000));
+                      if (await isOllamaRunning()) break;
+                    }
+                    if (!(await isOllamaRunning())) {
+                      setWizardPullError("Could not start Ollama server. Run 'ollama serve' manually, then press Enter.");
+                      return;
+                    }
+                  }
+
+                  await pullModel(selected.ollamaId, (p) => {
+                    setWizardPullProgress(p);
+                  });
+
+                  setWizardPullProgress({ status: "success", percent: 100 });
+
+                  // Wait briefly then connect
+                  await new Promise(r => setTimeout(r, 500));
+                  setWizardScreen(null);
+                  setWizardPullProgress(null);
+                  setWizardSelectedModel(null);
+                  addMsg("info", `✅ ${selected.name} installed! Connecting...`);
+                  await connectToProvider(true);
+                } catch (err: any) {
+                  setWizardPullError(err.message);
+                }
+              })();
+            }
+          }
+          return;
+        }
+        return;
+      }
+
+      if (wizardScreen === "install-ollama") {
+        if (key.escape) {
+          setWizardScreen("models");
+          setWizardIndex(0);
+          return;
+        }
+        if (key.return) {
+          // User says they installed it — check and proceed
+          if (isOllamaInstalled()) {
+            const selected = wizardSelectedModel;
+            if (selected) {
+              setWizardScreen("pulling");
+              setWizardPullProgress({ status: "starting", percent: 0 });
+              setWizardPullError(null);
+
+              (async () => {
+                try {
+                  const running = await isOllamaRunning();
+                  if (!running) {
+                    setWizardPullProgress({ status: "Starting Ollama server...", percent: 0 });
+                    startOllama();
+                    for (let i = 0; i < 15; i++) {
+                      await new Promise(r => setTimeout(r, 1000));
+                      if (await isOllamaRunning()) break;
+                    }
+                    if (!(await isOllamaRunning())) {
+                      setWizardPullError("Could not start Ollama server. Run 'ollama serve' manually, then press Enter.");
+                      return;
+                    }
+                  }
+                  await pullModel(selected.ollamaId, (p) => setWizardPullProgress(p));
+                  setWizardPullProgress({ status: "success", percent: 100 });
+                  await new Promise(r => setTimeout(r, 500));
+                  setWizardScreen(null);
+                  setWizardPullProgress(null);
+                  setWizardSelectedModel(null);
+                  addMsg("info", `✅ ${selected.name} installed! Connecting...`);
+                  await connectToProvider(true);
+                } catch (err: any) {
+                  setWizardPullError(err.message);
+                }
+              })();
+            }
+          } else {
+            addMsg("info", "Ollama not found yet. Install it and press Enter again.");
+          }
+          return;
+        }
+        return;
+      }
+
+      if (wizardScreen === "pulling") {
+        // Allow retry on error
+        if (wizardPullError && key.return) {
+          const selected = wizardSelectedModel;
+          if (selected) {
+            setWizardPullError(null);
+            setWizardPullProgress({ status: "retrying", percent: 0 });
+            (async () => {
+              try {
+                const running = await isOllamaRunning();
+                if (!running) {
+                  startOllama();
+                  for (let i = 0; i < 15; i++) {
+                    await new Promise(r => setTimeout(r, 1000));
+                    if (await isOllamaRunning()) break;
+                  }
+                }
+                await pullModel(selected.ollamaId, (p) => setWizardPullProgress(p));
+                setWizardPullProgress({ status: "success", percent: 100 });
+                await new Promise(r => setTimeout(r, 500));
+                setWizardScreen(null);
+                setWizardPullProgress(null);
+                setWizardSelectedModel(null);
+                addMsg("info", `✅ ${selected.name} installed! Connecting...`);
+                await connectToProvider(true);
+              } catch (err: any) {
+                setWizardPullError(err.message);
+              }
+            })();
+          }
+          return;
+        }
+        if (wizardPullError && key.escape) {
+          setWizardScreen("models");
+          setWizardIndex(0);
+          setWizardPullError(null);
+          setWizardPullProgress(null);
+          return;
+        }
+        return; // Ignore keys while pulling
+      }
+
+      return;
+    }
+
     // Theme picker navigation
     if (themePicker) {
       const themeKeys = listThemes();
@@ -1297,7 +1730,13 @@ function App() {
 
     if (key.ctrl && inputChar === "c") {
       if (ctrlCPressed) {
-        exit();
+        // Force quit on second Ctrl+C — don't block
+        const config = loadConfig();
+        if (config.defaults.stopOllamaOnExit) {
+          stopOllama().finally(() => exit());
+        } else {
+          exit();
+        }
       } else {
         setCtrlCPressed(true);
         addMsg("info", "Press Ctrl+C again to exit.");
@@ -1602,6 +2041,150 @@ function App() {
         </Box>
       )}
 
+      {/* ═══ OLLAMA DELETE CONFIRM ═══ */}
+      {ollamaDeleteConfirm && (
+        <Box flexDirection="column" borderStyle="single" borderColor={theme.colors.warning} paddingX={1} marginBottom={0}>
+          <Text bold color={theme.colors.warning}>Delete {ollamaDeleteConfirm.model} ({(ollamaDeleteConfirm.size / (1024 * 1024 * 1024)).toFixed(1)} GB)?</Text>
+          <Text>
+            <Text color={theme.colors.error} bold> [y]</Text><Text>es  </Text>
+            <Text color={theme.colors.success} bold>[n]</Text><Text>o</Text>
+          </Text>
+        </Box>
+      )}
+
+      {/* ═══ OLLAMA PULL PROGRESS ═══ */}
+      {ollamaPulling && (
+        <Box flexDirection="column" borderStyle="single" borderColor={theme.colors.border} paddingX={1} marginBottom={0}>
+          <Text bold color={theme.colors.secondary}>{"  Downloading "}{ollamaPulling.model}{"..."}</Text>
+          {ollamaPulling.progress.status === "downloading" || ollamaPulling.progress.percent > 0 ? (
+            <Text>
+              {"  "}
+              <Text color={theme.colors.primary}>
+                {"\u2588".repeat(Math.floor(ollamaPulling.progress.percent / 5))}
+                {"\u2591".repeat(20 - Math.floor(ollamaPulling.progress.percent / 5))}
+              </Text>
+              {"  "}<Text bold>{ollamaPulling.progress.percent}%</Text>
+              {ollamaPulling.progress.completed != null && ollamaPulling.progress.total != null ? (
+                <Text color={theme.colors.muted}>{" \u00B7 "}{formatBytes(ollamaPulling.progress.completed)}{" / "}{formatBytes(ollamaPulling.progress.total)}</Text>
+              ) : null}
+            </Text>
+          ) : (
+            <Text color={theme.colors.muted}>{"  "}{ollamaPulling.progress.status}...</Text>
+          )}
+        </Box>
+      )}
+
+      {/* ═══ OLLAMA EXIT PROMPT ═══ */}
+      {ollamaExitPrompt && (
+        <Box flexDirection="column" borderStyle="single" borderColor={theme.colors.warning} paddingX={1} marginBottom={0}>
+          <Text bold color={theme.colors.warning}>Ollama is still running.</Text>
+          <Text color={theme.colors.muted}>Stop it to free GPU memory?</Text>
+          <Text>
+            <Text color={theme.colors.success} bold> [y]</Text><Text>es  </Text>
+            <Text color={theme.colors.error} bold>[n]</Text><Text>o  </Text>
+            <Text color={theme.colors.primary} bold>[a]</Text><Text>lways</Text>
+          </Text>
+        </Box>
+      )}
+
+      {/* ═══ SETUP WIZARD ═══ */}
+      {wizardScreen === "connection" && (
+        <Box flexDirection="column" borderStyle="single" borderColor={theme.colors.border} paddingX={1} marginBottom={0}>
+          <Text bold color={theme.colors.secondary}>No LLM detected. How do you want to connect?</Text>
+          <Text>{""}</Text>
+          {[
+            { key: "local",      icon: "\uD83D\uDDA5\uFE0F",  label: "Set up a local model", desc: "free, runs on your machine" },
+            { key: "openrouter", icon: "\uD83C\uDF10", label: "OpenRouter", desc: "200+ cloud models, browser login" },
+            { key: "apikey",     icon: "\uD83D\uDD11", label: "Enter API key manually", desc: "" },
+            { key: "existing",   icon: "\u2699\uFE0F",  label: "I already have a server running", desc: "" },
+          ].map((item, i) => (
+            <Text key={item.key}>
+              {i === wizardIndex ? <Text color={theme.colors.suggestion} bold>{"  \u25B8 "}</Text> : <Text>{"    "}</Text>}
+              <Text color={i === wizardIndex ? theme.colors.suggestion : theme.colors.primary} bold>{item.icon}  {item.label}</Text>
+              {item.desc ? <Text color={theme.colors.muted}>{" ("}{item.desc}{")"}</Text> : null}
+            </Text>
+          ))}
+          <Text>{""}</Text>
+          <Text dimColor>{"  \u2191\u2193 navigate \u00B7 Enter to select"}</Text>
+        </Box>
+      )}
+
+      {wizardScreen === "models" && wizardHardware && (
+        <Box flexDirection="column" borderStyle="single" borderColor={theme.colors.border} paddingX={1} marginBottom={0}>
+          <Text bold color={theme.colors.secondary}>Your hardware:</Text>
+          <Text color={theme.colors.muted}>{"  CPU: "}{wizardHardware.cpu.name}{" ("}{wizardHardware.cpu.cores}{" cores)"}</Text>
+          <Text color={theme.colors.muted}>{"  RAM: "}{formatBytes(wizardHardware.ram)}</Text>
+          {wizardHardware.gpu ? (
+            <Text color={theme.colors.muted}>{"  GPU: "}{wizardHardware.gpu.name}{wizardHardware.gpu.vram > 0 ? ` (${formatBytes(wizardHardware.gpu.vram)})` : ""}</Text>
+          ) : (
+            <Text color={theme.colors.muted}>{"  GPU: none detected"}</Text>
+          )}
+          {!isLlmfitAvailable() && (
+            <Text dimColor>{"  Tip: Install llmfit for smarter recommendations: brew install llmfit"}</Text>
+          )}
+          <Text>{""}</Text>
+          <Text bold color={theme.colors.secondary}>Recommended models:</Text>
+          <Text>{""}</Text>
+          {wizardModels.map((m, i) => (
+            <Text key={m.ollamaId}>
+              {i === wizardIndex ? <Text color={theme.colors.suggestion} bold>{"  \u25B8 "}</Text> : <Text>{"    "}</Text>}
+              <Text>{getFitIcon(m.fit)} </Text>
+              <Text color={i === wizardIndex ? theme.colors.suggestion : theme.colors.primary} bold>{m.name}</Text>
+              <Text color={theme.colors.muted}>{"     ~"}{m.size}{" GB \u00B7 "}{m.quality === "best" ? "Best" : m.quality === "great" ? "Great" : "Good"}{" quality \u00B7 "}{m.speed}</Text>
+            </Text>
+          ))}
+          {wizardModels.length === 0 && (
+            <Text color={theme.colors.error}>{"  No suitable models found for your hardware."}</Text>
+          )}
+          <Text>{""}</Text>
+          <Text dimColor>{"  \u2191\u2193 navigate \u00B7 Enter to install \u00B7 Esc back"}</Text>
+        </Box>
+      )}
+
+      {wizardScreen === "install-ollama" && (
+        <Box flexDirection="column" borderStyle="single" borderColor={theme.colors.warning} paddingX={1} marginBottom={0}>
+          <Text bold color={theme.colors.warning}>Ollama is required for local models.</Text>
+          <Text>{""}</Text>
+          <Text color={theme.colors.primary}>{"  Install with: "}<Text bold>{getOllamaInstallCommand(wizardHardware?.os ?? "linux")}</Text></Text>
+          <Text>{""}</Text>
+          <Text dimColor>{"  Run the command above, then press Enter to continue..."}</Text>
+          <Text dimColor>{"  Esc to go back"}</Text>
+        </Box>
+      )}
+
+      {wizardScreen === "pulling" && wizardSelectedModel && (
+        <Box flexDirection="column" borderStyle="single" borderColor={theme.colors.border} paddingX={1} marginBottom={0}>
+          {wizardPullError ? (
+            <>
+              <Text color={theme.colors.error} bold>{"  \u274C Error: "}{wizardPullError}</Text>
+              <Text>{""}</Text>
+              <Text dimColor>{"  Press Enter to retry \u00B7 Esc to go back"}</Text>
+            </>
+          ) : wizardPullProgress ? (
+            <>
+              <Text bold color={theme.colors.secondary}>{"  Downloading "}{wizardSelectedModel.name}{"..."}</Text>
+              {wizardPullProgress.status === "downloading" || wizardPullProgress.percent > 0 ? (
+                <>
+                  <Text>
+                    {"  "}
+                    <Text color={theme.colors.primary}>
+                      {"\u2588".repeat(Math.floor(wizardPullProgress.percent / 5))}
+                      {"\u2591".repeat(20 - Math.floor(wizardPullProgress.percent / 5))}
+                    </Text>
+                    {"  "}<Text bold>{wizardPullProgress.percent}%</Text>
+                    {wizardPullProgress.completed != null && wizardPullProgress.total != null ? (
+                      <Text color={theme.colors.muted}>{" \u00B7 "}{formatBytes(wizardPullProgress.completed)}{" / "}{formatBytes(wizardPullProgress.total)}</Text>
+                    ) : null}
+                  </Text>
+                </>
+              ) : (
+                <Text color={theme.colors.muted}>{"  "}{wizardPullProgress.status}...</Text>
+              )}
+            </>
+          ) : null}
+        </Box>
+      )}
+
       {/* ═══ COMMAND SUGGESTIONS ═══ */}
       {showSuggestions && (
         <Box flexDirection="column" borderStyle="single" borderColor={theme.colors.muted} paddingX={1} marginBottom={0}>