@xuda.io/ai_module 1.1.4356 → 1.1.4358

package/index.mjs

@@ -5290,8 +5290,8 @@ setTimeout(async () => {
   //   },
   // });
 
-  // ret = await get_profile_image(uid, 'business', 'ioshka media', {}, account_profile_info);
-  // ret = await get_profile_image(
+  // ret = await get_profile_picture(uid, 'business', 'ioshka media', {}, account_profile_info);
+  // ret = await get_profile_picture(
   //   uid,
   //   'personal',
   //   'Glenn Willett',
@@ -5679,72 +5679,74 @@ 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 { url, uid, account_obj, prompt, is_profile_avatar, account_profile_info, docType = 'account', _id, metadata = {} } = req;
 
-    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();
+  // async function normalizeBase64To1024(base64Image) {
+  //   const inputBuffer = Buffer.from(base64Image, 'base64');
 
-    return normalizedBuffer.toString('base64'); // Return normalized base64 string
-  }
+  //   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();
 
-  async function hasTransparentBackground(filePath) {
-    const image = sharp(filePath);
-    const metadata = await image.metadata();
+  //   return normalizedBuffer.toString('base64'); // Return normalized base64 string
+  // }
 
-    // If the image doesn't have an alpha channel at all → no transparency possible
-    if (!metadata.hasAlpha) {
-      return false;
-    }
+  // async function hasTransparentBackground(filePath) {
+  //   const image = sharp(filePath);
+  //   const metadata = await image.metadata();
 
-    // For images with alpha channel, check if any pixel actually has alpha < 255
-    const { data, info } = await image.raw().toBuffer({ resolveWithObject: true });
+  //   // If the image doesn't have an alpha channel at all → no transparency possible
+  //   if (!metadata.hasAlpha) {
+  //     return false;
+  //   }
 
-    // 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;
+  //   // For images with alpha channel, check if any pixel actually has alpha < 255
+  //   const { data, info } = await image.raw().toBuffer({ resolveWithObject: true });
 
-    if (alphaOffset === -1) {
-      // Unusual channel count – be conservative and assume it might have transparency
-      return 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;
 
-    for (let i = alphaOffset; i < data.length; i += channels) {
-      if (data[i] < 255) {
-        return true; // Found at least one transparent/semi-transparent pixel
-      }
-    }
+  //   if (alphaOffset === -1) {
+  //     // Unusual channel count – be conservative and assume it might have transparency
+  //     return true;
+  //   }
 
-    return false; // Has alpha channel but all pixels are fully opaque
-  }
+  //   for (let i = alphaOffset; i < data.length; i += channels) {
+  //     if (data[i] < 255) {
+  //       return true; // Found at least one transparent/semi-transparent pixel
+  //     }
+  //   }
 
-  async function imageToPngBlob(filePath) {
-    const ext = path.extname(filePath).toLowerCase();
+  //   return false; // Has alpha channel but all pixels are fully opaque
+  // }
 
-    let buffer;
+  // async function imageToPngBlob(filePath) {
+  //   const ext = path.extname(filePath).toLowerCase();
 
-    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();
-    }
+  //   let buffer;
 
-    // Create a Blob with correct MIME type
+  //   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();
+  //   }
 
-    return { is_transparent, image_blob: new Blob([buffer], { type: 'image/png' }) };
-  }
+  //   // 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({
@@ -5777,59 +5779,60 @@ export const create_profile_avatar = async function (req, job_id, headers) {
   };
 
   // const drive_module = await import(`${module_path}/drive_module/index.mjs`);
-  const { url, uid, account_obj, prompt, is_profile_avatar, account_profile_info } = req;
+
   if (!is_profile_avatar) {
     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 +5840,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 +5854,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 +5902,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 +5925,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 +5948,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
@@ -6198,72 +6201,72 @@ const create_ai_agent_image = 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 { app_id, uid, ai_agent_id, tags, account_profile_info } = req;
 
@@ -6312,55 +6315,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. 
@@ -6742,7 +6745,7 @@ export const is_business_contact = async function (uid, email, name, subject, bo
   } catch (error) {}
 };
 
-const get_profile_image = async function (uid, account_type = 'business', name, facts = {}, account_profile_info, job_id, headers) {
+export const get_profile_picture = async function (uid, account_type = 'business', name, facts = {}, account_profile_info, job_id, headers) {
   const tempDir = os.tmpdir();
   const uniqueId = Date.now() + '_' + Math.random().toString(36).substring(7);
   const tempOutputPath = path.join(tempDir, `output_${uniqueId}.png`);
@@ -6756,7 +6759,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': {
@@ -6785,7 +6788,7 @@ const get_profile_image = async function (uid, account_type = 'business', name,
       uid,
       prompt,
       model: 'gpt-5-nano',
-      metadata: { func: 'get_profile_image' },
+      metadata: { func: 'get_profile_picture' },
       account_profile_info,
       tools: [{ type: 'web_search_preview' }],
     });
@@ -6808,7 +6811,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);
@@ -6827,7 +6830,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',
       });
@@ -6968,5 +6971,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.4356",
+  "version": "1.1.4358",
   "description": "Xuda AI Module",
   "main": "index.mjs",
   "type": "module",