FROM python:3.10-slim

WORKDIR /app

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

RUN pip install --no-cache-dir "typing-extensions==4.12.2" "numpy<2"

RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu

RUN pip install --no-cache-dir realesrgan==0.3.0 fastapi "uvicorn[standard]" python-multipart opencv-python-headless

RUN BASICSR_DEG=$(python -c "import basicsr; import os; print(os.path.join(os.path.dirname(basicsr.__file__), 'data', 'degradations.py'))" 2>/dev/null) || \
    BASICSR_DEG=$(find /usr/local/lib/python3.10 -path "*/basicsr/data/degradations.py" 2>/dev/null | head -1) && \
    if [ -n "$BASICSR_DEG" ]; then \
      sed -i 's/from torchvision.transforms.functional_tensor/from torchvision.transforms.functional/g' "$BASICSR_DEG" && \
      echo "basicsr patch applied to $BASICSR_DEG"; \
    else \
      echo "basicsr degradations.py not found, skipping patch"; \
    fi

COPY app.py .

ENV MODEL_PATH=/models/RealESRGAN_x2plus.pth

VOLUME /models

CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8080"]
