dockerfile

ARG TARGETARCH

# The base build, which should run on normal desktop computers
FROM python:3.11 AS build-amd64

WORKDIR /app

COPY pyproject.toml ./

# apt-update and install dependencies, with a cache
RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
	--mount=type=cache,target=/var/lib/apt,sharing=locked \
	apt-get update && apt-get --no-install-recommends install -y python3-opencv gcc

# pip install dependencies, with a cache
RUN --mount=type=cache,target=/root/.cache \
	pip install .

# Run last to not ruin the caches
COPY src/ ./src/
COPY models/best_float32.tflite ./models/

# Extra commands needed for arm64 (we're expecting it to be a Raspberry Pi)
FROM build-amd64 AS build-arm64

# install libcamera
RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
	--mount=type=cache,target=/var/lib/apt,sharing=locked \
	apt-get --no-install-recommends install -y libcamera

RUN --mount=type=cache,target=/root/.cache \
	pip install .[pi]


# Final build, using the detected architecture
FROM build-${TARGETARCH} AS final
EXPOSE 8080
CMD ["python", "-m", "src"]