FROM python:3.11-slim


RUN useradd -m -u 1000 user


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


RUN mkdir -p /home/user/.cache/huggingface \
 && chown -R user:user /home/user/.cache


USER user
ENV HOME=/home/user
WORKDIR $HOME/app


ENV HF_HOME=$HOME/.cache/huggingface
ENV TRANSFORMERS_CACHE=$HF_HOME


RUN python3 -c "from huggingface_hub import snapshot_download; import os; snapshot_download('MBZUAI/LaMini-Flan-T5-77M', cache_dir=os.environ.get('HF_HOME'))"


COPY --chown=user . $HOME/app


EXPOSE 7860
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]