#!/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
apt-get -y install rsync

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 natsort
python3.10 -m pip install httpcore
python3.10 -m pip install httpx==0.24.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
mkdir -p embeddings
\cp -f run.sh "../run.sh"
\cp -f pOut.py "../pOut.py"
\cp -f out.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/models
	cd models
	\cp -f config_go3.json "../config.json"
	\cp -f ui-config_go3.json "../ui-config.json"
	rsync -r "embeddings/" "../embeddings/"
	rsync -r "extensions/" "../extensions/"
	\mv -f /workspace/firstContainer/SD/repositories/BLIP/_git /workspace/firstContainer/SD/repositories/BLIP/.git
	\mv -f /workspace/firstContainer/SD/repositories/CodeFormer/_git /workspace/firstContainer/SD/repositories/CodeFormer/.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
	cd /workspace/*/SD/extensions/sd-webui-controlnet/annotator/oneformer/oneformer/data
	curl -L -O https://github.com/Mikubill/sd-webui-controlnet/blob/main/annotator/oneformer/oneformer/data/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 --opt-sdp-attention --no-download-sd-model --exit
pip cache purge
sudo apt clean