FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04

ENV DEBIAN_FRONTEND=noninteractive \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=1 \
    HF_HOME=/data/.huggingface

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

WORKDIR /app

RUN git clone --depth 1 https://github.com/Creative-Intelligence-Studio/GeoNeXt.git /app/GeoNeXt

RUN python3 -m pip install --upgrade pip \
    && python3 -m pip install torch==2.3.1 torchvision==0.18.1 \
       --index-url https://download.pytorch.org/whl/cu121 \
    && python3 -m pip install -r /app/GeoNeXt/requirements-wan.txt \
    && python3 -m pip install \
       gradio==4.44.1 pydantic==2.10.6 fastapi==0.115.6 \
    && python3 -m pip install -e /app/GeoNeXt

COPY app.py /app/app.py

EXPOSE 7860

CMD ["python3", "/app/app.py"]