FROM python:3.11

RUN apt-get update -y && \
    apt-get install -y libegl1-mesa-dev libgl1-mesa-dev libxi6 libxkbcommon-x11-0 git-lfs && \
    apt-get clean && \
    apt-get autoremove

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

WORKDIR $HOME
COPY --chown=user . .

RUN --mount=type=secret,id=GH_TOKEN,mode=0444,required=true \
    git config --global url.https://$(cat /run/secrets/GH_TOKEN)@github.com/.insteadOf https://github.com/ && \
    git clone --recursive https://github.com/jasongzy/3DShape2VecSet app
RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true \
    git config --global url.https://jasongzy:$(cat /run/secrets/HF_TOKEN)@huggingface.co/.insteadOf https://huggingface.co/ && \
    git lfs install && \
    git clone https://huggingface.co/jasongzy/Make-It-Animatable model && \
    GIT_LFS_SKIP_SMUDGE=1 git clone --recursive https://huggingface.co/datasets/jasongzy/Mixamo && \
    cd Mixamo && \
    git lfs pull -I animation && \
    git lfs pull -I bones.fbx && \
    git lfs pull -I bones_vroid.fbx

WORKDIR $HOME/app

RUN mv ../model/data . && \
    mv ../model/output . && \
    rm -r ../model && \
    python -m pip install --upgrade pip && \
    pip install -r requirements-demo.txt

CMD ["python", "app.py"]