Phase3: 三平台封面图生成

- cover_generator.py: Pillow生成知乎/微信/小红书封面图(渐变背景+标题)
- 知乎 1200×630 蓝调, 微信 900×500 绿调, 小红书 1080×1440 红调
- creator.py 流水线增加 cover_generator 作为最终回退
- db_helper.py 新增 save_cover_to_article
- main.py 挂载 /automation/images 静态目录
- articles.py preview 接口返回 images 字段
- topics.html 预览dialog展示封面图
This commit is contained in:
Yuzhiran Dev
2026-05-20 19:12:58 +08:00
parent 40a77cad2c
commit c8bee712d7
6 changed files with 237 additions and 5 deletions
+1 -1
View File
@@ -138,7 +138,7 @@ def preview_article(topic_id: str, platform: str = "zhihu", current_user: User =
article = db.query(Article).filter(Article.id == article_id).first() article = db.query(Article).filter(Article.id == article_id).first()
if not article or not article.html_content: if not article or not article.html_content:
raise HTTPException(status_code=404, detail=f"Article not found for {topic_id} on {platform}") raise HTTPException(status_code=404, detail=f"Article not found for {topic_id} on {platform}")
return {"topic_id": topic_id, "platform": platform, "html": article.html_content} return {"topic_id": topic_id, "platform": platform, "html": article.html_content, "images": article.images or {}}
@router.put("/{topic_id}/content") @router.put("/{topic_id}/content")
def update_article_content( def update_article_content(
+7
View File
@@ -95,6 +95,13 @@ app.include_router(platform_config.router)
app.include_router(collector_mgmt.router) app.include_router(collector_mgmt.router)
app.include_router(assistant.router) app.include_router(assistant.router)
# 挂载自动生成的图片(必须先于前端根挂载)
PROJECT_ROOT_DIR = Path(__file__).parent.parent.parent.parent
AUTOMATION_IMAGES_DIR = PROJECT_ROOT_DIR / "automation" / "images"
if AUTOMATION_IMAGES_DIR.exists():
app.mount("/automation/images", StaticFiles(directory=str(AUTOMATION_IMAGES_DIR)), name="images")
logger.info(f"Images mounted at /automation/images from {AUTOMATION_IMAGES_DIR}")
# 挂载前端 # 挂载前端
FRONTEND_DIR = Path(__file__).parent.parent.parent / "frontend" FRONTEND_DIR = Path(__file__).parent.parent.parent / "frontend"
if FRONTEND_DIR.exists() and (FRONTEND_DIR / "index.html").exists(): if FRONTEND_DIR.exists() and (FRONTEND_DIR / "index.html").exists():
+6 -3
View File
@@ -163,6 +163,9 @@
</div> </div>
</div> </div>
<div style="width:100%;"> <div style="width:100%;">
<div v-if="!editing && previewImages[previewPlatform] && previewImages[previewPlatform].cover" style="margin-bottom:16px; border:1px solid #ebeef5; border-radius:8px; overflow:hidden; text-align:center;">
<img :src="'/' + previewImages[previewPlatform].cover" alt="封面图" style="max-width:100%; max-height:360px; object-fit:contain;" @error="$event.target.style.display='none'" />
</div>
<div v-if="!editing" class="preview-body" style="border:1px solid #ebeef5; border-radius:8px; background:#fff; padding:20px; width:100%; line-height:1.8;" v-html="currentPreviewBody"></div> <div v-if="!editing" class="preview-body" style="border:1px solid #ebeef5; border-radius:8px; background:#fff; padding:20px; width:100%; line-height:1.8;" v-html="currentPreviewBody"></div>
<div v-else ref="editor" contenteditable="true" style="min-height:300px; height:68vh; border:1px solid #409eff; border-radius:8px; background:#fff; padding:20px; overflow-y:auto; width:100%; line-height:1.8; outline:none; box-shadow:0 0 0 2px rgba(64,158,255,0.2);" @input="editContent = $event.target.innerHTML"></div> <div v-else ref="editor" contenteditable="true" style="min-height:300px; height:68vh; border:1px solid #409eff; border-radius:8px; background:#fff; padding:20px; overflow-y:auto; width:100%; line-height:1.8; outline:none; box-shadow:0 0 0 2px rgba(64,158,255,0.2);" @input="editContent = $event.target.innerHTML"></div>
</div> </div>
@@ -196,7 +199,7 @@ const TopicsApp = {
stats: { total: 0, pending: 0, review: 0, ready: 0, published: 0, today: 0 }, stats: { total: 0, pending: 0, review: 0, ready: 0, published: 0, today: 0 },
topics: [], allTopics: [], todayTopics: [], todayCount: 0, topics: [], allTopics: [], todayTopics: [], todayCount: 0,
previewVisible: false, previewTopic: null, previewFullscreen: false, previewVisible: false, previewTopic: null, previewFullscreen: false,
previewPlatform: 'zhihu', platformContents: {}, editing: false, editContent: '', previewPlatform: 'zhihu', platformContents: {}, previewImages: {}, editing: false, editContent: '',
publishDialogVisible: false, publishTopic: null, publishDialogVisible: false, publishTopic: null,
publishPlatforms: { zhihu: true, wechat: true, xiaohongshu: true }, publishPlatforms: { zhihu: true, wechat: true, xiaohongshu: true },
publishing: false, publishing: false,
@@ -317,14 +320,14 @@ const TopicsApp = {
} catch (error) { this.$message.error(`批量审查失败: ${error.message}`); } } catch (error) { this.$message.error(`批量审查失败: ${error.message}`); }
}, },
async openPreview(topic) { async openPreview(topic) {
this.previewTopic = topic; this.previewPlatform = 'zhihu'; this.previewVisible = true; this.platformContents = {}; this.previewTopic = topic; this.previewPlatform = 'zhihu'; this.previewVisible = true; this.platformContents = {}; this.previewImages = {};
const token = this.getToken(); const token = this.getToken();
if (!token) return; if (!token) return;
const platforms = ['zhihu', 'wechat', 'xiaohongshu']; const platforms = ['zhihu', 'wechat', 'xiaohongshu'];
const names = { zhihu: '知乎', wechat: '微信公众号', xiaohongshu: '小红书' }; const names = { zhihu: '知乎', wechat: '微信公众号', xiaohongshu: '小红书' };
await Promise.all(platforms.map(p => await Promise.all(platforms.map(p =>
fetch(`/api/articles/${topic.id}/preview?platform=${p}`, { headers: { 'Authorization': 'Bearer ' + token } }) fetch(`/api/articles/${topic.id}/preview?platform=${p}`, { headers: { 'Authorization': 'Bearer ' + token } })
.then(r => r.ok ? r.json() : null).then(d => { if (d && d.html) this.platformContents[p] = d.html; }).catch(e => { console.error(`加载${p}预览失败:`, e); this.$message.error(`加载${names[p]}预览失败`); }) .then(r => r.ok ? r.json() : null).then(d => { if (d && d.html) { this.platformContents[p] = d.html; if (d.images) this.previewImages[p] = d.images; } }).catch(e => { console.error(`加载${p}预览失败:`, e); this.$message.error(`加载${names[p]}预览失败`); })
)); ));
}, },
togglePreviewFullscreen() { togglePreviewFullscreen() {
+199
View File
@@ -0,0 +1,199 @@
#!/usr/bin/env python3
"""
封面图生成器:为三平台文章生成文字封面图
依赖: Pillow + Noto Sans CJK 字体
用法: python3 scripts/cover_generator.py --topic-id A07
"""
import argparse, datetime, json, logging, sys, os
from pathlib import Path
from typing import Dict, Optional
from PIL import Image, ImageDraw, ImageFont
PROJECT_ROOT = Path(__file__).parent.parent
sys.path.insert(0, str(PROJECT_ROOT))
sys.path.insert(0, str(PROJECT_ROOT / "platform" / "backend"))
from db_helper import get_topic_by_id
LOGS_DIR = PROJECT_ROOT / "automation" / "logs"
OUTPUT_DIR = PROJECT_ROOT / "automation" / "images" / "generated"
TODAY = datetime.datetime.now().strftime("%Y-%m-%d")
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s',
handlers=[logging.FileHandler(LOGS_DIR / f"cover_{TODAY}.log"), logging.StreamHandler()])
logger = logging.getLogger(__name__)
# Find a suitable CJK font
_FONT_CANDIDATES = [
"/usr/share/fonts/google-noto-cjk/NotoSansCJK-Bold.ttc",
"/usr/share/fonts/google-noto-cjk/NotoSansCJK-Medium.ttc",
"/usr/share/fonts/google-noto-cjk/NotoSansCJK-Regular.ttc",
"/usr/share/fonts/opentype/noto/NotoSansCJK-Bold.ttc",
"/usr/share/fonts/truetype/noto/NotoSansCJK-Bold.ttc",
]
_FONT_PATH = None
for fp in _FONT_CANDIDATES:
if Path(fp).exists():
_FONT_PATH = fp
break
if not _FONT_PATH:
import subprocess
r = subprocess.run(["fc-match", "-f", "%{file}", "sans:lang=zh"], capture_output=True, text=True)
if r.returncode == 0 and Path(r.stdout.strip()).exists():
_FONT_PATH = r.stdout.strip()
COVER_SIZES = {
"zhihu": (1200, 630),
"wechat": (900, 500),
"xiaohongshu": (1080, 1440),
}
PLATFORM_COLORS = {
"zhihu": { "bg_start": "#0066cc", "bg_end": "#004d99", "title": "#ffffff", "accent": "#66b3ff", "watermark": "知乎专栏" },
"wechat": { "bg_start": "#07c160", "bg_end": "#06ad56", "title": "#ffffff", "accent": "#b3ffd9", "watermark": "微信公众号" },
"xiaohongshu": { "bg_start": "#ff2442", "bg_end": "#cc1a35", "title": "#ffffff", "accent": "#ffb3c1", "watermark": "小红书" },
}
def _get_font(size: int):
if _FONT_PATH:
return ImageFont.truetype(_FONT_PATH, size)
return ImageFont.load_default()
def _wrap_text(text: str, max_chars: int = 6) -> str:
"""Split long title into lines"""
lines = []
while text:
if len(text) <= max_chars:
lines.append(text)
break
idx = text.rfind("", 0, max_chars + 1)
if idx <= 0:
idx = text.rfind(" ", 0, max_chars + 1)
if idx <= 0:
idx = max_chars
lines.append(text[:idx])
text = text[idx:]
return "\n".join(lines)
def generate_cover(topic_id: str, platform: str) -> Optional[str]:
"""Generate cover image for a topic on a specific platform"""
topic = get_topic_by_id(topic_id)
if not topic:
logger.error(f"Topic {topic_id} not found")
return None
title = topic.get("title", "")
if not title:
logger.warning(f"Topic {topic_id} has no title")
return None
size = COVER_SIZES.get(platform)
if not size:
logger.error(f"Unknown platform: {platform}")
return None
colors = PLATFORM_COLORS[platform]
w, h = size
img = Image.new("RGB", (w, h))
draw = ImageDraw.Draw(img)
for y in range(h):
r = y / h
for c in range(3):
start = int(colors["bg_start"][1:3], 16) if c == 0 else (int(colors["bg_start"][3:5], 16) if c == 1 else int(colors["bg_start"][5:7], 16))
end = int(colors["bg_end"][1:3], 16) if c == 0 else (int(colors["bg_end"][3:5], 16) if c == 1 else int(colors["bg_end"][5:7], 16))
val = int(start + (end - start) * r)
color = list(img.getpixel((0, y)) if y > 0 else (0, 0, 0))
color[c] = val
img.putpixel((0, y), tuple(color))
for x in range(1, w):
color = img.getpixel((0, y))
draw.line([(x, y), (x, y)], fill=color)
if platform == "xiaohongshu":
title_font_size = 56
wrapped = _wrap_text(title, max_chars=7)
font = _get_font(title_font_size)
lines = wrapped.split("\n")
total_h = len(lines) * (title_font_size + 20)
start_y = (h - total_h) // 2 - 40
for i, line in enumerate(lines):
bbox = draw.textbbox((0, 0), line, font=font)
tw = bbox[2] - bbox[0]
x = (w - tw) // 2
y_pos = start_y + i * (title_font_size + 20)
draw.text((x, y_pos), line, fill=colors["title"], font=font)
font_small = _get_font(24)
wm = colors["watermark"]
bbox = draw.textbbox((0, 0), wm, font=font_small)
ww = bbox[2] - bbox[0]
draw.text(((w - ww) // 2, h - 80), wm, fill=colors["accent"], font=font_small)
else:
title_font_size = 52 if platform == "wechat" else 64
font = _get_font(title_font_size)
bbox = draw.textbbox((0, 0), title, font=font)
tw = bbox[2] - bbox[0]
if tw > w - 120:
wrapped = _wrap_text(title, max_chars=8)
lines = wrapped.split("\n")
total_h = len(lines) * (title_font_size + 16)
start_y = (h - total_h) // 2
for i, line in enumerate(lines):
bbox2 = draw.textbbox((0, 0), line, font=font)
tw2 = bbox2[2] - bbox2[0]
x = (w - tw2) // 2
draw.text((x, start_y + i * (title_font_size + 16)), line, fill=colors["title"], font=font)
else:
x = (w - tw) // 2
draw.text((x, h // 2 - title_font_size), title, fill=colors["title"], font=font)
font_small = _get_font(20)
wm = colors["watermark"]
bbox = draw.textbbox((0, 0), wm, font=font_small)
ww = bbox[2] - bbox[0]
draw.text((w - ww - 20, h - 40), wm, fill=colors["accent"], font=font_small)
out_dir = OUTPUT_DIR / TODAY
out_dir.mkdir(parents=True, exist_ok=True)
out_path = out_dir / f"cover_{platform}_{topic_id}.png"
img.save(out_path, "PNG")
relative_path = f"automation/images/generated/{TODAY}/cover_{platform}_{topic_id}.png"
logger.info(f"Cover saved: {out_path}")
return relative_path
def update_article_image(topic_id: str, platform: str, image_path: str):
"""Save cover path to Article.images JSON field"""
try:
from db_helper import save_cover_to_article
save_cover_to_article(topic_id, platform, image_path)
except Exception as e:
logger.warning(f"Failed to update article image in DB: {e}")
def main():
parser = argparse.ArgumentParser(description="生成文章封面图")
parser.add_argument("--topic-id", required=True)
parser.add_argument("--platform", choices=["zhihu", "wechat", "xiaohongshu", "all"], default="all")
args = parser.parse_args()
platforms = ["zhihu", "wechat", "xiaohongshu"] if args.platform == "all" else [args.platform]
images = {}
for plat in platforms:
path = generate_cover(args.topic_id, plat)
if path:
images[plat] = path
update_article_image(args.topic_id, plat, path)
print(json.dumps(images, ensure_ascii=False))
logger.info(f"Generated covers for {args.topic_id}: {list(images.keys())}")
if __name__ == "__main__":
main()
+3 -1
View File
@@ -103,10 +103,12 @@ def run_pipeline(topic_id: str = None) -> Dict:
update_topic_status(tid, 'pending') update_topic_status(tid, 'pending')
return {"ok": False, "error": "writer step failed"} return {"ok": False, "error": "writer step failed"}
# 4. 配图生成(AI版,失败时回退PIL版) # 4. 配图生成(AI版 → PIL版 → 封面文字版)
image_ok = run_step("ai_image_generator.py", tid) image_ok = run_step("ai_image_generator.py", tid)
if not image_ok: if not image_ok:
image_ok = run_step("image_generator.py", tid) image_ok = run_step("image_generator.py", tid)
if not image_ok:
image_ok = run_step("cover_generator.py", tid)
# 5. 合规优化(自动审核并标记为「待发布」) # 5. 合规优化(自动审核并标记为「待发布」)
if not run_optimizer_step(tid): if not run_optimizer_step(tid):
+21
View File
@@ -259,6 +259,27 @@ def save_article(topic_id: str, platform: str, html_content: str, db: Optional[S
if close_db: if close_db:
db.close() db.close()
def save_cover_to_article(topic_id: str, platform: str, cover_path: str, db: Optional[Session] = None):
"""保存封面图路径到 article.images 字段"""
close_db = False
if db is None:
db = SessionLocal()
close_db = True
try:
from app.models import Article
article_id = f"{platform}_{topic_id}"
article = db.query(Article).filter(Article.id == article_id).first()
if article:
images = article.images or {}
images["cover"] = cover_path
article.images = images
db.commit()
except Exception:
db.rollback()
finally:
if close_db:
db.close()
def get_article(topic_id: str, platform: str, db: Optional[Session] = None) -> Optional[Dict]: def get_article(topic_id: str, platform: str, db: Optional[Session] = None) -> Optional[Dict]:
"""从 articles 表获取文章 HTML""" """从 articles 表获取文章 HTML"""
close_db = False close_db = False