# Documentation: https://huggingface.co/docs/hub/spaces-sdks-docker

FROM python:3.9



RUN useradd -m -u 1000 user


COPY --chown=user ./requirements.txt /requirements.txt
COPY --chown=user ./train.sh /train.sh
RUN chmod +x /train.sh

RUN mkdir /app

RUN pip install --no-cache-dir --upgrade -r /requirements.txt
RUN git clone https://github.com/lee-ny/teaching_arithmetic.git 
#/teaching_arithmetic 
#&& cd teaching_arithmetic && pip install -e .

COPY --chown=user . /app
#COPY --chown=user /teaching_arithmetic /app

#&& cd teaching_arithmetic && pip install -e .
#COPY --chown=user /teaching_arithmetic /app/teaching_arithmetic

#RUN chmod +x /teaching_arithmetic

#COPY --chown=user . /app


USER user

WORKDIR /app

ENV PATH="/home/user/.local/bin:/opt/conda/bin:$PATH"
ENV HOME="/home/user"

#WORKDIR $HOME/app


#RUN pip install --no-cache-dir --upgrade pip

ARG PYTORCH_VERSION=2.1.0
ARG PYTHON_VERSION=3.9 #8.10
ARG CUDA_VERSION=11.8
ARG CU_DNN=8.5.0.96
ARG MAMBA_VERSION=24.3.0-0
ARG CUDA_CHANNEL=nvidia
ARG INSTALL_CHANNEL=pytorch
# Automatically set by buildx
ARG TARGETPLATFORM


# Update basic dependencies we'll be using.
#RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
#        build-essential \
#        ca-certificates \
#        ccache \
#        curl \
#        python3 \
#        python3-pip \
#        git && \
#        rm -rf /var/lib/apt/lists/*


#RUN teaching_arithmetic && pip install -e .
# Install conda
# translating Docker's TARGETPLATFORM into mamba arches
#RUN case ${TARGETPLATFORM} in \
#         "linux/arm64")  MAMBA_ARCH=aarch64  ;; \
#         *)              MAMBA_ARCH=x86_64   ;; \
#    esac && \
#    curl -fsSL -v -o ~/mambaforge.sh -O  "https://github.com/conda-forge/miniforge/releases/download/${MAMBA_VERSION}/Mambaforge-${MAMBA_VERSION}-Linux-${MAMBA_ARCH}.sh"
#RUN chmod +x ~/mambaforge.sh && \
#    bash ~/mambaforge.sh -b -p /opt/conda && \
#    rm ~/mambaforge.sh

# Install pytorch
# On arm64 we exit with an error code
#RUN case ${TARGETPLATFORM} in \
#         "linux/arm64")  exit 1 ;; \
#         *)              /opt/conda/bin/conda update -y conda &&  \
#                         /opt/conda/bin/conda install -c "${INSTALL_CHANNEL}" -c "${CUDA_CHANNEL}" -y "python=${PYTHON_VERSION}" "pytorch=$PYTORCH_VERSION" "pytorch-cuda=$(echo $CUDA_VERSION | cut -d'.' -f 1-2)"  ;; \
#    esac && \
#    /opt/conda/bin/conda clean -ya

#USER 1000
#COPY --chown=user ./requirements.txt requirements.txt
#COPY ./requirements.txt requirements.txt
#RUN pip install --no-cache-dir --upgrade -r requirements.txt

#COPY --chown=user teaching_arithmetic /teaching_arithmetic
#COPY . /app
#COPY ./out /out
#RUN chmod 777 -R /out

#RUN chmod +x /out
#COPY ./out/addition_train /out/
#RUN chmod +x /teaching_arithmetic/train.py
#COPY ./train.sh /train.sh
#RUN chmod +x /train.sh

# Expose the secret DEBUG at buildtime and use its value as git remote URL
#RUN --mount=type=secret,id=DEBUG,mode=0444,required=true \
# git init && \
# git remote add origin $(cat /run/secrets/DEBUG)


# USER 1000
RUN chmod +x train.sh

#CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
ENTRYPOINT ["/train.sh"]