commit code

scheduler/job_story_portal.py

@@ -0,0 +1,80 @@
+import json
+from utils.logger import logger
+import datetime
+import os
+import asyncio
+from models.script import Script
+from config.database import SessionLocal
+from llm.generate_daily_article import generate_daily_article
+
+project_name = "故事任意门"
+
+# for daily article generation
+def job_generate_daily_article():
+    """定时任务：生成每日文章并保存至数据库。"""
+
+    # 1. 调用 LLM 生成每日文章
+    content = generate_daily_article()
+    if not content:
+        logger.warning("No daily article generated.")
+        return
+    
+    # 2. 保存至数据库
+    # subject 以当前日期为准，格式 YYYY-MM-DD
+    today_str = datetime.datetime.now().strftime("%Y-%m-%d")
+    article_title= content["阶段4_今日文章"]["文章标题"]
+    db = SessionLocal()
+    try:
+        # 查询是否已存在 project+subject 唯一记录
+        script = db.query(Script).filter_by(project=project_name, subject=today_str).first()
+        if script:
+            # 存在则更新内容
+            script.content = json.dumps(content, ensure_ascii=False, separators=(",", ":"))
+            db.commit()
+            logger.info(f"Updated script for {today_str} with {article_title}.")
+        else:
+            # 不存在则新建
+            script = Script(
+                project=project_name,
+                subject=today_str,
+                content=json.dumps(content, ensure_ascii=False, separators=(",", ":"))
+            )
+            db.add(script)
+            db.commit()
+            logger.info(f"Saved script for {today_str} with {article_title}.")
+    except Exception as e:
+        db.rollback()
+        logger.error(f"Failed to save/update script for {today_str}: {e}")
+
+    # 3. 生成音频
+    try:
+        from tts.service import TTSService
+
+        article_text = content["阶段4_今日文章"]["文章正文"]
+        logger.debug(f"Synthesizing daily article audio for '{article_title}'")
+        article_audio = asyncio.run(TTSService.synthesize(
+            text=article_text,
+            voice="yanglan",
+            language="zh-CN"
+        ))
+
+        if not article_audio:
+            logger.warning("No audio synthesized for daily article.")
+            return
+
+        # 保存音频文件
+        out_dir = os.path.join("output", project_name)
+        os.makedirs(out_dir, exist_ok=True)
+        safe_title = "_".join(article_title.split())
+        audio_filename = f"{safe_title}_{today_str}.wav"
+        audio_path = os.path.join(out_dir, audio_filename)
+        with open(audio_path, "wb") as fw:
+            fw.write(article_audio.getvalue())
+        logger.info(f"Saved daily article audio to {audio_path}")
+    except Exception as e:
+        logger.error(f"Failed to synthesize/save daily article audio: {e}")
+
+# For manual testing
+if __name__ == "__main__":
+    # 每日文章生成
+    job_generate_daily_article()