@xuda.io/ai_module 1.1.4316 → 1.1.4318

package/index.mjs

@@ -5289,7 +5289,8 @@ setTimeout(async () => {
   //     Location: 'https://xuda-bucket-dev.nyc3.digitaloceanspaces.com/dev.xuda.io/user/d39126e0e2c51ffbd1aad10709fc8335/cb7353fc-4788-49b7-8e73-f7f1be108b69_drv_d39126e0e2c51ffbd1aad10709fc8335_b92337f79e090a2f287bc3dc35d0033e.png',
   //   },
   // });
-  ret = await get_business_logo(uid, 'bank of america', account_profile_info);
+
+  // ret = await get_business_logo(uid, 'bank of america', account_profile_info);
 
   console.log(ret);
 }, 1000);
@@ -6724,6 +6725,15 @@ export const is_business_contact = async function (uid, email, name, subject, bo
 };
 
 const get_business_logo = async function (uid, name, account_profile_info) {
+  function normalizeFilename(name) {
+    return name
+      .toLowerCase() // lowercase
+      .replace(/\s+/g, '_') // replace all spaces with single underscore
+      .replace(/[^a-z0-9._-]/g, '') // remove anything that's not letter, number, dot, underscore, or hyphen
+      .replace(/_+/g, '_') // collapse multiple underscores
+      .replace(/^_+|_+$/g, ''); // trim leading/trailing underscores
+  }
+
   const get_url = async function () {
     const ret = await submit_chat_gpt_prompt({
       uid,
@@ -6768,7 +6778,7 @@ const get_business_logo = async function (uid, name, account_profile_info) {
       prompt,
       background: 'transparent',
     });
-    debugger;
+
     let imageBase64 = await normalizeBase64To1024(logo_response.data[0].b64_json);
 
     const outputBuffer = Buffer.from(imageBase64, 'base64');
@@ -6778,6 +6788,15 @@ const get_business_logo = async function (uid, name, account_profile_info) {
     console.log('Success! Saved →', outputPath);
 
     console.log(file);
+
+    const file_obj = {
+      originalname: normalizeFilename(name) + '_xu_logo.png',
+      path: outputPath,
+      mimetype: `application/png`,
+    };
+
+    const drive_ret = await drive_module.upload_drive_file_user({ uid, path: '/Profile Images' }, job_id, headers, file_obj);
+
     record_chat_usage(uid, logo_response.usage.input_tokens, logo_response.usage.output_tokens, 'business logo', prompt, model, { url }, account_profile_info);
     // prompt: `find  "${name}" logo base on the following facts  "${facts}", Return only  png,jpg img tag , nothing else`,
     // try {
@@ -6786,6 +6805,8 @@ const get_business_logo = async function (uid, name, account_profile_info) {
     //     return data.is_business;
     //   }
     // } catch (error) {}
+
+    return drive_ret;
   } catch (err) {
     return console.error(err);
   }
@@ -6858,49 +6879,49 @@ const return_image_blob_from_downloaded_image = async function (url) {
 
     return { is_transparent, image_blob: new Blob([buffer], { type: 'image/png' }) };
   }
-  try {
-    if (!url) {
-      throw new Error('No URL provided in request');
-    }
+  // try {
+  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);
-    /////////////////////////////////
+  // 5. Write the downloaded image to disk
+  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 { image_blob: images[0].image_blob, is_transparent: images[0].is_transparent, tempOutputPath };
-  } catch (err) {
-    console.error(err);
-  }
+      return { is_transparent: img_ret.is_transparent, image_blob: blob };
+    }),
+  );
+  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.4316",
+  "version": "1.1.4318",
   "description": "Xuda AI Module",
   "main": "index.mjs",
   "type": "module",