@xuda.io/ai_module 1.1.4371 → 1.1.4373

package/index.mjs

@@ -5674,7 +5674,7 @@ export const create_account_profile_avatar = async function (req, job_id, header
   }
 };
 
-export const get_profile_avatar = async function (url, uid, prompt, account_profile_info, account_type, docType = 'account', _id, metadata = {}, profile_picture, business_size = 'unknown', name, email, job_id, headers) {
+export const get_profile_avatar = async function (profile_picture, uid, prompt, account_profile_info, account_type, docType = 'account', _id, metadata = {}, business_size = 'unknown', name, email, 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`);
@@ -5826,7 +5826,7 @@ export const get_profile_avatar = async function (url, uid, prompt, account_prof
     // );
 
     let imageBase64;
-    const image_blob_ret = await get_image_blob_from_downloaded_image(url);
+    const image_blob_ret = await get_image_blob_from_downloaded_image(profile_picture);
     let avatar_source = '';
     if (prompt) {
       const model = 'chatgpt-image-latest';
@@ -5837,7 +5837,7 @@ export const get_profile_avatar = async function (url, uid, prompt, account_prof
         background: 'transparent',
       });
       imageBase64 = ai_avatar_response.data[0].b64_json;
-      record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { url }, account_profile_info);
+      record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { profile_picture }, account_profile_info);
     } else {
       //////////////////
       imageBase64 = Buffer.from(await image_blob_ret.image_blob.arrayBuffer()).toString('base64');
@@ -5939,7 +5939,7 @@ export const get_profile_avatar = async function (url, uid, prompt, account_prof
           The background should be fully removed, remove trees,clouds,sky or any landscape objects`,
           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);
+        record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { profile_picture }, account_profile_info);
         imageBase64 = ai_avatar_response.data[0].b64_json;
         avatar_source = 'fictional';
       } else {
@@ -5956,7 +5956,7 @@ export const get_profile_avatar = async function (url, uid, prompt, account_prof
             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);
+          record_chat_usage(uid, ai_avatar_response.usage.input_tokens, ai_avatar_response.usage.output_tokens, 'profile avatar', prompt, model, { profile_picture }, account_profile_info);
           imageBase64 = ai_avatar_response.data[0].b64_json;
           avatar_source = 'ai profile';
         };
@@ -5966,7 +5966,7 @@ export const get_profile_avatar = async function (url, uid, prompt, account_prof
           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);
+          imageBase64 = await create_image(uid, prompt, undefined, undefined, 1, undefined, undefined, { url: profile_picture }, account_profile_info);
           avatar_source = 'fictional';
         };
         // personal account
@@ -7005,3 +7005,128 @@ const get_image_blob_from_downloaded_image = async function (url) {
 
   return images[0];
 };
+
+export const get_business_info = async function (uid, name, email, account_profile_info) {
+  // 1. Prepare the valid values from your categories list
+  const categories = [
+    {
+      id: 'ind_technology',
+      name: 'Technology',
+      sub: ['Software Development', 'AI & Machine Learning', 'Cybersecurity', 'Cloud Computing', 'IT Services', 'SaaS & Platforms', 'Hardware & Electronics'],
+    },
+    {
+      id: 'ind_marketing',
+      name: 'Marketing & Advertising',
+      sub: ['Digital Marketing', 'PPC & SEO', 'Content Creation', 'Branding', 'Social Media Management', 'Email Marketing', 'Market Research'],
+    },
+    {
+      id: 'ind_sales',
+      name: 'Sales',
+      sub: ['B2B Sales', 'B2C Sales', 'Real Estate Sales', 'Account Management', 'Lead Generation', 'Inside Sales', 'Field Sales'],
+    },
+    {
+      id: 'ind_finance',
+      name: 'Finance & Accounting',
+      sub: ['Banking', 'Investment', 'Financial Planning', 'Accounting', 'Bookkeeping', 'Insurance', 'FinTech'],
+    },
+    {
+      id: 'ind_healthcare',
+      name: 'Healthcare',
+      sub: ['Medical Clinics', 'Hospitals', 'Pharma', 'Mental Health', 'Healthcare IT', 'Home Care', 'Medical Research'],
+    },
+    {
+      id: 'ind_realestate',
+      name: 'Real Estate',
+      sub: ['Residential Real Estate', 'Commercial Real Estate', 'Property Management', 'Investment & BRRRR', 'Construction', 'Architecture', 'Interior Design'],
+    },
+    {
+      id: 'ind_ecommerce',
+      name: 'E-commerce',
+      sub: ['Online Store', 'Dropshipping', 'Amazon FBA', 'Marketplace Selling', 'Retail', 'Inventory Management', 'Fulfillment & Logistics'],
+    },
+    {
+      id: 'ind_education',
+      name: 'Education',
+      sub: ['Schools', 'Universities', 'Private Tutoring', 'Online Courses', 'EdTech', 'Training & Development'],
+    },
+    {
+      id: 'ind_hospitality',
+      name: 'Hospitality & Travel',
+      sub: ['Hotels', 'Restaurants', 'Cafes & Bars', 'Travel Agencies', 'Tourism', 'Events & Catering', 'Transportation'],
+    },
+    {
+      id: 'ind_professional',
+      name: 'Professional Services',
+      sub: ['Legal', 'Consulting', 'HR & Recruiting', 'Accounting Firms', 'Business Services', 'Management Consulting'],
+    },
+    {
+      id: 'ind_manufacturing',
+      name: 'Manufacturing',
+      sub: ['Automotive', 'Industrial', 'Textile', 'Electronics', 'Food Production', 'Machinery', 'Chemicals'],
+    },
+    {
+      id: 'ind_nonprofit',
+      name: 'Nonprofit & Government',
+      sub: ['NGOs', 'Municipality', 'Government Services', 'Public Health', 'Community Organizations', 'Education Programs'],
+    },
+    {
+      id: 'ind_media',
+      name: 'Media & Entertainment',
+      sub: ['Film & Video', 'Music', 'Publishing', 'Gaming', 'News & Journalism', 'Influencers', 'Production Studios'],
+    },
+    {
+      id: 'ind_logistics',
+      name: 'Logistics & Transportation',
+      sub: ['Shipping', 'Trucking', 'Last-Mile Delivery', 'Warehousing', 'Freight Forwarding', 'Route Optimization'],
+    },
+    {
+      id: 'ind_energy',
+      name: 'Energy & Utilities',
+      sub: ['Solar', 'Oil & Gas', 'Electricity Providers', 'Water Utilities', 'Environmental Services', 'Renewable Energy'],
+    },
+    {
+      id: 'ind_construction',
+      name: 'Construction & Trades',
+      sub: ['General Contracting', 'Roofing', 'Plumbing', 'HVAC', 'Electrical', 'Renovation', 'Engineering'],
+    },
+    {
+      id: 'ind_food',
+      name: 'Food & Beverage',
+      sub: ['Restaurants', 'Catering', 'Food Manufacturing', 'Bakeries', 'Bars & Nightlife', 'Food Delivery'],
+    },
+    {
+      id: 'ind_retail',
+      name: 'Retail',
+      sub: ['Physical Stores', 'Boutiques', 'Supermarkets', 'Consumer Goods', 'Fashion Retail', 'Home Goods'],
+    },
+  ];
+
+  const categoryNames = categories.map((c) => c.name);
+  const allSubCategories = categories.flatMap((c) => c.sub);
+
+  const business_info_ret = await submit_chat_gpt_prompt({
+    uid,
+    prompt: `Research the company "${name}" (${email}). Identify their business size, 
+             main industry category, and specific sub-category based on the provided lists.`,
+    model: 'gpt-5-nano',
+    response_format: z.object({
+      business_size: z.enum(['unknown', 'small', 'medium', 'large', 'enterprise']),
+      business_category: z.enum(categoryNames).describe('The primary industry category name'),
+      business_sub_category: z.enum(allSubCategories).describe('The specific niche within that category'),
+      business_country: z.string().describe('Primary country of operation'),
+    }),
+    metadata: { func: 'get_business_info' },
+    account_profile_info,
+    tools: [{ type: 'web_search_preview' }],
+  });
+
+  try {
+    if (business_info_ret.code > -1) {
+      const data = typeof business_info_ret.data === 'string' ? JSON.parse(business_info_ret.data) : business_info_ret.data;
+
+      return data;
+    }
+  } catch (error) {
+    console.error('Schema validation or parsing error', error);
+  }
+};

package/package.json

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