# Use the official Python 3.10.4 image
FROM python:3.10.4

# Install OS Libs
RUN apt-get update \
    && apt-get -y install ffmpeg

RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH

COPY --chown=user ./backend $HOME/backend

RUN mkdir $HOME/.cache

ENV HF_HUB_CACHE=/home/user/.cache

WORKDIR $HOME/backend

# Install requirements.txt 
RUN pip install --no-cache-dir --upgrade -r ./requirements.txt
RUN pip install git+https://github.com/huggingface/transformers.git
RUN pip install pydantic[email]

# Install spaCy language models
RUN python -m spacy download en_core_web_lg
RUN python -m spacy download es_core_news_sm
RUN python -m spacy download fr_core_news_sm
RUN python -m spacy download pl_core_news_sm
RUN python -m spacy download de_core_news_sm
RUN python -m spacy download it_core_news_sm
RUN python -m spacy download pt_core_news_sm
RUN python -m spacy download nl_core_news_sm
RUN python -m spacy download fi_core_news_sm
RUN python -m spacy download ro_core_news_sm
RUN python -m spacy download ru_core_news_sm

# Make all files and folders readable, writable, executable
# RUN chmod -R 777 ./

# Start the FastAPI app on port 7860, the default port expected by Spaces
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
# CMD ["gunicorn", "-b", "0.0.0.0:7860","main:app","--workers","1","-k","uvicorn.workers.UvicornWorker"]