Update Dockerfile

Dockerfile

@@ -1,43 +1,79 @@
 FROM python:3.10
 
-# 安装构建工具、git 和必要的库
+# 安装所有可能需要的构建工具
 RUN apt-get update && apt-get install -y \
     git \
     build-essential \
     cmake \
+    g++ \
+    gcc \
+    gfortran \
     libboost-all-dev \
-    && rm -rf /var/lib/apt/lists/* && \
-    python -m pip install --upgrade pip
+    python3-dev \
+    swig \
+    && rm -rf /var/lib/apt/lists/*
 
-# 创建非 root 用户
+# 升级 pip 和构建工具
+RUN python -m pip install --upgrade pip setuptools wheel
+
+# 先安装编译依赖
+RUN pip install --no-cache-dir \
+    Cython>=0.29.0 \
+    numpy>=1.21.0 \
+    scipy>=1.7.0
+
+WORKDIR /tmp
+
+# 复制 medpy source
+COPY ./medpy/source /tmp/medpy_source
+
+WORKDIR /tmp/medpy_source
+
+# 清理之前的编译产物
+RUN rm -rf build/ dist/ *.egg-info
+
+# 详细编译日志
+RUN echo "=== Starting medpy compilation ===" && \
+    python setup.py build_ext --inplace --force --verbose 2>&1 | tee /tmp/compile.log && \
+    echo "=== Compilation log ===" && \
+    cat /tmp/compile.log
+
+# 安装编译后的 medpy
+RUN pip install -e . --verbose
+
+# 关键：验证 maxflow 是否真的编译成功
+RUN echo "=== Verifying maxflow ===" && \
+    python -c "import sys; print('Python path:', sys.path)" && \
+    python -c "import medpy; print('medpy location:', medpy.__file__)" && \
+    python -c "from medpy.graphcut import maxflow; print('✓✓✓ maxflow loaded successfully'); print('GraphDouble:', maxflow.GraphDouble)" && \
+    echo "=== maxflow verification passed ==="
+
+# 创建用户
 RUN useradd -m -u 1000 user
+
+# 复制编译好的 maxflow 到用户可访问位置（如果需要）
+RUN find /tmp/medpy_source -name "*.so" -o -name "*.pyd" | xargs -I {} echo "Found compiled extension: {}"
+
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
 WORKDIR /app
 
-# 复制并安装依赖
+# 安装其他依赖
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 # 复制代码
 COPY --chown=user . /app
 
-# 编译 medpy 扩展模块
-WORKDIR /app/medpy/source
-RUN python setup.py build_ext --inplace || echo "Warning: Failed to build extensions"
-RUN pip install --no-cache-dir -e . || echo "Warning: Failed to install medpy"
-
-# 返回工作目录
-WORKDIR /app
-
-# 设置 Python 路径
-ENV PYTHONPATH=/app/medpy/source:$PYTHONPATH
+# Python 路径
+ENV PYTHONPATH=/app:$PYTHONPATH
 
 # 端口与环境变量
 EXPOSE 7860
 ENV MCP_TRANSPORT=http
 ENV MCP_PORT=7860
 
-# 启动命令
-CMD ["python", "medpy/mcp_output/start_mcp.py"]
+# 启动前再次验证
+CMD python -c "from medpy.graphcut.maxflow import GraphDouble; print('✓ maxflow available')" && \
+    python medpy/mcp_output/start_mcp.py