FROM 035363416972.dkr.ecr.us-east-1.amazonaws.com/minimal:latest

WORKDIR /opt

RUN apt update && apt install -y curl
RUN python3 -m venv venv
ADD composer/transformer/install_inference_requirements.sh ./
RUN ./install_inference_requirements.sh
ADD ./venv_optimizer.sh ./
RUN ./venv_optimizer.sh
RUN rm -f install_inference_requirements.sh venv_optimizer.sh

WORKDIR /opt/wavtool
ADD lib lib

WORKDIR /opt/wavtool/composer/transformer
# not in use
#ADD composer/transformer/t5-small-inference t5-small
ADD composer/transformer/byt5-small byt5-small
ARG COMPOSER_MODEL=model_2023-09-07_resurrect_aws
ADD composer/transformer/$COMPOSER_MODEL model
ADD composer/transformer/*.py ./
ADD composer/transformer/inference_container_entrypoint.sh /opt/entrypoint.sh
RUN ln -sf ../../lib lib
RUN bash -c ". /opt/venv/bin/activate ; python -m compileall ."

ENTRYPOINT ["/opt/entrypoint.sh"]