File size: 1,285 Bytes
55f57c9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | # report.py
"""
生成 HTML 与 Markdown 格式的报告，包含各维度得分和建议。
"""
from jinja2 import Template
HTML_TEMPLATE = """
<html>
<head><title>PRIVAL Prompt 验证报告</title></head>
<body>
<h2>PRIVAL 验证报告</h2>
<p>Overall Score: {{ overall }}</p>
<table border=1 cellpadding=5>
<tr><th>维度</th><th>分数</th><th>建议</th></tr>
{% for dim, score in scores.items() %}
<tr>
<td>{{ dim }}</td>
<td>{{ score }}</td>
<td>{{ suggestions[dim] | join('; ') }}</td>
</tr>
{% endfor %}
</table>
</body>
</html>
"""
MD_TEMPLATE = """
# PRIVAL Prompt 验证报告
**Overall Score:** {{ overall }}
| 维度 | 分数 | 建议 |
|-----|-----|------|
{% for dim, score in scores.items() %}
| {{ dim }} | {{ score }} | {{ suggestions[dim] | join('; ') }} |
{% endfor %}
"""
def generate_html_report(data: dict) -> str:
"""返回 HTML 格式报告字符串。"""
tmpl = Template(HTML_TEMPLATE)
return tmpl.render(scores=data['scores'], suggestions=data['suggestions'], overall=data['overall'])
def generate_md_report(data: dict) -> str:
"""返回 Markdown 格式报告字符串。"""
tmpl = Template(MD_TEMPLATE)
return tmpl.render(scores=data['scores'], suggestions=data['suggestions'], overall=data['overall']) |