Dockerfile

FROM python:3.10-slim

ENV DEBIAN_FRONTEND=noninteractive \
    PYTHONDONTWRITEBYTECODE=1 \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=1 \
    HF_HOME=/app/cache/huggingface \
    PYTHONIOENCODING=UTF-8 \
    PORT=7860 \
    MODEL_REPO="yujieq/MolScribe" \
    MODEL_FILE="swin_base_char_aux_1m.pth" \
    DEVICE="cpu" \
    AUTO_TRIM_WHITE="1" \
    WHITE_THRESHOLD="245" \
    WHITE_PADDING="16" \
    MAX_IMAGE_EDGE="1280" \
    MIN_IMAGE_EDGE="0" \
    REQUEST_TIMEOUT_SECONDS="180"

RUN apt-get update && apt-get install -y --no-install-recommends \
    curl ca-certificates \
    libgl1 libglib2.0-0 libsm6 libxext6 libxrender1 libgomp1 \
    tini \
 && rm -rf /var/lib/apt/lists/*

WORKDIR /app

RUN useradd -m -u 1000 user

COPY requirements.txt /app/requirements.txt
RUN pip install --upgrade pip && \
    pip install -r requirements.txt

COPY --chown=user:user . /app
RUN mkdir -p /app/cache/huggingface && chown -R user:user /app/cache

RUN python -c "from hf_loader import warmup; print(warmup())"

USER user
EXPOSE $PORT
ENTRYPOINT ["/usr/bin/tini", "--"]
CMD ["sh", "-c", "uvicorn app:app --host 0.0.0.0 --port ${PORT:-7860}"]