FROM python:3.10-slim

ENV PYTHONUNBUFFERED=1

WORKDIR /app

COPY api_server.py /app/

RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*

RUN pip install --no-cache-dir \
    torch torchvision torchaudio \
    --index-url https://download.pytorch.org/whl/cpu

RUN pip install --no-cache-dir \
    diffusers \
    transformers \
    safetensors \
    peft \
    flask \
    pillow \
    accelerate

RUN mkdir -p /app/model /app/generated_images

EXPOSE 7860

CMD ["python3", "-u", "/app/api_server.py"]