From 5a854479f931416c2857d5b60d02ba18df034a38 Mon Sep 17 00:00:00 2001
From: Oleg Proskurin <regx@usul.su>
Date: Tue, 24 Feb 2026 00:21:54 +0700
Subject: [PATCH] feat: banatie script

---
 CLAUDE.md               | 17 +++++++---
 src/scripts/banatie.mjs | 74 +++++++++++++++++++++++++----------------
 2 files changed, 59 insertions(+), 32 deletions(-)

diff --git a/CLAUDE.md b/CLAUDE.md
index f5dfcd6..38c04fc 100644
--- a/CLAUDE.md
+++ b/CLAUDE.md
@@ -150,12 +150,21 @@ When generating HTML worksheets:
 
 ## Banatie API
 
-REST API for generating images.
+REST API at `https://api.banatie.app` for generating images. Auth via `X-API-Key` header (reads `BANATIE_KEY` from `.env`).
 
-- **Backgrounds:** ~1200×1700px, themed illustrations (forest, space, ocean, etc.)
-- **Icons:** 128×128px, transparent PNG, simple collectible items (stars, gems, animals)
+**POST `/api/v1/generations`** — create generation:
+- `prompt` (string, required) — image description
+- `aspectRatio` — `1:1`, `16:9`, `9:16`, `3:2`, `21:9` (default: `1:1`)
+- `referenceImages` (string[]) — image IDs as references
+- `autoEnhance` (boolean) — prompt enhancement (default: true)
 
-Configuration is in `src/scripts/banatie.mjs`. Set the `BANATIE_API_KEY` environment variable for authentication.
+Response returns JSON with `data.outputImage.storageUrl` (CDN URL to download).
+
+Script `src/scripts/banatie.mjs` wraps the API with presets:
+- `background` → `aspectRatio: "9:16"` (vertical, for page backgrounds)
+- `icon` → `aspectRatio: "1:1"` (square, for problem card icons)
+
+Rate limit: 100 requests/hour per API key.
 
 ## Background Removal
 
diff --git a/src/scripts/banatie.mjs b/src/scripts/banatie.mjs
index d154d32..ae954cb 100644
--- a/src/scripts/banatie.mjs
+++ b/src/scripts/banatie.mjs
@@ -1,58 +1,76 @@
-import { writeFileSync, mkdirSync } from 'fs';
+import { writeFileSync, mkdirSync, readFileSync } from 'fs';
 import { resolve, dirname } from 'path';
 
-const API_ENDPOINT = process.env.BANATIE_API_URL || 'https://api.banatie.com/v1/generate';
-const API_KEY = process.env.BANATIE_API_KEY || '';
+const envPath = resolve(dirname(new URL(import.meta.url).pathname), '../../.env');
+try {
+  const envContent = readFileSync(envPath, 'utf-8');
+  for (const line of envContent.split('\n')) {
+    const match = line.match(/^([^#=]+)=(.*)$/);
+    if (match && !process.env[match[1].trim()]) {
+      process.env[match[1].trim()] = match[2].trim();
+    }
+  }
+} catch {}
 
-const PRESETS = {
-  background: { width: 1200, height: 1700, format: 'png' },
-  icon: { width: 128, height: 128, format: 'png', transparent: true },
-};
 
-export async function generateImage({ type = 'icon', prompt, output }) {
+const API_BASE = 'https://api.banatie.app/api/v1';
+const API_KEY = process.env.BANATIE_KEY || '';
+
+export async function generateImage({ prompt, output, aspectRatio = '1:1' }) {
   if (!API_KEY) {
-    console.error('BANATIE_API_KEY environment variable is not set');
+    console.error('BANATIE_KEY environment variable is not set');
     process.exit(1);
   }
 
-  const preset = PRESETS[type];
-  if (!preset) {
-    console.error(`Unknown type: ${type}. Use "background" or "icon".`);
-    process.exit(1);
-  }
+  const body = { prompt, aspectRatio };
 
-  const response = await fetch(API_ENDPOINT, {
+  console.log(`Generating: "${prompt}" (${body.aspectRatio})...`);
+
+  const response = await fetch(`${API_BASE}/generations`, {
     method: 'POST',
     headers: {
       'Content-Type': 'application/json',
-      'Authorization': `Bearer ${API_KEY}`,
+      'X-API-Key': API_KEY,
     },
-    body: JSON.stringify({
-      prompt,
-      ...preset,
-    }),
+    body: JSON.stringify(body),
   });
 
   if (!response.ok) {
-    const body = await response.text();
-    console.error(`API error ${response.status}: ${body}`);
+    const text = await response.text();
+    console.error(`API error ${response.status}: ${text}`);
     process.exit(1);
   }
 
-  const buffer = Buffer.from(await response.arrayBuffer());
+  const result = await response.json();
+  if (!result.success) {
+    console.error(`Generation failed:`, result.error);
+    process.exit(1);
+  }
+
+  const imageUrl = result.data.outputImage.storageUrl;
+  console.log(`Downloading from ${imageUrl}...`);
+
+  const imageResponse = await fetch(imageUrl);
+  if (!imageResponse.ok) {
+    console.error(`Failed to download image: ${imageResponse.status}`);
+    process.exit(1);
+  }
+
+  const buffer = Buffer.from(await imageResponse.arrayBuffer());
   const outputPath = resolve(output);
   mkdirSync(dirname(outputPath), { recursive: true });
   writeFileSync(outputPath, buffer);
-  console.log(`Image saved: ${outputPath}`);
-  return outputPath;
+
+  console.log(`Image saved: ${outputPath} (${result.data.outputImage.width}x${result.data.outputImage.height})`);
+  return { path: outputPath, generation: result.data };
 }
 
 function parseArgs(args) {
   const result = {};
   for (let i = 0; i < args.length; i++) {
-    if (args[i] === '--type') result.type = args[++i];
-    else if (args[i] === '--prompt') result.prompt = args[++i];
+    if (args[i] === '--prompt') result.prompt = args[++i];
     else if (args[i] === '--output') result.output = args[++i];
+    else if (args[i] === '--aspect-ratio') result.aspectRatio = args[++i];
   }
   return result;
 }
@@ -61,6 +79,6 @@ const args = parseArgs(process.argv.slice(2));
 if (args.prompt && args.output) {
   generateImage(args);
 } else if (process.argv.length > 2) {
-  console.error('Usage: node banatie.mjs --type <background|icon> --prompt "<description>" --output <path>');
+  console.error('Usage: node banatie.mjs --prompt "<description>" --output <path> [--aspect-ratio <ratio>]');
   process.exit(1);
 }