Update app.py

app.py

@@ -284,4 +284,212 @@ elif Model_Version=="V2.1-512px" or Model_Version=="V2.1-768px":
   model=V2()
 
 else:
-   model=inpmdl()
+   model=inpmdl()
+
+    #@markdown # ControlNet
+from torch.hub import download_url_to_file
+from urllib.parse import urlparse
+import re
+from subprocess import run
+
+Model = "Lineart_Anime" #@param [ "None", "All (21GB)", "Canny", "Depth", "Lineart", "MLSD", "Normal", "OpenPose", "Scribble", "Seg", "ip2p", "Shuffle", "Inpaint", "Softedge", "Lineart_Anime", "Tile", "T2iadapter_Models"]
+
+v2_Model = "All" #@param [ "None", "All", "Canny", "Depth", "HED", "OpenPose", "Scribble"]
+
+#@markdown - Download/update ControlNet extension and its models
+
+def download(url, model_dir):
+
+    filename = os.path.basename(urlparse(url).path)
+    pth = os.path.abspath(os.path.join(model_dir, filename))
+    if not os.path.exists(pth):
+        print('Downloading: '+os.path.basename(url))
+        download_url_to_file(url, pth, hash_prefix=None, progress=True)
+    else:
+      print(f"The model {filename} already exists")
+
+Canny='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_canny.pth'
+Depth='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11f1p_sd15_depth.pth'
+Lineart='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_lineart.pth'
+MLSD='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_mlsd.pth'
+Normal='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_normalbae.pth'
+OpenPose='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose.pth'
+Scribble='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_scribble.pth'
+Seg='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_seg.pth'
+ip2p='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11e_sd15_ip2p.pth'
+Shuffle='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11e_sd15_shuffle.pth'
+Inpaint='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_inpaint.pth'
+Softedge='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_softedge.pth'
+Lineart_Anime='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15s2_lineart_anime.pth'
+Tile='https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11u_sd15_tile.pth'
+
+with capture.capture_output() as cap:
+  %cd /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/extensions
+  if not os.path.exists("sd-webui-controlnet"):
+    !git clone https://github.com/Mikubill/sd-$blasphemy-controlnet.git
+    %cd /content
+  else:
+    %cd sd-webui-controlnet
+    !git reset --hard
+    !git pull
+    %cd /content
+
+mdldir='/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/extensions/sd-webui-controlnet/models'
+for filename in os.listdir(mdldir):
+  if "_sd14v1" in filename:
+    renamed = re.sub("_sd14v1", "-fp16", filename)
+    os.rename(os.path.join(mdldir, filename), os.path.join(mdldir, renamed))
+
+!wget -q -O CN_models.txt https://github.com/TheLastBen/fast-stable-diffusion/raw/main/AUTOMATIC1111_files/CN_models.txt
+!wget -q -O CN_models_v2.txt https://github.com/TheLastBen/fast-stable-diffusion/raw/main/AUTOMATIC1111_files/CN_models_v2.txt
+
+with open("CN_models.txt", 'r') as f:
+  mdllnk = f.read().splitlines()
+with open("CN_models_v2.txt", 'r') as d:
+  mdllnk_v2 = d.read().splitlines()
+
+!rm CN_models.txt CN_models_v2.txt
+
+with capture.capture_output() as cap:
+  cfgnames=[os.path.basename(url).split('.')[0]+'.yaml' for url in mdllnk_v2]
+  %cd /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/extensions/sd-webui-controlnet/models
+  for name in cfgnames:
+      run(['cp', 'cldm_v21.yaml', name])
+  %cd /content
+
+if Model == "All (21GB)":
+  for lnk in mdllnk:
+      download(lnk, mdldir)
+  clear_output()
+
+elif Model == "T2iadapter_Models":
+  mdllnk=list(filter(lambda x: 't2i' in x, mdllnk))
+  for lnk in mdllnk:
+      download(lnk, mdldir)
+  clear_output()
+
+elif Model == "None":
+    pass
+    clear_output()
+
+else:
+  download(globals()[Model], mdldir)
+  clear_output()
+
+Canny='https://huggingface.co/thibaud/controlnet-sd21/resolve/main/control_v11p_sd21_canny.safetensors'
+Depth='https://huggingface.co/thibaud/controlnet-sd21/resolve/main/control_v11p_sd21_depth.safetensors'
+HED='https://huggingface.co/thibaud/controlnet-sd21/resolve/main/control_v11p_sd21_hed.safetensors'
+OpenPose='https://huggingface.co/thibaud/controlnet-sd21/resolve/main/control_v11p_sd21_openpose.safetensors'
+Scribble='https://huggingface.co/thibaud/controlnet-sd21/resolve/main/control_v11p_sd21_scribble.safetensors'
+
+if v2_Model == "All":
+  for lnk_v2 in mdllnk_v2:
+      download(lnk_v2, mdldir)
+  clear_output()
+  inf('\u2714 Done','success', '50px')
+
+elif v2_Model == "None":
+    pass
+    clear_output()
+    inf('\u2714 Done','success', '50px')
+
+else:
+  download(globals()[v2_Model], mdldir)
+  clear_output()
+  inf('\u2714 Done','success', '50px')
+
+i#@markdown # Start Stable-Diffusion
+from IPython.utils import capture
+import time
+import sys
+import fileinput
+from pyngrok import ngrok, conf
+
+Use_Cloudflare_Tunnel = False #@param {type:"boolean"}
+#@markdown - Offers better gradio responsivity
+
+Ngrok_token = "" #@param {type:"string"}
+
+#@markdown - Input your ngrok token if you want to use ngrok server
+
+User = "" #@param {type:"string"}
+Password= "" #@param {type:"string"}
+#@markdown - Add credentials to your Gradio interface (optional)
+
+auth=f"--gradio-auth {User}:{Password}"
+if User =="" or Password=="":
+  auth=""
+
+
+with capture.capture_output() as cap:
+  %cd /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/modules/
+
+  if Use_Latest_Working_Commit:
+    !wget -q -O paths.py https://github.com/TheLastBen/fast-stable-diffusion/raw/5632d2ef7fffd940976538d270854ec4faf26855/AUTOMATIC1111_files/paths.py
+    !wget -q -O extras.py https://github.com/AUTOMATIC1111/stable-diffusion-$blasphemy/raw/a9eab236d7e8afa4d6205127904a385b2c43bb24/modules/extras.py
+    !wget -q -O sd_models.py https://github.com/AUTOMATIC1111/stable-diffusion-$blasphemy/raw/a9eab236d7e8afa4d6205127904a385b2c43bb24/modules/sd_models.py
+    !wget -q -O /usr/local/lib/python3.9/dist-packages/gradio/blocks.py https://github.com/TheLastBen/fast-stable-diffusion/raw/7ff88eaa1fb4997bacd9845bd487f9a14335d625/AUTOMATIC1111_files/blocks.py 
+  else:
+    !wget -q -O paths.py https://raw.githubusercontent.com/TheLastBen/fast-stable-diffusion/main/AUTOMATIC1111_files/paths.py
+    !wget -q -O extras.py https://raw.githubusercontent.com/AUTOMATIC1111/stable-diffusion-$blasphemy/master/modules/extras.py
+    !wget -q -O sd_models.py https://raw.githubusercontent.com/AUTOMATIC1111/stable-diffusion-$blasphemy/master/modules/sd_models.py
+    !wget -q -O /usr/local/lib/python3.9/dist-packages/gradio/blocks.py https://raw.githubusercontent.com/TheLastBen/fast-stable-diffusion/main/AUTOMATIC1111_files/blocks.py
+  %cd /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/
+
+  !sed -i "[email protected](checkpoint_file)@os.path.splitext(checkpoint_file); map_location='cuda'@" /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/modules/sd_models.py
+  !sed -i '[email protected]_ui().*@ui.create_ui();shared.demo.queue(concurrency_count=999999,status_update_rate=0.1)@' /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/webui.py
+  !sed -i "s@map_location='cpu'@map_location='cuda'@" /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/modules/extras.py
+  !sed -i 's@/content/gdrive/MyDrive/sd/stablediffusion@/content/gdrive/{mainpth}/sd/stablediffusion@' /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/modules/paths.py
+  !sed -i 's@print(\"No module.*@@' /content/gdrive/$mainpth/sd/stablediffusion/ldm/modules/diffusionmodules/model.py
+  !sed -i 's@\"quicksettings\": OptionInfo(.*@"quicksettings": OptionInfo("sd_model_checkpoint,  sd_vae, CLIP_stop_at_last_layers, inpainting_mask_weight, initial_noise_multiplier", "Quicksettings list"),@' /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/modules/shared.py
+
+share=''
+if Ngrok_token!="":
+  ngrok.kill()
+  srv=ngrok.connect(7860, pyngrok_config=conf.PyngrokConfig(auth_token=Ngrok_token) , bind_tls=True).public_url
+
+  for line in fileinput.input('/usr/local/lib/python3.9/dist-packages/gradio/blocks.py', inplace=True):
+    if line.strip().startswith('self.server_name ='):
+        line = f'            self.server_name = "{srv[8:]}"\n'
+    if line.strip().startswith('self.protocol = "https"'):
+        line = '            self.protocol = "https"\n'
+    if line.strip().startswith('if self.local_url.startswith("https") or self.is_colab'):
+        line = ''
+    if line.strip().startswith('else "http"'):
+        line = ''
+    sys.stdout.write(line)
+
+elif Use_Cloudflare_Tunnel:
+  with capture.capture_output() as cap:
+    !pkill cloudflared
+    time.sleep(4)
+    !nohup cloudflared tunnel --url http://localhost:7860 > /content/srv.txt 2>&1 &
+    time.sleep(4)
+    !grep -o 'https[^[:space:]]*\.trycloudflare.com' /content/srv.txt >/content/srvr.txt
+    time.sleep(2)
+    srv= getoutput('cat /content/srvr.txt')
+
+    for line in fileinput.input('/usr/local/lib/python3.9/dist-packages/gradio/blocks.py', inplace=True):
+      if line.strip().startswith('self.server_name ='):
+          line = f'            self.server_name = "{srv[8:]}"\n'
+      if line.strip().startswith('self.protocol = "https"'):
+          line = '            self.protocol = "https"\n'
+      if line.strip().startswith('if self.local_url.startswith("https") or self.is_colab'):
+          line = ''
+      if line.strip().startswith('else "http"'):
+          line = ''
+      sys.stdout.write(line)
+            
+    !rm /content/srv.txt /content/srvr.txt
+
+else:
+  share='--share'
+
+try:
+  model
+  if os.path.isfile(model):
+    !python /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/webui.py $share --api --disable-safe-unpickle --enable-insecure-extension-access --no-download-sd-model --no-half-vae  --ckpt "$model" --opt-sdp-attention $auth --disable-console-progressbars 
+  else:
+    !python /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/webui.py $share --api --disable-safe-unpickle --enable-insecure-extension-access --no-download-sd-model --no-half-vae  --ckpt-dir "$model" --opt-sdp-attention $auth --disable-console-progressbars 
+except:
+   !python /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/webui.py $share --api --disable-safe-unpickle --enable-insecure-extension-access --no-download-sd-model --no-half-vae --opt-sdp-attention $auth --disable-console-progressbars