Update Dockerfile

Dockerfile

@@ -1,40 +1,37 @@
-FROM python:3.10-slim
-
-# Create non-root user for HuggingFace Space
-RUN useradd -m -u 1000 user && \
-    apt-get update && \
-    apt-get install -y --no-install-recommends \
-        build-essential \
-        git \
-    && rm -rf /var/lib/apt/lists/* && \
-    python -m pip install --upgrade pip
-
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
-ENV HOME="/home/user"
-
-WORKDIR /app
-
-# Copy requirements first for better caching
-COPY --chown=user ./mcp_output/requirements.txt /app/requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
-
-# Copy the entire MatDeepLearn project
-COPY --chown=user . /app
-
-# Set Python path to include MatDeepLearn
-ENV PYTHONPATH=/app:$PYTHONPATH
-
-# Expose HuggingFace default port
-EXPOSE 7860
-
-# Set environment variables for MCP service
-ENV MCP_TRANSPORT=http
-ENV MCP_PORT=7860
-
-# Health check
-HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
-    CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:7860/health')" || exit 1
-
-# Start MCP service
-CMD ["python", "mcp_output/start_mcp.py"]
+FROM python:3.10-slim
+
+# Install system dependencies
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        build-essential \
+        git \
+    && rm -rf /var/lib/apt/lists/* && \
+    python -m pip install --upgrade pip
+
+WORKDIR /app
+
+# Install PyTorch first (CPU version)
+RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
+
+# Install PyTorch Geometric dependencies with pre-built wheels
+RUN pip install --no-cache-dir torch-scatter torch-sparse torch-cluster torch-spline-conv -f https://data.pyg.org/whl/torch-2.5.0+cpu.html
+
+# Copy and install remaining requirements
+COPY ./mcp_output/requirements.txt /app/requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the entire MatDeepLearn project
+COPY . /app
+
+# Set Python path to include MatDeepLearn
+ENV PYTHONPATH=/app:$PYTHONPATH
+
+# Expose HuggingFace default port
+EXPOSE 7860
+
+# Set environment variables for MCP service
+ENV MCP_TRANSPORT=http
+ENV MCP_PORT=7860
+
+# Start MCP service
+CMD ["python", "mcp_output/start_mcp.py"]