Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -291,44 +291,46 @@ def modeldwn(model_LINK):
 
 def loradwn(LoRA_LINK):
 
-    import gdown
-
-    if LoRA_LINK=='':
-        print('Nothing to do')
+    if LoRA_LINK == "":
+        print("[1;33mNothing to do")
     else:
-        os.makedirs('/workspace/sd/stable-diffusion-webui/models/Lora', exist_ok=True)
+        os.makedirs("/workspace/sd/stable-diffusion-webui/models/Lora", exist_ok=True)
 
-        src=getsrc(LoRA_LINK)
+        src = getsrc(LoRA_LINK)
 
-        if src=='civitai':
-            modelname=get_name(LoRA_LINK, False)
-            loramodel=f'/workspace/sd/stable-diffusion-webui/models/Lora/{modelname}'
+        if src == "civitai":
+            modelname = get_name(LoRA_LINK, False)
+            loramodel = f"/workspace/sd/stable-diffusion-webui/models/Lora/{modelname}"
             if not os.path.exists(loramodel):
-              dwn(LoRA_LINK, loramodel, 'Downloading the LoRA model')
-              clear_output()
+                dwn(LoRA_LINK, loramodel, "Downloading the LoRA model")
+                clear_output()
             else:
-              print('Model already exists')
-        elif src=='gdrive':
-            modelname=get_name(LoRA_LINK, True)
-            loramodel=f'/workspace/sd/stable-diffusion-webui/models/Lora/{modelname}'
+                print("[1;33mModel already exists")
+        elif src == "gdrive":
+            modelname = get_true_name(LoRA_LINK)
+            loramodel = f"/workspace/sd/stable-diffusion-webui/models/Lora/{modelname}"
             if not os.path.exists(loramodel):
-              gdown.download(url=LoRA_LINK, output=loramodel, quiet=False, fuzzy=True)
-              clear_output()
+                gdown.download(
+                    url=LoRA_LINK.replace("/file/d/", "/uc?id=").replace("/view", ""),
+                    output=loramodel,
+                    quiet=False,
+                )
+                clear_output()
             else:
-              print('Model already exists')
+                print("[1;33mModel already exists")
         else:
-            modelname=os.path.basename(LoRA_LINK)
-            loramodel=f'/workspace/sd/stable-diffusion-webui/models/Lora/{modelname}'
+            modelname = os.path.basename(LoRA_LINK)
+            loramodel = f"/workspace/sd/stable-diffusion-webui/models/Lora/{modelname}"
             if not os.path.exists(loramodel):
-              gdown.download(url=LoRA_LINK, output=loramodel, quiet=False, fuzzy=True)
-              clear_output()
+                gdown.download(url=LoRA_LINK, output=loramodel, quiet=False, fuzzy=True)
+                clear_output()
             else:
-              print('Model already exists')
+                print("[1;33mModel already exists")
 
-        if os.path.exists(loramodel) :
-          print('LoRA downloaded')
+        if os.path.exists(loramodel):
+            print("[1;32mLoRA downloaded")
         else:
-          print('Wrong link, check that the link is valid')
+            print("[1;31mWrong link, check that the link is valid")
 
 def CNet(ControlNet_Model, ControlNet_XL_Model):
     def download(url, model_dir):