FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04

ENV DEBIAN_FRONTEND=noninteractive
ENV PYTHONUNBUFFERED=1
ENV PYTHONDONTWRITEBYTECODE=1

# ---------------------------------------------------------------------------
# System dependencies + Python 3.11
# ---------------------------------------------------------------------------
RUN apt-get update && apt-get install -y --no-install-recommends \
    software-properties-common \
    curl \
    git \
    build-essential \
    libgl1-mesa-glx \
    libglib2.0-0 \
    libsm6 \
    libxext6 \
    libxrender-dev \
    ffmpeg \
    && add-apt-repository ppa:deadsnakes/ppa \
    && apt-get update && apt-get install -y --no-install-recommends \
    python3.11 \
    python3.11-venv \
    python3.11-dev \
    python3.11-distutils \
    && rm -rf /var/lib/apt/lists/*

# ---------------------------------------------------------------------------
# Set Python 3.11 as default + pip
# ---------------------------------------------------------------------------
RUN ln -sf /usr/bin/python3.11 /usr/bin/python \
    && ln -sf /usr/bin/python3.11 /usr/bin/python3 \
    && curl -sS https://bootstrap.pypa.io/get-pip.py | python3.11 \
    && python -m pip install --no-cache-dir --upgrade pip setuptools wheel

# ---------------------------------------------------------------------------
# Create non-root user (HF Spaces requirement)
# ---------------------------------------------------------------------------
RUN useradd -m -u 1000 user
ENV HOME=/home/user
ENV PATH=/home/user/.local/bin:$PATH

WORKDIR /app

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

# ---------------------------------------------------------------------------
# Copy app code
# ---------------------------------------------------------------------------
COPY . .

# ---------------------------------------------------------------------------
# Install realesrgan (pulls PyPI basicsr==1.4.2 as dependency)
# ---------------------------------------------------------------------------
RUN pip install --no-cache-dir realesrgan 2>&1 || true

# ---------------------------------------------------------------------------
# FIX: Uninstall broken PyPI basicsr (it imports removed
# torchvision.transforms.functional_tensor). CodeFormer's own local
# basicsr (at /app/CodeFormer/basicsr/) works fine and is used instead
# via PYTHONPATH below.
# ---------------------------------------------------------------------------
RUN pip uninstall -y basicsr 2>&1 || true

# ---------------------------------------------------------------------------
# Use CodeFormer's local basicsr + facelib via PYTHONPATH
# This makes `import basicsr` and `import facelib` resolve to the
# CodeFormer-bundled versions that are compatible with modern torchvision.
# ---------------------------------------------------------------------------
ENV PYTHONPATH="/app/CodeFormer:${PYTHONPATH}"

# ---------------------------------------------------------------------------
# Download pretrained models (now uses local basicsr — no import errors)
# ---------------------------------------------------------------------------
RUN python CodeFormer/scripts/download_pretrained_models.py facelib 2>&1 || true
RUN python CodeFormer/scripts/download_pretrained_models.py CodeFormer 2>&1 || true

# ---------------------------------------------------------------------------
# Directories + permissions
# ---------------------------------------------------------------------------
RUN mkdir -p /app/models /tmp/faceswap \
    && chown -R user:user /app /tmp/faceswap /home/user

# ---------------------------------------------------------------------------
# Switch to non-root
# ---------------------------------------------------------------------------
USER user

EXPOSE 7860

# ---------------------------------------------------------------------------
# PRODUCTION CMD (GPU SAFE)
# ---------------------------------------------------------------------------
CMD ["gunicorn","app:fastapi_app","--workers=1","--worker-class=uvicorn.workers.UvicornWorker","--bind=0.0.0.0:7860","--timeout=600","--graceful-timeout=300","--keep-alive=5","--max-requests=200","--max-requests-jitter=50","--access-logfile=-","--error-logfile=-","--log-level=info"]