Update Dockerfile

Dockerfile

@@ -1,10 +1,9 @@
 # ===============================
 # BackgroundFX Pro — Final Dockerfile
-# CUDA 12.1.1 + PyTorch 2.5.1 + MediaPipe 0.10.11 + Matplotlib 3.5.0 + NumPy 1.21.6
+# CUDA 12.1.1 + PyTorch 2.5.1 + Matplotlib 3.5.0 + NumPy 1.21.6
 # ===============================
 ARG PYTHON_VERSION=3.10
 FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
-
 # ---- Environment Setup ----
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONUNBUFFERED=1 \
@@ -29,12 +28,10 @@ ENV DEBIAN_FRONTEND=noninteractive \
     MATANY_DEVICE=cuda \
     TF_CPP_MIN_LOG_LEVEL=2 \
     SAM2_CHECKPOINT=/home/user/app/checkpoints/sam2.1_hiera_large.pt
-
 # ---- Non-root User ----
 RUN useradd -m -u 1000 user
 ENV HOME=/home/user
 WORKDIR $HOME/app
-
 # ---- System Dependencies ----
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git wget curl ca-certificates \
@@ -46,60 +43,44 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libblas-dev liblapack-dev gfortran \
     libfreetype6-dev libpng-dev libjpeg-dev \
     && rm -rf /var/lib/apt/lists/*
-
 # ---- Install FFmpeg ----
 RUN apt-get update && apt-get install -y --no-install-recommends ffmpeg \
     && rm -rf /var/lib/apt/lists/*
-
 # ---- Python Bootstrap ----
 RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel
-
-# ---- Install NumPy 1.21.6 (Required for MediaPipe) ----
+# ---- Install NumPy 1.21.6 ----
 RUN python3 -m pip install --no-cache-dir numpy==1.21.6
-
 # ---- Install Matplotlib 3.5.0 (Compatible with NumPy 1.21.6) ----
 RUN python3 -m pip install --no-cache-dir matplotlib==3.5.0
-
-# ---- 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())"
-
 # ---- Copy Requirements ----
 COPY requirements.txt .
 RUN python3 -m pip install --no-cache-dir -r requirements.txt
-
 # ---- Install MatAnyone ----
 RUN echo "Installing MatAnyone..." && \
     python3 -m pip install --no-cache-dir git+https://github.com/pq-yang/MatAnyone.git \
     && python3 -c "import matanyone; print('✅ MatAnyone installed')"
-
 # ---- Install SAM2 (Editable) ----
 RUN echo "Installing SAM2..." && \
     git clone --depth=1 https://github.com/facebookresearch/segment-anything-2.git third_party/sam2 && \
     cd third_party/sam2 && \
     python3 -m pip install --no-cache-dir -e .
-
 # ---- Download SAM2 Checkpoints ----
 RUN mkdir -p /home/user/app/checkpoints && \
     wget -q --show-progress -O /home/user/app/checkpoints/sam2.1_hiera_large.pt \
     https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt
-
 # ---- Create Directory Structure ----
 RUN mkdir -p /home/user/app/pipeline /home/user/app/models /home/user/app/storage && \
     chown -R user:user /home/user/app && \
     chmod -R 755 /home/user/app
-
 # ---- Copy Application Code ----
 COPY --chown=user:user . .
-
 # ---- Healthcheck ----
 HEALTHCHECK --interval=30s --timeout=8s --retries=3 CMD \
-    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__)"
-
+    python3 -c "import numpy; print('NumPy:', numpy.__version__); import matplotlib; print('Matplotlib:', matplotlib.__version__); import torch; print('PyTorch:', torch.__version__)"
 # ---- Runtime Startup ----
 USER user
 EXPOSE 7860
@@ -119,4 +100,4 @@ CMD ["sh", "-c", "\
     exec streamlit run --server.port=${PORT} --server.address=0.0.0.0 \
     --server.maxUploadSize=200 --server.enableCORS=false \
     --server.enableXsrfProtection=false --logger.level=debug \
-    streamlit_app.py"]
+    streamlit_app.py"]