"""
人物关系图谱可视化工具
"""
import json
from typing import Dict, List
def visualize_ascii(graph: RelationGraph):
entities = graph.get_all_entities()
for entity in entities:
relations = graph.get_relations(entity)
if not relations:
continue
print(f"┌─ {entity}")
for i, (rel_type, targets) in enumerate(relations.items()):
is_last = (i == len(relations) - 1)
prefix = "└─" if is_last else "├─"
for j, target in enumerate(targets):
is_last_target = (j == len(targets) - 1)
if j == 0:
print(f"│ {prefix} [{rel_type}] → {target}")
else:
continuation = " " if is_last else "│ "
print(f"│ {continuation} └─ {target}")
print("│")
print("=" * 60 + "\n")
def save_graphviz(graph: RelationGraph, output_file: str = "relation_graph.png"):
"""
使用 Graphviz 生成关系图谱图片
"""
try:
from graphviz import Digraph
except ImportError:
print("Graphviz not installed. Install with: pip install graphviz")
return
dot = Digraph(comment='人物关系图谱')
dot.attr(rankdir='LR')
dot.attr('node', shape='circle', style='filled', fillcolor='lightblue')
entities = graph.get_all_entities()
for entity in entities:
dot.node(entity, entity)
for head in graph.graph:
for relation in graph.graph[head]:
for tail in graph.graph[head][relation]:
dot.edge(head, tail, label=relation)
dot.render(output_file.replace('.png', ''), format='png', cleanup=True)
def generate_html_visualization(graph: RelationGraph, output_file: str = "relation_graph.html"):
"""
生成交互式网页可视化（使用 Cytoscape.js）
"""
cytoscape_data = graph.export_cytoscape()
html_template = """
人物关系图谱
人物关系图谱
"""
with open(output_file, 'w', encoding='utf-8') as f:
f.write(html_template)