Update Dockerfile

Dockerfile

@@ -1,52 +1,92 @@
+# syntax=docker/dockerfile:1
+##############################################################################
+# MuseTalk-ready Dockerfile
+# - clones upstream repo (default)
+# - overlays local build context if present (so local app.py / configs / requirements.txt override cloned ones)
+# - installs requirements (from local if provided, otherwise from cloned repo)
+# - optionally installs heavy pose/mm deps via build-arg INSTALL_POSE=true
+# - does NOT download large model weights during build (mount at runtime)
+##############################################################################
+
 FROM python:3.9-slim
 
-# Set working directory early
+# Repo to clone and branch (override at build time if needed)
+ARG REPO="https://github.com/TMElyralab/MuseTalk.git"
+ARG BRANCH="main"
+ARG INSTALL_POSE="false"         # set to "true" to install mmpose/mm packages (image grows)
+ARG PYTHONUNBUFFERED="1"
+ENV DEBIAN_FRONTEND=noninteractive
+ENV MODEL_DIR=/app/models
 WORKDIR /app
 
-# Install system dependencies with cleanup
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    ffmpeg \
-    git \
-    libgl1-mesa-glx \
-    libglib2.0-0 \
-    build-essential \
-    python3-dev \
-    libjpeg-dev \
-    libpng-dev \
+LABEL org.opencontainers.image.source=${REPO}
+
+# ----------------- system deps -----------------
+RUN apt-get update \
+ && apt-get install -y --no-install-recommends \
+      ca-certificates \
+      git \
+      ffmpeg \
+      build-essential \
+      python3-dev \
+      libgl1-mesa-glx \
+      libglib2.0-0 \
+      libjpeg-dev \
+      libpng-dev \
  && rm -rf /var/lib/apt/lists/*
 
-# Copy only what’s needed early for caching
-COPY requirements.txt .
-COPY scripts ./scripts
-COPY configs ./configs
+# ----------------- clone upstream repo into image -----------------
+# Shallow clone to keep image smaller (override REPO/BRANCH via --build-arg)
+RUN git clone --depth 1 --branch ${BRANCH} ${REPO} . \
+ && git submodule update --init --recursive || true
 
-# Upgrade pip + install Python deps
-RUN pip install --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt
-# Install LiveKit SDKs
-RUN pip install --no-cache-dir \
-    livekit==1.0.7 \
-    livekit-api==1.0.2 \
-    omegaconf \
-    transformers==4.39.3 \
- && pip uninstall -y protobuf && pip install --no-cache-dir protobuf==3.20.3
-
-# Install pose dependencies (with caching minimized)
-RUN pip install --no-cache-dir cython && \
-    pip install --no-cache-dir git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI
-
-RUN pip install --no-cache-dir mmengine==0.10.7 mmcv==2.0.0rc4 && \
-    pip install --no-cache-dir openmim && \
-    mim install mmpose && \
-    mim install mmdet
-
-# Copy rest of the code
+# ----------------- overlay local build context (if any) -----------------
+# If you run `docker build` from the repo root, your local files (app.py, configs/, requirements.txt, etc.)
+# will overwrite the cloned files. If you build from an empty folder, this COPY is effectively a no-op.
 COPY . .
 
-# Final cleanup (in case anything big remains)
-RUN apt-get clean && \
-    find /root/.cache -type f -delete && \
-    rm -rf /root/.cache/pip
+# ----------------- python tooling & install requirements -----------------
+RUN pip install --upgrade pip setuptools wheel
+
+# Install requirements if the file exists (it will come from local build context if present, otherwise from cloned repo)
+RUN if [ -f requirements.txt ]; then \
+      pip install --no-cache-dir -r requirements.txt ; \
+    else \
+      echo "No requirements.txt found; skipping pip install -r requirements.txt"; \
+    fi
+
+# Install common extra packages used by the project and pin protobuf to avoid compatibility issues
+RUN pip install --no-cache-dir \
+      livekit==1.0.7 \
+      livekit-api==1.0.2 \
+      omegaconf \
+      transformers==4.39.3 \
+  || true \
+ && pip uninstall -y protobuf || true \
+ && pip install --no-cache-dir protobuf==3.20.3 || true
+
+# ----------------- optional heavy pose/mm packages -----------------
+RUN if [ "${INSTALL_POSE}" = "true" ]; then \
+      pip install --no-cache-dir cython || true; \
+      pip install --no-cache-dir git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI || true; \
+      pip install --no-cache-dir openmim || true; \
+      pip install --no-cache-dir mmengine==0.10.7 mmcv==2.0.0rc4 || true; \
+      mim install mmpose || true; \
+      mim install mmdet || true; \
+    else \
+      echo "Skipping pose/mm installation (set --build-arg INSTALL_POSE=true to enable)"; \
+    fi
+
+# ----------------- housekeeping -----------------
+RUN find /root/.cache -type f -delete || true \
+ && rm -rf /root/.cache/pip || true
+
+# Make sure model dir exists (mount your weights here at runtime)
+RUN mkdir -p ${MODEL_DIR}
+
+# Expose typical ports (adjust if your app uses different ports)
+EXPOSE 7860 8000
 
-# Set entrypoint
-# CMD ["python3", "-m", "scripts.realtime_inference", "--version", "v15", "--inference_config", "configs/inference/realtime.yaml"]
+# ----------------- runtime command -----------------
+# Expect that /app/app.py exists in the repo (cloned or from local). If it's in a subfolder, update CMD.
+CMD ["python3", "app.py"]