@xuda.io/ai_module 1.1.4362 → 1.1.4364

package/index.mjs

@@ -5679,7 +5679,7 @@ 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`);
 
-  const { url, uid, account_obj, prompt, is_profile_avatar, account_profile_info, docType = 'account', _id, metadata = {}, profile_picture, business_size = 'unknown' } = req;
+  const { url, uid, account_obj, prompt, is_profile_avatar, account_profile_info, docType = 'account', _id, metadata = {}, profile_picture, business_size = 'unknown', name } = req;
 
   // async function normalizeBase64To1024(base64Image) {
   //   const inputBuffer = Buffer.from(base64Image, 'base64');
@@ -5843,7 +5843,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
       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, membership_plan } = metadata;
+        const { bio, country, mainCategory, subCategory } = metadata;
         // const ai_avatar_response = await client.images.edit({
         //   model: 'chatgpt-image-latest',
         //   image: image_blob_ret.image_blob, // only one image allowed
@@ -5928,7 +5928,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
           model,
           image: image_blob_ret.image_blob, //logo
           prompt: `
-          Create a transparent PNG profile image for a business account featuring building branded as ${business_name}. 
+          Create a transparent PNG profile image for a business account featuring building branded as ${name}. 
           Display the attached logo clearly as prominent exterior signage. 
           
           show ${obj?.[business_size] || 'office building'}
@@ -5945,7 +5945,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
       } else {
         /// PERSONAL
         const create_avatar_from_pic = async function () {
-          const { first_name, last_name, bio, country, mainCategory, subCategory } = account_obj.account_info;
+          // const { first_name, last_name, bio, country, mainCategory, subCategory } = account_obj.account_info;
           const model = 'chatgpt-image-latest';
           const ai_avatar_response = await client.images.edit({
             model,
@@ -5953,7 +5953,7 @@ export const create_profile_avatar = async function (req, job_id, headers) {
 
             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.
-            Context: ${JSON.stringify({ first_name, last_name, bio, country, mainCategory, subCategory })}`,
+            Context: ${JSON.stringify({ name, bio, country, mainCategory, subCategory })}`,
             background: 'transparent',
           });
           record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { url }, account_profile_info);
@@ -5961,9 +5961,9 @@ export const create_profile_avatar = async function (req, job_id, headers) {
           avatar_source = 'ai profile';
         };
         const create_fictional_avatar = async function () {
-          const { first_name, last_name, bio, country, mainCategory, subCategory } = account_obj.account_info;
+          // const { name, bio, country, mainCategory, subCategory } = account_obj.account_info;
 
-          const prompt = `analyze the the person info :  ${JSON.stringify({ first_name, last_name, bio, country, mainCategory, subCategory })} 
+          const prompt = `analyze the the person info :  ${JSON.stringify({ name, bio, country, mainCategory, subCategory })} 
             and create realistic fictional profile picture transparent background ,person in the center of the picture return only a centered head-and-shoulders portrait facing the camera. include no labels or text add top margin the person should cover the whole picture`;
 
           imageBase64 = await create_image(uid, prompt, undefined, undefined, 1, undefined, undefined, { url }, account_profile_info);
@@ -6004,35 +6004,39 @@ export const create_profile_avatar = async function (req, job_id, headers) {
       mimetype: `image/${ext}`,
     };
 
-    const drive_ret = await drive_module.upload_drive_file_user({ uid, path: '/Profile Images' }, job_id, headers, file_obj);
+    let drive_ret = await drive_module.upload_drive_file_user({ uid, path: '/Profile Images' }, job_id, headers, file_obj);
+    drive_ret.avatar_source = avatar_source;
 
-    if (drive_ret.code > 0) {
-      if (!is_profile_avatar) {
-        let db_doc = await db_module.get_couch_doc_native('xuda_accounts', account_obj._id);
-        db_doc.account_info.profile_avatar = drive_ret.data.file_url;
-        db_doc.account_info.profile_avatar_obj == drive_ret.data;
-        db_doc.account_info.avatar_source = avatar_source;
-        db_doc.image_request_ts = Date.now();
-        const save_ret = await db_module.save_couch_doc_native('xuda_accounts', db_doc);
-        console.log(save_ret);
-        await update_profile_avatar_status(uid, 3);
-        return save_ret;
-      } else {
-        return drive_ret;
-      }
-
-      // account_module.update_contacts_with_account_changes(db_doc._id, ['profile_avatar']);
-    } else {
-      throw new Error(drive_ret.data);
-    }
+    return drive_ret;
+    // if (drive_ret.code > 0) {
+    //   // if (!is_profile_avatar) {
+    //   //   let db_doc = await db_module.get_couch_doc_native('xuda_accounts', account_obj._id);
+    //   //   db_doc.account_info.profile_avatar = drive_ret.data.file_url;
+    //   //   db_doc.account_info.profile_avatar_obj == drive_ret.data;
+    //   //   db_doc.account_info.avatar_source = avatar_source;
+    //   //   db_doc.image_request_ts = Date.now();
+    //   //   const save_ret = await db_module.save_couch_doc_native('xuda_accounts', db_doc);
+    //   //   console.log(save_ret);
+    //   //   await update_profile_avatar_status(uid, 3);
+    //   //   return save_ret;
+    //   // } else {
+    //   //   return drive_ret;
+    //   // }
+
+    //   // account_module.update_contacts_with_account_changes(db_doc._id, ['profile_avatar']);
+    //   return drive_ret;
+    // } else {
+    //   throw new Error(drive_ret.data);
+    // }
   } catch (error) {
-    if (!is_profile_avatar) {
-      await update_profile_avatar_status(uid, 1);
-    }
+    // if (!is_profile_avatar) {
+    //   await update_profile_avatar_status(uid, 1);
+    // }
     console.error('Error:', error.message);
     if (error.response?.data) {
       console.error('OpenAI API response:', error.response.data);
     }
+    return { code: -56, data: error.message };
   }
 };
 

package/package.json

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