#!/bin/bash
cd /workspace/*
mkdir -p tmp
rm -rf README.md
rm -rf s
rm -rf ss.sh

apt-get update
sudo apt install -y python3.10
sudo apt-get install -y python3.10-distutils
sudo apt-get install git-lfs
sudo apt install -y libtcmalloc-minimal4
apt-get -y install rsync
apt-get -y install zip

curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10
python3.10 -m pip install --upgrade pip
python3.10 -m pip install setuptools
python3.10 -m pip install requests
python3.10 -m pip install wget
python3.10 -m pip install gdown
python3.10 -m pip install pcloud
python3.10 -m pip install natsort
python3.10 -m pip install httpcore
python3.10 -m pip install httpx==0.24.1
python3.10 -m pip uninstall -y setuptools
python3.10 -m pip install setuptools==69.5.1
python3.10 -m pip uninstall -y matplotlib
python3.10 -m pip install matplotlib==3.8.4

curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | sudo bash
git config --global user.name "anonymous"
git config --global user.email my@gmail.com
git lfs install --force
sudo apt update && sudo apt install -y libgl1-mesa-glx
sudo apt autoremove && sudo apt clean

cd /workspace/*/SD
\cp -f e "../e"
\cp -f run.sh "../run.sh"
#\cp -f pOut.py "../pOut.py"
#\cp -f out.sh "../out.sh"
\cp -f out_rp.sh "../out.sh"
\cp -f gc.sh "../gc.sh"
\cp -f forcePull.sh "../forcePull.sh"
\cp -f myRun.ipynb "../myRun.ipynb"
git gc
if [ ! -d "models" ]; then
	git clone https://huggingface.co/zuv0/modelsXL models
	cd models
	\mv -f /workspace/firstContainer/SD/repositories/BLIP/_git /workspace/firstContainer/SD/repositories/BLIP/.git
	\mv -f /workspace/firstContainer/SD/repositories/generative-models/_git /workspace/firstContainer/SD/repositories/generative-models/.git
	\mv -f /workspace/firstContainer/SD/repositories/k-diffusion/_git /workspace/firstContainer/SD/repositories/k-diffusion/.git
	\mv -f /workspace/firstContainer/SD/repositories/stable-diffusion-stability-ai/_git /workspace/firstContainer/SD/repositories/stable-diffusion-stability-ai/.git
	\mv -f /workspace/firstContainer/SD/repositories/stable-diffusion-webui-assets/_git /workspace/firstContainer/SD/repositories/stable-diffusion-webui-assets/.git
	cd /workspace/*/SD/extensions-builtin/forge_legacy_preprocessors/annotator/oneformer/oneformer/data
	curl -L -O https://huggingface.co/zuv0/test/resolve/main/bpe_simple_vocab_16e6.txt.gz
fi

cd /workspace/*/SD
sh downs.sh $1 $2

cd /workspace/*
apt-get install -y libgl1-mesa-glx
python3.10 ./SD/launch.py --reinstall-torch --no-download-sd-model --exit
sh forcePull.sh
sh gc.sh 1
pip cache purge
sudo apt clean