now works locally in docker

Dockerfile

@@ -3,26 +3,21 @@ FROM python:3.10-slim
 ENV PYTHONUNBUFFERED=1 \
     PIP_NO_CACHE_DIR=1
 
-# Install only the required system packages
+# System deps
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    ffmpeg \
-    libsm6 \
-    libxext6 \
-    libgl1 \
-    git \
-    git-lfs \
+    ffmpeg libsm6 libxext6 libgl1 git git-lfs \
     && rm -rf /var/lib/apt/lists/* \
     && git lfs install
 
 WORKDIR /app
 
-# Install Python deps first (better caching)
+# ---------- Cache heavy PIP layer ----------
 COPY requirements.txt .
-RUN pip install --upgrade pip \
-    && pip install -r requirements.txt \
-    && pip install "gradio[oauth,mcp]==6.0.1" uvicorn spaces
+RUN pip install --upgrade pip && \
+    pip install -r requirements.txt && \
+    pip install "gradio[oauth,mcp]==6.0.1" uvicorn spaces
 
-# Copy the rest of the app
+# ---------- Only copy app code ----------
 COPY . .
 
 EXPOSE 7860

app.py

@@ -1,3 +1,4 @@
+import os
 import json
 from pathlib import Path
 from datetime import datetime
@@ -449,24 +450,20 @@ with gr.Blocks() as ui:
             outputs=file_status,
         )
 
-
 # ----------------------------------------------------
 # Export ASGI app for Hugging Face Spaces
 # ----------------------------------------------------
-
-app = gr.mount_gradio_app(fastapi_app, ui, path="/")
-
-# ----------------------------------------------------
-# Export ASGI app for Hugging Face Spaces
+# Mount Gradio UI at /ui so FastAPI "/" stays alive
 # ----------------------------------------------------
 
-app = gr.mount_gradio_app(fastapi_app, ui, path="/")
+app = gr.mount_gradio_app(fastapi_app, ui, path="/ui")
+
 
 # ----------------------------------------------------
 # Local & Docker run mode
 # ----------------------------------------------------
-# HF Spaces sets SPACE_ID — we MUST NOT run uvicorn there,
-# but in local/docker we MUST run uvicorn so the process does not exit.
+# HF Spaces sets SPACE_ID — DO NOT run uvicorn in HF.
+# For local / Docker usage, run uvicorn manually.
 # ----------------------------------------------------
 
 if __name__ == "__main__" and not os.getenv("SPACE_ID"):
@@ -474,4 +471,3 @@ if __name__ == "__main__" and not os.getenv("SPACE_ID"):
     port = int(os.getenv("PORT", "7860"))
     print(f"Starting local uvicorn server on port {port}...")
     uvicorn.run(app, host="0.0.0.0", port=port)
-

build.sh

@@ -3,18 +3,23 @@ set -e
 
 IMAGE_NAME="avatar-mcp"
 
-echo "🧹 Removing old containers..."
-docker ps -a --filter "ancestor=$IMAGE_NAME" --format "{{.ID}}" | xargs -r docker rm -f
+if [[ "$1" == "clean" ]]; then
+    echo "🧹 CLEAN build requested – removing image & containers..."
+    docker ps -a --filter "ancestor=$IMAGE_NAME" --format "{{.ID}}" | xargs -r docker rm -f
+    docker images --filter "reference=$IMAGE_NAME" --format "{{.ID}}" | xargs -r docker rmi -f
+else
+    echo "🛑 Removing old containers only..."
+    docker ps -a --filter "ancestor=$IMAGE_NAME" --format "{{.ID}}" | xargs -r docker rm -f
+fi
 
-echo "🧹 Removing old image..."
-docker images --filter "reference=$IMAGE_NAME" --format "{{.ID}}" | xargs -r docker rmi -f
-
-echo "📦 Building new image..."
-docker build -t $IMAGE_NAME .
+echo "📦 Building image with cache…"
+docker build \
+    --cache-from=$IMAGE_NAME \
+    -t $IMAGE_NAME .
 
 echo "📏 Image size:"
 docker images | grep $IMAGE_NAME
 
 # Uncomment to run locally:
 # echo "🚀 Starting container on http://localhost:7860 ..."
-# docker run -p 7860:7860 avatar-mcp
+# docker run -p 7860:7860 $IMAGE_NAME