FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel

ENV DEBIAN_FRONTEND=noninteractive
WORKDIR /code

# Install uv
COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/

COPY . .

RUN apt update \
    && apt install -y git htop nano libpq-dev build-essential python3-opencv \
    && uv pip install --system hatchling \
    && uv pip install --system debugpy jupyter flake8 pytest parameterized \
    && uv pip install --system --no-build-isolation .

CMD ["jupyter", "notebook", "--ip=0.0.0.0", "--port=8888", "--no-browser", "--allow-root"]
