Update Dockerfile

Dockerfile

@@ -1,7 +1,6 @@
 # ===============================
 # BackgroundFX Pro — Final Dockerfile
-# CUDA 12.1.1 + PyTorch 2.5.1 (cu121) + Streamlit
-# Optimized for SAM2 + MatAnyone + MediaPipe
+# CUDA 12.1.1 + PyTorch 2.5.1 + MediaPipe 0.10.11 + Matplotlib 3.5.0 + NumPy 1.21.6
 # ===============================
 ARG PYTHON_VERSION=3.10
 FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
@@ -36,7 +35,7 @@ RUN useradd -m -u 1000 user
 ENV HOME=/home/user
 WORKDIR $HOME/app
 
-# ---- System Dependencies (Including NumPy/Matplotlib Requirements) ----
+# ---- System Dependencies ----
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git wget curl ca-certificates \
     python3 python3-pip python3-venv python3-dev \
@@ -55,13 +54,16 @@ RUN apt-get update && apt-get install -y --no-install-recommends ffmpeg \
 # ---- Python Bootstrap ----
 RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel
 
-# ---- Install NumPy First (Critical) ----
-RUN python3 -m pip install --no-cache-dir numpy==1.24.4
+# ---- Install NumPy 1.21.6 (Required for MediaPipe) ----
+RUN python3 -m pip install --no-cache-dir numpy==1.21.6
 
-# ---- Install Matplotlib with System Dependencies ----
+# ---- Install Matplotlib 3.5.0 (Compatible with NumPy 1.21.6) ----
 RUN python3 -m pip install --no-cache-dir matplotlib==3.5.0
 
-# ---- Install PyTorch (CUDA 12.1) ----
+# ---- Install MediaPipe 0.10.11 (After NumPy/Matplotlib) ----
+RUN python3 -m pip install --no-cache-dir mediapipe==0.10.11
+
+# ---- Install PyTorch 2.5.1 (CUDA 12.1) ----
 RUN python3 -m pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 \
     torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 \
     && python3 -c "import torch; print('PyTorch:', torch.__version__); print('CUDA available:', torch.cuda.is_available())"
@@ -70,9 +72,6 @@ RUN python3 -m pip install --no-cache-dir --index-url https://download.pytorch.o
 COPY requirements.txt .
 RUN python3 -m pip install --no-cache-dir -r requirements.txt
 
-# ---- Install MediaPipe (After NumPy/Matplotlib) ----
-RUN python3 -m pip install --no-cache-dir mediapipe==0.10.11
-
 # ---- Install MatAnyone ----
 RUN echo "Installing MatAnyone..." && \
     python3 -m pip install --no-cache-dir git+https://github.com/pq-yang/MatAnyone.git \
@@ -97,9 +96,9 @@ RUN mkdir -p /home/user/app/pipeline /home/user/app/models /home/user/app/storag
 # ---- Copy Application Code ----
 COPY --chown=user:user . .
 
-# ---- Healthcheck (Single Line) ----
+# ---- Healthcheck ----
 HEALTHCHECK --interval=30s --timeout=8s --retries=3 CMD \
-    python3 -c "import numpy; print('NumPy:', numpy.__version__); import matplotlib; print('Matplotlib:', matplotlib.__version__); import torch; print('PyTorch:', torch.__version__); import mediapipe; print('MediaPipe:', mediapipe.__version__); import subprocess; subprocess.run(['ffmpeg', '-version'], check=False)"
+    python3 -c "import numpy; print('NumPy:', numpy.__version__); import matplotlib; print('Matplotlib:', matplotlib.__version__); import mediapipe; print('MediaPipe:', mediapipe.__version__); import torch; print('PyTorch:', torch.__version__)"
 
 # ---- Runtime Startup ----
 USER user