@xuda.io/ai_module 1.1.4865 → 1.1.4867

package/index.mjs

@@ -5470,6 +5470,194 @@ export const get_profile_avatar = async function (profile_picture, uid, prompt,
   }
 };
 
+export const get_profile_picture = async function (uid, account_type = 'business', name, metadata = {}, 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`);
+
+  const { email, create_avatar } = metadata;
+
+  let prompt,
+    url,
+    factual_prompt,
+    originalname,
+    factual_image_mode,
+    imageBase64,
+    image_blob_ret,
+    file,
+    profile_picture_source,
+    drive_ret = { code: 1, data: {} },
+    resize = {};
+
+  switch (account_type) {
+    case 'personal': {
+      prompt = `find  "${name}" person profile image base on the following facts: "${JSON.stringify(metadata)}", Return only png,jpg image url single result e.g: "https://fb.com/person_image.png", nothing else`;
+      factual_prompt = `analyze the the person info base on the name and the facts: {"name": "${name || ''}", facts: "${JSON.stringify(
+        metadata,
+      )}" and create realistic fictional profile picture transparent background ,person in the center of the picture  add top margin the person should cover the whole picture`;
+      originalname = normalizeFilename(name) + '_xu_profile.png';
+      resize = {
+        fit: 'inside', // Scale up/down to fit inside 1024×1024 while preserving aspect ratio
+        withoutEnlargement: false, // Allow enlargement if original is smaller
+        background: { r: 0, g: 0, b: 0, alpha: 0 },
+      };
+      break;
+    }
+
+    case 'business': {
+      prompt = `Search for a large, high-quality "${name}" logo or icon in PNG or JPG format. Output **exclusively** a single valid image URL, like this: "https://example.com/large-logo.png". Do not include any other text, descriptions, or additional information.`;
+      factual_prompt = `create logo for the company named: "${name}" the logo should only be initials of the name or geometric shape`;
+      originalname = normalizeFilename(name) + '_xu_logo.png';
+      resize = {
+        fit: 'cover', // Fill the 1024×1024 area (may crop edges)
+        position: 'centre', // Center the crop
+        background: { r: 0, g: 0, b: 0, alpha: 0 },
+      };
+      break;
+    }
+
+    default:
+      throw new Error('account_type not supported');
+      break;
+  }
+
+  const getUserPhotoByEmail = async function (email) {
+    try {
+      const auth = admin.auth();
+      const userRecord = await auth.getUserByEmail(email);
+
+      // photoURL is the profile picture URL (or null if not set)
+      const photoURL = userRecord.photoURL;
+      // console.log(`User found: ${JSON.stringify(userRecord)}`);
+
+      // console.log(`User found: ${userRecord.uid}`);
+      // console.log(`Display name: ${userRecord.displayName || 'None'}`);
+      // console.log(`Email verified: ${userRecord.emailVerified}`);
+      // console.log(`Profile picture URL: ${photoURL || 'No photo set'}`);
+
+      return photoURL;
+    } catch (error) {
+      if (error.code === 'auth/user-not-found') {
+        console.log('getUserPhotoByEmail', `No user found with email: ${email}`);
+        return null;
+      }
+
+      console.error('Error fetching user:', error);
+      // throw error;
+    }
+  };
+
+  const get_url = async function () {
+    const ret = await submit_chat_gpt_prompt({
+      uid,
+      prompt,
+      model: 'gpt-5-nano',
+      metadata: { func: 'get_profile_picture' },
+      account_profile_info,
+      tools: [{ type: 'web_search_preview' }],
+    });
+
+    return ret.data;
+  };
+
+  // debugger;
+  try {
+    url = await getUserPhotoByEmail(email);
+    if (url) {
+      try {
+        image_blob_ret = await get_image_blob_from_downloaded_image(url);
+        profile_picture_source = 'google';
+        drive_ret.data = { file_url: url };
+      } catch (err) {
+        if (!err.message.includes('Failed to fetch image URL')) {
+          console.error(err);
+          return '';
+        }
+        url = '';
+      }
+    }
+
+    if (!url && create_avatar) {
+      // first attempt
+      if (account_type === 'business') {
+        url = await get_url();
+
+        if (url.substr(0, 8) !== 'https://') {
+          profile_picture_source = 'factual';
+          // second attempt
+          url = await get_url();
+          if (url.substr(0, 8) !== 'https://') {
+            factual_image_mode = true;
+            profile_picture_source = 'factual';
+          }
+        } else {
+          // success
+          profile_picture_source = 'internet search';
+          try {
+            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);
+              return '';
+            }
+            profile_picture_source = 'factual';
+            factual_image_mode = true;
+          }
+        }
+      } else {
+        factual_image_mode = true;
+        profile_picture_source = 'factual';
+      }
+
+      if (factual_image_mode) {
+        imageBase64 = await create_image(uid, factual_prompt, undefined, undefined, 1, undefined, undefined, {}, account_profile_info);
+      } else {
+        const model = 'gpt-image-1-mini'; // 'chatgpt-image-latest';
+        const prompt = `
+          extract the logo from the image
+          `;
+        let image_response;
+        try {
+          image_response = await client.images.edit({
+            model,
+            image: image_blob_ret.image_blob,
+            prompt,
+            background: 'transparent',
+          });
+          report_ai_status(model);
+        } catch (err) {
+          report_ai_status(model, err);
+          throw err;
+        }
+
+        imageBase64 = await normalizeBase64To1024(image_response.data[0].b64_json, resize);
+        account_module.record_ai_usage(uid, image_response.usage.input_tokens, image_response.usage.output_tokens, 'profile image', prompt, model, { url }, account_profile_info);
+      }
+
+      const outputBuffer = Buffer.from(imageBase64, 'base64');
+      const outputPath = tempOutputPath;
+      await fs.promises.writeFile(outputPath, outputBuffer);
+
+      console.log('Success! Saved →', outputPath);
+
+      console.log(file);
+
+      const file_obj = {
+        originalname,
+        path: outputPath,
+        mimetype: `image/png`,
+      };
+
+      drive_ret = await drive_module.upload_drive_file_user({ uid, path: '/Profile Images' }, job_id, headers, file_obj);
+      drive_ret.profile_picture_source = profile_picture_source;
+    }
+    return drive_ret;
+  } catch (err) {
+    console.error(err);
+    return;
+  }
+};
+
 export const conversation_actions = async function (req, job_id, headers) {
   const { conversation_id, item_id, action, uid, account_profile_info } = req;
   try {
@@ -6154,194 +6342,6 @@ function normalizeFilename(name) {
     .replace(/^_+|_+$/g, ''); // trim leading/trailing underscores
 }
 
-export const get_profile_picture = async function (uid, account_type = 'business', name, metadata = {}, 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`);
-
-  const { email, create_avatar } = metadata;
-
-  let prompt,
-    url,
-    factual_prompt,
-    originalname,
-    factual_image_mode,
-    imageBase64,
-    image_blob_ret,
-    file,
-    profile_picture_source,
-    drive_ret = { code: 1, data: {} },
-    resize = {};
-
-  switch (account_type) {
-    case 'personal': {
-      prompt = `find  "${name}" person profile image base on the following facts: "${JSON.stringify(metadata)}", Return only png,jpg image url single result e.g: "https://fb.com/person_image.png", nothing else`;
-      factual_prompt = `analyze the the person info base on the name and the facts: {"name": "${name || ''}", facts: "${JSON.stringify(
-        metadata,
-      )}" and create realistic fictional profile picture transparent background ,person in the center of the picture  add top margin the person should cover the whole picture`;
-      originalname = normalizeFilename(name) + '_xu_profile.png';
-      resize = {
-        fit: 'inside', // Scale up/down to fit inside 1024×1024 while preserving aspect ratio
-        withoutEnlargement: false, // Allow enlargement if original is smaller
-        background: { r: 0, g: 0, b: 0, alpha: 0 },
-      };
-      break;
-    }
-
-    case 'business': {
-      prompt = `Search for a large, high-quality "${name}" logo or icon in PNG or JPG format. Output **exclusively** a single valid image URL, like this: "https://example.com/large-logo.png". Do not include any other text, descriptions, or additional information.`;
-      factual_prompt = `create logo for the company named: "${name}" the logo should only be initials of the name or geometric shape`;
-      originalname = normalizeFilename(name) + '_xu_logo.png';
-      resize = {
-        fit: 'cover', // Fill the 1024×1024 area (may crop edges)
-        position: 'centre', // Center the crop
-        background: { r: 0, g: 0, b: 0, alpha: 0 },
-      };
-      break;
-    }
-
-    default:
-      throw new Error('account_type not supported');
-      break;
-  }
-
-  const getUserPhotoByEmail = async function (email) {
-    try {
-      const auth = admin.auth();
-      const userRecord = await auth.getUserByEmail(email);
-
-      // photoURL is the profile picture URL (or null if not set)
-      const photoURL = userRecord.photoURL;
-      // console.log(`User found: ${JSON.stringify(userRecord)}`);
-
-      // console.log(`User found: ${userRecord.uid}`);
-      // console.log(`Display name: ${userRecord.displayName || 'None'}`);
-      // console.log(`Email verified: ${userRecord.emailVerified}`);
-      // console.log(`Profile picture URL: ${photoURL || 'No photo set'}`);
-
-      return photoURL;
-    } catch (error) {
-      if (error.code === 'auth/user-not-found') {
-        console.log(`No user found with email: ${email}`);
-        return null;
-      }
-
-      console.error('Error fetching user:', error);
-      // throw error;
-    }
-  };
-
-  const get_url = async function () {
-    const ret = await submit_chat_gpt_prompt({
-      uid,
-      prompt,
-      model: 'gpt-5-nano',
-      metadata: { func: 'get_profile_picture' },
-      account_profile_info,
-      tools: [{ type: 'web_search_preview' }],
-    });
-
-    return ret.data;
-  };
-
-  // debugger;
-  try {
-    url = await getUserPhotoByEmail(email);
-    if (url) {
-      try {
-        image_blob_ret = await get_image_blob_from_downloaded_image(url);
-        profile_picture_source = 'google';
-        drive_ret.data = { file_url: url };
-      } catch (err) {
-        if (!err.message.includes('Failed to fetch image URL')) {
-          console.error(err);
-          return '';
-        }
-        url = '';
-      }
-    }
-
-    if (!url && create_avatar) {
-      // first attempt
-      if (account_type === 'business') {
-        url = await get_url();
-
-        if (url.substr(0, 8) !== 'https://') {
-          profile_picture_source = 'factual';
-          // second attempt
-          url = await get_url();
-          if (url.substr(0, 8) !== 'https://') {
-            factual_image_mode = true;
-            profile_picture_source = 'factual';
-          }
-        } else {
-          // success
-          profile_picture_source = 'internet search';
-          try {
-            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);
-              return '';
-            }
-            profile_picture_source = 'factual';
-            factual_image_mode = true;
-          }
-        }
-      } else {
-        factual_image_mode = true;
-        profile_picture_source = 'factual';
-      }
-
-      if (factual_image_mode) {
-        imageBase64 = await create_image(uid, factual_prompt, undefined, undefined, 1, undefined, undefined, {}, account_profile_info);
-      } else {
-        const model = 'gpt-image-1-mini'; // 'chatgpt-image-latest';
-        const prompt = `
-          extract the logo from the image
-          `;
-        let image_response;
-        try {
-          image_response = await client.images.edit({
-            model,
-            image: image_blob_ret.image_blob,
-            prompt,
-            background: 'transparent',
-          });
-          report_ai_status(model);
-        } catch (err) {
-          report_ai_status(model, err);
-          throw err;
-        }
-
-        imageBase64 = await normalizeBase64To1024(image_response.data[0].b64_json, resize);
-        account_module.record_ai_usage(uid, image_response.usage.input_tokens, image_response.usage.output_tokens, 'profile image', prompt, model, { url }, account_profile_info);
-      }
-
-      const outputBuffer = Buffer.from(imageBase64, 'base64');
-      const outputPath = tempOutputPath;
-      await fs.promises.writeFile(outputPath, outputBuffer);
-
-      console.log('Success! Saved →', outputPath);
-
-      console.log(file);
-
-      const file_obj = {
-        originalname,
-        path: outputPath,
-        mimetype: `image/png`,
-      };
-
-      drive_ret = await drive_module.upload_drive_file_user({ uid, path: '/Profile Images' }, job_id, headers, file_obj);
-      drive_ret.profile_picture_source = profile_picture_source;
-    }
-    return drive_ret;
-  } catch (err) {
-    console.error(err);
-    return;
-  }
-};
-
 async function normalizeBase64To1024(
   base64Image,
   resize = {

package/package.json

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