FROM python:3.11-slim

WORKDIR /app

RUN apt-get update && apt-get install -y \
    libglib2.0-0 \
    libgomp1 \
    && rm -rf /var/lib/apt/lists/*

COPY requirements.txt .


RUN pip install --no-cache-dir \
    flask==3.0.0 \
    pillow==10.0.0 \
    numpy==1.26.0 \
    gunicorn==21.2.0


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


RUN pip install --no-cache-dir \
    tensorflow==2.17.0 \
    keras==3.4.1

COPY . .

EXPOSE 7860

CMD ["gunicorn", "app:app", \
     "--bind", "0.0.0.0:7860", \
     "--workers", "1", \
     "--timeout", "120", \
     "--preload"]