Upload 2 files

Dockerfile

@@ -0,0 +1,30 @@
+# 使用官方 Python 轻量镜像
+FROM python:3.10-slim
+
+# 安装系统依赖，包括中文字体和其他必需的工具
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        fonts-noto-cjk \    # 中文字体
+        python3-pip \       # pip
+        git \               # git
+        curl \              # curl
+        && apt-get clean && \
+        rm -rf /var/lib/apt/lists/*
+
+# 设置 matplotlib 临时缓存目录（防止权限问题）
+ENV MPLCONFIGDIR=/tmp/matplotlib
+
+# 设置工作目录
+WORKDIR /app
+
+# 拷贝 Python 依赖文件
+COPY requirements.txt .
+
+# 安装 Python 库
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 拷贝项目代码
+COPY . .
+
+# 运行 Gradio 应用
+CMD ["python", "app.py"]

app.py

@@ -596,8 +596,12 @@ def generate_kg_image(entities, relations):
         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)
         plt.close()
 
@@ -607,8 +611,8 @@ def generate_kg_image(entities, relations):
         logging.error(f"[ERROR] 图谱生成失败: {str(e)}")
         return None
 
-
-
+        
+# ======================== 文件处理 ========================
 def process_file(file, model_type="bert"):
     try:
         with open(file.name, 'rb') as f: