FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime

WORKDIR /app

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

COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt && \
    python -c "import nltk; nltk.download('averaged_perceptron_tagger_eng'); nltk.download('cmudict')"

COPY wyoming_glados/ ./wyoming_glados/
COPY download_model.py .
COPY entrypoint.sh .
RUN chmod +x entrypoint.sh

RUN mkdir -p /data

EXPOSE 10200

ENTRYPOINT ["/app/entrypoint.sh"]