@xuda.io/ai_module 1.1.4355 → 1.1.4357

package/index.mjs

@@ -5679,72 +5679,72 @@ export const create_profile_avatar = async function (req, job_id, headers) {
   const uniqueId = Date.now() + '_' + Math.random().toString(36).substring(7);
   const tempOutputPath = path.join(tempDir, `output_${uniqueId}.png`);
 
-  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
-  }
-
-  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();
-
-    let buffer;
-
-    let is_transparent = false;
-    if (ext === '.png') {
-      buffer = await fs.promises.readFile(filePath);
-      is_transparent = await hasTransparentBackground(filePath);
-    } else {
-      console.log(`Converting ${path.basename(filePath)} to PNG...`);
-      buffer = await sharp(filePath).png({ quality: 100, compressionLevel: 9, force: true }).toBuffer();
-    }
-
-    // Create a Blob with correct MIME type
-
-    return { is_transparent, image_blob: new Blob([buffer], { type: 'image/png' }) };
-  }
+  // 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
+  // }
+
+  // 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();
+
+  //   let buffer;
+
+  //   let is_transparent = false;
+  //   if (ext === '.png') {
+  //     buffer = await fs.promises.readFile(filePath);
+  //     is_transparent = await hasTransparentBackground(filePath);
+  //   } else {
+  //     console.log(`Converting ${path.basename(filePath)} to PNG...`);
+  //     buffer = await sharp(filePath).png({ quality: 100, compressionLevel: 9, force: true }).toBuffer();
+  //   }
+
+  //   // Create a Blob with correct MIME type
+
+  //   return { is_transparent, image_blob: new Blob([buffer], { type: 'image/png' }) };
+  // }
 
   const detect_real_person_in_image = async function (base64) {
     const ret = await submit_chat_gpt_prompt({
@@ -5782,54 +5782,55 @@ export const create_profile_avatar = async function (req, job_id, headers) {
     await update_profile_avatar_status(account_obj._id, 2);
   }
   try {
-    if (!url) {
-      throw new Error('No URL provided in request');
-    }
+    // if (!url) {
+    //   throw new Error('No URL provided in request');
+    // }
 
     // 3. Download the image to a Buffer
-    const res = await fetch(url);
-    if (!res.ok) {
-      throw new Error(`Failed to fetch image URL: ${res.status} ${res.statusText}`);
-    }
+    // const res = await fetch(url);
+    // if (!res.ok) {
+    //   throw new Error(`Failed to fetch image URL: ${res.status} ${res.statusText}`);
+    // }
 
-    const arrayBuffer = await res.arrayBuffer();
-    const imageBuffer = Buffer.from(arrayBuffer);
+    // const arrayBuffer = await res.arrayBuffer();
+    // const imageBuffer = Buffer.from(arrayBuffer);
 
-    const tempDir = os.tmpdir();
-    const uniqueId = Date.now() + '_' + Math.random().toString(36).substring(7);
+    // const tempDir = os.tmpdir();
+    // const uniqueId = Date.now() + '_' + Math.random().toString(36).substring(7);
 
-    const inputExt = path.extname(url) || '.jpg';
-    const tempInputPath = path.join(tempDir, `input_${uniqueId}${inputExt}`);
-    const tempOutputPath = path.join(tempDir, `output_${uniqueId}.png`); // Output is always PNG (transparent)
+    // const inputExt = path.extname(url) || '.jpg';
+    // const tempInputPath = path.join(tempDir, `input_${uniqueId}${inputExt}`);
+    // const tempOutputPath = path.join(tempDir, `output_${uniqueId}.png`); // Output is always PNG (transparent)
 
     // 5. Write the downloaded image to disk
-    fs.writeFileSync(tempInputPath, imageBuffer);
+    // fs.writeFileSync(tempInputPath, imageBuffer);
     /////////////////////////////////
 
-    const imageFiles = [tempInputPath];
+    // const imageFiles = [tempInputPath];
 
-    const images = await Promise.all(
-      imageFiles.map(async (filePath) => {
-        const img_ret = await imageToPngBlob(filePath);
-        const blob = img_ret.image_blob;
+    // const images = await Promise.all(
+    //   imageFiles.map(async (filePath) => {
+    //     const img_ret = await imageToPngBlob(filePath);
+    //     const blob = img_ret.image_blob;
 
-        // OpenAI expects a filename for logging/debugging
-        const filename = path.basename(filePath, path.extname(filePath)) + '.png';
-        blob.name = filename; // Optional but helpful
-        // @ts-ignore – name is allowed in Node 20+
-        blob.lastModified = Date.now();
+    //     // OpenAI expects a filename for logging/debugging
+    //     const filename = path.basename(filePath, path.extname(filePath)) + '.png';
+    //     blob.name = filename; // Optional but helpful
+    //     // @ts-ignore – name is allowed in Node 20+
+    //     blob.lastModified = Date.now();
 
-        return { is_transparent: img_ret.is_transparent, image_blob: blob };
-      }),
-    );
+    //     return { is_transparent: img_ret.is_transparent, image_blob: blob };
+    //   }),
+    // );
 
     let imageBase64;
+    const image_blob_ret = await get_image_blob_from_downloaded_image(url);
     let avatar_source = '';
     if (prompt) {
       const model = 'chatgpt-image-latest';
       const ai_avatar_response = await client.images.edit({
         model, // 'gpt-image-1',
-        image: images[0].image_blob, //base photo
+        image: image_blob_ret.image_blob, //base photo
         prompt,
         background: 'transparent',
       });
@@ -5837,13 +5838,13 @@ export const create_profile_avatar = async function (req, job_id, headers) {
       record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { url }, account_profile_info);
     } else {
       //////////////////
-      imageBase64 = Buffer.from(await images[0].image_blob.arrayBuffer()).toString('base64');
+      imageBase64 = Buffer.from(await image_blob_ret.image_blob.arrayBuffer()).toString('base64');
       if (account_obj.account_info.account_type === 'business') {
         // create avatar from logo - BUSINESS
         const { bio, business_name, country, mainCategory, subCategory } = account_obj.account_info;
         // const ai_avatar_response = await client.images.edit({
         //   model: 'chatgpt-image-latest',
-        //   image: images[0].image_blob, // only one image allowed
+        //   image: image_blob_ret.image_blob, // only one image allowed
         //   prompt: `Generate a realistic fictional portrait wearing a hat. The portrait should be centered, head-and-shoulders only, on a transparent background. Place the provided logo on the hat, and ensure the subject is large enough to fill most of the frame. Use the supplied name, country, and context to guide the character’s appearance.
         // Context: ${JSON.stringify({ bio, business_name, country, mainCategory, subCategory })}`,
         //   background: 'transparent',
@@ -5851,7 +5852,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
         ///////////////////
         // const ai_avatar_response = await client.images.edit({
         //   model: 'chatgpt-image-latest',
-        //   image: images[0].image_blob, //base photo
+        //   image: image_blob_ret.image_blob, //base photo
         //   prompt: `Create a transparent PNG profile image for a business account featuring  building branded as ${business_name}.
         //   Display the attached logo clearly as prominent exterior signage.
         //   The building’s architecture and overall atmosphere should reflect the nature of the business, aligning with ${mainCategory} and ${subCategory}.
@@ -5899,7 +5900,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
         const model = 'chatgpt-image-latest';
         const ai_avatar_response = await client.images.edit({
           model,
-          image: images[0].image_blob, //logo
+          image: image_blob_ret.image_blob, //logo
           prompt: `
           Create a transparent PNG profile image for a business account featuring building branded as ${business_name}. 
           Display the attached logo clearly as prominent exterior signage. 
@@ -5922,7 +5923,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
           const model = 'chatgpt-image-latest';
           const ai_avatar_response = await client.images.edit({
             model,
-            image: images[0].image_blob,
+            image: image_blob_ret.image_blob,
 
             prompt: `Completely remove the background, enhance and color-correct the subject, and return only a centered head-and-shoulders portrait facing the camera. Add extra top margin, include no labels or text, and export the result as a clean transparent PNG.
              Ensure the subject is large enough to fill most of the frame. Use the supplied name, country, and context to guide the character’s appearance.
@@ -5945,9 +5946,9 @@ export const create_profile_avatar = async function (req, job_id, headers) {
         // personal account
         const is_real_person_in_picture = await detect_real_person_in_image(imageBase64);
         if (is_real_person_in_picture) {
-          if (images[0].is_transparent) {
+          if (image_blob_ret.is_transparent) {
             // keep the original
-            imageBase64 = Buffer.from(await images[0].image_blob.arrayBuffer()).toString('base64');
+            imageBase64 = Buffer.from(await image_blob_ret.image_blob.arrayBuffer()).toString('base64');
             avatar_source = 'authentic profile';
           } else {
             // create avatar from profile picture
@@ -6194,72 +6195,76 @@ const delete_depended_chats = async function (uid, reference_id) {
 };
 
 const create_ai_agent_image = async function (req, job_id, headers) {
-  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
-  }
-
-  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();
-
-    let buffer;
-
-    let is_transparent = false;
-    if (ext === '.png') {
-      buffer = await fs.promises.readFile(filePath);
-      is_transparent = await hasTransparentBackground(filePath);
-    } else {
-      console.log(`Converting ${path.basename(filePath)} to PNG...`);
-      buffer = await sharp(filePath).png({ quality: 100, compressionLevel: 9, force: true }).toBuffer();
-    }
-
-    // Create a Blob with correct MIME type
+  const tempDir = os.tmpdir();
+  const uniqueId = Date.now() + '_' + Math.random().toString(36).substring(7);
+  const tempOutputPath = path.join(tempDir, `output_${uniqueId}.png`);
 
-    return { is_transparent, image_blob: new Blob([buffer], { type: 'image/png' }) };
-  }
+  // 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
+  // }
+
+  // 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();
+
+  //   let buffer;
+
+  //   let is_transparent = false;
+  //   if (ext === '.png') {
+  //     buffer = await fs.promises.readFile(filePath);
+  //     is_transparent = await hasTransparentBackground(filePath);
+  //   } else {
+  //     console.log(`Converting ${path.basename(filePath)} to PNG...`);
+  //     buffer = await sharp(filePath).png({ quality: 100, compressionLevel: 9, force: true }).toBuffer();
+  //   }
+
+  //   // Create a Blob with correct MIME type
+
+  //   return { is_transparent, image_blob: new Blob([buffer], { type: 'image/png' }) };
+  // }
 
   const { app_id, uid, ai_agent_id, tags, account_profile_info } = req;
 
@@ -6308,55 +6313,55 @@ const create_ai_agent_image = async function (req, job_id, headers) {
   let url = account_doc.account_info.profile_avatar;
 
   try {
-    if (!url) {
-      throw new Error('No URL provided in request');
-    }
+    // if (!url) {
+    //   throw new Error('No URL provided in request');
+    // }
 
-    // 3. Download the image to a Buffer
-    const res = await fetch(url);
-    if (!res.ok) {
-      throw new Error(`Failed to fetch image URL: ${res.status} ${res.statusText}`);
-    }
+    // // 3. Download the image to a Buffer
+    // const res = await fetch(url);
+    // if (!res.ok) {
+    //   throw new Error(`Failed to fetch image URL: ${res.status} ${res.statusText}`);
+    // }
 
-    const arrayBuffer = await res.arrayBuffer();
-    const imageBuffer = Buffer.from(arrayBuffer);
+    // const arrayBuffer = await res.arrayBuffer();
+    // const imageBuffer = Buffer.from(arrayBuffer);
 
-    const tempDir = os.tmpdir();
-    const uniqueId = Date.now() + '_' + Math.random().toString(36).substring(7);
+    // const tempDir = os.tmpdir();
+    // const uniqueId = Date.now() + '_' + Math.random().toString(36).substring(7);
 
-    const inputExt = path.extname(url) || '.jpg';
-    const tempInputPath = path.join(tempDir, `input_${uniqueId}${inputExt}`);
-    const tempOutputPath = path.join(tempDir, `output_${uniqueId}.png`); // Output is always PNG (transparent)
+    // const inputExt = path.extname(url) || '.jpg';
+    // const tempInputPath = path.join(tempDir, `input_${uniqueId}${inputExt}`);
+    // const tempOutputPath = path.join(tempDir, `output_${uniqueId}.png`); // Output is always PNG (transparent)
 
     // 5. Write the downloaded image to disk
-    fs.writeFileSync(tempInputPath, imageBuffer);
+    // fs.writeFileSync(tempInputPath, imageBuffer);
     /////////////////////////////////
 
-    const imageFiles = [tempInputPath];
+    // const imageFiles = [tempInputPath];
 
-    const images = await Promise.all(
-      imageFiles.map(async (filePath) => {
-        const img_ret = await imageToPngBlob(filePath);
-        const blob = img_ret.image_blob;
+    // const images = await Promise.all(
+    //   imageFiles.map(async (filePath) => {
+    //     const img_ret = await imageToPngBlob(filePath);
+    //     const blob = img_ret.image_blob;
 
-        // OpenAI expects a filename for logging/debugging
-        const filename = path.basename(filePath, path.extname(filePath)) + '.png';
-        blob.name = filename; // Optional but helpful
-        // @ts-ignore – name is allowed in Node 20+
-        blob.lastModified = Date.now();
+    //     // OpenAI expects a filename for logging/debugging
+    //     const filename = path.basename(filePath, path.extname(filePath)) + '.png';
+    //     blob.name = filename; // Optional but helpful
+    //     // @ts-ignore – name is allowed in Node 20+
+    //     blob.lastModified = Date.now();
 
-        return { is_transparent: img_ret.is_transparent, image_blob: blob };
-      }),
-    );
+    //     return { is_transparent: img_ret.is_transparent, image_blob: blob };
+    //   }),
+    // );
 
     let imageBase64;
-
+    const image_blob_ret = await get_image_blob_from_downloaded_image(url);
     //////////////////
-    imageBase64 = Buffer.from(await images[0].image_blob.arrayBuffer()).toString('base64');
+    imageBase64 = Buffer.from(await image_blob_ret.image_blob.arrayBuffer()).toString('base64');
     const model = 'chatgpt-image-latest';
     const ai_avatar_response = await client.images.edit({
       model,
-      image: images[0].image_blob,
+      image: image_blob_ret.image_blob,
 
       prompt: `
                 Create a futuristic, profile portrait with a transparent background. 
@@ -6752,7 +6757,7 @@ const get_profile_image = async function (uid, account_type = 'business', name,
       .replace(/^_+|_+$/g, ''); // trim leading/trailing underscores
   }
 
-  let prompt, factual_prompt, originalname, factual_image_mode, imageBase64, image_blob, file;
+  let prompt, factual_prompt, originalname, factual_image_mode, imageBase64, image_blob_ret, file;
 
   switch (account_type) {
     case 'personal': {
@@ -6804,7 +6809,7 @@ const get_profile_image = async function (uid, account_type = 'business', name,
       // success
 
       try {
-        image_blob = await get_image_blob_from_downloaded_image(url);
+        image_blob_ret = await get_image_blob_from_downloaded_image(url);
       } catch (err) {
         if (!err.message.includes('Failed to fetch image URL')) {
           console.error(err);
@@ -6823,7 +6828,7 @@ const get_profile_image = async function (uid, account_type = 'business', name,
           `;
       const image_response = await client.images.edit({
         model,
-        image: image_blob,
+        image: image_blob_ret.image_blob,
         prompt,
         background: 'transparent',
       });
@@ -6964,5 +6969,5 @@ const get_image_blob_from_downloaded_image = async function (url) {
     }),
   );
 
-  return images[0].image_blob;
+  return images[0];
 };

package/package.json

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