FROM python:3.10-slim

ENV PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=1 \
    GRADIO_SERVER_NAME=0.0.0.0 \
    HF_HOME=/data/.huggingface \
    HF_HUB_CACHE=/data/.cache/huggingface/hub \
    TRANSFORMERS_CACHE=/data/.cache/huggingface/transformers

RUN apt-get update && apt-get install -y --no-install-recommends \
    ffmpeg \
    git \
    libgl1 \
    libglib2.0-0 \
    libsm6 \
    libxext6 \
    libxrender1 \
    && rm -rf /var/lib/apt/lists/*

RUN python -m pip install --upgrade pip
RUN python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

WORKDIR /opt
RUN git clone --depth 1 https://github.com/Stability-AI/generative-models.git

WORKDIR /opt/generative-models
RUN python -m pip install -r requirements/pt2.txt
RUN python -m pip install git+https://github.com/Stability-AI/datapipelines.git@main#egg=sdata
RUN python -m pip install . gradio huggingface_hub

WORKDIR /app
COPY app.py README.md /app/

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