模板化改造斗鱼日报与群聊总结图片渲染,支持HTML模板独立维护

变更项:\n1. 将 plugins/douyu/report_template.py 从内联HTML重构为模板渲染,新增 plugins/douyu/templates/daily_report.html 承载完整样式与结构,Python侧仅保留数据组装与安全注入。\n2. 修复斗鱼日报模板迁移后的样式缺失问题,补齐 metric-card、insight-card、badge-wall、active-user-grid、chart 等所有关键类样式,确保视觉与旧版一致。\n3. 在 plugins/message_summary/main.py 新增模板化图片渲染链路:优先使用 HtmlTemplateRenderer + html_to_image 生成总结图片,模板异常时自动回退 convert_md_str_to_image,保证稳定性。\n4. 新增 plugins/message_summary/templates/summary_card.html 作为群聊总结卡片模板,后续可仅改模板文件完成UI迭代。\n5. 扩展 plugins/message_summary/config.toml 输出配置,增加 summary_image_mode 与 summary_image_template_path,支持模板模式与回退模式按配置切换。\n6. 保持原有业务流程与发送逻辑不变,仅改造渲染层,降低后续维护成本。
This commit is contained in:
liuwei
2026-04-20 13:23:54 +08:00
parent f1a6f6d565
commit f719b10c4a
5 changed files with 849 additions and 621 deletions

View File

@@ -11,3 +11,9 @@ retry_delays_seconds = [10, 20]
[output]
output_dir = "output"
image_format = "png"
# 图片渲染模式:
# - template: 使用 HTML 模板渲染(推荐,便于后续只改模板)
# - markdown: 使用历史 md2image 样式
summary_image_mode = "template"
# 总结卡片模板路径(相对项目根目录)
summary_image_template_path = "plugins/message_summary/templates/summary_card.html"

View File

@@ -1,4 +1,5 @@
import asyncio
import html
import json
import re
import time
@@ -7,6 +8,7 @@ from pathlib import Path
from typing import Dict, Any, Tuple, Optional, List
from loguru import logger
from markupsafe import Markup
from base.plugin_common.message_plugin_interface import MessagePluginInterface
from base.plugin_common.plugin_interface import PluginStatus
@@ -15,7 +17,8 @@ from utils.compress_chat_data import compress_chat_data
from utils.decorator.plugin_decorators import plugin_stats_decorator
from utils.decorator.points_decorator import plugin_points_cost
from utils.decorator.rate_limit_decorator import group_feature_rate_limit
from utils.markdown_to_image import convert_md_str_to_image
from utils.html_template_renderer import HtmlTemplateRenderer
from utils.markdown_to_image import convert_md_str_to_image, html_to_image
from utils.revoke.message_auto_revoke import MessageAutoRevoke
from utils.robot_cmd.robot_command import GroupBotManager, PermissionStatus
from utils.string_utils import remove_reasoning_content, remove_trailing_content
@@ -94,6 +97,13 @@ class MessageSummaryPlugin(MessagePluginInterface):
self._image_render_timeout_seconds = int(output_config.get("image_render_timeout_seconds", 90))
# 默认只尝试 1 次,优先保证任务快速返回;需要更高成功率可在配置里提高。
self._image_render_retries = int(output_config.get("image_render_retries", 1))
# 输出模板配置:
# 1. summary_image_mode=template 时优先按 HTML 模板生图;
# 2. template 失败会自动回退到 markdown 模式,保证可用性。
self._summary_image_mode = str(output_config.get("summary_image_mode", "template")).strip().lower()
self._summary_image_template_path = str(
output_config.get("summary_image_template_path", "plugins/message_summary/templates/summary_card.html")
).strip()
self.llm_client = UnifiedLLMClient(api_config)
self._api_mode = self.llm_client.mode or self._api_mode
self._response_mode = self.llm_client.response_mode or self._response_mode
@@ -398,6 +408,91 @@ class MessageSummaryPlugin(MessagePluginInterface):
cleaned = re.sub(r'\n{3,}', '\n\n', cleaned).strip()
return cleaned
@staticmethod
def _summary_markdown_to_html(summary_text: str) -> str:
"""把总结 Markdown 转为基础 HTML 片段(模板内部展示用)。"""
# 这里不依赖第三方 markdown 库,保证在最小运行环境也能稳定渲染。
# 规则按“标题/列表/段落”三类做轻量转换,足够覆盖总结文本场景。
lines = str(summary_text or "").splitlines()
html_parts: List[str] = []
list_items: List[str] = []
def flush_list() -> None:
if not list_items:
return
html_parts.append("<ul>" + "".join(list_items) + "</ul>")
list_items.clear()
for raw_line in lines:
line = raw_line.strip()
if not line:
flush_list()
continue
if line.startswith("## "):
flush_list()
html_parts.append(f"<h2>{html.escape(line[3:].strip())}</h2>")
continue
if line.startswith("### "):
flush_list()
html_parts.append(f"<h3>{html.escape(line[4:].strip())}</h3>")
continue
if line.startswith("- "):
list_items.append(f"<li>{html.escape(line[2:].strip())}</li>")
continue
flush_list()
html_parts.append(f"<p>{html.escape(line)}</p>")
flush_list()
return "".join(html_parts)
def _render_summary_template_html(self, group_name: str, summary_text: str) -> str:
"""根据模板路径渲染总结图片 HTML。"""
# 约束:模板只负责展示,正文仍然由模型生成并在此做安全转义后注入。
renderer = HtmlTemplateRenderer()
summary_html = self._summary_markdown_to_html(summary_text)
return renderer.render(
self._summary_image_template_path,
{
"title": f"{group_name} 群聊总结",
"generated_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
"summary_html": Markup(summary_html),
},
)
async def _render_summary_image(
self,
answer: str,
group_name: str,
output_filename: str,
total_timeout: int,
) -> str:
"""生成总结图片:模板优先,失败自动回退 Markdown。"""
image_root = Path("temp") / "md2image"
image_root.mkdir(parents=True, exist_ok=True)
output_path = image_root / output_filename
# 优先模板渲染,便于后续仅改 HTML 文件完成样式迭代。
if self._summary_image_mode != "markdown":
try:
html_content = self._render_summary_template_html(group_name=group_name, summary_text=answer)
await asyncio.wait_for(html_to_image(html_content, str(output_path)), timeout=total_timeout)
if not output_path.exists() or output_path.stat().st_size < 1024:
raise RuntimeError("模板截图输出文件异常")
return str(output_path.resolve())
except Exception as template_error:
self.LOG.warning(f"模板渲染失败,回退 Markdown 转图: {template_error}")
# 回退逻辑:保持旧链路,确保模板异常时仍能发送图片。
return await asyncio.wait_for(
convert_md_str_to_image(
answer,
output_filename,
max_retries=self._image_render_retries,
render_timeout_seconds=self._image_render_timeout_seconds,
html_timeout_seconds=min(30, self._image_render_timeout_seconds),
),
timeout=total_timeout,
)
def _stringify_output(self, value: Any) -> str:
"""把 workflow 输出统一转成文本"""
if value is None:
@@ -546,15 +641,12 @@ class MessageSummaryPlugin(MessagePluginInterface):
self.LOG.info(f"开始生成图片: {output_path}")
# 额外包一层总超时,确保就算底层依赖异常也不会把整个任务拖住。
total_timeout = max(30, self._image_render_timeout_seconds * self._image_render_retries + 10)
spath = await asyncio.wait_for(
convert_md_str_to_image(
answer,
output_path,
max_retries=self._image_render_retries,
render_timeout_seconds=self._image_render_timeout_seconds,
html_timeout_seconds=min(30, self._image_render_timeout_seconds),
),
timeout=total_timeout,
# 统一走图片渲染入口:优先模板,失败自动降级到旧的 Markdown 转图链路。
spath = await self._render_summary_image(
answer=answer,
group_name=group_name,
output_filename=output_path,
total_timeout=total_timeout,
)
self.LOG.info(f"成功生成图片: {spath}")
except Exception as e:

View File

@@ -0,0 +1,118 @@
<html>
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<style>
:root {
--bg-top: #eef3fb;
--bg-bottom: #e5edf7;
--card: rgba(255, 255, 255, 0.96);
--title: #0f172a;
--text: #334155;
--muted: #64748b;
--line: rgba(148, 163, 184, 0.22);
--blue: #2563eb;
--cyan: #0f766e;
--shadow: 0 20px 48px rgba(15, 23, 42, 0.12);
}
* { box-sizing: border-box; }
body {
margin: 0;
padding: 26px;
background:
radial-gradient(circle at 10% 0%, rgba(37, 99, 235, 0.12), transparent 30%),
radial-gradient(circle at 90% 0%, rgba(15, 118, 110, 0.12), transparent 26%),
linear-gradient(180deg, var(--bg-top) 0%, var(--bg-bottom) 100%);
font-family: "Microsoft YaHei", "PingFang SC", "Segoe UI", sans-serif;
color: var(--text);
}
.sheet {
width: 820px;
margin: 0 auto;
background: var(--card);
border-radius: 24px;
border: 1px solid rgba(255, 255, 255, 0.7);
box-shadow: var(--shadow);
overflow: hidden;
}
.hero {
padding: 24px 28px 20px;
background:
radial-gradient(circle at 18% 18%, rgba(255,255,255,0.12), transparent 18%),
linear-gradient(135deg, #111827 0%, #1d4ed8 52%, #0f766e 100%);
color: #ffffff;
}
.hero-badge {
display: inline-block;
padding: 6px 12px;
border-radius: 999px;
font-size: 12px;
letter-spacing: .05em;
background: rgba(255, 255, 255, 0.12);
border: 1px solid rgba(255, 255, 255, 0.2);
}
.hero-title {
margin: 14px 0 8px;
font-size: 30px;
font-weight: 800;
line-height: 1.25;
}
.hero-time {
font-size: 13px;
color: rgba(240, 246, 255, 0.86);
}
.content {
padding: 24px 26px 26px;
}
h2 {
margin: 20px 0 10px;
font-size: 20px;
font-weight: 800;
color: var(--title);
}
h3 {
margin: 14px 0 8px;
font-size: 16px;
font-weight: 700;
color: #1e293b;
}
p {
margin: 8px 0;
font-size: 15px;
line-height: 1.78;
color: var(--text);
}
ul {
margin: 8px 0 12px;
padding-left: 22px;
}
li {
margin: 6px 0;
line-height: 1.72;
font-size: 14px;
color: var(--text);
}
.footer {
margin-top: 20px;
padding-top: 12px;
border-top: 1px dashed var(--line);
text-align: right;
font-size: 12px;
color: var(--muted);
}
</style>
</head>
<body>
<div class="sheet">
<div class="hero">
<div class="hero-badge">群聊总结</div>
<div class="hero-title">{{ title }}</div>
<div class="hero-time">生成时间:{{ generated_at }}</div>
</div>
<div class="content">
{{ summary_html }}
<div class="footer">ABOT · Message Summary Template</div>
</div>
</div>
</body>
</html>