################################################################
#
#   Project:    TTI Function
#
#   podman build -t tti-function .
#   podman run -v /<path_to_folder>/transcript_in/:/app/input/ -v /<path_to_folder>/transcript_out/:/app/images/ --name tti-function_container --rm -t tti-function
#
################################################################

FROM python:3.11-slim

# set the working directory
WORKDIR /app

# Create image folder
RUN mkdir /app/images
RUN mkdir /app/input
RUN mkdir /app/tasks

RUN mkdir -p /root/.cache/huggingface/hub/models--RunDiffusion--Juggernaut-XL-v8/snapshots/9022a900377ce2d3303d3e6d86b09f6874e1e2a7/text_encoder/
RUN mkdir -p /root/.cache/huggingface/hub/models--RunDiffusion--Juggernaut-XL-v8/snapshots/9022a900377ce2d3303d3e6d86b09f6874e1e2a7/text_encoder_2/
RUN mkdir -p /root/.cache/huggingface/hub/models--RunDiffusion--Juggernaut-XL-v8/snapshots/9022a900377ce2d3303d3e6d86b09f6874e1e2a7/vae/
RUN mkdir -p /root/.cache/huggingface/hub/models--RunDiffusion--Juggernaut-XL-v8/snapshots/9022a900377ce2d3303d3e6d86b09f6874e1e2a7/unet/

# Update Packagelist
RUN apt-get update

# install dependencies
COPY ./requirements.txt /app
RUN pip install --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host files.pythonhosted.org --no-cache-dir --upgrade -r requirements.txt

# copy files to folder
COPY ./runner.py /app
COPY ./init.sh /app
RUN chmod +x /app/init.sh
COPY ./styles.json /app

CMD ["bash", "init.sh"]