Dockerfile

FROM python:3.13-slim

# Set environment variables
ENV PYTHONDONTWRITEBYTECODE 1
ENV PYTHONUNBUFFERED 1
ENV BASE_DIR /app
ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility

# Install system dependencies (including libraries needed for PyTorch and curl for health checks)
RUN apt-get update && apt-get install -y --no-install-recommends \
    netcat-traditional \
    curl \
    gcc \
    g++ \
    build-essential \
    libc6-dev \
    libgl1-mesa-glx \
    libglib2.0-0 \
    libsm6 \
    libxext6 \
    libxrender-dev \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

# Set work directory
WORKDIR /app

# Install Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Install PyTorch with CPU-only support for reliability
RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu

# Copy project
COPY . .

# Add explicit permissions and fix module access
RUN mkdir -p /app/data/models
RUN touch /app/data/error.txt && chmod 666 /app/data/error.txt

# Make entrypoint scripts executable
RUN chmod +x /app/entrypoint.sh
RUN chmod +x /app/entrypoint.prod.sh
RUN chmod +x /app/docker-entrypoint-wrapper.sh

# Create necessary directories
RUN mkdir -p /app/media/profile_pics
RUN mkdir -p /app/media/xrays
RUN mkdir -p /app/data/models
COPY ./data/ /app/data/
RUN mkdir -p /app/staticfiles
RUN mkdir -p /app/prometheus

# Set permissions
RUN chmod -R 755 /app/media
RUN chmod -R 755 /app/staticfiles
RUN chmod -R 755 /app/data
RUN chmod -R 755 /app/prometheus

# Use the wrapper as the entrypoint
ENTRYPOINT ["/app/docker-entrypoint-wrapper.sh"]