FROM python:3.12-bullseye

RUN python3.12 -m venv /python_env/docker_small_fish
RUN apt-get update && \
    apt-get install -y --no-install-recommends \
    python3-tk \
    tk \
    libtk8.6 \
    libgl1-mesa-glx \
    libx11-xcb1 \
    libxcb-glx0 \
    libxrender1 \
    libxcb-shm0 \
    libxcb-icccm4 \
    libxcb-image0 \
    libxcb-keysyms1 \
    libxcb-randr0 \
    libxcb-shape0 \
    libxcb-xfixes0 \
    libxcb-xinerama0 \
    libxcb-dri3-0 \
    && rm -rf /var/lib/apt/lists/*

ENV PATH="/python_env/docker_small_fish/bin:$PATH"

RUN pip install --no-cache-dir small_fish_gui
RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu124 torch torchvision torchaudio

CMD ["python", "-m", "small_fish_gui"]
