Upload app.py

app.py

@@ -542,19 +542,18 @@ def generate_kg_image(entities, relations):
             font_prop = font_manager.FontProperties(fname=chinese_font)
             plt.rcParams['font.family'] = font_prop.get_name()
         else:
-            # 如果字体路径未找到，使用默认字体（DejaVu Sans）
             logging.warning("Using default font")
-            plt.rcParams['font.family'] = ['DejaVu Sans']
-
+            font_prop = font_manager.FontProperties()  # fallback
         plt.rcParams['axes.unicode_minus'] = False
 
         # === 2. 创建图谱 ===
         G = nx.DiGraph()
         entity_colors = {
-            'PER': '#FF6B6B',  # 人物-红色
-            'ORG': '#4ECDC4',  # 组织-青色
-            'LOC': '#45B7D1',  # 地点-蓝色
-            'TIME': '#96CEB4'  # 时间-绿色
+            'PER': '#FF6B6B',
+            'ORG': '#4ECDC4',
+            'LOC': '#45B7D1',
+            'TIME': '#96CEB4',
+            'TITLE': '#D3D3D3'
         }
 
         for entity in entities:
@@ -595,7 +594,7 @@ def generate_kg_image(entities, relations):
             G, pos,
             labels=node_labels,
             font_size=10,
-            font_family=font_prop.get_name() if chinese_font else 'SimHei',
+            font_properties=font_prop,
             font_weight='bold'
         )
 
@@ -604,17 +603,15 @@ def generate_kg_image(entities, relations):
             G, pos,
             edge_labels=edge_labels,
             font_size=8,
-            font_family=font_prop.get_name() if chinese_font else 'SimHei'
+            font_properties=font_prop
         )
 
         plt.axis('off')
         plt.tight_layout()
 
         # === 4. 保存图片 ===
-        temp_dir = tempfile.mkdtemp()  # 确保在 Docker 容器中有权限写入
+        temp_dir = tempfile.mkdtemp()
         output_path = os.path.join(temp_dir, "kg.png")
-
-        # 打印路径以方便调试
         logging.info(f"Saving graph image to {output_path}")
 
         plt.savefig(output_path, bbox_inches='tight', pad_inches=0.1)