Dockerfile.edge

# Multi-Stage NVIDIA L4T TensorRT Runtime for Jetson AGX Orin
FROM nvcr.io/nvidia/l4t-tensorrt:r10.0.0-runtime AS base

ENV DEBIAN_FRONTEND=noninteractive
ENV PYTHONUNBUFFERED=1

RUN apt-get update && apt-get install -y --no-install-recommends \
    python3-pip \
    python3-dev \
    libgl1 \
    libglib2.0-0 \
    libgstreamer1.0-0 \
    gstreamer1.0-plugins-base \
    gstreamer1.0-plugins-good \
    gstreamer1.0-plugins-bad \
    libgeos-dev \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /opt/sentinelzone

COPY pyproject.toml .
RUN pip3 install --no-cache-dir --upgrade pip && \
    pip3 install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cu121 && \
    pip3 install --no-cache-dir torch_geometric && \
    pip3 install --no-cache-dir opencv-python pydantic paho-mqtt shapely fastapi uvicorn pandas

COPY src/ /opt/sentinelzone/src/
COPY config/ /opt/sentinelzone/config/

ENTRYPOINT ["python3", "-m", "src.edge.edge_runtime"]