测试

src/service/article-service.ts

@@ -2,20 +2,28 @@ import fs from 'fs';
 import htmlDocx from 'html-docx-js';
 import path from 'path';
 
-import { delay, getImageBuffer, getSummaryText, getTranslatedText } from '@/utils';
+import { delay, getImageBuffer, getSummaryText, getTranslatedText, retryOnce } from '@/utils';
 
 const uploadDir = path.join(__dirname, '../../uploads');
 
 async function getImageBufferByText(text: string, apiKey: string) {
+  console.log(`------------ 开始获取图片，原文案：${text} ---------------`);
+  const translatedText = await retryOnce(() => getTranslatedText(text, apiKey));
+  console.log(`translatedText：${translatedText}`);
+  const summaryText = await retryOnce(() => getSummaryText(translatedText, apiKey));
+  console.log(`summaryText:${summaryText}`);
   // 使用axios以async/await形式调用API接口生成图片
-  const translatedText = await getTranslatedText(text, apiKey);
-  const summaryText = await getSummaryText(translatedText, apiKey);
-  console.log(summaryText);
-  // 使用axios以async/await形式调用API接口生成图片
-  const buffer = await getImageBuffer(summaryText, apiKey);
+  const buffer = await retryOnce(() => getImageBuffer(summaryText, apiKey));
+  console.log('------------- 图片buffer获取成功 ---------------');
   return buffer;
 }
 
+async function preheat(apiKey: string) {
+  try {
+    await retryOnce(() => getImageBufferByText('预热', apiKey));
+  } catch (error) {}
+}
+
 export const processArticleServe = async (data: { title: string; content: string; config: { output: string; hfApiKey: string } }) => {
   const { title, content, config } = data;
 
@@ -24,10 +32,7 @@ export const processArticleServe = async (data: { title: string; content: string
 
   let htmlWithImages = '';
 
-  // 预热
-  try {
-    await getImageBufferByText('预热', config.hfApiKey);
-  } catch (error) {}
+  await preheat(config.hfApiKey);
 
   // 遍历段落，根据段落内容调用接口生成图片并插入到HTML中
   for (let i = 0; i < paragraphs.length; i++) {
@@ -84,13 +89,8 @@ export const processArticleServe = async (data: { title: string; content: string
     const outputFilename = `docx-${Date.now()}.docx`;
     const outputPath = path.join(uploadDir, outputFilename);
 
-    const docxBlob = htmlDocx.asBlob(htmlContent) as Blob;
-    // 将 Blob 转换为 ArrayBuffer
-    const arrayBuffer = await docxBlob.arrayBuffer();
-    // 将 ArrayBuffer 转换为 Buffer
+    const arrayBuffer = htmlDocx.asBlob(htmlContent) as ArrayBuffer;
     const docxBuffer = Buffer.from(arrayBuffer);
-
-    // 将 DOCX Buffer 保存为文件
     fs.writeFileSync(outputPath, docxBuffer);
 
     return {

src/utils/common.ts

@@ -1,4 +1,3 @@
-import axios from 'axios';
 import util from 'util';
 const setTimeoutPromise = util.promisify(setTimeout);
 
@@ -7,38 +6,13 @@ export async function delay(time: number) {
   console.log(`延迟${time}毫秒后执行`);
 }
 
-
-export async function getTranslatedText(text:string, apiKey:string){
-  const translatedData = await axios.post(
-    'https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-zh-en',
-    { inputs: text, src_lang: 'ch', target_lang: 'en' },
-    {
-      headers: {
-        'Content-Type': 'application/json',
-        // 如果需要认证
-        Authorization: `Bearer ${apiKey}`,
-      }
-    },
-  );
-  const translatedText = translatedData.data[0].translation_text as string;
-
-  return translatedText
+export async function retryOnce(func: () => Promise<any>, timeout = 10 * 1000) {
+  try {
+    return await func();
+  } catch (error) {
+    await delay(timeout);
+    return await func().catch((error) => {
+      throw `重试失败：${error.message}`;
+    });
+  }
 }
-
-export async function getSummaryText(text:string, apiKey:string){
-  const summaryData = await axios.post(
-    'https://api-inference.huggingface.co/models/csebuetnlp/mT5_multilingual_XLSum',
-    { inputs: text },
-    {
-      headers: {
-        'Content-Type': 'application/json',
-        // 如果需要认证
-        Authorization: `Bearer ${apiKey}`,
-      }
-    },
-  );
-
-  const summaryText = summaryData.data[0].summary_text as string;
-
-  return summaryText
-}

src/utils/file-utils.ts

@@ -1,3 +1,4 @@
+import axios from 'axios';
 import fs from 'fs';
 import os from 'os';
 import path from 'path';
@@ -72,25 +73,22 @@ export async function downloadDocx(url: string, fileName: string) {
 }
 
 export async function getImageBuffer(text: string, apiKey: string) {
-  const response = await fetch('https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev', {
-    headers: {
-      'Content-Type': 'application/json',
-      // 如果需要认证
-      Authorization: `Bearer ${apiKey}`,
-    },
-    method: 'POST',
-    body: JSON.stringify({
+  const response = await axios.post(
+    'https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev',
+    {
       inputs: text,
       options: {
         use_cache: false,
       },
-    }),
-  });
-  const blob = await response.blob();
-  // 将 Blob 转换为 ArrayBuffer
-  const arrayBuffer = await blob.arrayBuffer();
-  // 将 ArrayBuffer 转换为 Buffer
-  const buffer = Buffer.from(arrayBuffer);
-
-  return buffer;
+    },
+    {
+      headers: {
+        'Content-Type': 'application/json',
+        Authorization: `Bearer ${apiKey}`,
+      },
+      responseType: 'arraybuffer', // 设置响应类型为 blob
+    },
+  );
+  const arrayBuffer = response.data as ArrayBuffer;
+  return Buffer.from(arrayBuffer);
 }

src/utils/hugging-face.ts

@@ -0,0 +1,36 @@
+import axios from 'axios';
+
+export async function getTranslatedText(text: string, apiKey: string) {
+  const translatedData = await axios.post(
+    'https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-zh-en',
+    { inputs: text, src_lang: 'ch', target_lang: 'en' },
+    {
+      headers: {
+        'Content-Type': 'application/json',
+        // 如果需要认证
+        Authorization: `Bearer ${apiKey}`,
+      },
+    },
+  );
+  const translatedText = translatedData.data[0].translation_text as string;
+
+  return translatedText;
+}
+
+export async function getSummaryText(text: string, apiKey: string) {
+  const summaryData = await axios.post(
+    'https://api-inference.huggingface.co/models/csebuetnlp/mT5_multilingual_XLSum',
+    { inputs: text },
+    {
+      headers: {
+        'Content-Type': 'application/json',
+        // 如果需要认证
+        Authorization: `Bearer ${apiKey}`,
+      },
+    },
+  );
+
+  const summaryText = summaryData.data[0].summary_text as string;
+
+  return summaryText;
+}

src/utils/index.ts

@@ -1,2 +1,3 @@
 export * from './common';
 export * from './file-utils';
+export * from './hugging-face';