Update Dockerfile

Dockerfile

@@ -1,20 +1,15 @@
 # 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)
+# MuseTalk-ready Dockerfile (fixed line continuations and comment placement)
 ##############################################################################
 
 FROM python:3.9-slim
 
-# Repo to clone and branch (override at build time if needed)
+# Build-time args
 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"
+ARG INSTALL_POSE="false"
+
 ENV DEBIAN_FRONTEND=noninteractive
 ENV MODEL_DIR=/app/models
 WORKDIR /app
@@ -22,8 +17,9 @@ WORKDIR /app
 LABEL org.opencontainers.image.source=${REPO}
 
 # ----------------- system deps -----------------
-RUN apt-get update \
- && apt-get install -y --no-install-recommends \
+# Install necessary system packages
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
       ca-certificates \
       git \
       ffmpeg \
@@ -32,12 +28,13 @@ RUN apt-get update \
       libgl1-mesa-glx \
       libglib2.0-0 \
       libjpeg-dev \
-      libpng-dev  && rm -rf /var/lib/apt/lists/*
+      libpng-dev && \
+    rm -rf /var/lib/apt/lists/*
 
 # ----------------- 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
+RUN git clone --depth 1 --branch ${BRANCH} ${REPO} . && \
+    git submodule update --init --recursive || true
 
 # ----------------- overlay local build context (if any) -----------------
 # If you run `docker build` from the repo root, your local files (app.py, configs/, requirements.txt, etc.)
@@ -47,24 +44,24 @@ COPY . .
 # ----------------- 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)
+# Install requirements from requirements.txt (if present)
 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
+# Install a few extra packages and pin protobuf to avoid common 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
+      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 -----------------
+# Set --build-arg INSTALL_POSE=true to enable these (image size will grow)
 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; \
@@ -77,15 +74,14 @@ RUN if [ "${INSTALL_POSE}" = "true" ]; then \
     fi
 
 # ----------------- housekeeping -----------------
-RUN find /root/.cache -type f -delete || true \
- && rm -rf /root/.cache/pip || true
+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)
+# Ensure model dir exists for mounting at runtime
 RUN mkdir -p ${MODEL_DIR}
 
-# Expose typical ports (adjust if your app uses different ports)
+# Expose ports commonly used by the project (adjust if needed)
 EXPOSE 7860 8000
 
-# ----------------- runtime command -----------------
-# Expect that /app/app.py exists in the repo (cloned or from local). If it's in a subfolder, update CMD.
+# Default command: run app.py (expects /app/app.py to exist, local copy overrides cloned if provided)
 CMD ["python3", "app.py"]