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:
@@ -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(
|
||||||
|
|||||||
@@ -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():
|
||||||
|
|||||||
@@ -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() {
|
||||||
|
|||||||
@@ -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
@@ -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):
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user