FROM python:3.10.0

WORKDIR /usr/src/
RUN git clone https://github.com/szymonrucinski/good-mood/
WORKDIR /usr/src/good-mood
RUN git checkout master
RUN ls /usr/src/good-mood
RUN  pip install --upgrade pip

RUN mkdir -m 777 /tmp/NUMBA_CACHE_DIR /tmp/MPLCONFIGDIR
ENV NUMBA_CACHE_DIR=/tmp/NUMBA_CACHE_DIR/
ENV MPLCONFIGDIR=/tmp/MPLCONFIGDIR/

RUN pip install -r /usr/src/good-mood/requirements.txt
RUN apt update
RUN apt-get install -y libsndfile1-dev
RUN pip install pysndfile
RUN pip install moviepy
RUN mkdir -m 777 flagged
RUN apt-get install git-lfs
RUN apt-get install ffmpeg -y
RUN apt-get update

RUN git clone https://huggingface.co/szymonrucinski/good-mood/
RUN apt-get update
RUN mv ./good-mood/alexnet-good-mood.pt ./model.pt

EXPOSE 7860
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
# CMD ["gradio","app.py"]