DockerFile

FROM python:3.11-slim AS base

ENV PYTHONDONTWRITEBYTECODE=1 \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=1 \
    PIP_DISABLE_PIP_VERSION_CHECK=1

WORKDIR /app

RUN apt-get update && apt-get install -y --no-install-recommends \
    gcc \
    && rm -rf /var/lib/apt/lists/*


FROM base AS builder

COPY requirements.txt .
RUN pip install --prefix=/install -r requirements.txt


FROM base AS runtime

COPY --from=builder /install /usr/local

RUN addgroup --system phishguard && \
    adduser  --system --ingroup phishguard --no-create-home phishguard

COPY --chown=phishguard:phishguard . .

RUN python scripts/train_model.py \
    --dataset dataset.csv \
    --output  phishing_model.joblib \
    --log-level INFO

USER phishguard

EXPOSE 5000

HEALTHCHECK --interval=30s --timeout=10s --start-period=15s --retries=3 \
    CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:5000/health')"

CMD ["gunicorn", \
     "--bind",        "0.0.0.0:5000", \
     "--workers",     "2", \
     "--threads",     "4", \
     "--timeout",     "120", \
     "--access-logfile", "-", \
     "--error-logfile",  "-", \
     "app:create_app()"]