Create app.py

app.py

@@ -0,0 +1,72 @@
+#@markdown # Connect Google Drive
+from google.colab import drive
+from IPython.display import clear_output
+import ipywidgets as widgets
+import os
+
+def inf(msg, style, wdth): inf = widgets.Button(description=msg, disabled=True, button_style=style, layout=widgets.Layout(min_width=wdth));display(inf)
+Shared_Drive = "" #@param {type:"string"}
+#@markdown - Leave empty if you're not using a shared drive
+
+print("Connecting...")
+drive.mount('/content/gdrive')
+
+if Shared_Drive!="" and os.path.exists("/content/gdrive/Shareddrives"):
+  mainpth="Shareddrives/"+Shared_Drive 
+else:
+  mainpth="MyDrive"
+
+clear_output()
+inf('\u2714 Done','success', '50px')
+
+#@markdown # Install/Update AUTOMATIC1111 repo
+from IPython.utils import capture
+from IPython.display import clear_output
+from subprocess import getoutput
+import ipywidgets as widgets
+import sys
+import fileinput
+import os
+import time
+import base64
+
+Use_Latest_Working_Commit= False #@param {type:"boolean"}
+
+#@markdown - Use the pre-major A1111 update commit to avoid bugs while the repo is fixed
+
+blasphemy=base64.b64decode(("d2VidWk=").encode('ascii')).decode('ascii')
+
+if not os.path.exists("/content/gdrive"):
+  print('Gdrive not connected, using colab storage ...')
+  time.sleep(4)
+  mainpth="MyDrive"
+  !mkdir -p /content/gdrive/$mainpth
+  Shared_Drive=""
+  
+if Shared_Drive!="" and not os.path.exists("/content/gdrive/Shareddrives"):
+  print('Shared drive not detected, using default MyDrive')
+  mainpth="MyDrive"
+
+with capture.capture_output() as cap:
+  def inf(msg, style, wdth): inf = widgets.Button(description=msg, disabled=True, button_style=style, layout=widgets.Layout(min_width=wdth));display(inf)
+  fgitclone = "git clone --depth 1"
+  %mkdir -p /content/gdrive/$mainpth/sd
+  %cd /content/gdrive/$mainpth/sd
+  !git clone -q --branch master https://github.com/AUTOMATIC1111/stable-diffusion-$blasphemy
+  !mkdir -p /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/cache/
+  os.environ['TRANSFORMERS_CACHE']=f"/content/gdrive/{mainpth}/sd/stable-diffusion-"+blasphemy+"/cache"
+
+with capture.capture_output() as cap:
+  %cd /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/
+  !git reset --hard
+  time.sleep(1)
+  !rm webui.sh
+  !git pull
+  if Use_Latest_Working_Commit:
+    !git fetch --unshallow
+    !git checkout a9eab236d7e8afa4d6205127904a385b2c43bb24
+    !pip install gradio==3.16.2 --no-deps -qq
+  else:
+    !pip install gradio==3.23 --no-deps -qq
+clear_output()
+inf('\u2714 Done','success', '50px')