FROM python:3.12-slim

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

WORKDIR /app

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

COPY pyproject.toml README.md /app/
COPY plapre /app/plapre

RUN pip install --upgrade pip setuptools wheel && \
    pip install ".[serve]"

ENV PLAPRE_CHECKPOINT=syvai/plapre-pico \
    PLAPRE_GPU_MEM=0.5 \
    PLAPRE_MAX_MODEL_LEN=512 \
    PLAPRE_ASYNC=1

EXPOSE 7860

CMD ["plapre-serve", "--host", "0.0.0.0", "--port", "7860"]