FROM ubuntu:jammy-20260217

ENV DEBIAN_FRONTEND=noninteractive

# System dependencies
RUN apt-get update && \
    apt-get install -y --no-install-recommends \
        ca-certificates \
        curl \
        python3 \
        python3-dev \
        python3-venv \
        python3-pip \
        build-essential && \
    rm -rf /var/lib/apt/lists/*

# Create a working directory
WORKDIR /workspace

# Install Python dependencies (separate to leverage caching)
COPY inference/requirements.txt /workspace/inference/requirements.txt
RUN python3 -m pip install --upgrade pip
RUN python3 -m pip install -r /workspace/inference/requirements.txt

# Copy the repository
COPY . /workspace

ENV PYTHONUNBUFFERED=1

# Default command: show help message about how to run
CMD ["bash","-lc","echo 'Environment ready. To run inference, use: python3 inference/generate.py --ckpt-path <path> --config <path>' && exec bash"]