feat: docx问题处理

package.json

@@ -17,6 +17,7 @@
   "license": "ISC",
   "dependencies": {
     "axios": "^1.7.7",
+    "docx": "^9.0.2",
     "fs": "0.0.1-security",
     "html-docx-js": "^0.3.1",
     "html-to-docx": "^1.8.0",

src/service/article-service.ts

@@ -1,13 +1,9 @@
-import fs from 'fs';
-import htmlDocx from 'html-docx-js';
-import path from 'path';
-import { blobToArrayBuffer, bufferToBase64ImageSrc, delay, getFluxImageBuffer, getRandomValueFromArray, retryAsync } from '@/utils';
+import { bufferToBase64ImageSrc, delay, getFluxImageBuffer, getRandomValueFromArray, retryAsync } from '@/utils';
 
 import { requestQw, uploadFile } from '@/utils';
 import { emotionalStoryTopic } from '@/const';
+const htmlToDocx = require('html-to-docx');
 
-// const uploadDir = path.join(__dirname, '../../uploads');
-const uploadDir = '/app/static'
 
 async function getImageBase64ByText(text: string, hfApiKey: string) {
   console.log(`------------ 开始获取图片，原文案：${text} ---------------`);
@@ -35,8 +31,6 @@ export const processArticleServe = async (data: { title: string; content: string
   const paragraphs = content.split(/\\n|\n/).filter((p) => p.trim() !== '');
   const imageParagraphCount = Math.max(3, Math.ceil(paragraphs.length / (3 * 2)));
 
-  // await preheat(config.hfApiKey);
-
   // 遍历段落，根据段落内容调用接口生成图片并插入到HTML中
   const htmlArray = [`<h1>${title}</h1>`];
   let promiseArray = [];
@@ -91,11 +85,11 @@ export const processArticleServe = async (data: { title: string; content: string
 
   if (config.output === 'docx') {
     const outputFilename = `docx-${Date.now()}.docx`;
-    const blob = htmlDocx.asBlob(htmlWithImages) as Blob;
-    const arrayBuffer = (await blobToArrayBuffer(blob)) as ArrayBuffer;
-    const docxBuffer = Buffer.from(arrayBuffer);
+    const docxBuffer = await htmlToDocx(htmlWithImages, {
+      orientation: 'portrait',
+    });
 
-    await uploadFile(docxBuffer, outputFilename)
+    await uploadFile(docxBuffer, outputFilename);
 
     return {
       article: outputFilename,
@@ -108,5 +102,5 @@ export const processArticleServe = async (data: { title: string; content: string
 };
 
 export const processArticleTopicServe = async () => {
-  return getRandomValueFromArray(emotionalStoryTopic)
+  return getRandomValueFromArray(emotionalStoryTopic);
 };