konjacpotato
2851a2806d
All checks were successful
Gitea Actions Demo / deploy (push) Successful in 12s
84 lines
3.0 KiB
Python
84 lines
3.0 KiB
Python
import json
|
|
from config.settings import settings
|
|
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():
|
|
"""定时任务：生成每日文章并保存至数据库。"""
|
|
logger.info("Starting 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(settings.OUTPUT_PATH, 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}")
|
|
|
|
logger.info("Completed job: Generate Daily Article")
|
|
|
|
# For manual testing
|
|
# if __name__ == "__main__":
|
|
# 每日文章生成
|
|
# job_generate_daily_article() |