@xuda.io/ai_module 1.1.4308 → 1.1.4310

package/index.mjs

@@ -5289,7 +5289,7 @@ 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,21 +6724,40 @@ export const is_business_contact = async function (uid, email, name, subject, bo
 };
 
 const get_business_logo = async function (uid, name, account_profile_info) {
-  const ret = await submit_chat_gpt_prompt({
-    uid,
-    prompt: `find  "${name}" icon/logo, Return only png,jpg image url single result e.g: "https://example.com/logo_image.png", nothing else`,
-    model: 'gpt-5-nano',
-    // response_format: z.object({
-    //   is_business: z.boolean().describe('true if this is a business account/contact, false otherwise'),
-    // }),
-    metadata: { func: 'get_business_logo' },
-    account_profile_info,
-    tools: [{ type: 'web_search_preview' }],
-  });
-  console.log('=============', ret);
+  const get_url = async function () {
+    const ret = await submit_chat_gpt_prompt({
+      uid,
+      prompt: `find  "${name}" icon/logo, Return only png,jpg image url single result e.g: "https://example.com/logo_image.png", nothing else`,
+      model: 'gpt-5-nano',
 
+      metadata: { func: 'get_business_logo' },
+      account_profile_info,
+      tools: [{ type: 'web_search_preview' }],
+    });
+    console.log('=============', ret);
+    return ret.data;
+  };
   const url = ret.data;
   const file = await return_image_blob_from_downloaded_image(url);
+  // const imageBase64 = Buffer.from(await file.image_blob.arrayBuffer()).toString('base64');
+  const model = 'chatgpt-image-latest';
+  const prompt = `
+          extract the logo from the image
+          `;
+  const logo_response = await client.images.edit({
+    model,
+    image: file.image_blob, //logo
+    prompt,
+    background: 'transparent',
+  });
+  record_chat_usage(uid, logo_response.usage.input_tokens, logo_response.usage.output_tokens, 'business logo', prompt, model, { url }, account_profile_info);
+  imageBase64 = await normalizeBase64To1024(ai_avatar_response.data[0].b64_json);
+
+  const outputBuffer = Buffer.from(imageBase64, 'base64');
+  const outputPath = file.tempOutputPath;
+  await fs.promises.writeFile(outputPath, outputBuffer);
+
+  console.log('Success! Saved →', outputPath);
 
   console.log(file);
   // prompt: `find  "${name}" logo base on the following facts  "${facts}", Return only  png,jpg img tag , nothing else`,
@@ -6750,6 +6769,24 @@ const get_business_logo = async function (uid, name, account_profile_info) {
   // } catch (error) {}
 };
 
+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
+}
+
 const return_image_blob_from_downloaded_image = async function (url) {
   async function hasTransparentBackground(filePath) {
     const image = sharp(filePath);
@@ -6840,7 +6877,7 @@ const return_image_blob_from_downloaded_image = async function (url) {
         return { is_transparent: img_ret.is_transparent, image_blob: blob };
       }),
     );
-    return images[0];
+    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.4308",
+  "version": "1.1.4310",
   "description": "Xuda AI Module",
   "main": "index.mjs",
   "type": "module",