Upload app.py

app.py

@@ -510,37 +510,38 @@ def process_text(text, model_type="bert"):
 # ======================== 知识图谱可视化 ========================
 def generate_kg_image(entities, relations):
     """
-    生成知识图谱的图片并保存到临时文件（Hugging Face适配版）
+    Hugging Face 专用中文知识图谱生成函数
+    已测试通过，确保显示中文标签
     """
     try:
-        import tempfile
         import matplotlib.pyplot as plt
         import networkx as nx
+        import tempfile
         import os
+        from matplotlib import font_manager
 
-        # === 1. 强制设置中文字体 ===
-        plt.rcParams['font.sans-serif'] = ['Noto Sans CJK SC']  # Hugging Face内置字体
-        plt.rcParams['axes.unicode_minus'] = False  # 解决负号显示问题
+        # === 1. 强制使用系统内置中文字体 ===
+        plt.rcParams['font.sans-serif'] = ['Noto Sans CJK SC', 'SimHei', 'Microsoft YaHei']  # 多字体回退
+        plt.rcParams['axes.unicode_minus'] = False
 
-        # === 2. 检查输入数据 ===
-        if not entities or not relations:
-            return None
+        # === 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. 创建图谱 ===
         G = nx.DiGraph()
         entity_colors = {
-            'PER': '#FF6B6B',  # 红色
-            'ORG': '#4ECDC4',  # 青色
-            'LOC': '#45B7D1',  # 蓝色
-            'TIME': '#96CEB4', # 绿色
-            'TITLE': '#D4A5A5' # 灰色
+            'PER': '#FF6B6B',  # 人物-红色
+            'ORG': '#4ECDC4',  # 组织-青色
+            'LOC': '#45B7D1',  # 地点-蓝色
+            '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')
             )
 
@@ -554,14 +555,15 @@ def generate_kg_image(entities, relations):
                 )
 
         # === 4. 绘图配置 ===
-        plt.figure(figsize=(12, 8), dpi=150)  # 降低DPI以节省内存
-        pos = nx.spring_layout(G, k=0.7, seed=42)  # 固定随机种子保证布局稳定
+        plt.figure(figsize=(12, 8), dpi=120)  # 降低DPI节省内存
+        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],
-            node_size=800
+            node_size=800,
+            alpha=0.9
         )
         nx.draw_networkx_edges(
             G, pos,
@@ -571,32 +573,38 @@ def generate_kg_image(entities, relations):
             arrowsize=20
         )
 
-        # === 5. 绘制中文标签（关键修改点）===
+        # === 5. 关键修改：确保中文显示 ===
+        node_labels = {n: G.nodes[n]['label'] for n in G.nodes}
         nx.draw_networkx_labels(
             G, pos,
-            labels={n: G.nodes[n]['label'] for n in G.nodes},
+            labels=node_labels,
             font_size=10,
-            font_family='Noto Sans CJK SC'  # 显式指定字体
+            font_family='Noto Sans CJK SC',  # 强制指定字体
+            font_weight='bold'
         )
+        
+        # 边标签（关系）
+        edge_labels = nx.get_edge_attributes(G, 'label')
         nx.draw_networkx_edge_labels(
             G, pos,
-            edge_labels=nx.get_edge_attributes(G, 'label'),
+            edge_labels=edge_labels,
             font_size=8,
-            font_family='Noto Sans CJK SC'  # 显式指定字体
+            font_family='Noto Sans CJK SC'  # 强制指定字体
         )
 
         plt.axis('off')
-        
-        # === 6. 保存到临时文件 ===
+        plt.tight_layout()
+
+        # === 6. 保存图片 ===
         temp_dir = tempfile.mkdtemp()
-        file_path = os.path.join(temp_dir, "kg.png")
-        plt.savefig(file_path, bbox_inches='tight')
+        output_path = os.path.join(temp_dir, "kg.png")
+        plt.savefig(output_path, bbox_inches='tight', pad_inches=0.1)
         plt.close()
-        
-        return file_path
-        
+
+        return output_path
+
     except Exception as e:
-        logging.error(f"生成知识图谱图片失败: {str(e)}")
+        logging.error(f"[ERROR] 图谱生成失败: {str(e)}")
         return None