ARG BASE_IMAGE=docker.m.daocloud.io/vllm/vllm-openai:v0.27.1-aarch64
FROM ${BASE_IMAGE}

WORKDIR /app
COPY vendor/diffusers-0.40.0-py3-none-any.whl /tmp/diffusers-0.40.0-py3-none-any.whl
RUN python3 -m pip install --no-cache-dir --no-deps /tmp/diffusers-0.40.0-py3-none-any.whl \
    && rm -f /tmp/diffusers-0.40.0-py3-none-any.whl \
    && python3 -c "from diffusers import ZImagePipeline; import fastapi, uvicorn, torch; print('ZImagePipeline OK', torch.__version__)"
COPY app.py /app/app.py

ENV PYTHONUNBUFFERED=1
ENTRYPOINT []
CMD ["python3", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
