Upload app.py

app.py

@@ -542,18 +542,19 @@ 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")
-            font_prop = font_manager.FontProperties()  # fallback
+            plt.rcParams['font.family'] = ['DejaVu Sans']
+
         plt.rcParams['axes.unicode_minus'] = False
 
         # === 2. 创建图谱 ===
         G = nx.DiGraph()
         entity_colors = {
-            'PER': '#FF6B6B',
-            'ORG': '#4ECDC4',
-            'LOC': '#45B7D1',
-            'TIME': '#96CEB4',
-            'TITLE': '#D3D3D3'
+            'PER': '#FF6B6B',  # 人物-红色
+            'ORG': '#4ECDC4',  # 组织-青色
+            'LOC': '#45B7D1',  # 地点-蓝色
+            'TIME': '#96CEB4'  # 时间-绿色
         }
 
         for entity in entities:
@@ -594,7 +595,7 @@ def generate_kg_image(entities, relations):
             G, pos,
             labels=node_labels,
             font_size=10,
-            font_properties=font_prop,
+            font_family=font_prop.get_name() if chinese_font else 'SimHei',
             font_weight='bold'
         )
 
@@ -603,15 +604,17 @@ def generate_kg_image(entities, relations):
             G, pos,
             edge_labels=edge_labels,
             font_size=8,
-            font_properties=font_prop
+            font_family=font_prop.get_name() if chinese_font else 'SimHei'
         )
 
         plt.axis('off')
         plt.tight_layout()
 
         # === 4. 保存图片 ===
-        temp_dir = tempfile.mkdtemp()
+        temp_dir = tempfile.mkdtemp()  # 确保在 Docker 容器中有权限写入
         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)
@@ -624,7 +627,6 @@ def generate_kg_image(entities, relations):
         return None
 
 
-
 # ======================== 文件处理 ========================
 def process_file(file, model_type="bert"):
     try: