FROM python:3.8.7-slim as builder
ARG HUGGINGFACE_TOKEN
ENV HUGGINGFACE_TOKEN=${HUGGINGFACE_TOKEN}
COPY docker/requirements.txt /app/

# install
RUN echo "==> Installing ..."  && \
    pip3 install --no-cache-dir --upgrade pip && \
    pip3 install virtualenv && \
    virtualenv -p /usr/local/bin/python /app/env && \
    /app/env/bin/pip install --no-cache-dir --upgrade pip && \
    /app/env/bin/pip install --no-cache-dir -r /app/requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu

COPY src/stable-diffusion-streamlit /app/
RUN cd /app/pages/model/ && \
    /app/env/bin/python prepare.py

FROM python:3.8.7-slim
COPY --from=builder /app /app
WORKDIR /app
CMD ["/app/env/bin/streamlit", "run", "主页.py"]