@xuda.io/ai_module 1.1.4313 → 1.1.4314

package/index.mjs

@@ -6735,52 +6735,54 @@ const get_business_logo = async function (uid, name, account_profile_info) {
       tools: [{ type: 'web_search_preview' }],
     });
     console.log('=============', ret);
+    if (!ret.data.includes('https://')) throw new Error('not valid url');
     return ret.data;
   };
-
-  let url = await get_url();
-  let file;
   try {
-    file = await return_image_blob_from_downloaded_image(url);
-  } catch (err) {
-    if (!err.message.includes('Failed to fetch image URL')) return console.error(err);
-
+    let url = await get_url();
+    let file;
     try {
-      url = await get_url();
       file = await return_image_blob_from_downloaded_image(url);
     } catch (err) {
-      return console.error(err);
+      if (!err.message.includes('Failed to fetch image URL')) return console.error(err);
+
+      try {
+        url = await get_url();
+        file = await return_image_blob_from_downloaded_image(url);
+      } catch (err) {
+        return console.error(err);
+      }
     }
-  }
-  // 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 = `
+    // 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(logo_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`,
-  // try {
-  //   if (ret.code > -1) {
-  //     const data = JSON.parse(ret.data);
-  //     return data.is_business;
-  //   }
-  // } catch (error) {}
+    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(logo_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`,
+    // try {
+    //   if (ret.code > -1) {
+    //     const data = JSON.parse(ret.data);
+    //     return data.is_business;
+    //   }
+    // } catch (error) {}
+  } catch (err) {}
 };
 
 async function normalizeBase64To1024(base64Image) {

package/package.json

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