Spaces:

Joey7938
/

Joey

Sleeping

App Files Files Community

yuanjiajun commited on Nov 1, 2024

Commit

7262592

1 Parent(s): 1660a7d

完成

Browse files

Files changed (4) hide show

package.json +1 -0
src/service/article-service.ts +24 -34
src/utils/common.ts +22 -26
src/utils/file-utils.ts +5 -0

package.json CHANGED Viewed

@@ -19,6 +19,7 @@
     "axios": "^1.7.7",
     "fs": "0.0.1-security",
     "html-docx-js": "^0.3.1",
     "koa": "^2.15.3",
     "koa-body": "^6.0.1",
     "koa-bodyparser": "^4.4.1",

     "axios": "^1.7.7",
     "fs": "0.0.1-security",
     "html-docx-js": "^0.3.1",
+    "html-to-docx": "^1.8.0",
     "koa": "^2.15.3",
     "koa-body": "^6.0.1",
     "koa-bodyparser": "^4.4.1",

src/service/article-service.ts CHANGED Viewed

@@ -1,29 +1,30 @@
 import fs from 'fs';
 import htmlDocx from 'html-docx-js';
 import path from 'path';
 import { requestQw, requestQwImage } from '@/utils';
 const uploadDir = path.join(__dirname, '../../uploads');
-async function getImageBufferByText(text: string, apiKey: string) {
   console.log(`------------ 开始获取图片，原文案：${text} ---------------`);
   const summary = await requestQw({
-    "messages": [
-        {
-            "role": "user",
-            "content": `简化内容提炼核心，控制在八个字内：${text}`
-        }
     ],
-    "model": "qwen-plus"
-})
   console.log(`summary:${summary}`);
   const imageBase64 = await requestQwImage({
-    "prompt":`中国现代：${summary}`,
-    "model":"qwen-vl-plus",
-    "response_format": "b64_json"
-  })
   console.log('------------- 图片Base64获取成功 ---------------');
   return imageBase64;
@@ -41,24 +42,22 @@ export const processArticleServe = async (data: { title: string; content: string
   // 将富文本内容分割成段落数组
   const paragraphs = content.split('\n').filter((p) => p.trim() !== '');
-  let htmlWithImages = '';
   // await preheat(config.hfApiKey);
   // 遍历段落，根据段落内容调用接口生成图片并插入到HTML中
   for (let i = 0; i < paragraphs.length; i++) {
-    // 确定用于生成图片的描述内容，除了第一段，其余为上一段的内容
-    const description = i === 0 ? paragraphs[i] : paragraphs[i - 1];
     try {
       // 如果是文章开头（i === 0）或者符合每两或三个段落插入图片的规则（且不在最后两段内）
       if (i === 0 || (i % 3 === 0 && i <= paragraphs.length - 2)) {
-        const imageBase64 = await getImageBufferByText(description, config.hfApiKey);
-        // 创建 Base64 数据 URI
         const base64ImageUri = `data:image/png;base64,${imageBase64}`; // 根据图像格式调整此处的 mime 类型
         htmlWithImages += `<img src="${base64ImageUri}" alt="Generated Image"><br>`;
       }
     } catch (error: any) {
       console.error('Error calling API to generate image:', error);
@@ -67,31 +66,22 @@ export const processArticleServe = async (data: { title: string; content: string
     htmlWithImages += `<p>${paragraphs[i]}</p>`;
   }
-  const htmlContent = `
-      <!DOCTYPE html>
-        <html>
-          <head>
-          <meta charset="utf-8">
-          <title>${title}</title>
-        </head>
-        <body>
-          ${htmlWithImages}
-        </body>
-      </html>
-  `;
   if (config.output === 'html') {
     return {
-      article: htmlContent,
     };
   }
   if (config.output === 'docx') {
     const outputFilename = `docx-${Date.now()}.docx`;
     const outputPath = path.join(uploadDir, outputFilename);
-    const arrayBuffer = htmlDocx.asBlob(htmlContent) as ArrayBuffer;
     const docxBuffer = Buffer.from(arrayBuffer);
     fs.writeFileSync(outputPath, docxBuffer);
     return {
       article: outputFilename,
     };

 import fs from 'fs';
 import htmlDocx from 'html-docx-js';
 import path from 'path';
+import { blobToArrayBuffer } from '@/utils';
 import { requestQw, requestQwImage } from '@/utils';
 const uploadDir = path.join(__dirname, '../../uploads');
+async function getImageBufferByText(text: string) {
   console.log(`------------ 开始获取图片，原文案：${text} ---------------`);
   const summary = await requestQw({
+    messages: [
+      {
+        role: 'user',
+        content: `简化内容提炼核心，控制在八个字内：${text}`,
+      },
     ],
+    model: 'qwen-plus',
+  });
   console.log(`summary:${summary}`);
   const imageBase64 = await requestQwImage({
+    prompt: `中国现代：${summary}`,
+    model: 'qwen-vl-plus',
+    response_format: 'b64_json',
+  });
   console.log('------------- 图片Base64获取成功 ---------------');
   return imageBase64;
   // 将富文本内容分割成段落数组
   const paragraphs = content.split('\n').filter((p) => p.trim() !== '');
+  let htmlWithImages = `<h1>${title}</h1>`;
   // await preheat(config.hfApiKey);
   // 遍历段落，根据段落内容调用接口生成图片并插入到HTML中
   for (let i = 0; i < paragraphs.length; i++) {
     try {
       // 如果是文章开头（i === 0）或者符合每两或三个段落插入图片的规则（且不在最后两段内）
       if (i === 0 || (i % 3 === 0 && i <= paragraphs.length - 2)) {
+        const text = i ? paragraphs[i - 3] + paragraphs[i - 2] + paragraphs[i - 1] : title;
+        const imageBase64 = await getImageBufferByText(text);
         const base64ImageUri = `data:image/png;base64,${imageBase64}`; // 根据图像格式调整此处的 mime 类型
+        // 创建 Base64 数据 URI
         htmlWithImages += `<img src="${base64ImageUri}" alt="Generated Image"><br>`;
+        console.log(htmlWithImages);
       }
     } catch (error: any) {
       console.error('Error calling API to generate image:', error);
     htmlWithImages += `<p>${paragraphs[i]}</p>`;
   }
+  console.log(htmlWithImages);
   if (config.output === 'html') {
     return {
+      article: htmlWithImages,
     };
   }
   if (config.output === 'docx') {
     const outputFilename = `docx-${Date.now()}.docx`;
     const outputPath = path.join(uploadDir, outputFilename);
+    const blob = htmlDocx.asBlob(htmlWithImages) as Blob;
+    const arrayBuffer = (await blobToArrayBuffer(blob)) as ArrayBuffer;
     const docxBuffer = Buffer.from(arrayBuffer);
     fs.writeFileSync(outputPath, docxBuffer);
     return {
       article: outputFilename,
     };

src/utils/common.ts CHANGED Viewed

@@ -20,36 +20,32 @@ export async function retryAsync(func: () => Promise<any>, x = 3, timeout = 10 *
   throw new Error(`Async function failed after ${x} attempts.`);
 }
-export async function requestQw(data:any) {
-  const qwToken = 'sBy1ogROHqapzX0CcdoyjCl$7wAX1NzRpOPRUMrQGgN8J7jSxmMQWreOgkeheTFbylzpf8Gz1g_n0'
-  const response = await axios.post(
-    'https://Joey7938-joe-qw-api.hf.space/api/chat/completions',
-    data,
-    {
-      headers: {
-        'Content-Type': 'application/json',
-        Authorization: `Bearer ${qwToken}`,
-      },
     },
-  );
   return response.data.choices[0].message.content;
 }
-export async function requestQwImage(data:any) {
-  const qwToken = 'sBy1ogROHqapzX0CcdoyjCl$7wAX1NzRpOPRUMrQGgN8J7jSxmMQWreOgkeheTFbylzpf8Gz1g_n0'
-  const response = await axios.post(
-    'https://Joey7938-joe-qw-api.hf.space/api/images/generations',
-    data,
-    {
-      headers: {
-        'Content-Type': 'application/json',
-        Authorization: `Bearer ${qwToken}`,
-      },
     },
-  );
-  return response.data.data[0].b64_json;
-}

   throw new Error(`Async function failed after ${x} attempts.`);
 }
+export async function requestQw(data: any) {
+  const qwToken = 'sBy1ogROHqapzX0CcdoyjCl$7wAX1NzRpOPRUMrQGgN8J7jSxmMQWreOgkeheTFbylzpf8Gz1g_n0';
+  const response = await axios.post('https://Joey7938-joe-qw-api.hf.space/api/chat/completions', data, {
+    headers: {
+      'Content-Type': 'application/json',
+      Authorization: `Bearer ${qwToken}`,
     },
+  });
   return response.data.choices[0].message.content;
 }
+export async function requestQwImage(data: any) {
+  const qwToken = 'sBy1ogROHqapzX0CcdoyjCl$7wAX1NzRpOPRUMrQGgN8J7jSxmMQWreOgkeheTFbylzpf8Gz1g_n0';
+  const { data: response } = await axios.post('https://Joey7938-joe-qw-api.hf.space/api/images/generations', data, {
+    headers: {
+      'Content-Type': 'application/json',
+      Authorization: `Bearer ${qwToken}`,
     },
+  });
+  if (response.code !== undefined && response.code !== 0) {
+    throw new Error(response.message);
+  }
+  return response.data[0].b64_json;
+}

src/utils/file-utils.ts CHANGED Viewed

@@ -90,3 +90,8 @@ export async function getImageBuffer(text: string, apiKey: string) {
     throw new Error(`生成图片失败，错误信息：${error}`);
   }
 }

     throw new Error(`生成图片失败，错误信息：${error}`);
   }
 }
+export async function blobToArrayBuffer(blob:Blob) {
+  const arrayBuffer = await blob.arrayBuffer();
+  return arrayBuffer;
+}