Dockerfile.gpu

# Multi-stage build: Stage 1 - Build React frontend
FROM node:20-slim AS frontend-builder

WORKDIR /app

# Copy everything and build in one layer to minimize size
COPY chainforge/react-server/ ./
RUN npm ci --legacy-peer-deps --prefer-offline --production=false && \
    npm run build && \
    rm -rf node_modules && \
    npm cache clean --force

# Stage 2 - Build Python dependencies (GPU version with CUDA support)
FROM python:3.12-slim AS python-builder

WORKDIR /build

# Copy all requirements and source files
COPY chainforge/requirements.txt chainforge/constraints.txt setup.py README.md ./
COPY chainforge/ ./chainforge/

# Do everything in one layer to minimize build context and layers
RUN apt-get --allow-releaseinfo-change update && \
    apt-get install -y --no-install-recommends build-essential git && \
    pip install --no-cache-dir --upgrade pip setuptools wheel && \
    pip install --no-cache-dir --prefix=/install \
        --extra-index-url https://download.pytorch.org/whl/cu121 \
        torch torchvision torchaudio && \
    pip install --no-cache-dir --prefix=/install \
        -r requirements.txt \
        -c constraints.txt && \
    pip install --no-cache-dir --prefix=/install . && \
    apt-get purge -y --auto-remove build-essential && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/* /root/.cache

# Stage 3 - Final minimal runtime image
FROM python:3.12-slim

WORKDIR /chainforge

# Copy Python packages and React build from builder stages
COPY --from=python-builder /install /usr/local
COPY --from=frontend-builder /app/build /usr/local/lib/python3.12/site-packages/chainforge/react-server/build

# Install runtime deps, clean up, and create user in one layer
RUN apt-get --allow-releaseinfo-change update && \
    apt-get install -y --no-install-recommends git libgomp1 && \
    rm -rf /var/lib/apt/lists/* && \
    apt-get clean && \
    find /usr/local -type d -name "tests" -exec rm -rf {} + 2>/dev/null || true && \
    find /usr/local -type d -name "__pycache__" -exec rm -rf {} + 2>/dev/null || true && \
    find /usr/local -type d -name "*.dist-info" -exec rm -rf {}/RECORD {} + 2>/dev/null || true && \
    find /usr/local -name "*.pyc" -delete && \
    find /usr/local -name "*.pyo" -delete && \
    find /usr/local -name "*.a" -delete && \
    find /usr/local -name "*.md" -delete 2>/dev/null || true && \
    find /usr/local -name "*.txt" -delete 2>/dev/null || true && \
    rm -rf /usr/local/lib/python3.12/site-packages/*/tests && \
    rm -rf /usr/local/lib/python3.12/site-packages/*/test && \
    rm -rf /tmp/* /var/tmp/* && \
    useradd -m -u 1000 chainforge && \
    mkdir -p /home/chainforge/.local/share/chainforge && \
    chown -R chainforge:chainforge /chainforge /home/chainforge

USER chainforge

EXPOSE 8000

ENTRYPOINT [ "chainforge", "serve", "--host", "0.0.0.0" ]