feat: 情感文章添加风格和结局设定

src/const/article.ts

@@ -80,8 +80,6 @@ const familyTopic = [
   '家庭成员一起为家庭的墙壁绘制壁画',
 ];
 
-  
-
 const householdTopic = [
   '家族内不同家庭间经济差距导致的心理落差',
   '家族内因某个成员发达后态度变化问题',
@@ -295,3 +293,168 @@ const socialTopic = [
 ];
 
 export const emotionalStoryTopic = [...familyTopic, ...householdTopic, ...socialTopic];
+
+export const emotionalStoryStyle = [
+  '用日常用语表达',
+  '像聊天一样说话',
+  '使用俗语或口头禅',
+  '模仿朋友间对话的语气',
+  '加入语气助词（呢、嘛、啦等）',
+  '以讲故事的口吻叙述',
+  '描述具体的颜色、形状、声音',
+  '提及当时的天气状况',
+  '讲讲周围人的反应',
+  '回忆自己的内心想法',
+  '说出身体的具体感受（累、兴奋、紧张等）',
+  '分享一个小插曲或意外事件',
+  '提到具体的时间点和地点',
+  '描述物品的质感',
+  '谈谈闻到的气味',
+  '讲讲尝到的味道',
+  '把 [X] 比作熟悉的事物',
+  '让 [Y] 像人一样行动',
+  '用排比强调 [具体内容]',
+  '夸张地形容 [某特点]',
+  '用反问增强语气',
+  '以设问引出话题',
+  '不要用生僻词汇',
+  '去掉过于正式的表达',
+  '避免大段理论阐述',
+  '杜绝专业术语堆砌',
+  '不使用复杂的长句',
+  '减少抽象概念的出现',
+  '别用格式化的开头结尾',
+  '去除模板化的语句结构',
+  '对话口语化表达',
+  '对话聊天式语气',
+  '讲故事口吻',
+  '真实感受分享',
+  '小插曲和意外',
+  '周围人反应',
+  '别那么正式',
+  '轻松点表达',
+  '非正式一点',
+  '摆脱官方语气',
+  '别像写报告一样',
+  '来点随意的表达',
+  '别长篇大论讲道理',
+  '少点理论阐述',
+  '不搞学术范儿',
+  '远离理论说教',
+  '拒绝枯燥理论',
+  '别太理论化',
+  '别用专业词汇',
+  '远离行话',
+  '普通人能懂的话',
+  '非专业人士也能明白',
+  '不用术语装高深',
+  '避开专业说法',
+  '采用轻松的表述',
+  '以诙谐的方式表达',
+  '运用幽默的语言',
+  '像朋友聊天那样说话',
+  '使用亲切的语气',
+  '保持简洁的表达',
+  '让语言通俗易懂',
+  '来点俏皮的说法',
+  '用随性的语言风格',
+  '打造活泼的氛围',
+  '巧妙运用比喻修辞',
+  '适时使用拟人手法',
+  '恰当进行夸张表达',
+  '合理运用排比增强气势',
+  '以反问引发思考',
+  '用设问吸引读者',
+  '借助对偶增添文采',
+  '善用借代丰富表达',
+  '通过对比突出特点',
+  '避免过于严肃的表达',
+  '杜绝生硬的语言',
+  '不要千篇一律的表述',
+  '去除模板化的痕迹',
+  '摆脱平淡无奇的语言',
+  '防止空洞无物的内容',
+  '舍弃华而不实的词藻',
+  '远离晦涩难懂的语句',
+  '拒绝机械呆板的风格',
+];
+
+export const emotionalStoryEnding = [
+  '结局要有意外的转折，让读者感到惊喜和意外。',
+  '结局要充满希望，给读者带来积极的感受。',
+  '结局可以是悲伤的，引发读者的深思。',
+  '结局要有温暖的情感，让人感到治愈。',
+  '结局可以带有一丝遗憾，增加故事的回味。',
+  '结局要充满悬念，让读者自行想象后续发展。',
+  '结局要有强烈的情感冲突，让读者的心被紧紧抓住。',
+  '结局可以是幽默的，给读者带来轻松的感觉。',
+  '结局要有一个意想不到的人物出现，改变故事的走向。',
+  '结局可以让主角做出一个与之前性格完全不同的决定。',
+  '结局可以引入一个新的事件，使故事更加复杂。',
+  '结局可以让之前的某个伏笔得到意想不到的揭示。',
+  '结局可以让主角面临一个全新的挑战。',
+  '结局可以有一个戏剧性的反转，颠覆读者的预期。',
+  '结局可以让主角在最后一刻发现自己的错误并改正。',
+  '结局可以让故事中的矛盾以一种独特的方式解决。',
+  '结局要深化故事的主题，让读者对主题有更深刻的理解。',
+  '结局可以通过主角的经历传达一个重要的人生道理。',
+  '结局要让读者对故事中的某个社会问题进行思考。',
+  '结局可以让主角在经历一系列事件后实现自我成长。',
+  '结局可以强调故事中的人性光辉或黑暗面。',
+  '结局要让读者对故事中的价值观进行反思。',
+  '结局可以以一种隐喻的方式表达故事的主题。',
+  '结局可以让主角的命运与故事的主题紧密相连。',
+  '结局可以受到外部环境的影响，如天气、自然灾害等。',
+  '结局可以让故事中的地点发生重大变化，影响主角的命运。',
+  '结局可以让主角适应或改变周围的环境。',
+  '结局可以让环境成为主角解决问题的关键因素。',
+  '结局可以让主角在特定的环境中找到新的机遇或挑战。',
+  '结局可以通过环境的描写营造出特定的氛围。',
+  '结局可以让环境与主角的内心世界相互呼应。',
+  '结局可以让环境的变化推动故事的发展。',
+  '结局充满喜悦，所有角色都实现了自己的梦想',
+  '结局带有淡淡的忧伤，主角不得不放弃一些重要的东西',
+  '结局是释然，主角放下了过去的执念',
+  '结局充满愤怒，主角被不公平的对待激发了反抗精神',
+  '结局有深深的愧疚，主角意识到自己的错误却无法弥补',
+  '结局是惊喜，主角获得了意想不到的财富或荣誉',
+  '结局是恐惧，新的威胁出现让主角陷入困境',
+  '结局是安心，危险过去，生活恢复了平静',
+  '结局是感动，一个温暖的举动让主角泪流满面',
+  '结局是孤独，主角独自面对未来',
+  '结局是兴奋，主角迎来了新的冒险',
+  '结局是绝望，所有的希望都破灭了',
+  '结局是满足，主角实现了一个小目标',
+  '结局是困惑，主角对未来感到迷茫',
+  '结局是骄傲，主角的成就得到了广泛认可',
+  '结局是焦虑，新的问题让主角坐立不安',
+  '结局是期待，未来充满了可能性',
+  '结局是沮丧，主角遭遇了重大挫折',
+  '结局是幸福，主角找到了真爱',
+  '结局是平静，主角内心达到了一种和谐状态',
+  '结局是主角发现了一个隐藏的秘密，改变了一切',
+  '结局有一个神秘人物出现，帮助主角解决了难题',
+  '结局主角发现自己一直被欺骗',
+  '结局主角必须在两个艰难的选择中做出决定',
+  '结局强调了友谊的力量，主角和朋友们一起克服了困难',
+  '结局探讨了人性的善恶，主角在困境中做出了艰难的选择',
+  '结局突出了家庭的重要性，主角回归家庭，找到了温暖',
+  '结局思考了人生的意义，主角开始重新审视自己的生活',
+  '结局揭示了社会的不公，主角决定为改变世界而努力',
+  '结局表达了对自然的敬畏，主角学会了与自然和谐相处',
+  '结局展现了科技的两面性，主角在科技的影响下做出了选择',
+  '结局传递了爱的力量，主角用爱化解了一切矛盾',
+  '结局反映了成长的痛苦，主角在经历挫折后变得更加坚强',
+  '结局探讨了梦想与现实的差距，主角在两者之间找到了平衡',
+  '结局强调了勇气的价值，主角勇敢地面对自己的恐惧',
+  '结局思考了时间的意义，主角学会了珍惜当下',
+  '结局揭示了命运的无常，主角在命运的捉弄下努力抗争',
+  '结局表达了对希望的坚持，主角在黑暗中找到了一丝光明',
+  '结局探讨了责任的重要性，主角承担起了自己的责任',
+  '结局展现了宽容的力量，主角学会了原谅他人',
+  '结局反映了文化的差异，主角在不同文化的碰撞中成长',
+  '结局思考了死亡的意义，主角在面对死亡时找到了生命的价值',
+  '结局强调了知识的力量，主角通过学习改变了自己的命运',
+  '结局探讨了自由的含义，主角为了追求自由付出了巨大的代价',
+  '结局是一个开放式的，让读者自己想象后续发展',
+];

src/controllers/article-controller.ts

@@ -2,17 +2,15 @@ import { Context } from 'koa';
 import Router from 'koa-router';
 
 import { handleControllerError } from '@/middleware';
-import { processArticleServe, processArticleTopicServe } from '@/service';
+import { processArticleServe, processArticleElementServe } from '@/service';
 
 const router = new Router({ prefix: '/api' });
 
-// 获取文章主题
-router.get('/generate-article-topic', async (ctx: Context) => {
+// 获取文章要素
+router.get('/generate-article-element', async (ctx: Context) => {
   try {
-    const topic = await processArticleTopicServe();
-    ctx.body = {
-      topic,
-    };
+    const result = await processArticleElementServe();
+    ctx.body = result;
   } catch (error) {
     // 将捕获到的异常传递给统一的错误处理中间件
     await handleControllerError(ctx, error);

src/service/article-service.ts

@@ -1,10 +1,9 @@
-import { bufferToBase64ImageSrc, delay, getFluxImageBuffer, getRandomValueFromArray, retryAsync } from '@/utils';
+import { bufferToBase64ImageSrc, delay, getFluxImageBuffer, getRandomUniqueElements, retryAsync } from '@/utils';
 
 import { requestQw, uploadFile } from '@/utils';
-import { emotionalStoryTopic } from '@/const';
+import { emotionalStoryTopic, emotionalStoryStyle, emotionalStoryEnding } from '@/const';
 const htmlToDocx = require('html-to-docx');
 
-
 async function getImageBase64ByText(text: string, hfApiKey: string) {
   console.log(`------------ 开始获取图片，原文案：${text} ---------------`);
 
@@ -101,6 +100,15 @@ export const processArticleServe = async (data: { title: string; content: string
   };
 };
 
-export const processArticleTopicServe = async () => {
-  return getRandomValueFromArray(emotionalStoryTopic);
+export const processArticleElementServe = async () => {
+  const topic = getRandomUniqueElements(emotionalStoryTopic, 1);
+  const styleList = getRandomUniqueElements(emotionalStoryStyle, 5);
+  const style = styleList.join(',');
+  const ending = getRandomUniqueElements(emotionalStoryEnding, 1);
+
+  return {
+    topic,
+    style,
+    ending,
+  };
 };

src/utils/common.ts

@@ -31,18 +31,19 @@ export async function requestQw(data: any) {
         Authorization: `Bearer ${qwToken}`,
       },
     });
-  
+
     return response.data.choices[0].message.content;
-  } catch (error:any) {
+  } catch (error: any) {
     throw new Error(error.response.data);
   }
 }
 
-export function getRandomValueFromArray<T>(array: T[]): T {
-  if (!Array.isArray(array) || array.length === 0) {
-      throw new Error("输入应该是一个非空数组");
+export function getRandomUniqueElements(arr: any[], x: number): any[] {
+  const result: any[] = [];
+  const copyArr = arr.slice();
+  while (result.length < x && copyArr.length > 0) {
+    const randomIndex = Math.floor(Math.random() * copyArr.length);
+    result.push(copyArr.splice(randomIndex, 1)[0]);
   }
-
-  const randomIndex = Math.floor(Math.random() * array.length);
-  return array[randomIndex];
+  return result;
 }