Dockerfile.backend

# ===== Builder stage =====
FROM nvidia/cuda:12.3.2-devel-ubuntu22.04 AS builder

ENV CUDA_HOME=/usr/local/cuda
ENV CARGO_HOME=/usr/local/cargo
ENV RUSTUP_HOME=/usr/local/rustup
ENV PATH=/usr/local/cargo/bin:/usr/local/cuda/bin:$PATH
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
ENV CUDA_COMPUTE_CAP=86

RUN apt-get update && apt-get install -y \
    curl \
    protobuf-compiler \
    build-essential \
    pkg-config \
    libssl-dev \
    libasound2-dev \
    ca-certificates \
    && rm -rf /var/lib/apt/lists/*

# Install Rust first (before WORKDIR for better caching)
RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs \
  | sh -s -- -y --no-modify-path --default-toolchain 1.93

WORKDIR /app
COPY . .

# Build
RUN cargo build --release -p server

# ===== Runtime stage =====
FROM nvidia/cuda:12.3.2-runtime-ubuntu22.04

ENV CUDA_HOME=/usr/local/cuda
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
ENV MODEL_NAME=whisper-tiny.en

RUN apt-get update && apt-get install -y \
    ca-certificates \
    libasound2 \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /app

COPY --from=builder /app/target/release/server /usr/local/bin/server

EXPOSE 3000

CMD ["sh", "-c", "server --model ${MODEL_NAME}"]