@vatzzza/botintern 1.0.0

package/.github/workflows/npm-publish.yml

@@ -0,0 +1,33 @@
+# This workflow will run tests using node and then publish a package to GitHub Packages when a release is created
+# For more information see: https://docs.github.com/en/actions/publishing-packages/publishing-nodejs-packages
+
+name: Node.js Package
+
+on: 
+  push:
+    # only trigger on branches, not on tags
+    branches: 'main'
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-node@v4
+        with:
+          node-version: 20
+      - run: npm ci
+
+  publish-npm:
+    needs: build
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-node@v4
+        with:
+          node-version: 20
+          registry-url: https://registry.npmjs.org/
+      - run: npm ci
+      - run: npm publish --access=public
+        env:
+          NODE_AUTH_TOKEN: ${{secrets.npm_token}}

package/README.md

@@ -0,0 +1,15 @@
+# botintern
+
+To install dependencies:
+
+```bash
+bun install
+```
+
+To run:
+
+```bash
+bun run index.ts
+```
+
+This project was created using `bun init` in bun v1.3.5. [Bun](https://bun.com) is a fast all-in-one JavaScript runtime.

package/bun.lock

@@ -0,0 +1,80 @@
+{
+  "lockfileVersion": 1,
+  "configVersion": 1,
+  "workspaces": {
+    "": {
+      "name": "botintern",
+      "dependencies": {
+        "@google/generative-ai": "^0.24.1",
+        "chalk": "^5.6.2",
+        "commander": "^14.0.2",
+        "dotenv": "^17.2.3",
+        "ora": "^9.0.0",
+        "playwright": "^1.57.0",
+      },
+      "devDependencies": {
+        "@types/bun": "latest",
+      },
+      "peerDependencies": {
+        "typescript": "^5",
+      },
+    },
+  },
+  "packages": {
+    "@google/generative-ai": ["@google/generative-ai@0.24.1", "", {}, "sha512-MqO+MLfM6kjxcKoy0p1wRzG3b4ZZXtPI+z2IE26UogS2Cm/XHO+7gGRBh6gcJsOiIVoH93UwKvW4HdgiOZCy9Q=="],
+
+    "@types/bun": ["@types/bun@1.3.6", "", { "dependencies": { "bun-types": "1.3.6" } }, "sha512-uWCv6FO/8LcpREhenN1d1b6fcspAB+cefwD7uti8C8VffIv0Um08TKMn98FynpTiU38+y2dUO55T11NgDt8VAA=="],
+
+    "@types/node": ["@types/node@25.0.9", "", { "dependencies": { "undici-types": "~7.16.0" } }, "sha512-/rpCXHlCWeqClNBwUhDcusJxXYDjZTyE8v5oTO7WbL8eij2nKhUeU89/6xgjU7N4/Vh3He0BtyhJdQbDyhiXAw=="],
+
+    "ansi-regex": ["ansi-regex@6.2.2", "", {}, "sha512-Bq3SmSpyFHaWjPk8If9yc6svM8c56dB5BAtW4Qbw5jHTwwXXcTLoRMkpDJp6VL0XzlWaCHTXrkFURMYmD0sLqg=="],
+
+    "bun-types": ["bun-types@1.3.6", "", { "dependencies": { "@types/node": "*" } }, "sha512-OlFwHcnNV99r//9v5IIOgQ9Uk37gZqrNMCcqEaExdkVq3Avwqok1bJFmvGMCkCE0FqzdY8VMOZpfpR3lwI+CsQ=="],
+
+    "chalk": ["chalk@5.6.2", "", {}, "sha512-7NzBL0rN6fMUW+f7A6Io4h40qQlG+xGmtMxfbnH/K7TAtt8JQWVQK+6g0UXKMeVJoyV5EkkNsErQ8pVD3bLHbA=="],
+
+    "cli-cursor": ["cli-cursor@5.0.0", "", { "dependencies": { "restore-cursor": "^5.0.0" } }, "sha512-aCj4O5wKyszjMmDT4tZj93kxyydN/K5zPWSCe6/0AV/AA1pqe5ZBIw0a2ZfPQV7lL5/yb5HsUreJ6UFAF1tEQw=="],
+
+    "cli-spinners": ["cli-spinners@3.4.0", "", {}, "sha512-bXfOC4QcT1tKXGorxL3wbJm6XJPDqEnij2gQ2m7ESQuE+/z9YFIWnl/5RpTiKWbMq3EVKR4fRLJGn6DVfu0mpw=="],
+
+    "commander": ["commander@14.0.2", "", {}, "sha512-TywoWNNRbhoD0BXs1P3ZEScW8W5iKrnbithIl0YH+uCmBd0QpPOA8yc82DS3BIE5Ma6FnBVUsJ7wVUDz4dvOWQ=="],
+
+    "dotenv": ["dotenv@17.2.3", "", {}, "sha512-JVUnt+DUIzu87TABbhPmNfVdBDt18BLOWjMUFJMSi/Qqg7NTYtabbvSNJGOJ7afbRuv9D/lngizHtP7QyLQ+9w=="],
+
+    "fsevents": ["fsevents@2.3.2", "", { "os": "darwin" }, "sha512-xiqMQR4xAeHTuB9uWm+fFRcIOgKBMiOBP+eXiyT7jsgVCq1bkVygt00oASowB7EdtpOHaaPgKt812P9ab+DDKA=="],
+
+    "get-east-asian-width": ["get-east-asian-width@1.4.0", "", {}, "sha512-QZjmEOC+IT1uk6Rx0sX22V6uHWVwbdbxf1faPqJ1QhLdGgsRGCZoyaQBm/piRdJy/D2um6hM1UP7ZEeQ4EkP+Q=="],
+
+    "is-interactive": ["is-interactive@2.0.0", "", {}, "sha512-qP1vozQRI+BMOPcjFzrjXuQvdak2pHNUMZoeG2eRbiSqyvbEf/wQtEOTOX1guk6E3t36RkaqiSt8A/6YElNxLQ=="],
+
+    "is-unicode-supported": ["is-unicode-supported@2.1.0", "", {}, "sha512-mE00Gnza5EEB3Ds0HfMyllZzbBrmLOX3vfWoj9A9PEnTfratQ/BcaJOuMhnkhjXvb2+FkY3VuHqtAGpTPmglFQ=="],
+
+    "log-symbols": ["log-symbols@7.0.1", "", { "dependencies": { "is-unicode-supported": "^2.0.0", "yoctocolors": "^2.1.1" } }, "sha512-ja1E3yCr9i/0hmBVaM0bfwDjnGy8I/s6PP4DFp+yP+a+mrHO4Rm7DtmnqROTUkHIkqffC84YY7AeqX6oFk0WFg=="],
+
+    "mimic-function": ["mimic-function@5.0.1", "", {}, "sha512-VP79XUPxV2CigYP3jWwAUFSku2aKqBH7uTAapFWCBqutsbmDo96KY5o8uh6U+/YSIn5OxJnXp73beVkpqMIGhA=="],
+
+    "onetime": ["onetime@7.0.0", "", { "dependencies": { "mimic-function": "^5.0.0" } }, "sha512-VXJjc87FScF88uafS3JllDgvAm+c/Slfz06lorj2uAY34rlUu0Nt+v8wreiImcrgAjjIHp1rXpTDlLOGw29WwQ=="],
+
+    "ora": ["ora@9.0.0", "", { "dependencies": { "chalk": "^5.6.2", "cli-cursor": "^5.0.0", "cli-spinners": "^3.2.0", "is-interactive": "^2.0.0", "is-unicode-supported": "^2.1.0", "log-symbols": "^7.0.1", "stdin-discarder": "^0.2.2", "string-width": "^8.1.0", "strip-ansi": "^7.1.2" } }, "sha512-m0pg2zscbYgWbqRR6ABga5c3sZdEon7bSgjnlXC64kxtxLOyjRcbbUkLj7HFyy/FTD+P2xdBWu8snGhYI0jc4A=="],
+
+    "playwright": ["playwright@1.57.0", "", { "dependencies": { "playwright-core": "1.57.0" }, "optionalDependencies": { "fsevents": "2.3.2" }, "bin": { "playwright": "cli.js" } }, "sha512-ilYQj1s8sr2ppEJ2YVadYBN0Mb3mdo9J0wQ+UuDhzYqURwSoW4n1Xs5vs7ORwgDGmyEh33tRMeS8KhdkMoLXQw=="],
+
+    "playwright-core": ["playwright-core@1.57.0", "", { "bin": { "playwright-core": "cli.js" } }, "sha512-agTcKlMw/mjBWOnD6kFZttAAGHgi/Nw0CZ2o6JqWSbMlI219lAFLZZCyqByTsvVAJq5XA5H8cA6PrvBRpBWEuQ=="],
+
+    "restore-cursor": ["restore-cursor@5.1.0", "", { "dependencies": { "onetime": "^7.0.0", "signal-exit": "^4.1.0" } }, "sha512-oMA2dcrw6u0YfxJQXm342bFKX/E4sG9rbTzO9ptUcR/e8A33cHuvStiYOwH7fszkZlZ1z/ta9AAoPk2F4qIOHA=="],
+
+    "signal-exit": ["signal-exit@4.1.0", "", {}, "sha512-bzyZ1e88w9O1iNJbKnOlvYTrWPDl46O1bG0D3XInv+9tkPrxrN8jUUTiFlDkkmKWgn1M6CfIA13SuGqOa9Korw=="],
+
+    "stdin-discarder": ["stdin-discarder@0.2.2", "", {}, "sha512-UhDfHmA92YAlNnCfhmq0VeNL5bDbiZGg7sZ2IvPsXubGkiNa9EC+tUTsjBRsYUAz87btI6/1wf4XoVvQ3uRnmQ=="],
+
+    "string-width": ["string-width@8.1.0", "", { "dependencies": { "get-east-asian-width": "^1.3.0", "strip-ansi": "^7.1.0" } }, "sha512-Kxl3KJGb/gxkaUMOjRsQ8IrXiGW75O4E3RPjFIINOVH8AMl2SQ/yWdTzWwF3FevIX9LcMAjJW+GRwAlAbTSXdg=="],
+
+    "strip-ansi": ["strip-ansi@7.1.2", "", { "dependencies": { "ansi-regex": "^6.0.1" } }, "sha512-gmBGslpoQJtgnMAvOVqGZpEz9dyoKTCzy2nfz/n8aIFhN/jCE/rCmcxabB6jOOHV+0WNnylOxaxBQPSvcWklhA=="],
+
+    "typescript": ["typescript@5.9.3", "", { "bin": { "tsc": "bin/tsc", "tsserver": "bin/tsserver" } }, "sha512-jl1vZzPDinLr9eUt3J/t7V6FgNEw9QjvBPdysz9KfQDD41fQrC2Y4vKQdiaUpFT4bXlb1RHhLpp8wtm6M5TgSw=="],
+
+    "undici-types": ["undici-types@7.16.0", "", {}, "sha512-Zz+aZWSj8LE6zoxD+xrjh4VfkIG8Ya6LvYkZqtUQGJPZjYl53ypCaUwWqo7eI0x66KBGeRo+mlBEkMSeSZ38Nw=="],
+
+    "yoctocolors": ["yoctocolors@2.1.2", "", {}, "sha512-CzhO+pFNo8ajLM2d2IW/R93ipy99LWjtwblvC1RsoSUMZgyLbYFr221TnSNT7GjGdYui6P459mw9JH/g/zW2ug=="],
+  }
+}

package/index.js

@@ -0,0 +1,425 @@
+#!/usr/bin/env node
+
+// Use bun run index.js to run this file single time.
+// Use bun run dev to run this file in watch mode.
+
+import { Command } from "commander";
+import chalk from "chalk";
+import exec from "child_process";
+import ora from "ora";
+import { fixBuild, generateYaml } from "./lib/ai.js";
+import fs from "fs";
+import { extractErrorFile } from "./utils/errorExtractor.js";
+import path from "path";
+import { getCriticalSourceCode, getProjectStructure } from "./lib/fileTreeParser.js";
+import { runTests } from "./lib/runTests.js";
+import { runPlaywright } from "./lib/runPlaywright.js";
+import { loop } from "./lib/loop.js";
+import { saveApiKey } from "./lib/registerAndFetchKey.js";
+import readline from "readline";
+
+const banner = `
+${chalk.bold.cyan('┌─────────────────────────────────────────────────────────────────┐')}
+${chalk.cyan('│')} ${chalk.bold.white('╔═════════════════════════════════════════════════════════════╗')} ${chalk.cyan('│')}
+${chalk.cyan('│')} ${chalk.cyan('║')} ${chalk.bold.white('█████▄ ▄████▄ ██████ ██ ███  ██ ██████ ██████ █████▄  ███  ██ ')} ${chalk.cyan('║')} ${chalk.cyan('│')}
+${chalk.cyan('│')} ${chalk.cyan('║')} ${chalk.bold.white('██▄▄██ ██  ██   ██   ██ ██ ▀▄██   ██   ██▄▄   ██▄▄██▄ ██ ▀▄██ ')} ${chalk.cyan('║')} ${chalk.cyan('│')}
+${chalk.cyan('│')} ${chalk.cyan('║')} ${chalk.bold.white('██▄▄█▀ ▀████▀   ██   ██ ██   ██   ██   ██▄▄▄▄ ██   ██ ██   ██  ')} ${chalk.cyan('║')} ${chalk.cyan('│')}
+${chalk.cyan('│')} ${chalk.cyan('║')} ${chalk.gray('         ' + chalk.italic.blue('AI-Powered Code Testing & Validation') + '         ')} ${chalk.cyan('║')} ${chalk.cyan('│')}
+${chalk.cyan('│')} ${chalk.cyan('║')} ${chalk.dim('      ' + chalk.yellow('⚡') + ' The intern that fixes more bugs than it creates ' + chalk.yellow('⚡') + '      ')} ${chalk.cyan('║')} ${chalk.cyan('│')}
+${chalk.cyan('│')} ${chalk.cyan('╚═════════════════════════════════════════════════════════════╝')} ${chalk.cyan('│')}
+${chalk.cyan('└─────────────────────────────────────────────────────────────────└')}
+
+${chalk.cyan.bold('🚀 Ready to validate your AI-generated code? 🚀')}
+
+${chalk.dim('✨ Scan • Fix • Test • Generate • Loop')}
+${chalk.dim('')}
+
+`;
+
+const program = new Command();
+
+program
+  .name('botintern')
+  .description(chalk.gray('Advanced CLI to validate and fix AI-generated code with intelligent testing.'))
+  .version('1.0.0')
+  .configureOutput({
+    writeErr: (str) => process.stderr.write(chalk.red(str)),
+    writeOut: (str) => process.stdout.write(str)
+  });
+
+program
+  .command('login')
+  .description('Authenticate BotIntern with your Gemini API Key')
+  .action(() => {
+    const rl = readline.createInterface({
+      input: process.stdin,
+      output: process.stdout
+    });
+
+    console.log(chalk.cyan('🤖 BotIntern Login'));
+    console.log(chalk.dim('Get your free key at: https://aistudio.google.com/app/apikey'));
+
+    rl.question('\n🔑 Paste your Gemini API Key: ', (key) => {
+      if (!key || key.trim().length === 0) {
+        console.log(chalk.red('❌ Invalid key.'));
+        rl.close();
+        return;
+      }
+
+      saveApiKey(key.trim());
+      console.log(chalk.green('\n✨ Success! Key saved globally. You can now use "botintern test" and "botintern loop" commands.'));
+      rl.close();
+    });
+  });
+
+program.command('scan')
+  .description('🔍 Scan a codebase for errors')
+  .action(() => {
+    console.log(chalk.cyan('┌─ ') + chalk.bold.white('SCAN MODE') + chalk.cyan(' ───────────────────────────'));
+    console.log(chalk.cyan('│') + ' ' + chalk.blue('🔍 Initializing AI code scanner...'));
+    console.log(chalk.cyan('│') + ' ' + chalk.yellow('⚡ Analyzing codebase for potential issues...'));
+    console.log(chalk.cyan('└─') + chalk.cyan('─'.repeat(45)));
+
+    const spinner = ora({
+      text: chalk.yellow('🔍 Scanning for AI hallucinations...'),
+      spinner: 'dots',
+      color: 'yellow'
+    }).start();
+
+    exec.exec("npm run build", function (error, stdout, stderr) {
+      if (error) {
+        spinner.fail({
+          text: chalk.red('❌ BUILD FAILED'),
+          symbol: '✖'
+        });
+        console.log('');
+        console.log(chalk.red('┌─ ') + chalk.bold.white('ERROR DETAILS') + chalk.red(' ──────────────────────'));
+        console.log(chalk.red('│') + ' ' + chalk.gray('Build output detected issues:'));
+        console.log(chalk.red('│') + chalk.dim('├─ STDOUT:'));
+        console.log(chalk.red('│') + chalk.dim('│ ' + stdout.trim()));
+        console.log(chalk.red('│') + chalk.dim('├─ STDERR:'));
+        console.log(chalk.red('│') + chalk.dim('│ ' + stderr.trim()));
+        console.log(chalk.red('└─') + chalk.red('─'.repeat(45)));
+      } else {
+        spinner.succeed({
+          text: chalk.green('✅ BUILD PASSED'),
+          symbol: '✔'
+        });
+        console.log(chalk.green('┌─ ') + chalk.white('SCAN RESULTS') + chalk.green(' ───────────────────────'));
+        console.log(chalk.green('│') + ' ' + chalk.white('✨ No errors detected! Code looks clean.'));
+        console.log(chalk.green('│') + ' ' + chalk.gray('🎯 AI code quality: EXCELLENT'));
+        console.log(chalk.green('└─') + chalk.green('─'.repeat(45)));
+      }
+    })
+  });
+
+
+program.command('fix')
+  .description('🔧 Automatically fix errors in a codebase')
+  .action(() => {
+    console.log(chalk.magenta('┌─ ') + chalk.bold.white('AUTO-FIX MODE') + chalk.magenta(' ──────────────────────'));
+    console.log(chalk.magenta('│') + ' ' + chalk.blue('🔧 Initializing intelligent code repair...'));
+    console.log(chalk.magenta('│') + ' ' + chalk.yellow('⚡ AI-powered error fixing enabled...'));
+    console.log(chalk.magenta('└─') + chalk.magenta('─'.repeat(45)));
+
+    const spinner = ora({
+      text: chalk.yellow('🔍 Detecting and analyzing build errors...'),
+      spinner: 'line',
+      color: 'yellow'
+    }).start();
+
+    exec.exec("npm run build", async function (error, stdout, stderr) {
+      if (error) {
+        spinner.fail({
+          text: chalk.red('❌ BUILD ERRORS FOUND'),
+          symbol: '✖'
+        });
+        console.log('');
+        console.log(chalk.red('┌─ ') + chalk.bold.white('ERROR ANALYSIS') + chalk.red(' ──────────────────────'));
+        console.log(chalk.red('│') + ' ' + chalk.gray('🔍 Processing build output...'));
+        console.log(chalk.red('│') + chalk.dim('├─ BUILD LOGS:'));
+        console.log(chalk.red('│') + chalk.dim('│ ' + (stdout + stderr).trim()));
+
+        const combinedLogs = stdout + "\n" + stderr;
+        let brokenFilePath = extractErrorFile(combinedLogs);
+
+        if (!brokenFilePath) {
+          console.log(chalk.yellow('│') + chalk.dim('├─ ') + chalk.yellow('⚠️  No specific file detected, using default: app/page.tsx'));
+          brokenFilePath = "app/page.tsx";
+        } else {
+          console.log(chalk.blue('│') + chalk.dim('├─ ') + chalk.blue(`🎯 Target identified: ${chalk.bold(brokenFilePath)}`));
+        }
+
+        console.log(chalk.magenta('│') + chalk.dim('└─ ') + chalk.magenta('🤖 AI attempting fix...'));
+        console.log(chalk.magenta('└─') + chalk.magenta('─'.repeat(45)));
+
+        try {
+          const absolutePath = path.resolve(process.cwd(), brokenFilePath);
+
+          if (fs.existsSync(absolutePath)) {
+            const fixSpinner = ora({
+              text: chalk.cyan('🧠 AI generating solution...'),
+              spinner: 'bouncingBar',
+              color: 'cyan'
+            }).start();
+
+            const pageContent = fs.readFileSync(absolutePath, "utf-8");
+            const packageContent = fs.readFileSync(path.join(process.cwd(), "package.json"), "utf-8");
+
+            const finalCode = await fixBuild([combinedLogs, pageContent, packageContent]);
+            fixSpinner.succeed({
+              text: chalk.green('✅ SOLUTION GENERATED'),
+              symbol: '✨'
+            });
+
+            const applySpinner = ora({
+              text: chalk.blue('📝 Applying fixes...'),
+              spinner: 'dots',
+              color: 'blue'
+            }).start();
+
+            fs.writeFileSync(absolutePath, finalCode);
+            applySpinner.succeed({
+              text: chalk.green('✨ FIXES APPLIED SUCCESSFULLY'),
+              symbol: '✔'
+            });
+
+            console.log(chalk.green('┌─ ') + chalk.white('FIX SUMMARY') + chalk.green(' ───────────────────────'));
+            console.log(chalk.green('│') + ' ' + chalk.white(`📁 File: ${chalk.gray(brokenFilePath)}`));
+            console.log(chalk.green('│') + ' ' + chalk.white('🔧 Status: ') + chalk.green('FIXED'));
+            console.log(chalk.green('│') + ' ' + chalk.gray('💡 Run "botintern scan" to verify fixes'));
+            console.log(chalk.green('└─') + chalk.green('─'.repeat(45)));
+          } else {
+            console.log(chalk.red(`❌ File not found: ${brokenFilePath}`));
+          }
+        } catch (err) {
+          console.log(chalk.red(`💥 Fix failed: ${err.message}`));
+        }
+
+      } else {
+        spinner.succeed({
+          text: chalk.green('✅ NO ERRORS TO FIX'),
+          symbol: '✨'
+        });
+        console.log(chalk.green('┌─ ') + chalk.white('CODE STATUS') + chalk.green(' ───────────────────────'));
+        console.log(chalk.green('│') + ' ' + chalk.white('🎉 Codebase is already error-free!'));
+        console.log(chalk.green('│') + ' ' + chalk.gray('💎 Quality: PERFECT'));
+        console.log(chalk.green('└─') + chalk.green('─'.repeat(45)));
+      }
+    })
+  });
+
+
+
+program.command("test")
+  .description("🧪 Run comprehensive playwright tests")
+  .action(async () => {
+    console.log(chalk.blue('┌─ ') + chalk.bold.white('TEST MODE') + chalk.blue(' ────────────────────────'));
+    console.log(chalk.blue('│') + ' ' + chalk.cyan('🧪 Initializing test environment...'));
+    console.log(chalk.blue('│') + ' ' + chalk.yellow('⚡ Launching browser automation...'));
+    console.log(chalk.blue('└─') + chalk.blue('─'.repeat(45)));
+
+    const spinner = ora({
+      text: chalk.cyan('🚀 Setting up test browser...'),
+      spinner: 'toggle',
+      color: 'cyan'
+    }).start();
+
+    try {
+      const { browser, page } = await runPlaywright();
+      spinner.succeed({
+        text: chalk.green('🌐 Browser ready'),
+        symbol: '✨'
+      });
+
+      const testSpinner = ora({
+        text: chalk.yellow('🧪 Running automated tests...'),
+        spinner: 'dots2',
+        color: 'yellow'
+      }).start();
+
+      const testResult = await runTests(page);
+      testSpinner.stop();
+
+      const cleanupSpinner = ora({
+        text: chalk.blue('🧹 Cleaning up...'),
+        spinner: 'pipe',
+        color: 'blue'
+      }).start();
+
+      await browser.close();
+      cleanupSpinner.succeed({
+        text: chalk.green('✨ Test session complete'),
+        symbol: '✔'
+      });
+
+      // Display results based on actual test outcome
+      if (testResult.success) {
+        console.log(chalk.green('┌─ ') + chalk.white('TEST RESULTS') + chalk.green(' ──────────────────────'));
+        console.log(chalk.green('│') + ' ' + chalk.white('🎯 Status: ') + chalk.green('PASSED'));
+        console.log(chalk.green('│') + ' ' + chalk.white('📊 Tests: ') + chalk.gray(`${testResult.passedTests}/${testResult.totalTests} passed`));
+        console.log(chalk.green('│') + ' ' + chalk.gray('💡 All checks passed successfully'));
+        console.log(chalk.green('└─') + chalk.green('─'.repeat(45)));
+      } else {
+        console.log(chalk.red('┌─ ') + chalk.white('TEST RESULTS') + chalk.red(' ──────────────────────'));
+        console.log(chalk.red('│') + ' ' + chalk.white('🎯 Status: ') + chalk.red('FAILED'));
+        if (testResult.totalTests) {
+          console.log(chalk.red('│') + ' ' + chalk.white('📊 Tests: ') + chalk.yellow(`${testResult.passedTests}/${testResult.totalTests} passed`));
+          console.log(chalk.red('│') + ' ' + chalk.white('❌ Failures: ') + chalk.red(`${testResult.failures?.length || 0}`));
+        } else if (testResult.error) {
+          console.log(chalk.red('│') + ' ' + chalk.white('❌ Error: ') + chalk.red(testResult.error));
+        }
+        console.log(chalk.red('└─') + chalk.red('─'.repeat(45)));
+        process.exit(1);
+      }
+    } catch (error) {
+      spinner.fail({
+        text: chalk.red('❌ Test execution failed'),
+        symbol: '✖'
+      });
+      console.log(chalk.red(`💥 Error: ${error.message}`));
+    }
+  });
+
+program.command("generate-yaml")
+  .description("📝 Generate intelligent YAML test cases")
+  .action(async () => {
+    console.log(chalk.yellow('┌─ ') + chalk.bold.white('YAML GENERATOR') + chalk.yellow(' ───────────────────'));
+    console.log(chalk.yellow('│') + ' ' + chalk.cyan('📝 Creating intelligent test configurations...'));
+    console.log(chalk.yellow('│') + ' ' + chalk.gray('🔍 Analyzing codebase structure...'));
+    console.log(chalk.yellow('└─') + chalk.yellow('─'.repeat(45)));
+
+    const analyzeSpinner = ora({
+      text: chalk.blue('🔍 Scanning project structure...'),
+      spinner: 'dots12',
+      color: 'blue'
+    }).start();
+
+    const cwd = process.cwd();
+    const fileTree = getProjectStructure(cwd);
+    const sourceCode = getCriticalSourceCode(cwd);
+
+    analyzeSpinner.succeed({
+      text: chalk.green('📊 Analysis complete'),
+      symbol: '✨'
+    });
+
+    const yamlSpinner = ora({
+      text: chalk.magenta('🤖 AI generating YAML configurations...'),
+      spinner: 'growHorizontal',
+      color: 'magenta'
+    }).start();
+
+    const yamlPath = `${cwd}/vibe.yaml`;
+    let currentYaml = "";
+    if (fs.existsSync(yamlPath)) {
+      currentYaml = fs.readFileSync(yamlPath, 'utf-8');
+    }
+
+    const yaml = await generateYaml([fileTree, sourceCode, currentYaml, ""]);
+    fs.writeFileSync(yamlPath, yaml);
+
+    yamlSpinner.succeed({
+      text: chalk.green('✨ YAML test cases generated'),
+      symbol: '📋'
+    });
+
+    console.log(chalk.green('┌─ ') + chalk.white('GENERATION SUMMARY') + chalk.green(' ───────────────'));
+    console.log(chalk.green('│') + ' ' + chalk.white('📁 Output: ') + chalk.gray('vibe.yaml'));
+    console.log(chalk.green('│') + ' ' + chalk.white('🧪 Test cases: ') + chalk.blue('Generated'));
+    console.log(chalk.green('│') + ' ' + chalk.gray('💡 Ready for automated testing'));
+    console.log(chalk.green('└─') + chalk.green('─'.repeat(45)));
+  });
+
+program.command("loop")
+  .description("🔄 AI-powered iterative development loop")
+  .argument('<string>', 'Custom prompt for AI generation')
+  .option("-p", "Prompt mode")
+  .action(async (prompt, options) => {
+    console.log(chalk.cyan('┌─ ') + chalk.bold.white('AI LOOP MODE') + chalk.cyan(' ───────────────────'));
+    console.log(chalk.cyan('│') + ' ' + chalk.blue('🔄 Initializing iterative development...'));
+    console.log(chalk.cyan('│') + ' ' + chalk.yellow('🤖 AI agent ready for continuous improvement...'));
+    console.log(chalk.cyan('└─') + chalk.cyan('─'.repeat(45)));
+
+    if (prompt) {
+      console.log(chalk.blue('┌─ ') + chalk.white('CUSTOM PROMPT') + chalk.blue(' ──────────────────────'));
+      console.log(chalk.blue('│') + ' ' + chalk.gray(`📝 ${prompt}`));
+      console.log(chalk.blue('└─') + chalk.blue('─'.repeat(45)));
+    }
+
+    const setupSpinner = ora({
+      text: chalk.cyan('🚀 Setting up development environment...'),
+      spinner: 'bouncingBall',
+      color: 'cyan'
+    }).start();
+
+    try {
+      const { browser, page } = await runPlaywright();
+      setupSpinner.succeed({
+        text: chalk.green('🌐 Environment ready'),
+        symbol: '✨'
+      });
+
+      const testSpinner = ora({
+        text: chalk.yellow('🧪 Running initial tests...'),
+        spinner: 'flip',
+        color: 'yellow'
+      }).start();
+
+      const testResult = await runTests(page);
+      testSpinner.succeed({
+        text: testResult.success ? chalk.green('✅ Tests passed') : chalk.yellow('⚠️  Tests need improvement'),
+        symbol: testResult.success ? '🎉' : '🔄'
+      });
+
+      if (testResult.success && !prompt) {
+        console.log(chalk.green('┌─ ') + chalk.white('LOOP STATUS') + chalk.green(' ──────────────────────'));
+        console.log(chalk.green('│') + ' ' + chalk.white('🎉 All tests passed!'));
+        console.log(chalk.green('│') + ' ' + chalk.gray('✨ No iterations needed'));
+        console.log(chalk.green('└─') + chalk.green('─'.repeat(45)));
+        process.exit(0);
+      } else {
+        const loopSpinner = ora({
+          text: chalk.magenta('🤖 AI improvement loop starting...'),
+          spinner: 'toggle8',
+          color: 'magenta'
+        }).start();
+
+        if (prompt) {
+          await loop(page, prompt);
+        } else {
+          await loop(page);
+        }
+
+        loopSpinner.succeed({
+          text: chalk.green('🎯 Loop completed'),
+          symbol: '✨'
+        });
+      }
+
+      await browser.close();
+    } catch (error) {
+      setupSpinner.fail({
+        text: chalk.red('❌ Loop initialization failed'),
+        symbol: '✖'
+      });
+      console.log(chalk.red(`💥 Error: ${error.message}`));
+    }
+  });
+
+// Show welcome banner and help when no arguments provided
+if (process.argv.length <= 2) {
+  console.log(banner);
+  console.log(chalk.cyan.bold('📖 Getting Started:'));
+  console.log(chalk.gray('  botintern scan') + chalk.dim('     • Scan codebase for errors'));
+  console.log(chalk.gray('  botintern fix') + chalk.dim('      • Auto-fix detected errors'));
+  console.log(chalk.gray('  botintern test') + chalk.dim('     • Run automated tests'));
+  console.log(chalk.gray('  botintern generate-yaml') + chalk.dim(' • Generate test configurations'));
+  console.log(chalk.gray('  botintern loop <prompt>') + chalk.dim(' • AI-powered iterative improvement'));
+  console.log('');
+  console.log(chalk.blue.bold('🤖 Need help?') + chalk.dim(' Run ' + chalk.gray('botintern --help')));
+  console.log(chalk.dim('─'.repeat(67)));
+  process.exit(0);
+}
+
+program.parse();