@xuda.io/ai_module 1.1.4308 → 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,6 +6769,24 @@ 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);
@@ -6840,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.4308",
+  "version": "1.1.4309",
   "description": "Xuda AI Module",
   "main": "index.mjs",
   "type": "module",