FROM python:3.11-slim

WORKDIR /app

RUN apt-get update && apt-get install -y \
    gcc g++ curl \
    && rm -rf /var/lib/apt/lists/*

COPY requirements.txt .
RUN pip install --no-cache-dir --upgrade pip && \
    pip install --no-cache-dir -r requirements.txt

RUN pip install https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.7.1/en_core_web_sm-3.7.1-py3-none-any.whl

COPY src/ ./src/

RUN mkdir -p models data

ARG HF_TOKEN
ENV HUGGING_FACE_HUB_TOKEN=$HF_TOKEN

RUN python -c "\
from huggingface_hub import snapshot_download; \
snapshot_download(repo_id='praveends/doculens-ner', local_dir='models/ner_model'); \
snapshot_download(repo_id='praveends/doculens-classifier', local_dir='models/classifier_model'); \
print('Models downloaded!')"

EXPOSE 7860

CMD ["uvicorn", "src.api:app", "--host", "0.0.0.0", "--port", "7860"]