Update app.py

app.py

@@ -1,21 +1,21 @@
-#@title - Instant (Simple)
-import requests
+#@title 1. Installation
+webui_version = "camenduru's fork (stable)" #@param ["camenduru's fork (stable)", "original automatic1111 repo (latest feature, might be buggy)"]
+import shutil
 import os
+import requests
 import subprocess
 cek = subprocess.getoutput('nvidia-smi')
-from google.colab.output import eval_js
-os.environ['colab_url'] = eval_js("google.colab.kernel.proxyPort(7860, {'cache': false})")
-
 if not 'T4' in cek:
-  print("[1;32m-----Your colab session is over. Use other google account or wait 12-24 hours-----")
+  print("-----Your colab session is over. Use other google account or wait 12-24 hours-----")
   print("-----Sesi colab elu udah habis bro. Saran gw tunggu 12-24 jam atau bikin/pake akun google lain-----")
   raise KeyboardInterrupt
+from google.colab.output import eval_js
+os.environ['colab_url'] = eval_js("google.colab.kernel.proxyPort(7860, {'cache': false})")
 
 
-#thanks to https://github.com/camenduru for the fork
-print('[1;32mInstalling dependencies...')
-print('[1;32m(This process will only happen once per session)')
-print('[0m')
+print('Installing dependencies...')
+print('(This process will only happen once per session)')
+print('')
 %cd /content
 #!if ! [ -f xformers ]; then pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230118-cp38-cp38-linux_x86_64.whl; pip install -q --pre triton; touch xformers; fi
 if not os.path.isfile('dependencies'):
@@ -45,26 +45,22 @@ if not os.path.isfile('dependencies'):
     # !pip install -q --pre triton
     !touch dependencies
 
-print('[1;32mInstalling webui...')
-print('[0m')
-!if ! [ -d stable-diffusion-webui ]; then git clone -b v2.1 https://github.com/camenduru/stable-diffusion-webui; touch forkedwebui; fi
+print('Installing webui...')
+print('')
+if webui_version=="camenduru's fork (stable)":
+  !if ! [ -f forkedwebui ]; then echo "cloning from camenduru's fork"; rm -rf stable-diffusion-webui; git clone -b v2.1 https://github.com/camenduru/stable-diffusion-webui; sed -i -e '''/    prepare_environment()/a\    os.system\(f\"""sed -i -e ''\"s/dict()))/dict())).cuda()/g\"'' /content/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/util.py""")''' /content/stable-diffusion-webui/launch.py; sed -i -e 's/fastapi==0.90.1/fastapi==0.89.1/g' /content/stable-diffusion-webui/requirements_versions.txt; touch forkedwebui; fi
+else:
+  !if [ -f forkedwebui ]; then rm forkedwebui; mv '/content/stable-diffusion-webui/models/Stable-diffusion' '/content/temp'; rm -rf stable-diffusion-webui; fi
+  !if ! [ -d stable-diffusion-webui ]; then echo "cloning from official repo"; git clone https://github.com/automatic1111/stable-diffusion-webui; cp -rf '/content/temp'/* '/content/stable-diffusion-webui/models/Stable-diffusion'; rm -rf '/content/temp'; fi
 %cd stable-diffusion-webui
 
-#AnythingV3 Model & Vae
-print('[1;32mDownloading AnythingV3 model...')
-print('[0m')
-%cd /content/stable-diffusion-webui/models/Stable-diffusion/
-!if ! [ -f Anything-V3.0.safetensors ]; then gdown https://huggingface.co/Linaqruf/anything-v3.0/resolve/main/anything-v3-fp32-pruned.safetensors -O /content/stable-diffusion-webui/models/Stable-diffusion/Anything-V3.0.safetensors; fi
-!if ! [ -f Anything-V3.0.vae.pt ]; then gdown https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/VAEs/orangemix.vae.pt -O /content/stable-diffusion-webui/models/Stable-diffusion/Anything-V3.0.vae.pt; fi
-
 #webui setup for nai
-print('[1;32mSetting UI configs...')
-print('[0m')
+print('Setting UI configs...')
+print('')
 %cd /content/stable-diffusion-webui
-!if [ -f configinstan ]; then rm configinstan; fi
-!if ! [ -f configinstanlite ]; then gdown https://pastebin.com/raw/fmN0i5Sd -O /content/stable-diffusion-webui/ui-config.json && gdown https://pastebin.com/raw/amVtB1NW -O /content/stable-diffusion-webui/config.json; fi
-!if ! [ -f configinstanlite ]; then mv /content/stable-diffusion-webui/modules/ui.py /content/stable-diffusion-webui/modules/ui_backup.py && gdown https://pastebin.com/raw/pKFWUfLM -O /content/stable-diffusion-webui/modules/ui.py; fi
-!touch configinstanlite
+!if [ -f configinstanlite ]; then rm configinstanlite && rm /content/stable-diffusion-webui/modules/ui.py && mv /content/stable-diffusion-webui/modules/ui_backup.py /content/stable-diffusion-webui/modules/ui.py; fi
+!if ! [ -f configinstan ]; then gdown https://pastebin.com/raw/gMTvTUUZ -O /content/stable-diffusion-webui/ui-config.json && gdown https://pastebin.com/raw/VviKmfiQ -O /content/stable-diffusion-webui/config.json; fi
+!touch configinstan
 
 #notification bell
 !gdown https://github.com/etherealxx/etherportal-webui-colab/raw/master/notification.mp3 -O /content/stable-diffusion-webui/notification.mp3
@@ -72,17 +68,17 @@ print('[0m')
 #foolhardy remacri
 ! [ ! -d models/ESRGAN ] && mkdir -p models/ESRGAN; [ ! -f models/ESRGAN/4x_foolhardy_Remacri.pth ] && gdown https://huggingface.co/etherealxx/whoseisthis/resolve/main/4x_foolhardy_Remacri.pth -O /content/stable-diffusion-webui/models/ESRGAN/4x_foolhardy_Remacri.pth
 
+#temporaryfix
+
 #extensions
-print('[1;32mInstalling extensions...')
-print('[0m')
+print('Installing extensions...')
+print('')
 %cd /content/stable-diffusion-webui/extensions
-!bash -c 'for repo in camenduru/stable-diffusion-webui-images-browser Extraltodeus/depthmap2mask ddPn08/kohya-sd-scripts-webui arenatemp/stable-diffusion-webui-model-toolkit; do if [ ! -d "
-repo" && true; fi; done' #DominikDoom/a1111-sd-webui-tagcomplete
-!bash -c 'for repo in sd-civitai-browser kohya-sd-scripts-webui stable-diffusion-webui-model-toolkit batchlinks-webui; do if [ -d "
-repo && true; fi; done'
-#%cd /content/stable-diffusion-webui/extensions/UnivAICharGen
-#!git checkout c56b5146049b91cfcd645c62bd356bd0631c4131 .
+!bash -c 'for repo in camenduru/stable-diffusion-webui-images-browser camenduru/sd-civitai-browser Extraltodeus/depthmap2mask kohya-ss/sd-webui-additional-networks etherealxx/batchlinks-webui; do if [ ! -d "$repo" ]; then git clone "https://github.com/$repo" && true; fi; done' #DominikDoom/a1111-sd-webui-tagcomplete
 !git clone -b pre-embedding --single-branch https://github.com/etherealxx/Umi-AI
+#%cd /content/stable-diffusion-webui/extensions/UnivAICharGen
+#!if [ -f forkedwebui ]; then git checkout c56b5146049b91cfcd645c62bd356bd0631c4131 .; fi
+# !git clone -b lora --single-branch https://github.com/etherealxx/sd-civitai-browser
 
 #badprompt & badartist
 %cd /content/stable-diffusion-webui/embeddings/
@@ -90,12 +86,5 @@ repo && true; fi; done'
 !if ! [ -f bad_artist.pt ]; then gdown https://huggingface.co/etherealxx/whoseisthis/resolve/main/bad_artist.pt -O /content/stable-diffusion-webui/embeddings/bad_artist.pt; fi
 !if ! [ -f bad_hands.pt ]; then gdown https://huggingface.co/etherealxx/whoseisthis/resolve/main/bad-hands-5.pt -O /content/stable-diffusion-webui/embeddings/bad_hands.pt; fi
 
-#some fix
-!sed -i -e '''/    prepare_environment()/a\    os.system\(f\"""sed -i -e ''\"s/dict()))/dict())).cuda()/g\"'' /content/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/util.py""")''' /content/stable-diffusion-webui/launch.py
-!sed -i -e 's/fastapi==0.90.1/fastapi==0.89.1/g' /content/stable-diffusion-webui/requirements_versions.txt
-
-print('[1;32mInstallation Success!')
-print('[1;32mLaunching the webui...')
-print('[0m')
-%cd /content/stable-diffusion-webui
-!python launch.py --ckpt /content/stable-diffusion-webui/models/Stable-diffusion/Anything-V3.0.safetensors --vae-path /content/stable-diffusion-webui
+print('Installation Success!')
+print('')