MatA fixing

Dockerfile

@@ -2,7 +2,6 @@
 # BackgroundFX Pro — Dockerfile (Hardened for Spaces GPU)
 # ===============================
 
-# Match PyTorch cu121 wheels (critical to avoid CUDA probe stalls)
 FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
 
 # --- Build args (optional pins) ---
@@ -18,7 +17,6 @@ RUN useradd -m -u 1000 user
 ENV HOME=/home/user
 ENV PATH=/home/user/.local/bin:$PATH
 
-# Pre-create app dir + caches with correct ownership (avoid permission denied)
 RUN mkdir -p /home/user/app/third_party /data/.cache && \
     chown -R user:user /home/user /data
 WORKDIR /home/user/app
@@ -31,7 +29,7 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y \
     libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender-dev libgomp1 \
  && rm -rf /var/lib/apt/lists/*
 
-# Switch to non-root for python/pip and repo work
+# Switch to non-root
 USER user
 
 # --- Python + Torch (cu121) ---
@@ -42,42 +40,33 @@ RUN python3 -m pip install --no-cache-dir --index-url https://download.pytorch.o
 # --- App deps ---
 COPY --chown=user requirements.txt ./requirements.txt
 RUN python3 -m pip install --no-cache-dir -r requirements.txt
-# Optional nice fallback
 RUN python3 -m pip install --no-cache-dir mediapipe==0.10.14
 
-# --- Third-party repos (build-time, never at runtime) ---
+# --- Third-party repos ---
 # SAM2
 RUN git clone --depth=1 https://github.com/facebookresearch/segment-anything-2.git third_party/sam2 && \
     cd third_party/sam2 && \
     if [ "${SAM2_SHA}" != "__PIN_ME__" ]; then git fetch --depth=1 origin ${SAM2_SHA} && git checkout ${SAM2_SHA}; fi
-
-# Show what we got
-RUN echo "=== DEBUG: SAM2 contents ===" && \
-    ls -la third_party/sam2/ && \
-    echo "=== DEBUG: SAM2 configs ===" && \
-    (ls -la third_party/sam2/configs/sam2/ || echo "configs missing")
-
-# Install SAM2 (editable ok)
 RUN cd third_party/sam2 && python3 -m pip install --no-cache-dir -e .
 
-# --- App code last (so code changes don't invalidate heavy layers) ---
+# --- App code last ---
 COPY --chown=user . /home/user/app
 
-# Verify SAM2 clone not overwritten by COPY
+# Verify SAM2 after COPY
 RUN echo "=== DEBUG: After COPY (SAM2) ===" && \
     ls -la third_party/sam2/ && \
     (ls -la third_party/sam2/configs/sam2/ || echo "SAM2 configs missing")
 
-# 🛠️ MatAnyone AFTER COPY so it persists
+# ✅ MatAnyone AFTER COPY so it persists
 RUN rm -rf third_party/matanyone && \
     git clone --depth=1 https://github.com/pq-yang/MatAnyone.git third_party/matanyone && \
     cd third_party/matanyone && \
     git fetch --tags --depth=1 && \
     git checkout ${MATANYONE_SHA} && \
     if [ -f requirements.txt ]; then python3 -m pip install --no-cache-dir -r requirements.txt; fi
-
-# Double-check MatAnyone exists
-RUN echo "=== DEBUG: MatAnyone contents ===" && ls -la third_party/matanyone/ || true
+# Sanity check: package folder must exist
+RUN echo "=== DEBUG: MatAnyone package ===" && \
+    ls -la third_party/matanyone/matanyone || (echo "MISSING matanyone package dir!" && exit 2)
 
 # --- Runtime environment ---
 ENV PYTHONUNBUFFERED=1 \
@@ -87,7 +76,7 @@ ENV PYTHONUNBUFFERED=1 \
     TORCH_HOME=/data/.cache/torch \
     MPLCONFIGDIR=/data/.cache/matplotlib \
     PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128 \
-    PYTHONPATH="$PYTHONPATH:/home/user/app/third_party/sam2:/home/user/app/third_party/matanyone" \
+    PYTHONPATH="/home/user/app:/home/user/app/third_party:/home/user/app/third_party/sam2:/home/user/app/third_party/matanyone:${PYTHONPATH}" \
     FFMPEG_BIN=ffmpeg \
     THIRD_PARTY_SAM2_DIR=/home/user/app/third_party/sam2 \
     THIRD_PARTY_MATANY_DIR=/home/user/app/third_party/matanyone \
@@ -95,11 +84,6 @@ ENV PYTHONUNBUFFERED=1 \
     MATANY_CHECKPOINT="" \
     ENABLE_MATANY=1
 
-# Do NOT set PORT here; Spaces injects it.
 EXPOSE 7860
-
-# Optional: basic health check to see if the server bound
 HEALTHCHECK --interval=30s --timeout=5s --retries=5 CMD wget -qO- "http://127.0.0.1:${PORT:-7860}/" || exit 1
-
-# Use exec form + unbuffered
 CMD ["python3","-u","app.py"]