brains-cli 0.1.0 → 1.0.2

package/src/commands/run.js

@@ -5,8 +5,17 @@ const chalk = require('chalk');
 const ora = require('ora');
 const path = require('path');
 const fs = require('fs');
+const readline = require('readline');
 const { getBrainById } = require('../registry');
-const { showSuccess, showError, showInfo, showBox, ACCENT, DIM, SUCCESS, WARN } = require('../utils/ui');
+const {
+  getActiveProvider, getProviderApiKey, getProviderModel,
+  setProviderApiKey, setActiveProvider,
+  getModel, saveSession,
+} = require('../config');
+const { getProviderDef, getAllProviderNames } = require('../providers/registry');
+const { streamMessage } = require('../api');
+const { readFilesForReview, parseFileBlocks, writeFileBlocks, generateTreeView } = require('../utils/files');
+const { showSuccess, showError, showInfo, showWarning, ACCENT, DIM, SUCCESS, WARN } = require('../utils/ui');
 
 const BRAINS_DIR = path.join(require('os').homedir(), '.brains');
 const INSTALLED_FILE = path.join(BRAINS_DIR, 'installed.json');
@@ -28,6 +37,138 @@ function loadBrainConfig(brainId) {
   return null;
 }
 
+/**
+ * Ensure API key is available for the active provider.
+ * Prompts user if missing. Offers provider switching if no key.
+ */
+async function ensureApiKey() {
+  const providerName = getActiveProvider();
+  const providerDef = getProviderDef(providerName);
+
+  // Ollama doesn't need a key
+  if (providerDef && providerDef.requires_key === false) {
+    return 'not-required';
+  }
+
+  let key = getProviderApiKey(providerName);
+  if (key) return key;
+
+  console.log('');
+  showWarning(`No API key found for ${chalk.bold(providerDef ? providerDef.name : providerName)}.`);
+
+  if (providerDef && providerDef.key_url) {
+    showInfo(`Get one at ${ACCENT(providerDef.key_url)}\n`);
+  }
+
+  const keyHint = providerDef && providerDef.key_prefix
+    ? `Key should start with ${providerDef.key_prefix}`
+    : '';
+
+  const { apiKey } = await inquirer.prompt([
+    {
+      type: 'password',
+      name: 'apiKey',
+      message: `Enter your ${providerDef ? providerDef.name : providerName} API key:`,
+      mask: '*',
+      validate: (v) => {
+        if (!v || v.length < 5) return 'Please enter a valid API key';
+        if (providerDef && providerDef.key_prefix && !v.startsWith(providerDef.key_prefix)) {
+          return keyHint;
+        }
+        return true;
+      },
+    },
+  ]);
+
+  const { save } = await inquirer.prompt([
+    {
+      type: 'confirm',
+      name: 'save',
+      message: 'Save this key to ~/.brains/config.json for future use?',
+      default: true,
+    },
+  ]);
+
+  if (save) {
+    setProviderApiKey(providerName, apiKey);
+    showSuccess('API key saved.');
+  }
+
+  // Set in environment for this session
+  if (providerDef && providerDef.key_env) {
+    process.env[providerDef.key_env] = apiKey;
+  }
+
+  return apiKey;
+}
+
+/**
+ * Build a merged system prompt when stacking brains.
+ */
+function buildSystemPrompt(brain, stackedBrains) {
+  let prompt = brain.systemPrompt;
+
+  if (stackedBrains.length > 0) {
+    prompt += '\n\n--- ADDITIONAL EXPERTISE ---\n';
+    prompt += 'You also have the following additional expertise available. ';
+    prompt += 'Incorporate this knowledge when relevant:\n\n';
+
+    for (const extra of stackedBrains) {
+      prompt += `## ${extra.name} (${extra.category})\n`;
+      prompt += extra.systemPrompt + '\n\n';
+    }
+  }
+
+  return prompt;
+}
+
+/**
+ * Format user answers as context for Claude.
+ */
+function formatContext(answers, brain) {
+  let context = `The user provided the following context:\n\n`;
+  for (const [key, value] of Object.entries(answers)) {
+    if (value && value.trim()) {
+      context += `**${key}**: ${value}\n`;
+    }
+  }
+  return context;
+}
+
+// ─── Category color map ───
+const CATEGORY_COLORS = {
+  builder: '#FF3366',
+  role: '#7B2FFF',
+  reviewer: '#00AAFF',
+  domain: '#00CC88',
+};
+
+function showSessionHeader(brain, stackedBrains) {
+  const color = CATEGORY_COLORS[brain.category] || '#ffffff';
+
+  console.log('');
+  console.log(chalk.hex(color).bold('  ┌─────────────────────────────────────────────────────'));
+  console.log(
+    chalk.hex(color).bold(`  │  🧠 ${brain.name}`) + DIM(` — ${brain.category}`)
+  );
+  if (stackedBrains.length > 0) {
+    console.log(
+      chalk.hex(color).bold('  │  ') + DIM(`+ ${stackedBrains.map((b) => b.name).join(', ')}`)
+    );
+  }
+  const providerName = getActiveProvider();
+  const providerDef = getProviderDef(providerName);
+  const providerLabel = providerDef ? providerDef.name : providerName;
+
+  console.log(chalk.hex(color).bold('  │  ') + DIM(`Stack: ${brain.stack.join(', ')}`));
+  console.log(chalk.hex(color).bold('  │  ') + DIM(`Provider: ${providerLabel}  •  Model: ${getModel()}`));
+  console.log(chalk.hex(color).bold('  └─────────────────────────────────────────────────────'));
+  console.log('');
+}
+
+// ═══════════════════════════════════════════
+//  MAIN RUN FUNCTION
+// ═══════════════════════════════════════════
 async function run(brainName, options) {
   const brain = getBrainById(brainName);
 
@@ -57,6 +198,9 @@ async function run(brainName, options) {
     }
   }
 
+  // Ensure API key
+  await ensureApiKey();
+
   const config = loadBrainConfig(brainName) || brain;
 
   // Handle brain stacking
@@ -71,82 +215,36 @@ async function run(brainName, options) {
         stackedBrains.push(extra);
         showInfo(`  ${SUCCESS('+')} ${extra.name} loaded`);
       } else {
-        showInfo(`  ${WARN('⚠')} ${extraBrain} not found, skipping`);
+        showWarning(`  ${extraBrain} not found, skipping`);
       }
     }
   }
 
-  // Show brain activation
-  console.log('');
-  const spinner = ora({
-    text: `Activating ${chalk.bold(brain.name)}...`,
-    color: 'magenta',
-    spinner: 'dots12',
-  }).start();
-  await new Promise((r) => setTimeout(r, 1200));
-  spinner.succeed(`${chalk.bold(brain.name)} is active`);
-
-  if (stackedBrains.length > 0) {
-    const stackSpinner = ora({
-      text: 'Merging stacked brain contexts...',
-      color: 'cyan',
-      spinner: 'dots12',
-    }).start();
-    await new Promise((r) => setTimeout(r, 800));
-    stackSpinner.succeed('Brain stack ready');
-  }
-
   // Show session header
-  const categoryColors = {
-    builder: '#FF3366',
-    role: '#7B2FFF',
-    reviewer: '#00AAFF',
-    domain: '#00CC88',
-  };
-  const color = categoryColors[brain.category] || '#ffffff';
-
-  console.log('');
-  console.log(
-    chalk.hex(color).bold('  ┌─────────────────────────────────────────────────────')
-  );
-  console.log(
-    chalk.hex(color).bold(`  │  🧠 ${brain.name}`) +
-      DIM(` — ${brain.category}`)
-  );
-  if (stackedBrains.length > 0) {
-    console.log(
-      chalk.hex(color).bold('  │  ') +
-        DIM(`+ ${stackedBrains.map((b) => b.name).join(', ')}`)
-    );
-  }
-  console.log(
-    chalk.hex(color).bold('  │  ') + DIM(`Stack: ${brain.stack.join(', ')}`)
-  );
-  console.log(
-    chalk.hex(color).bold('  └─────────────────────────────────────────────────────')
-  );
-  console.log('');
+  showSessionHeader(brain, stackedBrains);
 
-  // ─── Brain-specific workflows ───
-
-  if (brain.category === 'builder') {
-    await runBuilderBrain(brain, config, options);
-  } else if (brain.category === 'reviewer') {
-    await runReviewerBrain(brain, config, options);
-  } else if (brain.category === 'role') {
-    await runRoleBrain(brain, config, options);
-  } else if (brain.category === 'domain') {
-    await runDomainBrain(brain, config, options);
+  // ─── Route to brain-specific workflow ───
+  try {
+    if (brain.category === 'builder') {
+      await runBuilderBrain(brain, config, stackedBrains, options);
+    } else if (brain.category === 'reviewer') {
+      await runReviewerBrain(brain, config, stackedBrains, options);
+    } else if (brain.category === 'role' || brain.category === 'domain') {
+      await runConversationalBrain(brain, config, stackedBrains, options);
+    }
+  } catch (err) {
+    handleApiError(err);
   }
 }
 
-// ─── Builder Brain Runner ───
-async function runBuilderBrain(brain, config, options) {
+// ═══════════════════════════════════════════
+//  BUILDER BRAIN — Ask questions → Generate → Write files
+// ═══════════════════════════════════════════
+async function runBuilderBrain(brain, config, stackedBrains, options) {
   showInfo(`${brain.name} will ask you a few questions, then generate your project.\n`);
 
+  // Run interview
   const answers = {};
-
-  // Run through the brain's interview questions
   for (const q of config.questions || brain.questions) {
     const { answer } = await inquirer.prompt([
       {
@@ -158,61 +256,111 @@ async function runBuilderBrain(brain, config, options) {
     answers[q.key] = answer;
   }
 
+  const projectName = answers.product || answers.name || answers.idea || brain.id;
+  const projectDir = path.resolve(options.dir || `./${projectName.toLowerCase().replace(/\s+/g, '-')}-project`);
+
   console.log('');
-  showInfo('Got it! Generating your project...\n');
-
-  // Simulate generation steps based on brain type
-  const genSteps = getBuilderSteps(brain.id);
-  for (const step of genSteps) {
-    const spinner = ora({
-      text: step.text,
-      color: 'magenta',
-      spinner: 'dots12',
-    }).start();
-    await new Promise((r) => setTimeout(r, step.time));
-    spinner.succeed(chalk.dim(step.text));
+  showInfo('Generating your project... (this may take a moment)\n');
+
+  const systemPrompt = buildSystemPrompt(brain, stackedBrains) + `
+
+CRITICAL OUTPUT FORMAT:
+You MUST output every file using this exact format so the CLI can parse and write them:
+
+FILE: relative/path/to/file.ext
+\`\`\`language
+file contents here
+\`\`\`
+
+Output ALL files needed for a complete, working project. Include package.json, config files, and every source file.
+Do NOT skip any files. Do NOT use placeholders like "// ... rest of code". Output complete file contents.`;
+
+  const userMessage = formatContext(answers, brain) +
+    `\n\nGenerate the complete project. Output every file using the FILE: format specified in your instructions.`;
+
+  // Stream the response — accumulate silently, then parse files
+  const spinner = ora({
+    text: 'Generating project files (streaming from Claude)...',
+    color: 'magenta',
+    spinner: 'dots12',
+  }).start();
+
+  let receivedBytes = 0;
+  let fullResponse = '';
+  try {
+    fullResponse = await streamMessage({
+      systemPrompt,
+      messages: [{ role: 'user', content: userMessage }],
+      maxTokens: 8096,
+      onText: (text) => {
+        receivedBytes += text.length;
+        if (receivedBytes % 200 < text.length) {
+          spinner.text = `Generating... (${(receivedBytes / 1024).toFixed(0)}KB received)`;
+        }
+      },
+      onStart: () => {},
+      onEnd: () => {
+        spinner.succeed(`Generation complete (${(receivedBytes / 1024).toFixed(1)}KB)`);
+      },
+    });
+  } catch (err) {
+    spinner.stop();
+    throw err;
   }
 
-  // Show output
-  const dir = options.dir || `./${answers.product || answers.name || brain.id}-project`;
+  // Parse file blocks from response
+  const fileBlocks = parseFileBlocks(fullResponse);
 
+  if (fileBlocks.length === 0) {
+    console.log('');
+    showWarning('No file blocks detected in response. Showing raw output:\n');
+    console.log(chalk.dim(fullResponse.slice(0, 2000)));
+    if (fullResponse.length > 2000) {
+      console.log(chalk.dim(`\n... (${fullResponse.length} chars total)`));
+    }
+    return;
+  }
+
+  // Write files to disk
   console.log('');
-  showSuccess('Project generated successfully!\n');
+  const writeSpinner = ora({
+    text: `Writing ${fileBlocks.length} files to ${projectDir}...`,
+    color: 'magenta',
+    spinner: 'dots12',
+  }).start();
 
-  const tree = getProjectTree(brain.id, answers);
-  console.log(chalk.dim(tree));
+  const { written, errors } = writeFileBlocks(projectDir, fileBlocks);
+  writeSpinner.succeed(`${written.length} files written`);
 
+  if (errors.length > 0) {
+    for (const err of errors) {
+      showWarning(`  Failed: ${err}`);
+    }
+  }
+
+  // Show project tree
   console.log('');
-  showInfo(`Project created at: ${ACCENT(dir)}`);
-  showInfo(`Next steps:`);
-  console.log(`    ${DIM('$')} ${ACCENT(`cd ${dir}`)}`);
+  showSuccess('Project generated successfully!\n');
+  const tree = generateTreeView(written, path.basename(projectDir));
+  console.log(chalk.dim('  ' + tree.split('\n').join('\n  ')));
+
+  // Show next steps
+  console.log('');
+  showInfo(`Project created at: ${ACCENT(projectDir)}`);
+  showInfo('Next steps:');
+  console.log(`    ${DIM('$')} ${ACCENT(`cd ${path.basename(projectDir)}`)}`);
   console.log(`    ${DIM('$')} ${ACCENT('npm install')}`);
   console.log(`    ${DIM('$')} ${ACCENT('npm run dev')}`);
   console.log('');
-
-  // Show what the brain generated
-  const systemPromptInfo = `
-${chalk.bold('What was generated:')}
-
-This project was scaffolded by the ${chalk.hex('#FF3366').bold(brain.name)} Brain.
-The brain used the following context to customize your project:
-${Object.entries(answers)
-  .map(([k, v]) => `  ${chalk.hex('#FF3366')('▸')} ${k}: ${DIM(v)}`)
-  .join('\n')}
-
-${chalk.bold('Brain System Prompt:')}
-${DIM('The brain operated with a specialized system prompt that ensures')}
-${DIM('best practices, modern patterns, and production-grade output.')}
-${DIM(`View it at: ~/.brains/${brain.id}/BRAIN.md`)}
-`;
-  console.log(systemPromptInfo);
 }
 
-// ─── Reviewer Brain Runner ───
-async function runReviewerBrain(brain, config, options) {
-  const targetDir = options.dir || process.cwd();
+// ═══════════════════════════════════════════
+//  REVIEWER BRAIN — Read codebase → Analyze → Report
+// ═══════════════════════════════════════════
+async function runReviewerBrain(brain, config, stackedBrains, options) {
+  const targetDir = path.resolve(options.dir || process.cwd());
 
-  showInfo(`${brain.name} will analyze your codebase.\n`);
+  showInfo(`${brain.name} will analyze your codebase at ${ACCENT(targetDir)}\n`);
 
   // Ask focus questions
   const answers = {};
@@ -227,55 +375,81 @@ async function runReviewerBrain(brain, config, options) {
     answers[q.key] = answer;
   }
 
+  // Scan files
   console.log('');
+  const scanSpinner = ora({
+    text: 'Scanning project files...',
+    color: 'cyan',
+    spinner: 'dots12',
+  }).start();
+
+  const { content: filesContent, fileCount, totalSize, files } = readFilesForReview(targetDir);
 
-  const scanSteps = [
-    { text: 'Scanning project structure', time: 600 },
-    { text: 'Analyzing source files', time: 1000 },
-    { text: 'Checking dependencies', time: 500 },
-    { text: 'Running pattern analysis', time: 800 },
-    { text: 'Evaluating security posture', time: 700 },
-    { text: 'Generating review report', time: 600 },
-  ];
-
-  for (const step of scanSteps) {
-    const spinner = ora({
-      text: step.text,
-      color: 'cyan',
-      spinner: 'dots12',
-    }).start();
-    await new Promise((r) => setTimeout(r, step.time));
-    spinner.succeed(chalk.dim(step.text));
+  if (fileCount === 0) {
+    scanSpinner.fail('No source files found in this directory');
+    showInfo(`Make sure you're in a project directory with source code.`);
+    return;
   }
 
+  scanSpinner.succeed(`Found ${fileCount} files (${(totalSize / 1024).toFixed(1)}KB)`);
+
+  // Show scanned files
+  console.log(chalk.dim(`  Files: ${files.slice(0, 10).join(', ')}${files.length > 10 ? ` ... +${files.length - 10} more` : ''}`));
   console.log('');
-  showSuccess('Review complete!\n');
 
-  // Show sample review output
-  console.log(chalk.bold('  Review Summary\n'));
-  console.log(`  ${chalk.red('🔴 CRITICAL')}  ${chalk.dim('0 issues')}`);
-  console.log(`  ${chalk.hex('#FF8800')('🟡 WARNING')}   ${chalk.dim('3 issues')}`);
-  console.log(`  ${chalk.hex('#00CC88')('🟢 SUGGESTION')} ${chalk.dim('7 suggestions')}`);
-  console.log(`  ${chalk.blue('📝 NOTE')}      ${chalk.dim('2 notes')}`);
+  const reviewSpinner = ora({
+    text: 'Analyzing codebase with Claude...',
+    color: 'cyan',
+    spinner: 'dots12',
+  }).start();
 
-  console.log(`\n${chalk.dim('─'.repeat(56))}\n`);
+  const systemPrompt = buildSystemPrompt(brain, stackedBrains);
+  const focusArea = answers.focus || answers.type || 'general review';
 
-  console.log(
-    `  ${chalk.bold('In production, this brain would:')}\n` +
-      `  ${chalk.hex('#00AAFF')('▸')} Read every file in your project\n` +
-      `  ${chalk.hex('#00AAFF')('▸')} Apply the review methodology from its system prompt\n` +
-      `  ${chalk.hex('#00AAFF')('▸')} Use Claude to analyze patterns, bugs, and security\n` +
-      `  ${chalk.hex('#00AAFF')('▸')} Output a detailed, prioritized review report\n` +
-      `  ${chalk.hex('#00AAFF')('▸')} Optionally auto-fix issues with --fix flag\n`
-  );
+  const userMessage = `Review the following codebase. Focus area: ${focusArea}
+
+${filesContent}
+
+Provide a structured review using the severity format (🔴 CRITICAL, 🟡 WARNING, 🟢 SUGGESTION, 📝 NOTE). Be specific with file names and line references. Start with a summary, then list findings.`;
+
+  let fullResponse = '';
+  try {
+    fullResponse = await streamMessage({
+      systemPrompt,
+      messages: [{ role: 'user', content: userMessage }],
+      maxTokens: 8096,
+      onStart: () => {
+        reviewSpinner.stop();
+        console.log(chalk.bold('\n  Review Report\n'));
+        console.log(chalk.dim('─'.repeat(56)) + '\n');
+      },
+      onText: (text) => process.stdout.write(text),
+      onEnd: () => {
+        console.log('\n\n' + chalk.dim('─'.repeat(56)));
+      },
+    });
+  } catch (err) {
+    reviewSpinner.stop();
+    throw err;
+  }
 
-  showInfo(`Brain config: ${DIM(`~/.brains/${brain.id}/BRAIN.md`)}`);
+  // Save session
+  const sessionFile = saveSession(brain.id, [
+    { role: 'user', content: `Review ${fileCount} files from ${targetDir}` },
+    { role: 'assistant', content: fullResponse },
+  ]);
+  console.log('');
+  showInfo(`Review saved to: ${DIM(sessionFile)}`);
   console.log('');
 }
 
-// ─── Role Brain Runner ───
-async function runRoleBrain(brain, config, options) {
-  showInfo(`${brain.name} is ready. Starting conversational session.\n`);
+// ═══════════════════════════════════════════
+//  CONVERSATIONAL BRAIN (Role + Domain) — REPL Loop
+// ═══════════════════════════════════════════
+async function runConversationalBrain(brain, config, stackedBrains, options) {
+  const color = CATEGORY_COLORS[brain.category] || '#ffffff';
+
+  showInfo(`Starting conversational session with ${brain.name}.\n`);
 
   // Ask initial context questions
   const answers = {};
@@ -284,204 +458,190 @@ async function runRoleBrain(brain, config, options) {
       {
         type: 'input',
         name: 'answer',
-        message: chalk.hex('#7B2FFF')(q.question),
+        message: chalk.hex(color)(q.question),
       },
     ]);
     answers[q.key] = answer;
   }
 
-  console.log('');
+  const systemPrompt = buildSystemPrompt(brain, stackedBrains);
+  const conversationHistory = [];
 
-  const spinner = ora({
-    text: 'Loading context and preparing response...',
+  // Build initial context message from answers
+  const contextMessage = formatContext(answers, brain) +
+    '\n\nPlease acknowledge this context and let me know you\'re ready. Keep it brief.';
+
+  conversationHistory.push({ role: 'user', content: contextMessage });
+
+  // Get initial response
+  console.log('');
+  const initSpinner = ora({
+    text: 'Loading context...',
     color: 'magenta',
     spinner: 'dots12',
   }).start();
-  await new Promise((r) => setTimeout(r, 1500));
-  spinner.succeed(chalk.dim('Context loaded'));
 
-  console.log(`\n${chalk.dim('─'.repeat(56))}\n`);
+  let initialResponse = '';
+  try {
+    initialResponse = await streamMessage({
+      systemPrompt,
+      messages: conversationHistory,
+      maxTokens: 2048,
+      onStart: () => {
+        initSpinner.stop();
+        console.log(`\n  ${chalk.hex(color).bold(`[${brain.name}]`)}\n`);
+      },
+      onText: (text) => process.stdout.write(text),
+      onEnd: () => console.log('\n'),
+    });
+  } catch (err) {
+    initSpinner.stop();
+    throw err;
+  }
 
-  console.log(
-    `  ${chalk.bold('In production, this brain would now:')}\n` +
-      `  ${chalk.hex('#7B2FFF')('▸')} Enter a REPL-style conversational loop\n` +
-      `  ${chalk.hex('#7B2FFF')('▸')} Use the ${brain.name} system prompt to guide every response\n` +
-      `  ${chalk.hex('#7B2FFF')('▸')} Generate code, review PRs, design systems interactively\n` +
-      `  ${chalk.hex('#7B2FFF')('▸')} Maintain context across the conversation\n` +
-      `  ${chalk.hex('#7B2FFF')('▸')} Write files directly to your project\n`
-  );
+  conversationHistory.push({ role: 'assistant', content: initialResponse });
 
-  console.log(
-    `  ${chalk.bold('The system prompt for this brain includes:')}\n` +
-      `  ${DIM('Personality, methodology, tech preferences, and output format.')}\n` +
-      `  ${DIM(`View it at: ~/.brains/${brain.id}/BRAIN.md`)}\n`
-  );
+  // Enter REPL loop
+  console.log(chalk.dim('  Type your message and press Enter. Commands: exit, clear, save\n'));
 
-  // Interactive loop demo
-  const { continueChat } = await inquirer.prompt([
-    {
-      type: 'confirm',
-      name: 'continueChat',
-      message: 'Want to see a sample interaction?',
-      default: true,
-    },
-  ]);
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+    prompt: chalk.hex(color).bold('  You: '),
+  });
 
-  if (continueChat) {
-    console.log(`\n  ${chalk.hex('#7B2FFF').bold(`[${brain.name}]`)} ${chalk.white("I've reviewed your request. Here's my approach:")}\n`);
-    console.log(DIM('  (In production, Claude would respond here using the brain\'s system prompt'));
-    console.log(DIM('   as its personality and methodology, generating real code and advice.)\n'));
-  }
-}
+  rl.prompt();
 
-// ─── Domain Brain Runner ───
-async function runDomainBrain(brain, config, options) {
-  showInfo(`${brain.name} expert session started.\n`);
+  rl.on('line', async (line) => {
+    const input = line.trim();
 
-  // Ask what they need
-  for (const q of config.questions || brain.questions) {
-    const { answer } = await inquirer.prompt([
-      {
-        type: 'input',
-        name: 'answer',
-        message: chalk.hex('#00CC88')(q.question),
-      },
-    ]);
-  }
+    if (!input) {
+      rl.prompt();
+      return;
+    }
 
-  console.log('');
+    // Handle commands
+    if (input === 'exit' || input === 'quit') {
+      const sessionFile = saveSession(brain.id, conversationHistory);
+      console.log('');
+      showSuccess('Session ended.');
+      showInfo(`Conversation saved to: ${DIM(sessionFile)}`);
+      console.log('');
+      rl.close();
+      return;
+    }
 
-  const spinner = ora({
-    text: `Consulting ${brain.name} knowledge base...`,
-    color: 'green',
-    spinner: 'dots12',
-  }).start();
-  await new Promise((r) => setTimeout(r, 1200));
-  spinner.succeed(chalk.dim('Ready'));
+    if (input === 'clear') {
+      conversationHistory.length = 0;
+      console.log('');
+      showInfo('Conversation history cleared.');
+      console.log('');
+      rl.prompt();
+      return;
+    }
 
-  console.log(`\n${chalk.dim('─'.repeat(56))}\n`);
+    if (input === 'save') {
+      const sessionFile = saveSession(brain.id, conversationHistory);
+      showInfo(`Saved to: ${DIM(sessionFile)}`);
+      rl.prompt();
+      return;
+    }
 
-  console.log(
-    `  ${chalk.bold('In production, this brain would:')}\n` +
-      `  ${chalk.hex('#00CC88')('▸')} Use deep ${brain.stack[0]} expertise from its system prompt\n` +
-      `  ${chalk.hex('#00CC88')('▸')} Generate best-practice code specific to your question\n` +
-      `  ${chalk.hex('#00CC88')('▸')} Include explanations, trade-offs, and gotchas\n` +
-      `  ${chalk.hex('#00CC88')('▸')} Reference latest patterns and APIs\n` +
-      `  ${chalk.hex('#00CC88')('▸')} Work interactively in a conversational loop\n`
-  );
+    // Send to Claude
+    conversationHistory.push({ role: 'user', content: input });
+
+    try {
+      console.log('');
+      const response = await streamMessage({
+        systemPrompt,
+        messages: conversationHistory,
+        maxTokens: 8096,
+        onStart: () => {
+          console.log(`  ${chalk.hex(color).bold(`[${brain.name}]`)}\n`);
+        },
+        onText: (text) => process.stdout.write(text),
+        onEnd: () => console.log('\n'),
+      });
+
+      conversationHistory.push({ role: 'assistant', content: response });
+    } catch (err) {
+      handleApiError(err);
+    }
 
-  showInfo(`Brain config: ${DIM(`~/.brains/${brain.id}/BRAIN.md`)}`);
-  console.log('');
-}
+    rl.prompt();
+  });
+
+  rl.on('close', () => {
+    // Ensure clean exit
+  });
 
-// ─── Helper: Builder generation steps ───
-function getBuilderSteps(brainId) {
-  const common = [
-    { text: 'Analyzing requirements', time: 500 },
-    { text: 'Selecting architecture', time: 400 },
-  ];
-
-  const specific = {
-    resume: [
-      { text: 'Generating Next.js project', time: 700 },
-      { text: 'Creating resume components', time: 600 },
-      { text: 'Applying theme and typography', time: 500 },
-      { text: 'Adding responsive styles', time: 400 },
-      { text: 'Configuring SEO metadata', time: 300 },
-      { text: 'Setting up Vercel deployment', time: 300 },
-    ],
-    mvp: [
-      { text: 'Generating Next.js + Prisma project', time: 800 },
-      { text: 'Creating database schema', time: 600 },
-      { text: 'Building API routes', time: 700 },
-      { text: 'Setting up authentication', time: 500 },
-      { text: 'Generating dashboard pages', time: 600 },
-      { text: 'Creating landing page', time: 500 },
-      { text: 'Configuring Docker', time: 400 },
-    ],
-    landing: [
-      { text: 'Generating page structure', time: 500 },
-      { text: 'Writing conversion copy', time: 700 },
-      { text: 'Building hero section', time: 500 },
-      { text: 'Creating feature sections', time: 400 },
-      { text: 'Adding social proof', time: 400 },
-      { text: 'Setting up email capture', time: 300 },
-      { text: 'Adding animations', time: 500 },
-    ],
-  };
-
-  return [...common, ...(specific[brainId] || specific.mvp)];
+  // Return a promise that resolves when REPL ends
+  return new Promise((resolve) => {
+    rl.on('close', resolve);
+  });
 }
 
-// ─── Helper: Project tree output ───
-function getProjectTree(brainId, answers) {
-  const name = answers.product || answers.name || brainId;
-  const trees = {
-    resume: `
-  ${chalk.bold(`${name}-project/`)}
-  ├── app/
-  │   ├── layout.tsx
-  │   ├── page.tsx
-  │   └── globals.css
-  ├── components/
-  │   ├── Hero.tsx
-  │   ├── Experience.tsx
-  │   ├── Skills.tsx
-  │   ├── Projects.tsx
-  │   └── Contact.tsx
-  ├── lib/
-  │   └── data.ts
-  ├── public/
-  │   └── resume.pdf
-  ├── package.json
-  ├── tailwind.config.ts
-  ├── tsconfig.json
-  └── next.config.js`,
-    mvp: `
-  ${chalk.bold(`${name}-project/`)}
-  ├── app/
-  │   ├── layout.tsx
-  │   ├── page.tsx              ${DIM('← Landing page')}
-  │   ├── (auth)/
-  │   │   ├── login/page.tsx
-  │   │   └── register/page.tsx
-  │   ├── dashboard/
-  │   │   ├── layout.tsx
-  │   │   └── page.tsx
-  │   └── api/
-  │       └── [...routes].ts
-  ├── components/
-  │   ├── ui/
-  │   └── features/
-  ├── lib/
-  │   ├── db.ts
-  │   └── auth.ts
-  ├── prisma/
-  │   ├── schema.prisma
-  │   └── migrations/
-  ├── package.json
-  ├── docker-compose.yml
-  └── .env.example`,
-    landing: `
-  ${chalk.bold(`${name}-project/`)}
-  ├── app/
-  │   ├── layout.tsx
-  │   ├── page.tsx
-  │   └── globals.css
-  ├── components/
-  │   ├── Hero.tsx
-  │   ├── Features.tsx
-  │   ├── HowItWorks.tsx
-  │   ├── Testimonials.tsx
-  │   ├── Pricing.tsx
-  │   ├── FAQ.tsx
-  │   └── CTA.tsx
-  ├── package.json
-  ├── tailwind.config.ts
-  └── next.config.js`,
-  };
-
-  return trees[brainId] || trees.mvp;
+// ═══════════════════════════════════════════
+//  ERROR HANDLING
+// ═══════════════════════════════════════════
+function handleApiError(err) {
+  console.log('');
+
+  const providerName = getActiveProvider();
+  const providerDef = getProviderDef(providerName);
+  const providerLabel = providerDef ? providerDef.name : providerName;
+
+  if (err.message === 'NO_API_KEY') {
+    showError(`No API key configured for ${providerLabel}.`);
+    showInfo(`Run ${ACCENT('brains config set api_key <your-key>')}`);
+    if (providerDef && providerDef.key_env) {
+      showInfo(`Or set env var: ${ACCENT(`export ${providerDef.key_env}=<your-key>`)}`);
+    }
+    return;
+  }
+
+  if (err.status === 401 || err.code === 'invalid_api_key') {
+    showError(`Invalid API key for ${providerLabel}.`);
+    if (providerDef && providerDef.key_url) {
+      showInfo(`Check your key at ${ACCENT(providerDef.key_url)}`);
+    }
+    showInfo(`Update with: ${ACCENT('brains config set api_key <your-key>')}`);
+    return;
+  }
+
+  if (err.status === 429) {
+    showError('Rate limit exceeded.');
+    showInfo('Wait a moment and try again, or check your API usage limits.');
+    if (providerDef && providerDef.free) {
+      showInfo(`${providerLabel} free tier has rate limits. Consider upgrading or switching providers.`);
+      showInfo(`Switch: ${ACCENT('brains config set provider <name>')}`);
+    }
+    return;
+  }
+
+  if (err.status === 529 || err.status === 503) {
+    showError(`${providerLabel} is temporarily overloaded.`);
+    showInfo('Try again in a few seconds.');
+    return;
+  }
+
+  if (err.code === 'ENOTFOUND' || err.code === 'ECONNREFUSED' || err.code === 'ETIMEDOUT') {
+    showError(`Network error — cannot reach ${providerLabel}.`);
+    if (providerName === 'ollama') {
+      showInfo('Is Ollama running? Start it with: ollama serve');
+    } else {
+      showInfo('Check your internet connection and try again.');
+    }
+    return;
+  }
+
+  // Generic error
+  showError(`API Error: ${err.message || err}`);
+  if (err.status) {
+    showInfo(`Status: ${err.status}`);
+  }
+  showInfo(`Provider: ${providerLabel} | Model: ${getModel()}`);
+  console.log('');
 }
 
 module.exports = { run };