Upload 2 files

Dockerfile

@@ -5,9 +5,10 @@ FROM python:3.10-slim
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \
         fonts-noto-cjk \    # 中文字体
+        fontconfig \         # 字体配置工具
         python3-pip \       # pip
-        git \               # git
-        curl \              # curl
+        git \               # git（如果有需要拉取代码）
+        curl \              # curl（如果有需要下载文件）
         && apt-get clean && \
         rm -rf /var/lib/apt/lists/*
 

app.py

@@ -508,29 +508,33 @@ def process_text(text, model_type="bert"):
     return ent_text, rel_text, kg_text, f"{total_duration:.2f} 秒"
 
 # ======================== 知识图谱可视化 ========================
+import matplotlib.pyplot as plt
+import networkx as nx
+import tempfile
+import os
+import logging
+from matplotlib import font_manager
+
 def generate_kg_image(entities, relations):
     """
     中文知识图谱生成函数，支持自动匹配系统中的中文字体，避免中文显示为方框。
     """
     try:
-        import matplotlib.pyplot as plt
-        import networkx as nx
-        import tempfile
-        import os
-        import logging
-        from matplotlib import font_manager
-
-        # === 1. 自动查找可用中文字体 ===
+        # === 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
+            # 使用字体路径而非字体名，这里硬编码路径
+            font_path = "/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc"
+            if os.path.exists(font_path):
+                return font_path
+            return None
 
         chinese_font = find_chinese_font()
-        plt.rcParams['font.sans-serif'] = [chinese_font]
+        if chinese_font:
+            plt.rcParams['font.sans-serif'] = [chinese_font]
+        else:
+            # 如果字体路径未找到，使用默认字体
+            plt.rcParams['font.sans-serif'] = ['SimHei']
+
         plt.rcParams['axes.unicode_minus'] = False
 
         # === 2. 创建图谱 ===
@@ -580,7 +584,7 @@ def generate_kg_image(entities, relations):
             G, pos,
             labels=node_labels,
             font_size=10,
-            font_family=chinese_font,
+            font_family=chinese_font or 'SimHei',  # 若字体路径为空，使用 SimHei 作为备选
             font_weight='bold'
         )
 
@@ -589,7 +593,7 @@ def generate_kg_image(entities, relations):
             G, pos,
             edge_labels=edge_labels,
             font_size=8,
-            font_family=chinese_font
+            font_family=chinese_font or 'SimHei'  # 同上
         )
 
         plt.axis('off')
@@ -611,7 +615,8 @@ def generate_kg_image(entities, relations):
         logging.error(f"[ERROR] 图谱生成失败: {str(e)}")
         return None
 
-        
+
+
 # ======================== 文件处理 ========================
 def process_file(file, model_type="bert"):
     try: