356 lines
17 KiB
Python
356 lines
17 KiB
Python
import base64
|
||
import asyncio
|
||
from io import BytesIO
|
||
from pathlib import Path
|
||
from PIL import Image, ImageDraw, ImageFont, ImageOps
|
||
from openai import AsyncOpenAI
|
||
|
||
from app.core.config import settings
|
||
from app.models.palm_report import PalmReport
|
||
from app.models.uploaded_image import UploadedImage
|
||
from app.services.image_service import ImageService
|
||
|
||
|
||
POSTER_W = 1080
|
||
POSTER_H = 2400
|
||
INK = "#080d10"
|
||
PANEL = "#10191c"
|
||
PANEL_2 = "#162326"
|
||
TEXT = "#f2e9d8"
|
||
MUTED = "#9db0b4"
|
||
DIM = "#728389"
|
||
CYAN = "#00e0b8"
|
||
GOLD = "#d8a84e"
|
||
WARN = "#ff6b4a"
|
||
|
||
|
||
class SharePosterService:
|
||
def __init__(self) -> None:
|
||
self.image_service = ImageService()
|
||
self.client = (
|
||
AsyncOpenAI(api_key=settings.openai_api_key, base_url=settings.openai_base_url)
|
||
if settings.openai_api_key
|
||
else None
|
||
)
|
||
self.font_regular = _font(34)
|
||
self.font_small = _font(26)
|
||
self.font_tiny = _font(22)
|
||
self.font_medium = _font(40)
|
||
self.font_title = _font(70)
|
||
self.font_score = _font(98)
|
||
|
||
async def render_ai_or_fallback(self, report: PalmReport, image: UploadedImage | None) -> bytes:
|
||
if settings.share_image_mode == "ai" and self.client:
|
||
try:
|
||
background = await asyncio.wait_for(self.render_ai_background(), timeout=120)
|
||
return self.render(report, image, background=background)
|
||
except Exception:
|
||
return self.render(report, image)
|
||
return self.render(report, image)
|
||
|
||
async def render_ai_background(self) -> Image.Image:
|
||
response = await self.client.images.generate(
|
||
model=settings.openai_image_model,
|
||
prompt=(
|
||
"生成一张竖版高端移动端分享海报背景，不要任何文字、不要数字、不要 logo。"
|
||
"主题：赛博先生 AI 命理实验室，手相报告。"
|
||
"必须包含精美的发光手掌、掌纹扫描线、AI 电路线、东方玄学圆形符号。"
|
||
"设计风格：深墨黑背景 #080d10，青绿色 AI 发光线条 #00e0b8，少量金色点缀 #d8a84e；"
|
||
"现代、东方玄学、AI 扫描终端感；不要卡通，不要恐怖，不要传统庙宇风，不要紫色渐变。"
|
||
"画面中预留干净的暗色区域，方便后续叠加中文报告文字。"
|
||
),
|
||
size="1024x1536",
|
||
quality="high",
|
||
output_format="png",
|
||
response_format="b64_json",
|
||
)
|
||
b64 = response.data[0].b64_json
|
||
if not b64:
|
||
raise RuntimeError("Image model returned no background data")
|
||
return Image.open(BytesIO(base64.b64decode(b64))).convert("RGB")
|
||
|
||
async def render_ai(self, report: PalmReport) -> bytes:
|
||
data = report.report_data or {}
|
||
response = await self.client.images.generate(
|
||
model=settings.openai_image_model,
|
||
prompt=self._build_ai_prompt(report, data),
|
||
size="1024x1536",
|
||
quality="high",
|
||
output_format="png",
|
||
response_format="b64_json",
|
||
)
|
||
b64 = response.data[0].b64_json
|
||
if not b64:
|
||
raise RuntimeError("Image model returned no image data")
|
||
return base64.b64decode(b64)
|
||
|
||
def render(self, report: PalmReport, image: UploadedImage | None, background: Image.Image | None = None) -> bytes:
|
||
data = report.report_data or {}
|
||
poster_h = self._estimate_height(data)
|
||
poster = self._base_canvas(poster_h, background)
|
||
draw = ImageDraw.Draw(poster)
|
||
self._draw_background(draw, poster_h)
|
||
|
||
y = 70
|
||
draw.text((64, y), "CYBER FORTUNE REPORT", font=self.font_tiny, fill=GOLD)
|
||
y += 42
|
||
draw.text((64, y), "赛博先生手相报告", font=self.font_title, fill=TEXT)
|
||
y += 96
|
||
|
||
score = self._score(data)
|
||
y = self._draw_score_card(poster, draw, y, score, data)
|
||
y = self._draw_palm_card(poster, draw, y, image)
|
||
y = self._draw_dimensions(draw, y, data)
|
||
y = self._draw_suggestions(draw, y, data)
|
||
self._draw_footer(draw, poster_h)
|
||
|
||
out = BytesIO()
|
||
poster.save(out, format="PNG", optimize=True)
|
||
return out.getvalue()
|
||
|
||
def _base_canvas(self, poster_h: int, background: Image.Image | None) -> Image.Image:
|
||
if not background:
|
||
return Image.new("RGB", (POSTER_W, poster_h), INK)
|
||
base = ImageOps.fit(background, (POSTER_W, poster_h), method=Image.Resampling.LANCZOS)
|
||
overlay = Image.new("RGBA", (POSTER_W, poster_h), (8, 13, 16, 188))
|
||
base = base.convert("RGBA")
|
||
base.alpha_composite(overlay)
|
||
return base.convert("RGB")
|
||
|
||
def _build_ai_prompt(self, report: PalmReport, data: dict) -> str:
|
||
score = self._score(data)
|
||
dimensions = data.get("dimensions") or []
|
||
dimension_lines = []
|
||
for item in dimensions[:6]:
|
||
name = item.get("name", "维度")
|
||
confidence = int((item.get("confidence") or 0) * 100)
|
||
interpretation = _trim(item.get("interpretation") or "", 48)
|
||
advice = _trim(item.get("advice") or "", 34)
|
||
dimension_lines.append(f"- {name}｜{confidence}分｜{interpretation}｜建议：{advice}")
|
||
|
||
suggestions = data.get("suggestions") or []
|
||
keywords = data.get("lucky_keywords") or []
|
||
summary = _trim(data.get("overall_summary") or "", 90)
|
||
hand_side = {"left": "左手", "right": "右手", "unknown": "未知手"}.get(report.hand_side, "未知手")
|
||
|
||
return (
|
||
"生成一张竖版中文移动端分享海报，比例 2:3，精美、高级、可读性很高。"
|
||
"品牌是“赛博先生”，定位是 AI 命理实验室，功能是手相报告。"
|
||
"设计风格：深墨黑背景 #080d10，青绿色 AI 发光线条 #00e0b8，少量金色点缀 #d8a84e；"
|
||
"现代、东方玄学、AI 扫描终端感；不要卡通，不要恐怖，不要传统庙宇风，不要紫色渐变。"
|
||
"必须有一个清晰的手掌视觉元素，可以是发光掌纹图、掌心扫描线或手掌轮廓。"
|
||
"文字必须清楚、中文可读、排版整齐，像高端 App 分享长图。"
|
||
"禁止使用省略号，禁止把任何维度分析或建议截断；如果内容多，就把版面做成长图。"
|
||
"请在画面中包含以下内容："
|
||
f"标题：赛博先生手相报告；副标题：{hand_side} · AI 掌纹解读；"
|
||
f"综合能量分：{score}/100；"
|
||
f"先生结论：{summary}；"
|
||
f"幸运关键词：{'、'.join(keywords[:5])}；"
|
||
"核心维度分析："
|
||
+ "；".join(dimension_lines)
|
||
+ "；近期建议："
|
||
+ "；".join(_trim(item, 42) for item in suggestions[:4])
|
||
+ "；底部小字：仅供娱乐与自我反思，不构成现实决策建议。"
|
||
"版式要求：顶部品牌标题，中部综合分和手掌图，下面是 6 个维度卡片，底部是建议和免责声明。"
|
||
"不要遗漏分数、手掌图、维度分析和建议，不要出现“...”或“…”省略。"
|
||
)
|
||
|
||
def _estimate_height(self, data: dict) -> int:
|
||
dimensions = data.get("dimensions") or []
|
||
suggestions = data.get("suggestions") or []
|
||
dim_h = 0
|
||
scratch = Image.new("RGB", (POSTER_W, 200), INK)
|
||
draw = ImageDraw.Draw(scratch)
|
||
for item in dimensions:
|
||
interpretation_lines = self._wrap_text(draw, item.get("interpretation") or "", 850, self.font_tiny)
|
||
advice_lines = self._wrap_text(draw, f"建议：{item.get('advice') or ''}", 850, self.font_tiny)
|
||
dim_h += 102 + len(interpretation_lines) * 33 + len(advice_lines) * 33 + 30
|
||
suggestion_h = 120
|
||
for suggestion in suggestions:
|
||
suggestion_h += len(self._wrap_text(draw, suggestion, 824, self.font_small)) * 40 + 12
|
||
summary_lines = self._wrap_text(draw, data.get("overall_summary") or "", 440, self.font_small)
|
||
score_h = max(300, 168 + len(summary_lines) * 42 + 68)
|
||
return max(POSTER_H, 70 + 42 + 96 + score_h + 32 + 422 + 72 + dim_h + suggestion_h + 180)
|
||
|
||
def _draw_background(self, draw: ImageDraw.ImageDraw, poster_h: int) -> None:
|
||
for x in range(0, POSTER_W, 64):
|
||
draw.line((x, 0, x, poster_h), fill=(15, 34, 35), width=1)
|
||
for y in range(0, poster_h, 64):
|
||
draw.line((0, y, POSTER_W, y), fill=(15, 34, 35), width=1)
|
||
draw.rectangle((0, 0, POSTER_W, 280), fill=(8, 21, 22))
|
||
draw.line((64, 232, POSTER_W - 64, 232), fill=CYAN, width=2)
|
||
|
||
def _draw_score_card(self, poster: Image.Image, draw: ImageDraw.ImageDraw, y: int, score: int, data: dict) -> int:
|
||
summary = (data.get("overall_summary") or "这份报告正在整理中。").replace("\n", "")
|
||
summary_lines = self._wrap_text(draw, summary, 440, self.font_small)
|
||
card_h = max(300, 168 + len(summary_lines) * 42 + 68)
|
||
self._round_rect(draw, (64, y, POSTER_W - 64, y + card_h), 34, PANEL)
|
||
draw.text((104, y + 40), "综合能量分", font=self.font_small, fill=MUTED)
|
||
draw.text((104, y + 78), str(score), font=self.font_score, fill=CYAN)
|
||
draw.text((250, y + 120), "/ 100", font=self.font_small, fill=DIM)
|
||
|
||
self._multiline(draw, summary, 520, y + 42, 440, self.font_small, TEXT, 99, line_gap=10)
|
||
|
||
keywords = data.get("lucky_keywords") or []
|
||
x = 104
|
||
ky = y + card_h - 68
|
||
for keyword in keywords:
|
||
w = self._text_w(draw, keyword, self.font_tiny) + 34
|
||
if x + w > 480:
|
||
break
|
||
self._pill(draw, (x, ky, x + w, ky + 42), keyword, self.font_tiny, CYAN, "#06201c")
|
||
x += w + 12
|
||
return y + card_h + 32
|
||
|
||
def _draw_palm_card(self, poster: Image.Image, draw: ImageDraw.ImageDraw, y: int, image: UploadedImage | None) -> int:
|
||
card_h = 390
|
||
self._round_rect(draw, (64, y, POSTER_W - 64, y + card_h), 34, PANEL)
|
||
draw.text((104, y + 34), "掌纹照片", font=self.font_small, fill=GOLD)
|
||
|
||
box = (104, y + 86, 486, y + 344)
|
||
palm = self._load_palm_image(image)
|
||
if palm:
|
||
palm = ImageOps.fit(palm.convert("RGB"), (box[2] - box[0], box[3] - box[1]), method=Image.Resampling.LANCZOS)
|
||
mask = Image.new("L", palm.size, 0)
|
||
mask_draw = ImageDraw.Draw(mask)
|
||
mask_draw.rounded_rectangle((0, 0, palm.size[0], palm.size[1]), radius=26, fill=255)
|
||
poster.paste(palm, box[:2], mask)
|
||
else:
|
||
self._round_rect(draw, box, 26, PANEL_2)
|
||
draw.text((216, y + 190), "掌", font=self.font_score, fill=CYAN)
|
||
|
||
draw.text((532, y + 94), "先生观察", font=self.font_medium, fill=TEXT)
|
||
copy = "掌纹主线会被用于生成娱乐向解读。报告更适合当作近期生活、学习和工作节奏的提醒。"
|
||
self._multiline(draw, copy, 532, y + 156, 390, self.font_small, MUTED, 4, line_gap=10)
|
||
return y + card_h + 32
|
||
|
||
def _draw_dimensions(self, draw: ImageDraw.ImageDraw, y: int, data: dict) -> int:
|
||
draw.text((64, y), "核心维度", font=self.font_medium, fill=TEXT)
|
||
y += 62
|
||
dimensions = data.get("dimensions") or []
|
||
for idx, item in enumerate(dimensions, start=1):
|
||
interpretation = item.get("interpretation") or ""
|
||
advice = item.get("advice") or ""
|
||
interpretation_lines = self._wrap_text(draw, interpretation, 850, self.font_tiny)
|
||
advice_lines = self._wrap_text(draw, f"建议：{advice}", 850, self.font_tiny)
|
||
card_h = 102 + len(interpretation_lines) * 33 + len(advice_lines) * 33 + 24
|
||
self._round_rect(draw, (64, y, POSTER_W - 64, y + card_h), 28, PANEL)
|
||
draw.text((104, y + 28), f"0{idx}", font=self.font_tiny, fill=DIM)
|
||
draw.text((156, y + 22), item.get("name", "维度"), font=self.font_medium, fill=TEXT)
|
||
confidence = int((item.get("confidence") or 0) * 100)
|
||
self._pill(draw, (820, y + 28, 976, y + 70), f"{confidence}%", self.font_tiny, CYAN, "#06201c")
|
||
self._multiline(draw, interpretation, 104, y + 78, 850, self.font_tiny, MUTED, 99, line_gap=7)
|
||
self._multiline(draw, f"建议：{advice}", 104, y + 78 + len(interpretation_lines) * 33 + 10, 850, self.font_tiny, "#eadcc1", 99, line_gap=7)
|
||
y += card_h + 18
|
||
return y + 10
|
||
|
||
def _draw_suggestions(self, draw: ImageDraw.ImageDraw, y: int, data: dict) -> int:
|
||
suggestions = data.get("suggestions") or []
|
||
scratch_lines = [self._wrap_text(draw, suggestion, 824, self.font_small) for suggestion in suggestions]
|
||
bottom = y + 110 + sum(len(lines) * 40 + 12 for lines in scratch_lines)
|
||
self._round_rect(draw, (64, y, POSTER_W - 64, bottom), 30, PANEL)
|
||
draw.text((104, y + 30), "近期建议", font=self.font_medium, fill=GOLD)
|
||
sy = y + 92
|
||
for suggestion in suggestions:
|
||
draw.ellipse((104, sy + 10, 116, sy + 22), fill=CYAN)
|
||
sy = self._multiline(draw, suggestion, 132, sy, 824, self.font_small, TEXT, 99, line_gap=8) + 12
|
||
return sy
|
||
|
||
def _draw_footer(self, draw: ImageDraw.ImageDraw, poster_h: int) -> None:
|
||
draw.line((64, poster_h - 118, POSTER_W - 64, poster_h - 118), fill=(34, 56, 58), width=1)
|
||
draw.text((64, poster_h - 88), "赛博先生 · AI 命理实验室", font=self.font_small, fill=TEXT)
|
||
draw.text((64, poster_h - 48), "仅供娱乐与自我反思，不构成现实决策建议", font=self.font_tiny, fill=DIM)
|
||
draw.text((POSTER_W - 180, poster_h - 88), "掌", font=self.font_medium, fill=CYAN)
|
||
|
||
def _score(self, data: dict) -> int:
|
||
confidences = [item.get("confidence", 0) for item in data.get("dimensions", []) if isinstance(item, dict)]
|
||
base = sum(confidences) / len(confidences) if confidences else 0.68
|
||
quality = (data.get("quality_check") or {}).get("confidence", 0.7)
|
||
return max(60, min(96, round((base * 0.65 + quality * 0.35) * 100)))
|
||
|
||
def _load_palm_image(self, image: UploadedImage | None) -> Image.Image | None:
|
||
if not image:
|
||
return None
|
||
try:
|
||
return Image.open(BytesIO(self.image_service.read_bytes(image.storage_key)))
|
||
except Exception:
|
||
return None
|
||
|
||
def _round_rect(self, draw: ImageDraw.ImageDraw, xy: tuple[int, int, int, int], radius: int, fill: str) -> None:
|
||
draw.rounded_rectangle(xy, radius=radius, fill=fill, outline=(0, 224, 184, 54), width=1)
|
||
|
||
def _pill(
|
||
self,
|
||
draw: ImageDraw.ImageDraw,
|
||
xy: tuple[int, int, int, int],
|
||
text: str,
|
||
font: ImageFont.ImageFont,
|
||
fill: str,
|
||
bg: str,
|
||
) -> None:
|
||
draw.rounded_rectangle(xy, radius=(xy[3] - xy[1]) // 2, fill=bg, outline=fill, width=1)
|
||
tw = self._text_w(draw, text, font)
|
||
th = self._text_h(draw, text, font)
|
||
draw.text((xy[0] + (xy[2] - xy[0] - tw) / 2, xy[1] + (xy[3] - xy[1] - th) / 2 - 1), text, font=font, fill=fill)
|
||
|
||
def _multiline(
|
||
self,
|
||
draw: ImageDraw.ImageDraw,
|
||
text: str,
|
||
x: int,
|
||
y: int,
|
||
max_width: int,
|
||
font: ImageFont.ImageFont,
|
||
fill: str,
|
||
max_lines: int,
|
||
line_gap: int,
|
||
) -> int:
|
||
lines = self._wrap_text(draw, text, max_width, font)
|
||
if len(lines) > max_lines:
|
||
lines = lines[:max_lines]
|
||
lines[-1] = lines[-1].rstrip("，。,. ") + "..."
|
||
line_h = self._text_h(draw, "国", font) + line_gap
|
||
for i, line in enumerate(lines):
|
||
draw.text((x, y + i * line_h), line, font=font, fill=fill)
|
||
return y + len(lines) * line_h
|
||
|
||
def _wrap_text(self, draw: ImageDraw.ImageDraw, text: str, max_width: int, font: ImageFont.ImageFont) -> list[str]:
|
||
lines: list[str] = []
|
||
current = ""
|
||
for char in text:
|
||
candidate = current + char
|
||
if self._text_w(draw, candidate, font) <= max_width:
|
||
current = candidate
|
||
else:
|
||
if current:
|
||
lines.append(current)
|
||
current = char
|
||
if current:
|
||
lines.append(current)
|
||
return lines or [""]
|
||
|
||
def _text_w(self, draw: ImageDraw.ImageDraw, text: str, font: ImageFont.ImageFont) -> int:
|
||
return int(draw.textbbox((0, 0), text, font=font)[2])
|
||
|
||
def _text_h(self, draw: ImageDraw.ImageDraw, text: str, font: ImageFont.ImageFont) -> int:
|
||
box = draw.textbbox((0, 0), text, font=font)
|
||
return int(box[3] - box[1])
|
||
|
||
|
||
def _font(size: int) -> ImageFont.ImageFont:
|
||
candidates = [
|
||
"/System/Library/Fonts/Hiragino Sans GB.ttc",
|
||
"/System/Library/Fonts/STHeiti Medium.ttc",
|
||
"/System/Library/Fonts/Supplemental/Arial Unicode.ttf",
|
||
"/System/Library/Fonts/Supplemental/Songti.ttc",
|
||
]
|
||
for candidate in candidates:
|
||
if Path(candidate).exists():
|
||
return ImageFont.truetype(candidate, size=size)
|
||
return ImageFont.load_default(size=size)
|
||
|
||
|
||
def _trim(text: str, max_chars: int) -> str:
|
||
text = text.replace("\n", " ").strip()
|
||
return text if len(text) <= max_chars else text[: max_chars - 1] + "…"
|