#!/bin/bash  
  
if [ ! -f "${ROOT}/prepare-v1.ready" ]; then

    python -m pip install --upgrade pip
    python -m pip install rich
    python -m pip install numexpr
    python -m pip install insightface
    python -m pip install onnx
    python -m pip install onnxruntime
    python -m pip install onnxruntime-gpu
    python -m pip install opencv-python
    python -m pip install ifnude
    python -m pip install cython
    python -m pip install pytorch-lightning
    python -m pip install blobfile

    touch "${ROOT}/prepare-v1.ready"

fi

  
if [ ! -f "${ROOT}/prepare-v2.ready" ]; then

    python -m pip install wandb
    python -m pip install fastai==1.0.60
    python -m pip install tensorboardX
    python -m pip install ffmpeg
    python -m pip install ffmpeg-python
    python -m pip install yt-dlp
    python -m pip install opencv-python
    python -m pip install Pillow

    touch "${ROOT}/prepare-v2.ready"

fi

if [ ! -f "${ROOT}/prepare-v3.ready" ]; then

    python -m pip install scikit-learn
    python -m pip install diffusers

    touch "${ROOT}/prepare-v3.ready"

fi

if [ ! -f "${VOLUME_DIR_LOCAL}/styles.csv" ]; then

    wget "https://huggingface.co/spaces/Moldwebs/srv/raw/main/sd-webui/styles.csv" -O ${VOLUME_DIR_LOCAL}/styles.csv

fi

if [ ! -f "${ROOT}/styles.csv" ]; then

    cp ${VOLUME_DIR_LOCAL}/styles.csv ${ROOT}/styles.csv

fi