feat: add gemini api support

Author: mukulboro

.gitignore

@@ -33,4 +33,5 @@ build/
 
 # Optional: lock files (if you want to avoid committing them)
 # package-lock.json
-# yarn.lock
+# yarn.lock
+bun.lock

.npmignore

@@ -64,3 +64,5 @@ build/
 # Temporary folders
 tmp/
 temp/
+
+bun.lock

src/ai/ai-analyzer.js

@@ -9,35 +9,52 @@ dotenv.config();
  * AI-Powered Website Analysis using OpenAI GPT-4o
  */
 export class AIAnalyzer {
-  constructor() {
+  constructor(aiModel = 'gpt-4o') {
     this.openai = null;
     this.isEnabled = false;
+    this.aiModel = aiModel;
     this.initializeOpenAI();
   }
 
   initializeOpenAI() {
-    const apiKey = process.env.OPENAI_API_KEY;
+    const isGemini = this.aiModel.toLowerCase().includes('gemini');
+    let apiKey = isGemini ? process.env.GEMINI_API_KEY : process.env.OPENAI_API_KEY;
 
+    // Fallback if specific key is missing
     if (!apiKey) {
+      apiKey = process.env.OPENAI_API_KEY || process.env.GEMINI_API_KEY;
+    }
+
+    if (!apiKey) {
+      const keyName = isGemini ? 'GEMINI_API_KEY' : 'OPENAI_API_KEY';
       console.log(
         chalk.yellow(
-          '⚠️ OpenAI API key not found. AI analysis will be disabled.',
+          `⚠️ AI analysis API key not found. AI analysis will be disabled.`,
         ),
       );
       console.log(
         chalk.gray(
-          '   Set OPENAI_API_KEY environment variable to enable AI features.',
+          `   Set ${keyName} environment variable to enable AI features.`,
         ),
       );
       return;
     }
 
     try {
-      this.openai = new OpenAI({ apiKey });
+      const config = { apiKey };
+      
+      // Use Gemini OpenAI-compatible endpoint if model is gemini or GEMINI_API_KEY is used
+      if (isGemini || process.env.GEMINI_API_KEY === apiKey) {
+        config.baseURL = 'https://generativelanguage.googleapis.com/v1beta/openai/';
+        console.log(chalk.green(`✅ AI analysis enabled with Gemini model: ${this.aiModel}`));
+      } else {
+        console.log(chalk.green(`✅ AI analysis enabled with OpenAI model: ${this.aiModel}`));
+      }
+
+      this.openai = new OpenAI(config);
       this.isEnabled = true;
-      console.log(chalk.green('✅ AI analysis enabled with OpenAI GPT-4o'));
     } catch (error) {
-      console.log(chalk.red('❌ Failed to initialize OpenAI:'), error.message);
+      console.log(chalk.red('❌ Failed to initialize AI client:'), error.message);
     }
   }
 
@@ -72,7 +89,7 @@ export class AIAnalyzer {
     );
 
     const response = await this.openai.chat.completions.create({
-      model: 'gpt-4o',
+      model: this.aiModel,
       messages: [
         {
           role: 'system',
@@ -225,7 +242,7 @@ Provide actionable optimization steps in JSON format:
 }`;
 
       const response = await this.openai.chat.completions.create({
-        model: 'gpt-4o',
+        model: this.aiModel,
         messages: [
           {
             role: 'system',

src/cli.js

@@ -48,33 +48,48 @@ function loadEnvWithPriority() {
 loadEnvWithPriority();
 
 /**
- * Validate and configure OpenAI API key for AI features.
- * Uses OPENAI_API_KEY from environment (shell, .env.local, .env) or --openai-key if provided.
+ * Validate and configure AI API key for AI features.
+ * Priority: 
+ * 1. CLI flag (--openai-key)
+ * 2. GEMINI_API_KEY (if model is gemini-*)
+ * 3. OPENAI_API_KEY
  */
-function validateAISetup(aiEnabled, openaiApiKey) {
+function validateAISetup(aiEnabled, openaiApiKey, aiModel) {
   if (!aiEnabled) return false;
 
-  // CLI flag has highest priority
-  const finalApiKey = openaiApiKey || process.env.OPENAI_API_KEY;
+  const isGemini = aiModel.toLowerCase().includes('gemini');
+  
+  // CLI flag has highest priority (for OpenAI)
+  let finalApiKey = openaiApiKey;
+  
+  if (!finalApiKey) {
+    if (isGemini) {
+      finalApiKey = process.env.GEMINI_API_KEY || process.env.OPENAI_API_KEY;
+    } else {
+      finalApiKey = process.env.OPENAI_API_KEY;
+    }
+  }
 
   if (!finalApiKey) {
+    const keyName = isGemini ? 'GEMINI_API_KEY or OPENAI_API_KEY' : 'OPENAI_API_KEY';
     console.log('');
     console.log(
       chalk.yellow(
-        '⚠️  AI features requested but no OPENAI_API_KEY found (checked env, .env.local, and .env).',
+        `⚠️  AI features requested but no ${keyName} found (checked env, .env.local, and .env).`,
       ),
     );
     console.log(
       chalk.white(
-        'Add OPENAI_API_KEY to your .env.local or .env file, export it in your shell, or pass --openai-key "sk-..."',
+        'Add the API key to your .env.local or .env file, export it in your shell, or pass --openai-key "sk-..."',
       ),
     );
     console.log(chalk.dim('Continuing without AI features...'));
     console.log('');
     return false;
   }
 
-  if (!finalApiKey.startsWith('sk-')) {
+  // Basic format check (OpenAI starts with sk-, Gemini often doesn't but we'll be flexible)
+  if (!isGemini && !finalApiKey.startsWith('sk-')) {
     console.log('');
     console.log(
       chalk.yellow('⚠️  Invalid OpenAI API key format (must start with "sk-")'),
@@ -84,9 +99,11 @@ function validateAISetup(aiEnabled, openaiApiKey) {
     return false;
   }
 
-  // If provided via flag, ensure it’s in process.env for downstream code
+  // Ensure key is in process.env for downstream code if provided via flag or specific var
   if (openaiApiKey) {
     process.env.OPENAI_API_KEY = openaiApiKey;
+  } else if (isGemini && process.env.GEMINI_API_KEY) {
+    // Already in env
   }
 
   return true;
@@ -112,12 +129,17 @@ program
   )
   .option(
     '--ai',
-    'Enable AI-powered website analysis (reads OPENAI_API_KEY from env, .env.local, or .env)',
+    'Enable AI-powered website analysis (reads OPENAI_API_KEY or GEMINI_API_KEY from env)',
     false,
   )
+  .option(
+    '--ai-model <model>',
+    'AI model to use for analysis',
+    'gemini-3-flash-preview',
+  )
   .option(
     '--openai-key <key>',
-    'OpenAI API key for AI features (overrides env/.env.local/.env for this run)',
+    'OpenAI API key for AI features (overrides env for this run)',
   )
   .option('--debug', 'Enable detailed debug logging and error traces', false)
   .option(
@@ -135,12 +157,13 @@ program
       if (!url.startsWith('http')) url = 'https://' + url;
       new URL(url); // validate
 
-      const aiEnabled = validateAISetup(options.ai, options.openaiKey);
+      const aiEnabled = validateAISetup(options.ai, options.openaiKey, options.aiModel);
 
       const config = {
         outputDir: options.output,
         clean: options.clean,
         ai: aiEnabled,
+        aiModel: options.aiModel,
         debug: options.debug,
         timeout: parseInt(options.timeout),
         headless: options.headless !== 'false',

src/core/mirror-cloner.js

@@ -60,7 +60,7 @@ export class MirrorCloner {
     this.logger = new Logger(this.options);
 
     // Optional AI
-    this.aiAnalyzer = this.options.ai ? new AIAnalyzer() : null;
+    this.aiAnalyzer = this.options.ai ? new AIAnalyzer(this.options.aiModel) : null;
 
     // Data
     this.$ = null; // Cheerio DOM instance