FROM hyzhou404/hugsim_server:0.1

ENV DEBIAN_FRONTEND=noninteractive \
    TZ=UTC \
    HF_HUB_ENABLE_HF_TRANSFER=1

ENV PATH="${HOME}/miniconda3/bin:${PATH}"
ARG PATH="${HOME}/miniconda3/bin:${PATH}"

WORKDIR /app

USER 1000

ENV PATH /app/miniconda/bin:$PATH

SHELL ["conda", "run","--no-capture-output", "-p","/app/env", "/bin/bash", "-c"]

COPY --chown=1000:1000 ./code /app/code
COPY --chown=1000:1000 ./web_server.py /app/code/web_server.py
COPY --chown=1000:1000 ./download_public_datas.py /app/code/download_public_datas.py
COPY --chown=1000:1000 ./download_private_datas.py /app/code/download_private_datas.py
COPY --chown=1000:1000 ./launch.sh /app/launch.sh
RUN chmod +x /app/launch.sh

ENV TCNN_CUDA_ARCHITECTURES 75

ENV TORCH_CUDA_ARCH_LIST "7.5"

ENV http_proxy=
ENV https_proxy=

RUN ./.pixi/envs/default/bin/python3 -m ensurepip

RUN ./.pixi/envs/default/bin/python3 -m pip install psutil

RUN ./.pixi/envs/default/bin/python3 -m pip install moviepy

RUN ./.pixi/envs/default/bin/python /app/code/download_public_datas.py

CMD ["/app/launch.sh"]