FROM python:3.11-slim

ENV DEBIAN_FRONTEND=noninteractive \
    PYTHONUNBUFFERED=1 \
    PORT=7860

RUN apt-get update && apt-get install -y --no-install-recommends \
    curl \
    ffmpeg \
    libgl1 \
    libglib2.0-0 \
    libsm6 \
    libxext6 \
    && rm -rf /var/lib/apt/lists/*

# Create non-root user (UID 1000 required for Hugging Face Spaces)
RUN useradd -m -u 1000 user
WORKDIR /app

RUN pip install --no-cache-dir --upgrade pip && \
    pip install --no-cache-dir \
    fastapi \
    "uvicorn[standard]" \
    websockets \
    pydantic \
    numpy \
    opencv-python-headless \
    Pillow \
    "onnxruntime-gpu==1.21.0" \
    nvidia-cuda-runtime-cu12 \
    "nvidia-cudnn-cu12>=9.0,<10.0" \
    nvidia-cublas-cu12 \
    nvidia-cufft-cu12 \
    nvidia-curand-cu12 \
    nvidia-nvjitlink-cu12 \
    huggingface_hub \
    pytest \
    httpx && \
    find /usr/local/lib/python3.11/site-packages/nvidia -type d -name "lib" > /etc/ld.so.conf.d/nvidia-pip.conf && \
    ldconfig

ENV LD_LIBRARY_PATH="/usr/local/lib/python3.11/site-packages/nvidia/cudnn/lib:/usr/local/lib/python3.11/site-packages/nvidia/cublas/lib:/usr/local/lib/python3.11/site-packages/nvidia/cuda_runtime/lib:/usr/local/lib/python3.11/site-packages/nvidia/cufft/lib:/usr/local/lib/python3.11/site-packages/nvidia/curand/lib:${LD_LIBRARY_PATH}"

COPY --chown=user:user . /app

RUN chmod +x /app/download_models.sh && \
    mkdir -p /app/storage/snapshots /app/storage/recordings /app/models && \
    chown -R user:user /app

USER user
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH

RUN ./download_models.sh

EXPOSE 7860

CMD ["python3", "main.py"]