Upload app.py

app.py

@@ -510,25 +510,30 @@ def process_text(text, model_type="bert"):
 # ======================== 知识图谱可视化 ========================
 def generate_kg_image(entities, relations):
     """
-    Hugging Face 专用中文知识图谱生成函数
-    已测试通过，确保显示中文标签
+    中文知识图谱生成函数，支持自动匹配系统中的中文字体，避免中文显示为方框。
     """
     try:
         import matplotlib.pyplot as plt
         import networkx as nx
         import tempfile
         import os
+        import logging
         from matplotlib import font_manager
 
-        # === 1. 强制使用系统内置中文字体 ===
-        plt.rcParams['font.sans-serif'] = ['Noto Sans CJK SC', 'SimHei', 'Microsoft YaHei']  # 多字体回退
+        # === 1. 自动查找可用中文字体 ===
+        def find_chinese_font():
+            preferred_fonts = ['Noto Sans CJK SC', 'Microsoft YaHei', 'SimHei', 'WenQuanYi Zen Hei', 'PingFang SC']
+            available_fonts = set(f.name for f in font_manager.fontManager.ttflist)
+            for font in preferred_fonts:
+                if font in available_fonts:
+                    return font
+            return 'sans-serif'  # fallback
+
+        chinese_font = find_chinese_font()
+        plt.rcParams['font.sans-serif'] = [chinese_font]
         plt.rcParams['axes.unicode_minus'] = False
 
-        # === 2. 检查可用字体（调试用）===
-        if os.environ.get('DISPLAY_FONTS', '0') == '1':  # 设置环境变量DISPLAY_FONTS=1启用
-            print("可用字体:", [f.name for f in font_manager.fontManager.ttflist if 'CJK' in f.name or 'Hei' in f.name])
-
-        # === 3. 创建图谱 ===
+        # === 2. 创建图谱 ===
         G = nx.DiGraph()
         entity_colors = {
             'PER': '#FF6B6B',  # 人物-红色
@@ -537,15 +542,13 @@ def generate_kg_image(entities, relations):
             'TIME': '#96CEB4'  # 时间-绿色
         }
 
-        # 添加节点（实体）
         for entity in entities:
             G.add_node(
                 entity["text"],
-                label=f"{entity['text']} ({entity['type']})",  # 保留英文类型
+                label=f"{entity['text']} ({entity['type']})",
                 color=entity_colors.get(entity['type'], '#D3D3D3')
             )
 
-        # 添加边（关系）
         for relation in relations:
             if relation["head"] in G.nodes and relation["tail"] in G.nodes:
                 G.add_edge(
@@ -554,11 +557,10 @@ def generate_kg_image(entities, relations):
                     label=relation["relation"]
                 )
 
-        # === 4. 绘图配置 ===
-        plt.figure(figsize=(12, 8), dpi=120)  # 降低DPI节省内存
-        pos = nx.spring_layout(G, k=0.7, seed=42)  # 固定随机种子
+        # === 3. 绘图配置 ===
+        plt.figure(figsize=(12, 8), dpi=150)
+        pos = nx.spring_layout(G, k=0.7, seed=42)
 
-        # 绘制节点和边
         nx.draw_networkx_nodes(
             G, pos,
             node_color=[G.nodes[n]['color'] for n in G.nodes],
@@ -573,29 +575,27 @@ def generate_kg_image(entities, relations):
             arrowsize=20
         )
 
-        # === 5. 关键修改：确保中文显示 ===
         node_labels = {n: G.nodes[n]['label'] for n in G.nodes}
         nx.draw_networkx_labels(
             G, pos,
             labels=node_labels,
             font_size=10,
-            font_family='Noto Sans CJK SC',  # 强制指定字体
+            font_family=chinese_font,
             font_weight='bold'
         )
-        
-        # 边标签（关系）
+
         edge_labels = nx.get_edge_attributes(G, 'label')
         nx.draw_networkx_edge_labels(
             G, pos,
             edge_labels=edge_labels,
             font_size=8,
-            font_family='Noto Sans CJK SC'  # 强制指定字体
+            font_family=chinese_font
         )
 
         plt.axis('off')
         plt.tight_layout()
 
-        # === 6. 保存图片 ===
+        # === 4. 保存图片 ===
         temp_dir = tempfile.mkdtemp()
         output_path = os.path.join(temp_dir, "kg.png")
         plt.savefig(output_path, bbox_inches='tight', pad_inches=0.1)
@@ -608,6 +608,7 @@ def generate_kg_image(entities, relations):
         return None
 
 
+
 def process_file(file, model_type="bert"):
     try:
         with open(file.name, 'rb') as f: