@xuda.io/ai_module 1.1.4307 → 1.1.4309

package/index.mjs

@@ -6739,6 +6739,25 @@ const get_business_logo = async function (uid, name, account_profile_info) {
 
   const url = ret.data;
   const file = await return_image_blob_from_downloaded_image(url);
+  // const imageBase64 = Buffer.from(await file.image_blob.arrayBuffer()).toString('base64');
+  const model = 'chatgpt-image-latest';
+  const prompt = `
+          extract the logo from the image
+          `;
+  const logo_response = await client.images.edit({
+    model,
+    image: file.image_blob, //logo
+    prompt,
+    background: 'transparent',
+  });
+  record_chat_usage(uid, logo_response.usage.input_tokens, logo_response.usage.output_tokens, 'business logo', prompt, model, { url }, account_profile_info);
+  imageBase64 = await normalizeBase64To1024(ai_avatar_response.data[0].b64_json);
+
+  const outputBuffer = Buffer.from(imageBase64, 'base64');
+  const outputPath = file.tempOutputPath;
+  await fs.promises.writeFile(outputPath, outputBuffer);
+
+  console.log('Success! Saved →', outputPath);
 
   console.log(file);
   // prompt: `find  "${name}" logo base on the following facts  "${facts}", Return only  png,jpg img tag , nothing else`,
@@ -6750,7 +6769,55 @@ const get_business_logo = async function (uid, name, account_profile_info) {
   // } catch (error) {}
 };
 
+async function normalizeBase64To1024(base64Image) {
+  const inputBuffer = Buffer.from(base64Image, 'base64');
+
+  const normalizedBuffer = await sharp(inputBuffer)
+    .resize(1024, 1024, {
+      fit: 'contain', // Preserve aspect ratio, fit within 211x211
+      background: { r: 0, g: 0, b: 0, alpha: 0 }, // Transparent background for padding
+    })
+    .png({
+      quality: 100,
+      compressionLevel: 9,
+      force: true,
+    })
+    .toBuffer();
+
+  return normalizedBuffer.toString('base64'); // Return normalized base64 string
+}
+
 const return_image_blob_from_downloaded_image = async function (url) {
+  async function hasTransparentBackground(filePath) {
+    const image = sharp(filePath);
+    const metadata = await image.metadata();
+
+    // If the image doesn't have an alpha channel at all → no transparency possible
+    if (!metadata.hasAlpha) {
+      return false;
+    }
+
+    // For images with alpha channel, check if any pixel actually has alpha < 255
+    const { data, info } = await image.raw().toBuffer({ resolveWithObject: true });
+
+    // Determine position of alpha channel based on number of channels
+    const channels = info.channels; // 2 (grayscale+alpha), 4 (RGBA), etc.
+    const alphaOffset = channels === 2 ? 1 : channels === 4 ? 3 : -1;
+
+    if (alphaOffset === -1) {
+      // Unusual channel count – be conservative and assume it might have transparency
+      return true;
+    }
+
+    for (let i = alphaOffset; i < data.length; i += channels) {
+      if (data[i] < 255) {
+        return true; // Found at least one transparent/semi-transparent pixel
+      }
+    }
+
+    return false; // Has alpha channel but all pixels are fully opaque
+  }
+
   async function imageToPngBlob(filePath) {
     const ext = path.extname(filePath).toLowerCase();
 
@@ -6810,7 +6877,7 @@ const return_image_blob_from_downloaded_image = async function (url) {
         return { is_transparent: img_ret.is_transparent, image_blob: blob };
       }),
     );
-    return images[0];
+    return { image_blob: images[0].image_blob, is_transparent: images[0].is_transparent, tempOutputPath };
   } catch (err) {
     console.error(err);
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xuda.io/ai_module",
-  "version": "1.1.4307",
+  "version": "1.1.4309",
   "description": "Xuda AI Module",
   "main": "index.mjs",
   "type": "module",