Update Dockerfile

Dockerfile

@@ -5,7 +5,6 @@ WORKDIR /app
 ENV PYTHONUNBUFFERED=1 \
     PYTHONDONTWRITEBYTECODE=1
 
-# 安装系统依赖
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     git \
@@ -24,22 +23,26 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
-# 安装Python和gradio
-RUN conda install -c conda-forge \
-    python=3.10 \
-    gradio \
-    -y && \
+RUN conda install -c conda-forge python=3.10 gradio=5.0.0 -y && \
     conda clean --all -y
 
-# 从PyPI安装cadquery-ocp
-RUN pip install --no-cache-dir cadquery-ocp==7.8.1.1.post1
+RUN pip install --no-cache-dir \
+    cadquery-ocp==7.8.1.1.post1 \
+    gmsh==4.15.0 \
+    numpy==2.2.6 \
+    pandas==2.3.3 \
+    scikit-learn==1.7.2 \
+    torch==2.9.1 \
+    matplotlib==3.10.7 \
+    joblib==1.5.2
 
 COPY . /app/
 
 EXPOSE 7860
 
 # 验证安装
-RUN python -c "import sys; print('Python:', sys.version)" && \
-    python -c "from OCP.STEPControl import STEPControl_Reader; print('OCP installed successfully')"
+RUN python -c "from OCP.STEPControl import STEPControl_Reader; print('OCP OK')" && \
+    python -c "import gmsh; print('gmsh OK')" && \
+    python -c "import torch; print('PyTorch OK')"
 
 CMD ["python", "/app/app.py"]