@djangocfg/imgai 2.1.42 → 2.1.44

package/dist/cli/index.js

@@ -1,257 +1,14 @@
 import chalk from 'chalk';
 import inquirer from 'inquirer';
 import ora from 'ora';
-import fs from 'fs-extra';
-import path2 from 'path';
-import sharp from 'sharp';
+import fs3 from 'fs-extra';
+import path from 'path';
+import { glob } from 'glob';
+import sharp2 from 'sharp';
 import OpenAI from 'openai';
 import Anthropic from '@anthropic-ai/sdk';
-import { glob } from 'glob';
 
 // src/cli/index.ts
-var ImageGenerator = class {
-  config;
-  openai;
-  anthropic;
-  constructor(config) {
-    this.config = config;
-    this.initializeProviders();
-  }
-  initializeProviders() {
-    const openaiKey = this.config.openaiApiKey || process.env.OPENAI_API_KEY;
-    const anthropicKey = this.config.anthropicApiKey || process.env.ANTHROPIC_API_KEY;
-    if (openaiKey) {
-      this.openai = new OpenAI({ apiKey: openaiKey });
-    }
-    if (anthropicKey) {
-      this.anthropic = new Anthropic({ apiKey: anthropicKey });
-    }
-    if (!this.openai && this.config.provider === "openai") {
-      throw new Error("OpenAI API key required. Set OPENAI_API_KEY or pass openaiApiKey in config.");
-    }
-  }
-  // ──────────────────────────────────────────────────────────────────────────
-  // SINGLE IMAGE GENERATION
-  // ──────────────────────────────────────────────────────────────────────────
-  async generate(options) {
-    const startTime = Date.now();
-    try {
-      const fullPrompt = this.buildPrompt(options.prompt);
-      const imageData = await this.generateWithOpenAI(fullPrompt);
-      const filename = options.filename || this.generateFilename(options.prompt);
-      const category = options.category || "general";
-      const outputDir = path2.join(
-        this.config.projectRoot,
-        this.config.outputDir,
-        category
-      );
-      await fs.ensureDir(outputDir);
-      const originalPath = path2.join(outputDir, `${filename}.png`);
-      await fs.writeFile(originalPath, Buffer.from(imageData, "base64"));
-      const resizeOptions = options.resize || this.config.resize;
-      let finalPath = originalPath;
-      if (resizeOptions) {
-        finalPath = await this.resizeImage(originalPath, resizeOptions);
-        if (finalPath !== originalPath) {
-          await fs.remove(originalPath);
-        }
-      }
-      const imageInfo = await this.getImageInfo(finalPath, category);
-      imageInfo.metadata = {
-        prompt: options.prompt,
-        caption: options.metadata?.caption,
-        tags: options.tags,
-        category,
-        generatedAt: /* @__PURE__ */ new Date(),
-        model: "dall-e-3"
-      };
-      const duration = Date.now() - startTime;
-      return {
-        success: true,
-        imagePath: finalPath,
-        imageUrl: imageInfo.url,
-        imageInfo,
-        duration
-      };
-    } catch (error) {
-      return {
-        success: false,
-        error: error instanceof Error ? error.message : "Unknown error",
-        duration: Date.now() - startTime
-      };
-    }
-  }
-  // ──────────────────────────────────────────────────────────────────────────
-  // BATCH GENERATION
-  // ──────────────────────────────────────────────────────────────────────────
-  async generateBatch(options) {
-    const startTime = Date.now();
-    const results = [];
-    const concurrency = options.concurrency || 2;
-    const delayBetween = options.delayBetween || 2e3;
-    for (let i = 0; i < options.items.length; i += concurrency) {
-      const batch = options.items.slice(i, i + concurrency);
-      const batchPromises = batch.map(async (item) => {
-        const result = await this.generate({
-          prompt: item.prompt,
-          filename: item.filename,
-          category: item.category
-        });
-        return result;
-      });
-      const batchResults = await Promise.all(batchPromises);
-      for (const result of batchResults) {
-        results.push(result);
-        if (options.onProgress) {
-          options.onProgress(results.length, options.items.length, result);
-        }
-      }
-      if (i + concurrency < options.items.length) {
-        await this.delay(delayBetween);
-      }
-    }
-    return {
-      total: results.length,
-      success: results.filter((r) => r.success).length,
-      failed: results.filter((r) => !r.success).length,
-      results,
-      duration: Date.now() - startTime
-    };
-  }
-  // ──────────────────────────────────────────────────────────────────────────
-  // PROMPT ENHANCEMENT (using Claude/GPT)
-  // ──────────────────────────────────────────────────────────────────────────
-  async enhancePrompt(basePrompt, context) {
-    const systemPrompt = `You are an expert at creating image generation prompts.
-Given a description, create:
-1. A detailed visual prompt (max 150 words) for DALL-E 3
-2. A descriptive filename (lowercase, hyphens, max 30 chars)
-3. A short caption (max 50 words)
-
-${this.config.prefix ? `Style prefix: ${this.config.prefix}` : ""}
-${this.config.suffix ? `Style suffix: ${this.config.suffix}` : ""}
-
-Return ONLY valid JSON: {"prompt": "...", "filename": "...", "caption": "..."}`;
-    const userPrompt = context ? `Description: ${basePrompt}
-Context: ${context}` : `Description: ${basePrompt}`;
-    if (this.anthropic && this.config.provider === "anthropic") {
-      const response = await this.anthropic.messages.create({
-        model: "claude-3-5-sonnet-20241022",
-        max_tokens: 500,
-        messages: [
-          { role: "user", content: `${systemPrompt}
-
-${userPrompt}` }
-        ]
-      });
-      const text = response.content[0].type === "text" ? response.content[0].text : "";
-      return JSON.parse(text);
-    }
-    if (this.openai) {
-      const response = await this.openai.chat.completions.create({
-        model: "gpt-4-turbo-preview",
-        messages: [
-          { role: "system", content: systemPrompt },
-          { role: "user", content: userPrompt }
-        ],
-        max_tokens: 500,
-        temperature: 0.7,
-        response_format: { type: "json_object" }
-      });
-      const text = response.choices[0]?.message?.content || "{}";
-      return JSON.parse(text);
-    }
-    throw new Error("No AI provider configured for prompt enhancement");
-  }
-  // ──────────────────────────────────────────────────────────────────────────
-  // PRIVATE METHODS
-  // ──────────────────────────────────────────────────────────────────────────
-  async generateWithOpenAI(prompt) {
-    if (!this.openai) {
-      throw new Error("OpenAI client not initialized");
-    }
-    const response = await this.openai.images.generate({
-      model: "dall-e-3",
-      prompt,
-      n: 1,
-      size: this.config.size,
-      quality: this.config.quality,
-      style: this.config.dalleStyle || "natural",
-      response_format: "b64_json"
-    });
-    const imageData = response.data?.[0]?.b64_json;
-    if (!imageData) {
-      throw new Error("No image data received from OpenAI");
-    }
-    return imageData;
-  }
-  async resizeImage(inputPath, options) {
-    const { width, height, quality = 85, format = "webp", fit = "inside" } = options;
-    let pipeline = sharp(inputPath);
-    if (width || height) {
-      pipeline = pipeline.resize(width, height, {
-        fit,
-        withoutEnlargement: true
-      });
-    }
-    const outputPath = inputPath.replace(/\.[^.]+$/, `.${format}`);
-    switch (format) {
-      case "webp":
-        pipeline = pipeline.webp({ quality });
-        break;
-      case "jpeg":
-        pipeline = pipeline.jpeg({ quality });
-        break;
-      case "png":
-        pipeline = pipeline.png({ quality });
-        break;
-      case "avif":
-        pipeline = pipeline.avif({ quality });
-        break;
-    }
-    await pipeline.toFile(outputPath);
-    return outputPath;
-  }
-  async getImageInfo(imagePath, category) {
-    const stats = await fs.stat(imagePath);
-    const metadata = await sharp(imagePath).metadata();
-    const filename = path2.basename(imagePath);
-    const extension = path2.extname(filename).slice(1);
-    const id = path2.basename(filename, path2.extname(filename));
-    const publicDir = path2.join(this.config.projectRoot, this.config.publicDir);
-    const relativePath = path2.relative(publicDir, imagePath);
-    const url = "/" + relativePath.replace(/\\/g, "/");
-    return {
-      id,
-      filename,
-      extension,
-      path: relativePath,
-      url,
-      size: stats.size,
-      width: metadata.width,
-      height: metadata.height,
-      createdAt: stats.birthtime,
-      modifiedAt: stats.mtime
-    };
-  }
-  buildPrompt(basePrompt) {
-    const parts = [];
-    if (this.config.prefix) {
-      parts.push(this.config.prefix);
-    }
-    parts.push(basePrompt);
-    if (this.config.suffix) {
-      parts.push(this.config.suffix);
-    }
-    return parts.join(" ");
-  }
-  generateFilename(prompt) {
-    return prompt.toLowerCase().replace(/[^a-z0-9\s]/g, "").replace(/\s+/g, "-").substring(0, 30);
-  }
-  delay(ms) {
-    return new Promise((resolve) => setTimeout(resolve, ms));
-  }
-};
 var DEFAULT_EXTENSIONS = ["png", "jpg", "jpeg", "webp", "avif", "gif", "svg"];
 var DEFAULT_DIRECTORIES = ["static/images", "images", "assets"];
 var ImageScanner = class {
@@ -270,15 +27,15 @@ var ImageScanner = class {
       includeDimensions = true,
       recursive = true
     } = options;
-    const publicDir = path2.join(this.config.projectRoot, this.config.publicDir);
+    const publicDir = path.join(this.config.projectRoot, this.config.publicDir);
     const images = [];
     const byCategory = {};
     const byExtension = {};
     let totalSize = 0;
     const patterns = directories.flatMap((dir) => {
-      const basePath = path2.join(publicDir, dir);
+      const basePath = path.join(publicDir, dir);
       const extPattern = `*.{${extensions.join(",")}}`;
-      return recursive ? [path2.join(basePath, "**", extPattern)] : [path2.join(basePath, extPattern)];
+      return recursive ? [path.join(basePath, "**", extPattern)] : [path.join(basePath, extPattern)];
     });
     for (const pattern of patterns) {
       const files = await glob(pattern, { nodir: true });
@@ -345,17 +102,17 @@ var ImageScanner = class {
   // PRIVATE METHODS
   // ──────────────────────────────────────────────────────────────────────────
   async processImage(filePath, publicDir, includeDimensions) {
-    const stats = await fs.stat(filePath);
-    const filename = path2.basename(filePath);
-    const extension = path2.extname(filename).slice(1).toLowerCase();
-    const id = path2.basename(filename, path2.extname(filename));
-    const relativePath = path2.relative(publicDir, filePath).replace(/\\/g, "/");
+    const stats = await fs3.stat(filePath);
+    const filename = path.basename(filePath);
+    const extension = path.extname(filename).slice(1).toLowerCase();
+    const id = path.basename(filename, path.extname(filename));
+    const relativePath = path.relative(publicDir, filePath).replace(/\\/g, "/");
     const url = "/" + relativePath;
     let width;
     let height;
     if (includeDimensions && extension !== "svg") {
       try {
-        const metadata = await sharp(filePath).metadata();
+        const metadata = await sharp2(filePath).metadata();
         width = metadata.width;
         height = metadata.height;
       } catch {
@@ -393,6 +150,8 @@ function formatDuration(ms) {
   if (ms < 1e3) return `${ms}ms`;
   return `${(ms / 1e3).toFixed(2)}s`;
 }
+
+// src/config/generator.ts
 var ConfigGenerator = class {
   config;
   scanner;
@@ -405,10 +164,10 @@ var ConfigGenerator = class {
   // ──────────────────────────────────────────────────────────────────────────
   async generate() {
     const catalog = await this.scanner.buildCatalog();
-    const outputPath = path2.join(this.config.projectRoot, this.config.configOutputPath);
-    await fs.ensureDir(path2.dirname(outputPath));
+    const outputPath = path.join(this.config.projectRoot, this.config.configOutputPath);
+    await fs3.ensureDir(path.dirname(outputPath));
     const content = this.generateTypeScript(catalog);
-    await fs.writeFile(outputPath, content, "utf-8");
+    await fs3.writeFile(outputPath, content, "utf-8");
     return outputPath;
   }
   // ──────────────────────────────────────────────────────────────────────────
@@ -524,6 +283,249 @@ export const IMAGE_CATALOG_META = {
     return `  '${category}': [${imageRefs.join(", ")}]`;
   }
 };
+var ImageGenerator = class {
+  config;
+  openai;
+  anthropic;
+  constructor(config) {
+    this.config = config;
+    this.initializeProviders();
+  }
+  initializeProviders() {
+    const openaiKey = this.config.openaiApiKey || process.env.OPENAI_API_KEY;
+    const anthropicKey = this.config.anthropicApiKey || process.env.ANTHROPIC_API_KEY;
+    if (openaiKey) {
+      this.openai = new OpenAI({ apiKey: openaiKey });
+    }
+    if (anthropicKey) {
+      this.anthropic = new Anthropic({ apiKey: anthropicKey });
+    }
+    if (!this.openai && this.config.provider === "openai") {
+      throw new Error("OpenAI API key required. Set OPENAI_API_KEY or pass openaiApiKey in config.");
+    }
+  }
+  // ──────────────────────────────────────────────────────────────────────────
+  // SINGLE IMAGE GENERATION
+  // ──────────────────────────────────────────────────────────────────────────
+  async generate(options) {
+    const startTime = Date.now();
+    try {
+      const fullPrompt = this.buildPrompt(options.prompt);
+      const imageData = await this.generateWithOpenAI(fullPrompt);
+      const filename = options.filename || this.generateFilename(options.prompt);
+      const category = options.category || "general";
+      const outputDir = path.join(
+        this.config.projectRoot,
+        this.config.outputDir,
+        category
+      );
+      await fs3.ensureDir(outputDir);
+      const originalPath = path.join(outputDir, `${filename}.png`);
+      await fs3.writeFile(originalPath, Buffer.from(imageData, "base64"));
+      const resizeOptions = options.resize || this.config.resize;
+      let finalPath = originalPath;
+      if (resizeOptions) {
+        finalPath = await this.resizeImage(originalPath, resizeOptions);
+        if (finalPath !== originalPath) {
+          await fs3.remove(originalPath);
+        }
+      }
+      const imageInfo = await this.getImageInfo(finalPath, category);
+      imageInfo.metadata = {
+        prompt: options.prompt,
+        caption: options.metadata?.caption,
+        tags: options.tags,
+        category,
+        generatedAt: /* @__PURE__ */ new Date(),
+        model: "dall-e-3"
+      };
+      const duration = Date.now() - startTime;
+      return {
+        success: true,
+        imagePath: finalPath,
+        imageUrl: imageInfo.url,
+        imageInfo,
+        duration
+      };
+    } catch (error) {
+      return {
+        success: false,
+        error: error instanceof Error ? error.message : "Unknown error",
+        duration: Date.now() - startTime
+      };
+    }
+  }
+  // ──────────────────────────────────────────────────────────────────────────
+  // BATCH GENERATION
+  // ──────────────────────────────────────────────────────────────────────────
+  async generateBatch(options) {
+    const startTime = Date.now();
+    const results = [];
+    const concurrency = options.concurrency || 2;
+    const delayBetween = options.delayBetween || 2e3;
+    for (let i = 0; i < options.items.length; i += concurrency) {
+      const batch = options.items.slice(i, i + concurrency);
+      const batchPromises = batch.map(async (item) => {
+        const result = await this.generate({
+          prompt: item.prompt,
+          filename: item.filename,
+          category: item.category
+        });
+        return result;
+      });
+      const batchResults = await Promise.all(batchPromises);
+      for (const result of batchResults) {
+        results.push(result);
+        if (options.onProgress) {
+          options.onProgress(results.length, options.items.length, result);
+        }
+      }
+      if (i + concurrency < options.items.length) {
+        await this.delay(delayBetween);
+      }
+    }
+    return {
+      total: results.length,
+      success: results.filter((r) => r.success).length,
+      failed: results.filter((r) => !r.success).length,
+      results,
+      duration: Date.now() - startTime
+    };
+  }
+  // ──────────────────────────────────────────────────────────────────────────
+  // PROMPT ENHANCEMENT (using Claude/GPT)
+  // ──────────────────────────────────────────────────────────────────────────
+  async enhancePrompt(basePrompt, context) {
+    const systemPrompt = `You are an expert at creating image generation prompts.
+Given a description, create:
+1. A detailed visual prompt (max 150 words) for DALL-E 3
+2. A descriptive filename (lowercase, hyphens, max 30 chars)
+3. A short caption (max 50 words)
+
+${this.config.prefix ? `Style prefix: ${this.config.prefix}` : ""}
+${this.config.suffix ? `Style suffix: ${this.config.suffix}` : ""}
+
+Return ONLY valid JSON: {"prompt": "...", "filename": "...", "caption": "..."}`;
+    const userPrompt = context ? `Description: ${basePrompt}
+Context: ${context}` : `Description: ${basePrompt}`;
+    if (this.anthropic && this.config.provider === "anthropic") {
+      const response = await this.anthropic.messages.create({
+        model: "claude-3-5-sonnet-20241022",
+        max_tokens: 500,
+        messages: [
+          { role: "user", content: `${systemPrompt}
+
+${userPrompt}` }
+        ]
+      });
+      const text = response.content[0].type === "text" ? response.content[0].text : "";
+      return JSON.parse(text);
+    }
+    if (this.openai) {
+      const response = await this.openai.chat.completions.create({
+        model: "gpt-4-turbo-preview",
+        messages: [
+          { role: "system", content: systemPrompt },
+          { role: "user", content: userPrompt }
+        ],
+        max_tokens: 500,
+        temperature: 0.7,
+        response_format: { type: "json_object" }
+      });
+      const text = response.choices[0]?.message?.content || "{}";
+      return JSON.parse(text);
+    }
+    throw new Error("No AI provider configured for prompt enhancement");
+  }
+  // ──────────────────────────────────────────────────────────────────────────
+  // PRIVATE METHODS
+  // ──────────────────────────────────────────────────────────────────────────
+  async generateWithOpenAI(prompt) {
+    if (!this.openai) {
+      throw new Error("OpenAI client not initialized");
+    }
+    const response = await this.openai.images.generate({
+      model: "dall-e-3",
+      prompt,
+      n: 1,
+      size: this.config.size,
+      quality: this.config.quality,
+      style: this.config.dalleStyle || "natural",
+      response_format: "b64_json"
+    });
+    const imageData = response.data?.[0]?.b64_json;
+    if (!imageData) {
+      throw new Error("No image data received from OpenAI");
+    }
+    return imageData;
+  }
+  async resizeImage(inputPath, options) {
+    const { width, height, quality = 85, format = "webp", fit = "inside" } = options;
+    let pipeline = sharp2(inputPath);
+    if (width || height) {
+      pipeline = pipeline.resize(width, height, {
+        fit,
+        withoutEnlargement: true
+      });
+    }
+    const outputPath = inputPath.replace(/\.[^.]+$/, `.${format}`);
+    switch (format) {
+      case "webp":
+        pipeline = pipeline.webp({ quality });
+        break;
+      case "jpeg":
+        pipeline = pipeline.jpeg({ quality });
+        break;
+      case "png":
+        pipeline = pipeline.png({ quality });
+        break;
+      case "avif":
+        pipeline = pipeline.avif({ quality });
+        break;
+    }
+    await pipeline.toFile(outputPath);
+    return outputPath;
+  }
+  async getImageInfo(imagePath, category) {
+    const stats = await fs3.stat(imagePath);
+    const metadata = await sharp2(imagePath).metadata();
+    const filename = path.basename(imagePath);
+    const extension = path.extname(filename).slice(1);
+    const id = path.basename(filename, path.extname(filename));
+    const publicDir = path.join(this.config.projectRoot, this.config.publicDir);
+    const relativePath = path.relative(publicDir, imagePath);
+    const url = "/" + relativePath.replace(/\\/g, "/");
+    return {
+      id,
+      filename,
+      extension,
+      path: relativePath,
+      url,
+      size: stats.size,
+      width: metadata.width,
+      height: metadata.height,
+      createdAt: stats.birthtime,
+      modifiedAt: stats.mtime
+    };
+  }
+  buildPrompt(basePrompt) {
+    const parts = [];
+    if (this.config.prefix) {
+      parts.push(this.config.prefix);
+    }
+    parts.push(basePrompt);
+    if (this.config.suffix) {
+      parts.push(this.config.suffix);
+    }
+    return parts.join(" ");
+  }
+  generateFilename(prompt) {
+    return prompt.toLowerCase().replace(/[^a-z0-9\s]/g, "").replace(/\s+/g, "-").substring(0, 30);
+  }
+  delay(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+  }
+};
 
 // src/cli/index.ts
 var ImageAICLI = class {