Update Dockerfile

Dockerfile

@@ -8,36 +8,37 @@ ENV DEBIAN_FRONTEND=noninteractive \
     TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;8.9" \
     CMAKE_BUILD_PARALLEL_LEVEL=8
 
-# Python 3.11 + toolchain + **Cairo dev**
+# Python 3.11 + headers + toolchain + Cairo dev (for svglib/rlpycairo)
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3.11 python3.11-venv python3-pip \
+    python3.11 python3.11-venv python3.11-dev python3-pip \
     git git-lfs build-essential ninja-build cmake pkg-config \
     libgl1 libsm6 libxext6 ffmpeg \
     libcairo2-dev libpango1.0-dev libffi-dev \
  && rm -rf /var/lib/apt/lists/* && git lfs install \
  && ln -sf /usr/bin/python3.11 /usr/bin/python \
- && python -m pip install -U pip
-
-# Python build deps needed by flash-attn setup
-RUN pip install -U setuptools wheel packaging pybind11 ninja cmake psutil
+ && python -m pip install -U pip wheel setuptools packaging pybind11
 
 WORKDIR /app
 COPY requirements.txt /app/requirements.txt
 
-# 1) Install Torch (CUDA 12.1) first
+# 1) Install Torch (CUDA 12.1) first so flash-attn can find headers/libs
 RUN pip install --index-url https://download.pytorch.org/whl/cu121 \
     torch==2.5.1 torchvision==0.20.1
 
 # 2) Build/install flash-attn against existing Torch/CUDA
 RUN pip install --no-build-isolation --no-cache-dir flash-attn==2.7.3
 
-# 3) Rest of deps (no torch here)
-RUN pip install -r requirements.txt
+# 3) Pre-install Cairo stack from wheels to avoid building pycairo
+#    (if wheels exist for cp311/manylinux, this short-circuits Meson)
+RUN pip install --no-cache-dir pycairo==1.26.0 rlpycairo==0.4.0 svglib==1.6.0
+
+# 4) Rest of your deps (no torch here)
+RUN pip install --no-cache-dir -r requirements.txt
 
-# 4) Install starvector (pulls svglib/rlpycairo, now succeeds with cairo dev)
-RUN pip install git+https://github.com/joanrod/star-vector.git@main
+# 5) Install StarVector package last (now its extras resolve cleanly)
+RUN pip install --no-cache-dir git+https://github.com/joanrod/star-vector.git@main
 
-# 5) Copy app
+# 6) Copy app code after deps (better caching)
 COPY . /app
 
 ENV PORT=7860