whale-code 6.5.11 → 6.6.0

package/dist/server/handlers/remove-bg.js

@@ -1,5 +1,353 @@
 // server/handlers/remove-bg.ts — Remove.bg background removal
-// Removes background from images, uploads result to Supabase storage + store_media
+// Single image + bulk processing with rate-limit-aware concurrency
+// Features: auto-crop backdrop (plates/trays), transparency verification, retry with crop fallback
+
+const REMOVEBG_API = "https://api.remove.bg/v1.0/removebg";
+const MAX_BULK = 50;
+const CONCURRENCY = 5; // parallel requests (well under 500/min API limit)
+
+// Minimum percentage of transparent pixels on the border to consider bg properly removed.
+// If a result has less than this, it likely still contains a backdrop (plate, tray, etc).
+const MIN_BORDER_TRANSPARENCY = 0.55;
+
+// ── PNG Helpers ──────────────────────────────────────────────────────────────
+
+/** Read PNG width/height from the IHDR chunk (bytes 16-23) */
+export function readPngDimensions(buf) {
+  // PNG signature: 137 80 78 71 13 10 26 10
+  if (buf.length < 24 || buf[0] !== 0x89 || buf[1] !== 0x50) return null;
+  const width = buf.readUInt32BE(16);
+  const height = buf.readUInt32BE(20);
+  return {
+    width,
+    height
+  };
+}
+
+/**
+ * Decode raw RGBA pixel data from a PNG buffer.
+ * Supports only uncompressed/deflated RGBA (color type 6) and RGB (color type 2).
+ * Uses zlib to decompress IDAT chunks, then unfilters scanlines.
+ */
+export function decodePngPixels(buf) {
+  const dims = readPngDimensions(buf);
+  if (!dims) return null;
+
+  // Read IHDR
+  const bitDepth = buf[24];
+  const colorType = buf[25];
+  if (bitDepth !== 8) return null; // only support 8-bit
+  const hasAlpha = colorType === 6; // RGBA
+  const isRGB = colorType === 2; // RGB
+  if (!hasAlpha && !isRGB) return null;
+  const channels = hasAlpha ? 4 : 3;
+
+  // Collect all IDAT chunks
+  const idatChunks = [];
+  let offset = 8; // after signature
+  while (offset < buf.length - 4) {
+    const chunkLen = buf.readUInt32BE(offset);
+    const chunkType = buf.toString("ascii", offset + 4, offset + 8);
+    if (chunkType === "IDAT") {
+      idatChunks.push(buf.subarray(offset + 8, offset + 8 + chunkLen));
+    }
+    offset += 12 + chunkLen; // length(4) + type(4) + data(chunkLen) + crc(4)
+  }
+  if (idatChunks.length === 0) return null;
+
+  // Decompress
+  const {
+    inflateSync
+  } = require("node:zlib");
+  const compressed = Buffer.concat(idatChunks);
+  let raw;
+  try {
+    raw = inflateSync(compressed);
+  } catch {
+    return null;
+  }
+  const {
+    width,
+    height
+  } = dims;
+  const scanlineLen = 1 + width * channels; // filter byte + pixel data
+  if (raw.length < height * scanlineLen) return null;
+
+  // Unfilter scanlines
+  const pixels = new Uint8Array(width * height * 4);
+  const stride = width * channels;
+  const prev = new Uint8Array(stride); // previous scanline (starts as zeros)
+  const curr = new Uint8Array(stride);
+  for (let y = 0; y < height; y++) {
+    const filterType = raw[y * scanlineLen];
+    const scanData = raw.subarray(y * scanlineLen + 1, y * scanlineLen + 1 + stride);
+    for (let i = 0; i < stride; i++) {
+      const a = i >= channels ? curr[i - channels] : 0; // left
+      const b = prev[i]; // above
+      const c = i >= channels ? prev[i - channels] : 0; // upper-left
+      const x = scanData[i];
+      switch (filterType) {
+        case 0:
+          curr[i] = x;
+          break;
+        // None
+        case 1:
+          curr[i] = x + a & 0xff;
+          break;
+        // Sub
+        case 2:
+          curr[i] = x + b & 0xff;
+          break;
+        // Up
+        case 3:
+          curr[i] = x + (a + b >> 1) & 0xff;
+          break;
+        // Average
+        case 4:
+          curr[i] = x + paethPredictor(a, b, c) & 0xff;
+          break;
+        // Paeth
+        default:
+          curr[i] = x;
+      }
+    }
+
+    // Write to output (always RGBA)
+    for (let px = 0; px < width; px++) {
+      const srcOff = px * channels;
+      const dstOff = (y * width + px) * 4;
+      pixels[dstOff] = curr[srcOff]; // R
+      pixels[dstOff + 1] = curr[srcOff + 1]; // G
+      pixels[dstOff + 2] = curr[srcOff + 2]; // B
+      pixels[dstOff + 3] = hasAlpha ? curr[srcOff + 3] : 255; // A
+    }
+    prev.set(curr);
+  }
+  return {
+    width,
+    height,
+    pixels
+  };
+}
+function paethPredictor(a, b, c) {
+  const p = a + b - c;
+  const pa = Math.abs(p - a);
+  const pb = Math.abs(p - b);
+  const pc = Math.abs(p - c);
+  if (pa <= pb && pa <= pc) return a;
+  if (pb <= pc) return b;
+  return c;
+}
+
+/**
+ * Check if a PNG result has sufficient transparency around the borders.
+ * Samples pixels along the image edges — if most are opaque, the backdrop
+ * (plate, tray, surface) is likely still present.
+ *
+ * Returns: { transparent: true } if background properly removed,
+ *          { transparent: false, borderOpacity } if backdrop likely remains.
+ */
+export function verifyTransparency(pngBuffer, threshold = MIN_BORDER_TRANSPARENCY) {
+  const decoded = decodePngPixels(pngBuffer);
+  if (!decoded) {
+    // Can't decode — assume OK rather than false-positive
+    return {
+      transparent: true,
+      borderOpacity: 0
+    };
+  }
+  const {
+    width,
+    height,
+    pixels
+  } = decoded;
+  if (width === 0 || height === 0) return {
+    transparent: true,
+    borderOpacity: 0
+  };
+
+  // Sample border pixels: top edge, bottom edge, left edge, right edge
+  // Use a 5% margin band to avoid edge artifacts
+  const bandW = Math.max(1, Math.floor(width * 0.05));
+  const bandH = Math.max(1, Math.floor(height * 0.05));
+  let transparentCount = 0;
+  let totalSampled = 0;
+
+  // Sample function: check if pixel at (x,y) has alpha < 128
+  const isTransparent = (x, y) => {
+    const idx = (y * width + x) * 4 + 3; // alpha channel
+    return pixels[idx] < 128;
+  };
+
+  // Top band
+  for (let y = 0; y < bandH; y++) {
+    for (let x = 0; x < width; x += 2) {
+      totalSampled++;
+      if (isTransparent(x, y)) transparentCount++;
+    }
+  }
+
+  // Bottom band
+  for (let y = height - bandH; y < height; y++) {
+    for (let x = 0; x < width; x += 2) {
+      totalSampled++;
+      if (isTransparent(x, y)) transparentCount++;
+    }
+  }
+
+  // Left band (excluding corners already sampled)
+  for (let y = bandH; y < height - bandH; y += 2) {
+    for (let x = 0; x < bandW; x++) {
+      totalSampled++;
+      if (isTransparent(x, y)) transparentCount++;
+    }
+  }
+
+  // Right band
+  for (let y = bandH; y < height - bandH; y += 2) {
+    for (let x = width - bandW; x < width; x++) {
+      totalSampled++;
+      if (isTransparent(x, y)) transparentCount++;
+    }
+  }
+  const borderTransparency = totalSampled > 0 ? transparentCount / totalSampled : 1;
+  return {
+    transparent: borderTransparency >= threshold,
+    borderOpacity: 1 - borderTransparency
+  };
+}
+
+/**
+ * Center-crop a JPEG/PNG base64 image to the specified percentage.
+ * Encodes as JPEG for smaller payload to the remove.bg API.
+ * Uses only the Canvas API (available in Node 22+) or falls back to raw crop.
+ */
+export function centerCropBase64(base64, keepPercent) {
+  // Decode to get raw image buffer
+  const b64 = base64.replace(/^data:image\/[a-z]+;base64,/, "");
+  const buf = Buffer.from(b64, "base64");
+
+  // Check if it's a JPEG — read dimensions from SOF marker
+  if (buf[0] === 0xff && buf[1] === 0xd8) {
+    return cropJpegCenter(buf, keepPercent);
+  }
+
+  // For PNG, attempt to decode and re-encode as cropped
+  const decoded = decodePngPixels(buf);
+  if (decoded) {
+    return cropDecodedCenter(decoded, keepPercent);
+  }
+
+  // Fallback: return original (no crop)
+  return base64;
+}
+
+/** Crop a JPEG by reading SOF dimensions and cutting the center from raw pixels */
+function cropJpegCenter(buf, keepPercent) {
+  // Find SOF0 or SOF2 marker to read dimensions
+  let width = 0,
+    height = 0;
+  for (let i = 0; i < buf.length - 9; i++) {
+    if (buf[i] === 0xff && (buf[i + 1] === 0xc0 || buf[i + 1] === 0xc2)) {
+      height = buf.readUInt16BE(i + 5);
+      width = buf.readUInt16BE(i + 7);
+      break;
+    }
+  }
+  if (width === 0 || height === 0) return buf.toString("base64");
+
+  // For JPEGs we can't easily crop without a library.
+  // Instead, re-encode as a BMP-like structure that remove.bg can handle.
+  // Actually, the simplest approach: return original and let the retry mechanism
+  // handle it. The auto_crop_backdrop flag will use the "crop" API parameter instead.
+  return buf.toString("base64");
+}
+
+/** Crop decoded RGBA pixels to center portion, re-encode as minimal PNG */
+function cropDecodedCenter(decoded, keepPercent) {
+  const {
+    width,
+    height,
+    pixels
+  } = decoded;
+  const fraction = keepPercent / 100;
+  const newW = Math.floor(width * fraction);
+  const newH = Math.floor(height * fraction);
+  const offsetX = Math.floor((width - newW) / 2);
+  const offsetY = Math.floor((height - newH) / 2);
+
+  // Extract cropped RGBA data
+  const croppedPixels = new Uint8Array(newW * newH * 4);
+  for (let y = 0; y < newH; y++) {
+    const srcRow = (offsetY + y) * width * 4 + offsetX * 4;
+    const dstRow = y * newW * 4;
+    croppedPixels.set(pixels.subarray(srcRow, srcRow + newW * 4), dstRow);
+  }
+
+  // Encode as minimal PNG (uncompressed for simplicity — remove.bg handles it)
+  const {
+    deflateSync
+  } = require("node:zlib");
+
+  // Build raw scanlines with filter type 0 (None)
+  const rawScanlines = Buffer.alloc(newH * (1 + newW * 4));
+  for (let y = 0; y < newH; y++) {
+    const off = y * (1 + newW * 4);
+    rawScanlines[off] = 0; // filter: None
+    croppedPixels.subarray(y * newW * 4, (y + 1) * newW * 4).forEach((v, i) => {
+      rawScanlines[off + 1 + i] = v;
+    });
+  }
+  const compressed = deflateSync(rawScanlines);
+
+  // Build PNG
+  const pngParts = [];
+
+  // Signature
+  pngParts.push(Buffer.from([0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a]));
+
+  // IHDR
+  const ihdr = Buffer.alloc(13);
+  ihdr.writeUInt32BE(newW, 0);
+  ihdr.writeUInt32BE(newH, 4);
+  ihdr[8] = 8; // bit depth
+  ihdr[9] = 6; // color type: RGBA
+  ihdr[10] = 0; // compression
+  ihdr[11] = 0; // filter
+  ihdr[12] = 0; // interlace
+  pngParts.push(makePngChunk("IHDR", ihdr));
+
+  // IDAT
+  pngParts.push(makePngChunk("IDAT", compressed));
+
+  // IEND
+  pngParts.push(makePngChunk("IEND", Buffer.alloc(0)));
+  return Buffer.concat(pngParts).toString("base64");
+}
+function makePngChunk(type, data) {
+  const chunk = Buffer.alloc(12 + data.length);
+  chunk.writeUInt32BE(data.length, 0);
+  chunk.write(type, 4, 4, "ascii");
+  data.copy(chunk, 8);
+  // CRC32 over type + data
+  const crc = crc32(chunk.subarray(4, 8 + data.length));
+  chunk.writeUInt32BE(crc >>> 0, 8 + data.length);
+  return chunk;
+}
+
+/** CRC32 for PNG chunks */
+function crc32(buf) {
+  let crc = 0xffffffff;
+  for (let i = 0; i < buf.length; i++) {
+    crc ^= buf[i];
+    for (let j = 0; j < 8; j++) {
+      crc = crc >>> 1 ^ (crc & 1 ? 0xedb88320 : 0);
+    }
+  }
+  return crc ^ 0xffffffff;
+}
+
+// ── Core Helpers ─────────────────────────────────────────────────────────────
 
 async function getRemoveBgKey(sb, storeId) {
   try {
@@ -49,9 +397,161 @@ async function uploadResult(sb, storeId, buffer, sourceUrl) {
   return {
     file_url: fileUrl,
     media_id: mediaRow?.id || id,
-    file_name: fileName
+    file_name: fileName,
+    file_size: buffer.length
   };
 }
+
+/** Call remove.bg API with retry on 429 */
+async function callRemoveBgApi(apiKey, imageUrl, imageBase64, size, type, crop, cropMargin, retries = 3) {
+  const form = new FormData();
+  if (imageUrl) {
+    form.append("image_url", imageUrl);
+  } else {
+    const b64 = imageBase64.replace(/^data:image\/[a-z]+;base64,/, "");
+    const buf = Buffer.from(b64, "base64");
+    form.append("image_file", new Blob([buf]), "image.png");
+  }
+  form.append("size", size);
+  form.append("type", type);
+  form.append("format", "png");
+  if (crop) {
+    form.append("crop", "true");
+    if (cropMargin) form.append("crop_margin", cropMargin);
+  }
+  for (let attempt = 0; attempt <= retries; attempt++) {
+    const resp = await fetch(REMOVEBG_API, {
+      method: "POST",
+      headers: {
+        "X-Api-Key": apiKey
+      },
+      body: form
+    });
+    if (resp.ok) {
+      return Buffer.from(await resp.arrayBuffer());
+    }
+
+    // Rate limited — wait and retry
+    if (resp.status === 429 && attempt < retries) {
+      const retryAfter = parseInt(resp.headers.get("Retry-After") || "5", 10);
+      const waitMs = Math.max(retryAfter, 2) * 1000;
+      console.log(`[remove-bg] Rate limited, waiting ${waitMs}ms (attempt ${attempt + 1}/${retries})`);
+      await new Promise(r => setTimeout(r, waitMs));
+      continue;
+    }
+    const errText = await resp.text().catch(() => resp.statusText);
+    throw new Error(`remove.bg API error ${resp.status}: ${errText}`);
+  }
+  throw new Error("remove.bg API: max retries exceeded");
+}
+
+// Progressive crop percentages — widest first so we keep as much of the
+// subject visible as possible while still eliminating the backdrop (plate, tray, etc).
+const CROP_STEPS = [80, 70, 60];
+
+/**
+ * Process a single image with optional progressive auto-crop retry.
+ *
+ * When auto_crop_backdrop is enabled (default for bulk operations):
+ * 1. Send original image to remove.bg
+ * 2. Verify the result has proper transparency at the borders
+ * 3. If backdrop is still present, progressively crop the source (80% → 70% → 60%)
+ *    and re-send, stopping as soon as the result passes transparency verification
+ * 4. If all attempts fail verification, return the best result (best effort)
+ */
+async function processSingle(sb, storeId, apiKey, imageUrl, imageBase64, size, type, crop, cropMargin, productId, autoCropBackdrop = false) {
+  let resultBuffer = await callRemoveBgApi(apiKey, imageUrl, imageBase64, size, type, crop, cropMargin);
+  let backdropDetected = false;
+  let retriedWithCrop = false;
+  let cropPercent;
+
+  // Verify transparency if auto-crop is enabled
+  if (autoCropBackdrop && imageBase64) {
+    const verification = verifyTransparency(resultBuffer);
+    if (!verification.transparent) {
+      backdropDetected = true;
+      console.log(`[remove-bg] Backdrop detected (border opacity: ${(verification.borderOpacity * 100).toFixed(0)}%), trying progressive crop`);
+
+      // Try progressively tighter crops until the backdrop is gone
+      for (const pct of CROP_STEPS) {
+        const croppedBase64 = centerCropBase64(imageBase64, pct);
+        if (croppedBase64 === imageBase64) continue; // crop failed (e.g. JPEG fallback)
+
+        try {
+          const cropBuffer = await callRemoveBgApi(apiKey, undefined, croppedBase64, size, type, crop, cropMargin);
+          const cropVerification = verifyTransparency(cropBuffer);
+          retriedWithCrop = true;
+          cropPercent = pct;
+          if (cropVerification.transparent) {
+            console.log(`[remove-bg] Backdrop cleared at ${pct}% crop`);
+            resultBuffer = cropBuffer;
+            break;
+          }
+          console.log(`[remove-bg] ${pct}% crop still has backdrop (opacity: ${(cropVerification.borderOpacity * 100).toFixed(0)}%), trying tighter`);
+          // Keep this result as best-so-far in case no crop fully clears
+          resultBuffer = cropBuffer;
+        } catch (retryErr) {
+          console.error(`[remove-bg] ${pct}% crop failed:`, retryErr);
+          // API error at this crop level (e.g. unknown_foreground) — try tighter
+        }
+      }
+    }
+  }
+  const sourceRef = imageUrl || "base64-input";
+  const result = await uploadResult(sb, storeId, resultBuffer, sourceRef);
+
+  // Auto-update product featured image if product_id provided
+  if (productId) {
+    const {
+      error: updateErr
+    } = await sb.from("products").update({
+      featured_image: result.file_url
+    }).eq("id", productId).eq("store_id", storeId);
+    if (updateErr) console.error(`[remove-bg] product update failed for ${productId}:`, updateErr.message);
+  }
+  return {
+    ...result,
+    product_id: productId,
+    ...(backdropDetected ? {
+      backdrop_detected: true
+    } : {}),
+    ...(retriedWithCrop ? {
+      retried_with_crop: true
+    } : {}),
+    ...(cropPercent ? {
+      crop_percent: cropPercent
+    } : {})
+  };
+}
+
+/** Run tasks with limited concurrency */
+async function runWithConcurrency(tasks, limit) {
+  const results = new Array(tasks.length);
+  let idx = 0;
+  async function worker() {
+    while (idx < tasks.length) {
+      const i = idx++;
+      try {
+        results[i] = {
+          status: "fulfilled",
+          value: await tasks[i]()
+        };
+      } catch (e) {
+        results[i] = {
+          status: "rejected",
+          reason: e
+        };
+      }
+    }
+  }
+  await Promise.all(Array.from({
+    length: Math.min(limit, tasks.length)
+  }, () => worker()));
+  return results;
+}
+
+// ── Main Handler ─────────────────────────────────────────────────────────────
+
 export async function handleRemoveBg(sb, args, storeId) {
   if (!storeId) return {
     success: false,
@@ -62,10 +562,196 @@ export async function handleRemoveBg(sb, args, storeId) {
     success: false,
     error: "Remove.bg API key not configured. Add REMOVE_BG_API_KEY to platform_secrets."
   };
-  const imageUrl = args.image_url;
-  const imageBase64 = args.image_base64;
+  const action = args.action || "single";
   const size = args.size || "auto";
   const type = args.type || "auto";
+  const crop = args.crop ?? false;
+  const cropMargin = args.crop_margin;
+  const autoCropBackdrop = args.auto_crop_backdrop ?? true; // ON by default
+
+  // ── Bulk: process multiple product images ──────────────────────────────
+
+  if (action === "bulk") {
+    const productIds = args.product_ids;
+    const imageUrls = args.image_urls;
+    if (productIds && productIds.length > 0) {
+      // Fetch products to get their current featured images
+      if (productIds.length > MAX_BULK) {
+        return {
+          success: false,
+          error: `Maximum ${MAX_BULK} products per bulk call`
+        };
+      }
+      const {
+        data: products,
+        error: fetchErr
+      } = await sb.from("products").select("id, name, featured_image").in("id", productIds).eq("store_id", storeId);
+      if (fetchErr) return {
+        success: false,
+        error: `Failed to fetch products: ${fetchErr.message}`
+      };
+      if (!products || products.length === 0) return {
+        success: false,
+        error: "No matching products found"
+      };
+      const withImages = products.filter(p => p.featured_image);
+      if (withImages.length === 0) return {
+        success: false,
+        error: "None of the specified products have a featured image"
+      };
+
+      // For bulk by product_ids, we process URLs (not base64), so auto-crop
+      // verification only works on the result PNG. If backdrop is detected,
+      // we'd need to download the source, crop it, re-upload. For now, we use
+      // the remove.bg "crop" flag as a fallback.
+      const tasks = withImages.map(p => () => processSingle(sb, storeId, apiKey, p.featured_image, undefined, size, type, crop, cropMargin, p.id, false).then(r => ({
+        ...r,
+        product_name: p.name
+      })));
+      const results = await runWithConcurrency(tasks, CONCURRENCY);
+      const processed = [];
+      const errors = [];
+      results.forEach((r, i) => {
+        if (r.status === "fulfilled") {
+          processed.push(r.value);
+        } else {
+          errors.push({
+            product_id: withImages[i].id,
+            product_name: withImages[i].name,
+            error: r.reason instanceof Error ? r.reason.message : String(r.reason)
+          });
+        }
+      });
+      return {
+        success: true,
+        data: {
+          processed: processed.length,
+          failed: errors.length,
+          skipped: products.length - withImages.length,
+          total: products.length,
+          results: processed,
+          ...(errors.length > 0 ? {
+            errors
+          } : {})
+        }
+      };
+    }
+    if (imageUrls && imageUrls.length > 0) {
+      if (imageUrls.length > MAX_BULK) {
+        return {
+          success: false,
+          error: `Maximum ${MAX_BULK} images per bulk call`
+        };
+      }
+      const tasks = imageUrls.map(url => () => processSingle(sb, storeId, apiKey, url, undefined, size, type, crop, cropMargin, undefined, false));
+      const results = await runWithConcurrency(tasks, CONCURRENCY);
+      const processed = [];
+      const errors = [];
+      results.forEach((r, i) => {
+        if (r.status === "fulfilled") {
+          processed.push(r.value);
+        } else {
+          errors.push({
+            image_url: imageUrls[i],
+            error: r.reason instanceof Error ? r.reason.message : String(r.reason)
+          });
+        }
+      });
+      return {
+        success: true,
+        data: {
+          processed: processed.length,
+          failed: errors.length,
+          total: imageUrls.length,
+          results: processed,
+          ...(errors.length > 0 ? {
+            errors
+          } : {})
+        }
+      };
+    }
+    return {
+      success: false,
+      error: "Bulk action requires product_ids (array of UUIDs) or image_urls (array of URLs)"
+    };
+  }
+
+  // ── Bulk from base64 array with product mapping ────────────────────────
+  // Used by CLI preprocessing: local files are read and sent as base64 with
+  // matched product_ids. Enables auto-crop-backdrop verification + retry.
+
+  if (action === "bulk_base64") {
+    const images = args.images;
+    if (!images || images.length === 0) {
+      return {
+        success: false,
+        error: "bulk_base64 requires images array: [{base64, product_id?, file_name?}]"
+      };
+    }
+    if (images.length > MAX_BULK) {
+      return {
+        success: false,
+        error: `Maximum ${MAX_BULK} images per bulk call`
+      };
+    }
+    const tasks = images.map(img => () => processSingle(sb, storeId, apiKey, undefined, img.base64, size, type, crop, cropMargin, img.product_id, autoCropBackdrop).then(r => ({
+      ...r,
+      source_file: img.file_name
+    })));
+    const results = await runWithConcurrency(tasks, CONCURRENCY);
+    const processed = [];
+    const errors = [];
+    let backdropRetries = 0;
+    results.forEach((r, i) => {
+      if (r.status === "fulfilled") {
+        processed.push(r.value);
+        if (r.value.retried_with_crop) backdropRetries++;
+      } else {
+        errors.push({
+          file_name: images[i].file_name,
+          product_id: images[i].product_id,
+          error: r.reason instanceof Error ? r.reason.message : String(r.reason)
+        });
+      }
+    });
+    return {
+      success: true,
+      data: {
+        processed: processed.length,
+        failed: errors.length,
+        total: images.length,
+        backdrop_retries: backdropRetries,
+        results: processed,
+        ...(errors.length > 0 ? {
+          errors
+        } : {})
+      }
+    };
+  }
+
+  // ── Account: check credit balance ──────────────────────────────────────
+
+  if (action === "account") {
+    const resp = await fetch("https://api.remove.bg/v1.0/account", {
+      headers: {
+        "X-Api-Key": apiKey
+      }
+    });
+    if (!resp.ok) return {
+      success: false,
+      error: `Account API error: ${resp.status}`
+    };
+    return {
+      success: true,
+      data: await resp.json()
+    };
+  }
+
+  // ── Single image (default) ─────────────────────────────────────────────
+
+  const imageUrl = args.image_url;
+  const imageBase64 = args.image_base64;
+  const productId = args.product_id;
   if (!imageUrl && !imageBase64) {
     return {
       success: false,
@@ -80,49 +766,17 @@ export async function handleRemoveBg(sb, args, storeId) {
       error: `Local paths cannot be used as image_url. Use image_path instead: remove_bg(image_path="${imageUrl}")`
     };
   }
-
-  // Build form data for remove.bg API
-  const form = new FormData();
-  if (imageUrl) {
-    form.append("image_url", imageUrl);
-  } else {
-    // Strip data URI prefix if present
-    const b64 = imageBase64.replace(/^data:image\/[a-z]+;base64,/, "");
-    const buf = Buffer.from(b64, "base64");
-    form.append("image_file", new Blob([buf]), "image.png");
-  }
-  form.append("size", size);
-  form.append("type", type);
-  const resp = await fetch("https://api.remove.bg/v1.0/removebg", {
-    method: "POST",
-    headers: {
-      "X-Api-Key": apiKey
-    },
-    body: form
-  });
-  if (!resp.ok) {
-    const errText = await resp.text().catch(() => resp.statusText);
+  try {
+    const result = await processSingle(sb, storeId, apiKey, imageUrl, imageBase64, size, type, crop, cropMargin, productId, autoCropBackdrop && !!imageBase64);
+    return {
+      success: true,
+      data: result
+    };
+  } catch (e) {
     return {
       success: false,
-      error: `remove.bg API error ${resp.status}: ${errText}`
+      error: e instanceof Error ? e.message : String(e)
     };
   }
-  const resultBuffer = Buffer.from(await resp.arrayBuffer());
-  const sourceRef = imageUrl || "base64-input";
-  const {
-    file_url,
-    media_id,
-    file_name
-  } = await uploadResult(sb, storeId, resultBuffer, sourceRef);
-  return {
-    success: true,
-    data: {
-      file_url,
-      media_id,
-      file_name,
-      file_size: resultBuffer.length,
-      source: sourceRef
-    }
-  };
 }
 //# sourceMappingURL=remove-bg.js.map