FROM python:3.11-slim

ENV DEBIAN_FRONTEND=noninteractive \
    PIP_NO_CACHE_DIR=1 \
    PYTHONUNBUFFERED=1 \
    PORT=7860 \
    GRADIO_SERVER_NAME=0.0.0.0 \
    SATA_DEVICE=cpu \
    SATA_GEN_DEVICE=cpu

RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential \
    git \
    curl \
    ffmpeg \
    libglib2.0-0 \
    libgl1 \
    libgomp1 \
    libsm6 \
    libx11-6 \
    libxext6 \
    libxrender1 \
    libegl1 \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /app
COPY . /app

RUN python -m pip install --upgrade pip setuptools wheel
RUN python -m pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu121
RUN python -m pip install torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.5.1+cu121.html
RUN python -m pip install torch_geometric==2.7.0
RUN python -m pip install -r requirements-hf.txt
RUN python -m pip install https://github.com/openai/CLIP/archive/refs/heads/main.zip
RUN python -m pip install --no-build-isolation chumpy==0.70 && python scripts/patch_chumpy.py
RUN python -m pip install -e .
RUN mkdir -p /app/gradio/output /app/gradio/.gradio_tmp

EXPOSE 7860
CMD ["python", "app.py"]