ll

VideoBackgroundReplacer2/Dockerfile

@@ -1,11 +1,13 @@
 # ===============================
 # Hugging Face Space — Stable Dockerfile
-# CUDA 12.1.1 + PyTorch 2.5.1 (cu121) + Streamlit 1.32.0
-# SAM2 installed from source; MatAnyone via pip (repo)
+# CUDA 12.1.1 + PyTorch 2.5.1 (cu121) + Streamlit 1.32+
+# SAM2 from source; MatAnyone via pip (repo)
 # ===============================
 
-# Set Python version explicitly
+# ---- Build args (cache buster) ----
 ARG PYTHON_VERSION=3.10
+ARG BUILD_TS
+
 FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
 
 # ---- Environment (runtime hygiene) ----
@@ -26,11 +28,6 @@ ENV DEBIAN_FRONTEND=noninteractive \
     STREAMLIT_SERVER_HEADLESS=true \
     STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
 
-# ---- System cleanup ----
-RUN rm -rf /var/lib/apt/lists/* && \
-    apt-get clean && \
-    rm -rf /tmp/* /var/tmp/*
-
 # ---- Non-root user ----
 RUN useradd -m -u 1000 user
 ENV HOME=/home/user
@@ -38,15 +35,17 @@ WORKDIR $HOME/app
 
 # ---- System deps ----
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    git ffmpeg wget curl \
+    git ffmpeg wget curl ca-certificates \
     python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python3-pip python${PYTHON_VERSION}-venv \
     build-essential gcc g++ pkg-config \
     libffi-dev libssl-dev libc6-dev \
     libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender1 libgomp1 \
-    && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 \
-    && update-alternatives --set python3 /usr/bin/python${PYTHON_VERSION} \
-    && ln -sf /usr/bin/python${PYTHON_VERSION} /usr/bin/python \
-    && rm -rf /var/lib/apt/lists/*
+    # WebRTC deps for aiortc / streamlit-webrtc (Jammy uses libvpx6)
+    libopus0 libvpx6 libsrtp2-1 \
+ && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 \
+ && update-alternatives --set python3 /usr/bin/python${PYTHON_VERSION} \
+ && ln -sf /usr/bin/python${PYTHON_VERSION} /usr/bin/python \
+ && rm -rf /var/lib/apt/lists/*
 
 # ---- Python bootstrap ----
 RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel
@@ -54,7 +53,7 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel
 # ---- Install PyTorch (CUDA 12.1 wheels) ----
 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 - <<'PY'
+ && python3 - <<'PY'
 import torch
 print("PyTorch:", torch.__version__)
 print("CUDA available:", torch.cuda.is_available())
@@ -62,18 +61,28 @@ print("torch.version.cuda:", getattr(torch.version, "cuda", None))
 PY
 
 # ---- Copy deps first (better caching) ----
+# Cache-buster echo shows up in build logs to prove fresh layer
+ENV BUILD_TS_ENV=${BUILD_TS}
 COPY --chown=user:user requirements.txt ./
 
-# ---- Install remaining Python deps with force reinstall ----
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt --force-reinstall
+# ---- Guard & print requirements.txt (diagnostics) ----
+RUN echo "BUILD_TS=${BUILD_TS_ENV}" && \
+    echo "=== requirements.txt (as copied) ===" && ls -l requirements.txt && echo "----" && sed -n '1,200p' requirements.txt && echo "---- END ----" && \
+    # HARD GUARD: fail if the bad pin exists anywhere in the context copy
+    (! grep -R "annotated-types==0\.60\.0" -n .) || (echo "❌ Found forbidden pin annotated-types==0.60.0" && exit 1)
+
+# ---- Install remaining Python deps ----
+RUN pip install --no-cache-dir -r requirements.txt
 
 # ---- MatAnyone (pip install from repo with retry) ----
 RUN echo "Installing MatAnyone..." && \
     (python3 -m pip install --no-cache-dir -v git+https://github.com/pq-yang/MatAnyone@main#egg=matanyone || \
      (echo "Retrying MatAnyone..." && \
       python3 -m pip install --no-cache-dir -v git+https://github.com/pq-yang/MatAnyone@main#egg=matanyone)) && \
-    python3 -c "import matanyone; print('MatAnyone import OK')"
+    python3 - <<'PY'
+import matanyone
+print("✅ MatAnyone import OK:", getattr(matanyone, "__version__", "unknown"))
+PY
 
 # ---- App code ----
 COPY --chown=user:user . .
@@ -93,6 +102,36 @@ ENV PYTHONPATH=/home/user/app:/home/user/app/third_party:/home/user/app/third_pa
     TF_CPP_MIN_LOG_LEVEL=2 \
     SAM2_CHECKPOINT=/home/user/app/checkpoints/sam2_hiera_large.pt
 
+# ---- Optional: fetch SAM2 checkpoint if missing ----
+RUN python3 - <<'PY'
+import os, pathlib, urllib.request
+ckpt = os.environ.get("SAM2_CHECKPOINT","/home/user/app/checkpoints/sam2_hiera_large.pt")
+pathlib.Path(ckpt).parent.mkdir(parents=True, exist_ok=True)
+if not os.path.exists(ckpt):
+    url = "https://dl.fbaipublicfiles.com/segment_anything_2/checkpoints/sam2_hiera_large.pt"
+    try:
+        print("Downloading SAM2 checkpoint:", url)
+        urllib.request.urlretrieve(url, ckpt)
+        print("Saved to:", ckpt)
+    except Exception as e:
+        print("⚠️ Could not auto-download SAM2 checkpoint:", e)
+else:
+    print("SAM2 checkpoint already present at", ckpt)
+PY
+
+# ---- Quick sanity imports (catch cv2/mediapipe surprises early) ----
+RUN python3 - <<'PY'
+import cv2, sys
+print("✅ OpenCV", cv2.__version__)
+import mediapipe as mp
+print("✅ MediaPipe", mp.__version__)
+import annotated_types, pydantic, pydantic_core
+print("✅ annotated_types", annotated_types.__version__)
+print("✅ pydantic", pydantic.__version__, "| pydantic-core", pydantic_core.__version__)
+import torch
+print("✅ torch", torch.__version__, "| cuda avail =", torch.cuda.is_available())
+PY
+
 # ---- Create writable dirs (caches + checkpoints) ----
 RUN mkdir -p /home/user/app/checkpoints /home/user/app/.hf /home/user/app/.torch && \
     chown -R user:user /home/user/app && \
@@ -102,17 +141,17 @@ RUN mkdir -p /home/user/app/checkpoints /home/user/app/.hf /home/user/app/.torch
     chmod +x /home/user/app/app.py || true
 
 # ---- Healthcheck ----
-HEALTHCHECK --interval=30s --timeout=5s --retries=3 CMD \
-  ["python3","-c","import torch; print('torch', torch.__version__, '| cuda', getattr(torch.version,'cuda',None), '| ok=', torch.cuda.is_available())"]
+HEALTHCHECK --interval=30s --timeout=8s --retries=3 CMD \
+  ["python3","-c","import torch; import sys; print('torch', torch.__version__, '| cuda', getattr(torch.version,'cuda',None), '| ok=', torch.cuda.is_available())"]
 
 # ---- Runtime ----
 USER user
 EXPOSE 8501
 
-# Streamlit server command
+# ---- Startup: verify imports, then run Streamlit ----
 CMD ["sh", "-c", "\
   echo '===========================================' && \
-  echo '=== MYAVATAR STREAMLIT CONTAINER STARTUP ===' && \
+  echo '=== STREAMLIT CONTAINER STARTUP ===' && \
   echo '===========================================' && \
   echo 'Timestamp:' $(date) && \
   echo 'Current directory:' $(pwd) && \
@@ -120,7 +159,6 @@ CMD ["sh", "-c", "\
   echo 'User ID:' $(id) && \
   echo '' && \
   echo '=== FILE SYSTEM CHECK ===' && \
-  echo 'Files in app directory:' && \
   ls -la && \
   echo '' && \
   echo '=== PYTHON ENVIRONMENT ===' && \
@@ -135,7 +173,7 @@ CMD ["sh", "-c", "\
       echo 'Testing Python imports...' && \
       python3 -B -c 'import streamlit; print(\"✅ Streamlit:\", streamlit.__version__)' && \
       python3 -B -c 'import torch; print(\"✅ Torch:\", torch.__version__)' && \
-      echo 'Testing app.py import...' && \
+      python3 -B -c 'import cv2; import mediapipe as mp; print(\"✅ cv2 & mediapipe OK\")' && \
       python3 -B -c 'import sys; sys.path.insert(0, \".\"); import app; print(\"✅ app.py imports successfully\")' && \
       echo '✅ All checks passed!'; \
   else \
@@ -145,4 +183,4 @@ CMD ["sh", "-c", "\
   echo '' && \
   echo '=== STARTING STREAMLIT SERVER ===' && \
   streamlit run --server.port=8501 --server.address=0.0.0.0 app.py \
-"]
+"]

VideoBackgroundReplacer2/requirements.txt

@@ -1,11 +1,13 @@
-# Update requirements.txt with correct versions
-cat > requirements.txt << 'EOL'
-# Core Dependencies
+# ===============================
+# Requirements for HF Space (Py3.10 + Torch 2.5.1 cu121)
+# ===============================
+
+# --- Core Dependencies ---
 numpy>=1.24.0,<2.0.0
 Pillow>=10.0.0,<11.0.0
 protobuf>=4.25.0,<5.0.0
 
-# Image/Video Processing
+# --- Image / Video Processing ---
 opencv-python-headless>=4.8.0,<5.0.0
 imageio>=2.25.0,<3.0.0
 imageio-ffmpeg>=0.4.7,<1.0.0
@@ -13,10 +15,10 @@ moviepy>=1.0.3,<2.0.0
 decord>=0.6.0,<0.7.0
 scikit-image>=0.19.3,<0.22.0
 
-# MediaPipe
+# --- MediaPipe ---
 mediapipe>=0.10.0,<0.11.0
 
-# SAM2 Dependencies
+# --- SAM2 Dependencies ---
 hydra-core>=1.3.2,<2.0.0
 omegaconf>=2.3.0,<3.0.0
 einops>=0.6.0,<0.9.0
@@ -25,28 +27,28 @@ pyyaml>=6.0.0,<7.0.0
 matplotlib>=3.5.0,<4.0.0
 iopath>=0.1.10,<0.2.0
 
-# MatAnyone Dependencies
-kornia>=0.7.0,<0.8.0
+# --- MatAnyone Dependencies ---
+kornia>=0.7.2,<0.8.0     # 0.7.2 is stable with Torch 2.5.x
 tqdm>=4.60.0,<5.0.0
 
-# Streamlit UI
+# --- Streamlit UI ---
 streamlit>=1.32.0,<2.0.0
 streamlit-webrtc>=0.50.0,<1.0.0
 
-# Web stack
+# --- Web stack ---
 fastapi>=0.104.0,<1.0.0
 uvicorn>=0.24.0,<1.0.0
 httpx>=0.25.0,<1.0.0
 anyio>=4.0.0,<5.0.0
 orjson>=3.10.0,<4.0.0
 
-# Pydantic (with fixed versions for Python 3.10)
+# --- Pydantic (Py3.10 safe pins) ---
 pydantic>=2.6.0,<2.9.0
 pydantic-core>=2.20.0,<2.22.0
 annotated-types>=0.6.0,<0.8.0
 typing-extensions>=4.0.0,<5.0.0
 
-# Helpers and Utilities
+# --- Helpers / Utilities ---
 huggingface-hub>=0.20.0,<1.0.0
 ffmpeg-python>=0.2.0,<1.0.0
 psutil>=5.8.0,<7.0.0
@@ -54,6 +56,3 @@ requests>=2.25.0,<3.0.0
 scikit-learn>=1.3.0,<2.0.0
 loguru>=0.6.0,<1.0.0
 python-multipart>=0.0.5,<1.0.0
-EOL
-
-echo "requirements.txt has been updated with fixed dependency versions."