File size: 2,115 Bytes
cd5198d 1fc3e51 cd5198d 538e429 cd5198d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 | # ==========================================
# Stage 1: Build the React + Vite Frontend
# ==========================================
FROM node:20-slim AS frontend-builder
WORKDIR /app/frontend
# Copy frontend source files
COPY frontend/package*.json ./
RUN npm ci
COPY frontend/ ./
RUN npm run build
# ==========================================
# Stage 2: Build the FastAPI + PyTorch Backend
# ==========================================
FROM python:3.10-slim
# Install system dependencies (libsndfile1 is required by soundfile, espeak for pyttsx3 Linux speech)
RUN apt-get update && apt-get install -y --no-install-recommends \
libsndfile1 \
espeak \
ffmpeg \
git \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /app
# Create a non-root user with UID 1000 (Hugging Face Spaces requirement)
RUN useradd -m -u 1000 user
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH \
PYTHONUNBUFFERED=1
# Pre-create writable directories for audio caches and temp audio files
RUN mkdir -p /app/static/temp && \
mkdir -p /home/user/.cache/huggingface && \
chown -R user:user /app /home/user
# Switch to the non-root user
USER user
# Install CPU-only PyTorch first to keep the container extremely lightweight (~1.5GB instead of 5GB)
RUN pip3 install --no-cache-dir --user torch torchaudio --index-url https://download.pytorch.org/whl/cpu
# Copy backend requirements and install dependencies
COPY --chown=user:user requirements.txt .
RUN pip3 install --no-cache-dir --user -r requirements.txt
# Install VibeVoice package with --no-deps to prevent compiling unused/heavy packages (like aiortc and av)
RUN pip3 install --no-cache-dir --user --no-deps git+https://github.com/shijincai/VibeVoice.git
# Copy FastAPI backend code and assets
COPY --chown=user:user app.py .
COPY --chown=user:user static/ static/
# Copy the built React static bundles from Stage 1
COPY --chown=user:user --from=frontend-builder /app/frontend/dist ./frontend/dist
# Expose Hugging Face Space port
EXPOSE 7860
ENV PORT=7860
# Start the FastAPI server dynamically binding to port 7860
CMD ["python3", "app.py"]
|