Upload 3 files

Dockerfile

@@ -1,10 +1,10 @@
-# 使用 python:3.10-slim 作为基础镜像
-FROM python:3.10-slim
+# 使用更小的基础镜像 python:3.10-slim-buster，减少镜像体积
+FROM python:3.10-slim-buster
 
 # 设置工作目录
 WORKDIR /app
 
-# 安装必要的系统依赖
+# 安装系统依赖，包括图形库和中文字体
 RUN apt-get update && apt-get install -y \
     libglib2.0-0 \
     libx11-6 \
@@ -17,17 +17,20 @@ RUN apt-get update && apt-get install -y \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
-# 复制当前目录到容器内
-COPY . /app
+# 复制 requirements.txt 文件先安装 Python 依赖，减少不必要的层
+COPY requirements.txt /app/
 
 # 安装 Python 依赖
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 暴露端口（假设你的 Gradio 应用在 7860 端口运行）
+# 复制剩余应用代码到容器中
+COPY . /app
+
+# 暴露端口（假设 Gradio 应用在 7860 端口运行）
 EXPOSE 7860
 
-# 设置环境变量
+# 设置环境变量，确保 Python 输出不缓冲
 ENV PYTHONUNBUFFERED=1
 
-# 设置默认命令
+# 设置默认命令，启动应用
 CMD ["python", "app.py"]

app.py

@@ -515,103 +515,115 @@ import os
 import logging
 from matplotlib import font_manager
 
-# 强制重新加载字体缓存（关键步骤！）
-import matplotlib
-import os
-from matplotlib import font_manager
-
-# 确保字体缓存重建
-font_cache_path = os.path.join(matplotlib.get_cachedir(), 'fontlist-v330.json')
-if os.path.exists(font_cache_path):
-    os.remove(font_cache_path)
-matplotlib.font_manager._rebuild()
-
-# 验证字体是否安装成功
-print("=== 可用中文字体 ===")
-cjk_fonts = [f.name for f in font_manager.fontManager.ttflist 
-            if 'Noto Sans CJK' in f.name]
-print(cjk_fonts if cjk_fonts else "未找到Noto Sans CJK字体！")
+# 这个函数用于查找并验证中文字体路径
+def find_chinese_font():
+    # 尝试查找 Noto Sans CJK 字体
+    font_paths = [
+        "/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc",  # Noto CJK 字体
+        "/usr/share/fonts/truetype/wqy/wqy-microhei.ttc"         # 微软雅黑
+    ]
+    
+    for font_path in font_paths:
+        if os.path.exists(font_path):
+            logging.info(f"Found font at {font_path}")
+            return font_path
+    
+    logging.error("No Chinese font found!")
+    return None
 
 def generate_kg_image(entities, relations):
+    """
+    中文知识图谱生成函数，支持自动匹配系统中的中文字体，避免中文显示为方框。
+    """
     try:
-        import matplotlib.pyplot as plt
-        import networkx as nx
-        import tempfile
-        import os
-        from matplotlib import font_manager
-        
-        # === 1. 核弹级字体保障 ===
-        # 方法1：使用Noto Sans CJK
-        plt.rcParams['font.sans-serif'] = ['Noto Sans CJK SC']
+        # === 1. 确保使用合适的中文字体 ===
+        chinese_font = find_chinese_font()  # 调用查找字体函数
+        if chinese_font:
+            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']
+
         plt.rcParams['axes.unicode_minus'] = False
-        
-        # 方法2：如果仍然失败，使用最后手段
-        try:
-            # 测试字体是否有效
-            plt.text(0.5, 0.5, "中文测试", fontsize=12)
-            plt.close()
-        except:
-            print("主字体失效，启用备用方案...")
-            # 使用系统默认字体（至少能显示部分字符）
-            plt.rcParams['font.sans-serif'] = ['DejaVu Sans']
-        
+
         # === 2. 创建图谱 ===
         G = nx.DiGraph()
-        entity_colors = {'PER': '#FF6B6B', 'ORG': '#4ECDC4', 'LOC': '#45B7D1'}
-        
-        # 添加节点（简化标签，只显示中文）
-        for entity in entities:
-            G.add_node(entity["text"], 
-                      label=entity["text"],  # 只显示中文名，不显示类型
-                      color=entity_colors.get(entity['type'], 'gray'))
+        entity_colors = {
+            'PER': '#FF6B6B',  # 人物-红色
+            'ORG': '#4ECDC4',  # 组织-青色
+            'LOC': '#45B7D1',  # 地点-蓝色
+            'TIME': '#96CEB4'  # 时间-绿色
+        }
 
-        # 添加边
-        for rel in relations:
-            if rel["head"] in G.nodes and rel["tail"] in G.nodes:
-                G.add_edge(rel["head"], rel["tail"], label=rel["relation"])
+        for entity in entities:
+            G.add_node(
+                entity["text"],
+                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(
+                    relation["head"],
+                    relation["tail"],
+                    label=relation["relation"]
+                )
 
         # === 3. 绘图配置 ===
-        plt.figure(figsize=(10, 6), dpi=120)
-        pos = nx.spring_layout(G, k=0.8, seed=42)
-        
-        # 绘制节点（不带标签）
-        nx.draw_networkx_nodes(G, pos, 
-                             node_color=[G.nodes[n]['color'] for n in G.nodes],
-                             node_size=700)
-        
-        # 绘制边
-        nx.draw_networkx_edges(G, pos, 
-                             arrows=True,
-                             arrowsize=15,
-                             width=1.2)
-        
-        # === 4. 手动添加文本标签（绕过字体限制）===
-        for node, (x, y) in pos.items():
-            plt.text(x, y, 
-                    node, 
-                    fontsize=10,
-                    ha='center', 
-                    va='center',
-                    bbox=dict(facecolor='white', alpha=0.7, edgecolor='none'))
-        
-        # 边标签（英文关系可能能显示）
+        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],
+            node_size=800,
+            alpha=0.9
+        )
+        nx.draw_networkx_edges(
+            G, pos,
+            edge_color='#888888',
+            width=1.5,
+            arrows=True,
+            arrowsize=20
+        )
+
+        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=font_prop.get_name() if chinese_font else 'SimHei',
+            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)
+        nx.draw_networkx_edge_labels(
+            G, pos,
+            edge_labels=edge_labels,
+            font_size=8,
+            font_family=font_prop.get_name() if chinese_font else 'SimHei'
+        )
 
         plt.axis('off')
-        
-        # === 5. 保存图片 ===
-        temp_dir = tempfile.mkdtemp()
+        plt.tight_layout()
+
+        # === 4. 保存图片 ===
+        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)
         plt.close()
-        
+
         return output_path
 
     except Exception as e:
-        print(f"[FATAL ERROR] {str(e)}")
+        logging.error(f"[ERROR] 图谱生成失败: {str(e)}")
         return None
 
 

requirements.txt

@@ -22,7 +22,7 @@ python-json-logger>=2.0.0
 # 知识图谱可视化
 networkx>=3.0
 pyvis>=0.3.0
-matplotlib>=3.7.0
+matplotlib>=3.5.0
 plotly>=5.10.0
 
 # 可选依赖（如果使用ChatGLM）
@@ -32,6 +32,4 @@ accelerate>=0.20.0
 
 # 其他依赖
 requests>=2.25.0
-beautifulsoup4>=4.10.0
-
-fonts-noto-cjk
+beautifulsoup4>=4.10.0