feat: 单图

src/const/article.ts

@@ -458,3 +458,106 @@ export const emotionalStoryEnding = [
   '结局探讨了自由的含义，主角为了追求自由付出了巨大的代价',
   '结局是一个开放式的，让读者自己想象后续发展',
 ];
+
+export const emotionalStoryCover = [
+  '乡村土路',
+  '山间小溪',
+  '麦田间的田埂',
+  '村头的老槐树',
+  '柴火垛旁的空地',
+  '农家小院的篱笆墙',
+  '黄昏下的打谷场',
+  '清晨的露珠挂在草叶上',
+  '河边的垂柳',
+  '雨后的乡间小道',
+  '田边的灌溉渠',
+  '村口的石磨',
+  '破旧的茅草屋',
+  '山脚下的梯田',
+  '堆满干草的草棚',
+  '菜园里的蔬菜架',
+  '田野里的稻草人',
+  '夏季的瓜棚',
+  '冬季的雪压树枝',
+  '秋天的落叶满地',
+  '村后的小山坡',
+  '河边的洗衣石',
+  '田边的野花丛',
+  '村口的老井',
+  '屋顶的烟囱',
+  '被雨水冲刷的土墙',
+  '麦收后的麦茬地',
+  '稻穗低垂的稻田',
+  '菜园里的水井',
+  '村中的石板路',
+  '院门口的石狮子',
+  '老房子的木窗',
+  '黄昏的炊烟',
+  '夏日的蝉鸣树',
+  '秋天的柿子树',
+  '冬日的冰挂树枝',
+  '春天的桃花林',
+  '夏日的荷花塘（边缘部分，不拍荷花）',
+  '雨后的水洼',
+  '村边的小树林',
+  '田边的灌溉水车',
+  '废弃的石碾',
+  '堆满农具的角落',
+  '乡村小学的操场',
+  '村里的祠堂',
+  '晒谷场上的竹席',
+  '被藤蔓缠绕的电线杆',
+  '田边的稻草人',
+  '夏日的丝瓜架',
+  '冬日的干草堆',
+  '秋天的葡萄园',
+  '夏日的向日葵地',
+  '河边的菖蒲',
+  '田边的艾草',
+  '村中的老戏台',
+  '夏日的葡萄藤架',
+  '秋日的玉米地',
+  '冬天的枯树林',
+  '春天的油菜花田',
+  '夏日的竹林',
+  '河边的芦苇荡',
+  '田边的荆棘丛',
+  '村中的石拱桥',
+  '夏日的稻田蛙鸣处',
+  '秋日的山楂树',
+  '冬日的雪地脚印（如果脚印不太明显可算）',
+  '春天的榆钱树',
+  '村中的老柳树下',
+  '田边的排水沟',
+  '夏日的豆角架',
+  '秋日的南瓜地',
+  '冬日的树挂雾凇',
+  '春天的杏花林',
+  '村中的晒谷场角落',
+  '田边的简易瓜棚',
+  '夏日的薄荷地',
+  '秋日的板栗树',
+  '冬日的冻土裂缝',
+  '春天的杨树林',
+  '河边的苔藓',
+  '田边的废弃犁具',
+  '村中的古井台',
+  '夏日的稻田飞虫',
+  '秋日的酸枣树',
+  '冬日的秃树鸟巢',
+  '春天的果园小径',
+  '村中的柴火堆',
+  '田边的灌溉水阀',
+  '夏日的薄荷丛',
+  '秋日的核桃树',
+  '冬日的篱笆上的冰柱',
+  '春天的蒲公英地',
+  '村中的老榆树下',
+  '田边的水洼倒影',
+  '夏日的韭菜地',
+  '秋日的柿子树红果',
+  '冬日的雪地枯草',
+  '春天的柳树新芽',
+  '村中的石墩',
+  '田边的野花小径',
+];

src/service/article-service.ts

@@ -1,39 +1,24 @@
-import { bufferToBase64ImageSrc, delay, getFluxImageBuffer, getRandomUniqueElements, retryAsync } from '@/utils';
+import { delay, getImageBase64ByText, getRandomUniqueElements, retryAsync } from '@/utils';
 
-import { requestQw, uploadFile } from '@/utils';
-import { emotionalStoryTopic, emotionalStoryStyle, emotionalStoryEnding } from '@/const';
+import { uploadFile } from '@/utils';
+import { emotionalStoryTopic, emotionalStoryStyle, emotionalStoryEnding, emotionalStoryCover } from '@/const';
 const htmlToDocx = require('html-to-docx');
 
-async function getImageBase64ByText(text: string, hfApiKey: string) {
-  console.log(`------------ 开始获取图片，原文案：${text} ---------------`);
-
-  const summary = await requestQw({
-    messages: [
-      {
-        role: 'user',
-        content: `翻译成英语并简化内容提炼核心，控制在十五个单词内：${text}`,
-      },
-    ],
-    model: 'qwen-plus',
-  });
-  console.log(`summary:${summary}`);
-  const buffer = await getFluxImageBuffer(`modern China：${summary}`, hfApiKey);
-  const imageBase64 = bufferToBase64ImageSrc(buffer, 'image/png');
-  console.log('------------- 图片Base64获取成功 ---------------');
-  return imageBase64;
-}
-
-export const processArticleServe = async (data: { title: string; content: string; config: { output: string; hfApiKey: string } }) => {
+export const processArticleServe = async (data: {
+  title: string;
+  content: string;
+  config: { output: string; hfApiKey: string; imageCount: number };
+}) => {
   const { title, content, config } = data;
-
+  const { output = 'docx', hfApiKey, imageCount = 6 } = config;
   // 将富文本内容分割成段落数组
   const paragraphs = content.split(/\\n|\n/).filter((p) => p.trim() !== '');
-  const imageParagraphCount = Math.max(3, Math.ceil(paragraphs.length / (3 * 2)));
+  const imageParagraphCount = Math.ceil(paragraphs.length / imageCount);
 
   // 遍历段落，根据段落内容调用接口生成图片并插入到HTML中
   const htmlArray = [`<h1>${title}</h1>`];
   let promiseArray = [];
-  let imageCount = 0;
+  let currentImageCount = 0;
 
   for (let i = 0; i < paragraphs.length; i++) {
     const isGeneImage = i === 0 || (i % imageParagraphCount === 0 && i <= paragraphs.length - (imageParagraphCount - 1));
@@ -41,29 +26,24 @@ export const processArticleServe = async (data: { title: string; content: string
 
     promiseArray.push(
       new Promise(async (resolve) => {
-        try {
-          // 如果是文章开头（i === 0）或者符合每两或三个段落插入图片的规则（且不在最后两段内）
-          if (isGeneImage) {
-            imageCount++;
-
-            const text = i ? paragraphs[i - 3] + paragraphs[i - 2] + paragraphs[i - 1] : title;
-            const base64ImageUri = await retryAsync(() => getImageBase64ByText(text, config.hfApiKey), 1, 5 * 1000);
-            // 创建 Base64 数据 URI
-            htmlArray[i + 1] += `<img src="${base64ImageUri}" alt="Generated Image"><br>`;
-          }
-        } catch (error: any) {
-          console.error('Error calling API to generate image:', error);
+        // 如果是文章开头（i === 0）或者符合每两或三个段落插入图片的规则（且不在最后两段内）
+        if (isGeneImage) {
+          currentImageCount++;
+
+          const text = i ? paragraphs[i - 3] + paragraphs[i - 2] + paragraphs[i - 1] : title;
+          const base64ImageUri = await retryAsync(() => getImageBase64ByText(text, hfApiKey), 1, 5 * 1000);
+          // 创建 Base64 数据 URI
+          htmlArray[i + 1] += `<img src="${base64ImageUri}" alt="Generated Image"><br>`;
         }
 
         htmlArray[i + 1] += `<p>${paragraphs[i]}</p>`;
-
         resolve(1);
       }),
     );
 
     if (isGeneImage) {
-      imageCount++;
-      if (imageCount % 3 === 0) {
+      currentImageCount++;
+      if (currentImageCount % 3 === 0) {
         const start = Date.now();
         await Promise.all(promiseArray);
         await delay(Math.max(65 * 1000 - (Date.now() - start), 0));
@@ -76,13 +56,13 @@ export const processArticleServe = async (data: { title: string; content: string
 
   const htmlWithImages = htmlArray.join('');
 
-  if (config.output === 'html') {
+  if (output === 'html') {
     return {
       article: htmlWithImages,
     };
   }
 
-  if (config.output === 'docx') {
+  if (output === 'docx') {
     const outputFilename = `docx-${Date.now()}.docx`;
     const docxBuffer = await htmlToDocx(htmlWithImages, {
       orientation: 'portrait',
@@ -104,7 +84,7 @@ export const processArticleElementServe = async () => {
   const topic = getRandomUniqueElements(emotionalStoryTopic, 1)[0];
   const styleList = getRandomUniqueElements(emotionalStoryStyle, 5);
   const style = styleList.join(',');
-  const ending = getRandomUniqueElements(emotionalStoryEnding, 1)[0];;
+  const ending = getRandomUniqueElements(emotionalStoryEnding, 1)[0];
 
   return {
     topic,

src/utils/common.ts

@@ -1,4 +1,3 @@
-import axios from 'axios';
 import util from 'util';
 const setTimeoutPromise = util.promisify(setTimeout);
 
@@ -21,23 +20,6 @@ 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';
-
-  try {
-    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;
-  } catch (error: any) {
-    throw new Error(error.response.data);
-  }
-}
-
 export function getRandomUniqueElements(arr: any[], x: number): any[] {
   const result: any[] = [];
   const copyArr = arr.slice();

src/utils/file-utils.ts

@@ -3,6 +3,9 @@ import fs from 'fs';
 import os from 'os';
 import path from 'path';
 import sharp from 'sharp';
+import { getRandomUniqueElements } from './common';
+import { requestQwImage } from './hugging-face';
+import { emotionalStoryCover } from '@/const';
 
 // 创建上传文件夹，如果不存在
 export const createUploadDir = (): string => {
@@ -100,3 +103,30 @@ export async function blobToArrayBuffer(blobOrArrayBuffer: Blob | ArrayBuffer){
   }
 }
 
+
+export async function getImageBase64ByText(text: string, hfApiKey: string) {
+  console.log(`------------ 开始获取图片，原文案：${text} ---------------`);
+
+  // const summary = await requestQw({
+  //   messages: [
+  //     {
+  //       role: 'user',
+  //       content: `翻译成英语并简化内容提炼核心，控制在十五个单词内：${text}`,
+  //     },
+  //   ],
+  //   model: 'qwen-plus',
+  // });
+  // console.log(`summary:${summary}`);
+  // const buffer = await getFluxImageBuffer(`modern China：${summary}`, hfApiKey);
+  // const imageBase64 = bufferToBase64ImageSrc(buffer, 'image/png');
+
+  const coverTheme = getRandomUniqueElements(emotionalStoryCover, 1)[0];
+  const base64 = await requestQwImage({
+    prompt: `手机拍摄，贴近现实生活，灰土气息，纯风景，不要人物和动物：${coverTheme}`,
+    model: 'qwen-plus',
+    response_format: 'b64_json',
+  });
+  const imageBase64 = `data:image/png;base64,${base64}`;
+  console.log('------------- 图片Base64获取成功 ---------------');
+  return imageBase64;
+}

src/utils/hugging-face.ts

@@ -14,7 +14,7 @@ export async function getTranslatedText(text: string, apiKey: string) {
       },
     );
     const translatedText = translatedData.data[0].translation_text as string;
-  
+
     return translatedText;
   } catch (error) {
     throw new Error(`翻译失败，错误信息：${error}`);
@@ -34,15 +34,32 @@ export async function getSummaryText(text: string, apiKey: string) {
         },
       },
     );
-  
+
     const summaryText = summaryData.data[0].summary_text as string;
-  
+
     return summaryText;
   } catch (error) {
     throw new Error(`总结失败，错误信息：${error}`);
   }
 }
 
+export async function requestQw(data: any) {
+  const qwToken = 'sBy1ogROHqapzX0CcdoyjCl$7wAX1NzRpOPRUMrQGgN8J7jSxmMQWreOgkeheTFbylzpf8Gz1g_n0';
+
+  try {
+    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;
+  } catch (error: any) {
+    throw new Error(error.response.data);
+  }
+}
+
 export async function requestQwImage(data: any) {
   const qwToken = 'sBy1ogROHqapzX0CcdoyjCl$7wAX1NzRpOPRUMrQGgN8J7jSxmMQWreOgkeheTFbylzpf8Gz1g_n0';