Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -148,9 +148,31 @@ def mdl(Original_Model_Version, Path_to_MODEL, MODEL_LINK):
     
     src=getsrc(MODEL_LINK)  
 
+    if not os.path.exists("/workspace/sd/stable-diffusion-webui/models/VAE"):
+        call("mkdir -p /workspace/sd/stable-diffusion-webui/models/VAE", shell=True)
+
+    # Получаем список файлов в директории /workspace/auto-VAE
+    files = os.listdir("/workspace/auto-VAE")
+
+    for file in files:
+        source_path = os.path.join("/workspace/auto-VAE", file)
+        target_path = os.path.join(
+            "/workspace/sd/stable-diffusion-webui/models/VAE", file
+        )
+        # Проверяем, существует ли уже символическая ссылка
+        if not os.path.exists(target_path):
+            call(f"ln -s {source_path} {target_path}", shell=True)
+
     if not os.path.exists('/workspace/sd/stable-diffusion-webui/models/Stable-diffusion/SDv1-5.ckpt'):
         call('ln -s /workspace/auto-models/* /workspace/sd/stable-diffusion-webui/models/Stable-diffusion', shell=True)
-
+    
+    if not os.path.exists("/workspace/sd/stable-diffusion-webui/models/Lora"):
+        call("mkdir -p /workspace/sd/stable-diffusion-webui/models/Lora", shell=True)
+        call(
+            "ln -s /workspace/auto-lora/* /workspace/sd/stable-diffusion-webui/models/Lora",
+            shell=True,
+        )
+        
     if Path_to_MODEL !='':
       if os.path.exists(str(Path_to_MODEL)):
         print('Using the custom model')
@@ -261,97 +283,120 @@ def loradwn(LoRA_LINK):
         else:
           print('Wrong link, check that the link is valid')
 
-
-
 def CNet(ControlNet_Model, ControlNet_XL_Model):
-    
     def download(url, model_dir):
+        try:
+            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"[1;32mThe model {filename} already exists[0m")
+        except Exception as e:
+            print(f"Ошибка при скачивании {url}: {e}")
 
-        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")    
-
-    wrngv1=False
-    os.chdir('/workspace/sd/stable-diffusion-webui/extensions')
+    wrngv1 = False
+    os.chdir("/workspace/sd/stable-diffusion-webui/extensions")
     if not os.path.exists("sd-webui-controlnet"):
-      call('git clone https://github.com/Mikubill/sd-webui-controlnet.git', shell=True)
-      os.chdir('/workspace')
+        call(
+            "git clone https://github.com/Mikubill/sd-webui-controlnet.git", shell=True
+        )
+        os.chdir("/workspace")
     else:
-      os.chdir('sd-webui-controlnet')
-      call('git reset --hard', shell=True, stdout=open('/dev/null', 'w'), stderr=open('/dev/null', 'w'))
-      call('git pull', shell=True, stdout=open('/dev/null', 'w'), stderr=open('/dev/null', 'w'))
-      os.chdir('/workspace')
+        os.chdir("sd-webui-controlnet")
+        call(
+            "git reset --hard",
+            shell=True,
+            stdout=open("/dev/null", "w"),
+            stderr=open("/dev/null", "w"),
+        )
+        call(
+            "git pull",
+            shell=True,
+            stdout=open("/dev/null", "w"),
+            stderr=open("/dev/null", "w"),
+        )
+        os.chdir("/workspace")
+
+    mdldir = (
+        "/workspace/sd/stable-diffusion-webui/extensions/sd-webui-controlnet/models"
+    )
+    call(
+        "ln -s /workspace/auto-controlnet-models/* /workspace/sd/stable-diffusion-webui/extensions/sd-webui-controlnet/models",
+        shell=True,
+    )
 
-    mdldir="/workspace/sd/stable-diffusion-webui/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))
+        if "_sd14v1" in filename:
+            renamed = re.sub("_sd14v1", "-fp16", filename)
+            os.rename(os.path.join(mdldir, filename), os.path.join(mdldir, renamed))
 
-    call('wget -q -O CN_models.txt https://github.com/TheLastBen/fast-stable-diffusion/raw/main/AUTOMATIC1111_files/CN_models.txt', shell=True)
-    call('wget -q -O CN_models_XL.txt https://github.com/TheLastBen/fast-stable-diffusion/raw/main/AUTOMATIC1111_files/CN_models_XL.txt', shell=True)
-      
-    with open("CN_models.txt", 'r') as f:
+    call(
+        "wget -q -O CN_models.txt https://huggingface.co/spaces/stlaurentjr/RNPD/raw/main/config/CN_models.txt",
+        shell=True,
+    )
+    call(
+        "wget -q -O CN_models_XL.txt https://github.com/TheLastBen/fast-stable-diffusion/raw/main/AUTOMATIC1111_files/CN_models_XL.txt",
+        shell=True,
+    )
+
+    with open("CN_models.txt", "r") as f:
         mdllnk = f.read().splitlines()
-    with open("CN_models_XL.txt", 'r') as d:
+    with open("CN_models_XL.txt", "r") as d:
         mdllnk_XL = d.read().splitlines()
-    call('rm CN_models.txt CN_models_XL.txt', shell=True)
-    
-    os.chdir('/workspace')
+    call("rm CN_models.txt CN_models_XL.txt", shell=True)
 
-    if ControlNet_Model == "All" or ControlNet_Model == "all" :     
-      for lnk in mdllnk:
-          download(lnk, mdldir)
-      clear_output()
+    os.chdir("/workspace")
 
-      
-    elif ControlNet_Model == "15":
-      mdllnk=list(filter(lambda x: 't2i' in x, mdllnk))
-      for lnk in mdllnk:
-          download(lnk, mdldir)
-      clear_output()        
+    if ControlNet_Model == "All" or ControlNet_Model == "all":
+        for lnk in mdllnk:
+            download(lnk, mdldir)
+        clear_output()
 
+    elif ControlNet_Model == "17":
+        mdllnk = list(filter(lambda x: "t2i" in x, mdllnk))
+        for lnk in mdllnk:
+            download(lnk, mdldir)
+        clear_output()
+
+    elif (
+        ControlNet_Model.isdigit()
+        and int(ControlNet_Model) - 1 < 16
+        and int(ControlNet_Model) > 0
+    ):
+        download(mdllnk[int(ControlNet_Model) - 1], mdldir)
+        clear_output()
 
-    elif ControlNet_Model.isdigit() and int(ControlNet_Model)-1<14 and int(ControlNet_Model)>0:
-      download(mdllnk[int(ControlNet_Model)-1], mdldir)
-      clear_output()
-      
     elif ControlNet_Model == "none":
-       pass
-       clear_output()
+        pass
+        clear_output()
 
     else:
-      print('Wrong ControlNet V1 choice, try again')
-      wrngv1=True
+        print("[1;31mWrong ControlNet V1 choice, try again")
+        wrngv1 = True
 
+    if ControlNet_XL_Model == "All" or ControlNet_XL_Model == "all":
+        for lnk_XL in mdllnk_XL:
+            download(lnk_XL, mdldir)
+        if not wrngv1:
+            clear_output()
+        done()
 
-    if ControlNet_XL_Model == "All" or ControlNet_XL_Model == "all" :
-      for lnk_XL in mdllnk_XL:
-          download(lnk_XL, mdldir)
-      if not wrngv1:
-        clear_output()
-      done()
+    elif ControlNet_XL_Model.isdigit() and int(ControlNet_XL_Model) - 1 < 5:
+        download(mdllnk_XL[int(ControlNet_XL_Model) - 1], mdldir)
+        if not wrngv1:
+            clear_output()
+        done()
 
-    elif ControlNet_XL_Model.isdigit() and int(ControlNet_XL_Model)-1<5:
-      download(mdllnk_XL[int(ControlNet_XL_Model)-1], mdldir)
-      if not wrngv1:
-        clear_output()
-      done()
-    
     elif ControlNet_XL_Model == "none":
-       pass
-       if not wrngv1:
-        clear_output()
-       done()       
+        pass
+        if not wrngv1:
+            clear_output()
+        done()
 
     else:
-      print('Wrong ControlNet XL choice, try again')
-
-
+        print("[1;31mWrong ControlNet XL choice, try again")
 
 def sd(User, Password, model):
 
@@ -398,8 +443,8 @@ def sd(User, Password, model):
             mdlpth="--ckpt "+model
         else:
             mdlpth="--ckpt-dir "+model
-
-    configf="--disable-console-progressbars --no-half-vae --disable-safe-unpickle --api --no-download-sd-model --opt-sdp-attention --enable-insecure-extension-access  --skip-version-check --listen --port 3000 "+auth+" "+mdlpth
+    vae_path = "--vae-path /workspace/sd/stable-diffusion-webui/models/VAE/vae-ft-mse-840000-ema-pruned.safetensors"
+    configf="--disable-console-progressbars --no-half-vae --disable-safe-unpickle --api --no-download-sd-model --opt-sdp-attention --enable-insecure-extension-access  --skip-version-check --listen --port 3000 "+auth+" "+mdlpth+" "+vae_path
 
     return configf
 
@@ -488,7 +533,163 @@ def get_name(url, gdrive):
         content_disposition = six.moves.urllib_parse.unquote(res.headers["Content-Disposition"])
         filenm = re.search('attachment; filename="(.*?)"', content_disposition).groups()[0]
         return filenm   
+def download_and_install_config():
+    target_directory = "/workspace/sd/stable-diffusion-webui/"
+    config_url = (
+        "https://huggingface.co/spaces/stlaurentjr/RNPD/raw/main/config/config.json"
+    )
+    config_file_path = os.path.join(target_directory, "config.json")
+
+    # Check if the config file already exists
+    if not os.path.exists(config_file_path):
+        # Change the directory
+        os.chdir(target_directory)
+        # Download the file using curl command
+        call(f"curl -o config.json {config_url}", shell=True)
+        print(f"Config file downloaded successfully.")
+    else:
+        print("Config file already exists, download skipped.")
+
+
+def InstallDeforum():
+    # Переход в директорию расширений
+    os.chdir("/workspace/sd/stable-diffusion-webui/extensions")
 
+    # Проверка наличия директории с расширением
+    if not os.path.exists("deforum-for-automatic1111-webui"):
+        # Клонирование репозитория, если он не существует
+        call(
+            "git clone https://github.com/deforum-art/sd-webui-deforum.git", shell=True
+        )
+        # Возврат в корневую директорию
+        os.chdir("/workspace")
+    else:
+        # Если директория существует, переходим в нее
+        # os.chdir('deforum-for-automatic1111-webui')
+        # Сброс изменений и обновление репозитория
+        # call('git reset --hard', shell=True, stdout=open(os.devnull, 'w'), stderr=open(os.devnull, 'w'))
+        # call('git pull', shell=True, stdout=open(os.devnull, 'w'), stderr=open(os.devnull, 'w'))
+        # Возврат в корневую директорию
+        os.chdir("/workspace")
+        
+def InstallReActor():
+    # Переход в директорию расширений
+    os.chdir("/workspace/sd/stable-diffusion-webui/extensions")
+    # Проверка наличия директории с расширением
+    if not os.path.exists("sd-webui-reactor"):
+        # Установка библиотек
+        from subprocess import call, DEVNULL
+        call(
+            'pip install "onnxruntime-gpu>=1.16.1"', 
+             shell=True
+        )
+        # Клонирование репозитория, если он не существует
+        call(
+            "git clone https://github.com/Gourieff/sd-webui-reactor.git",
+            shell=True,
+        )
+        # Создание папки под модели
+        call(
+            "mkdir /workspace/sd/stable-diffusion-webui/models/insightface",
+            shell=True
+        )
+        os.chdir("/workspace/sd/stable-diffusion-webui/models/insightface")
+        # Скачивание модели для reActor
+        call(
+            "wget https://github.com/facefusion/facefusion-assets/releases/download/models/inswapper_128.onnx", 
+            shell=True, 
+            stdout=DEVNULL, 
+            stderr=DEVNULL
+        )
+        # Вовзрат в корневую директорию
+        clear_output()
+        os.chdir("/workspace")
+    else:
+        #Возврат в корневую директорию
+        os.chdir("/workspace")
+
+def InstallAnimateDiff():
+    # Переход в директорию расширений
+    os.chdir("/workspace/sd/stable-diffusion-webui/extensions")
+
+    # Проверка наличия директории с расширением
+    if not os.path.exists("sd-webui-animatediff"):
+        from subprocess import call, DEVNULL
+        # Клонирование репозитория, если он не существует
+        call(
+            "git clone https://github.com/continue-revolution/sd-webui-animatediff.git",
+            shell=True,
+        )
+        # Переход в папку с моделями
+        os.chdir(
+            "/workspace/sd/stable-diffusion-webui/extensions/sd-webui-animatediff/model"
+        )
+        # Скачивание моделей для animatediff
+        call(
+            "wget https://huggingface.co/guoyww/animatediff/resolve/refs%2Fpr%2F3/mm_sd_v15_v2.ckpt",
+            shell=True, 
+            stdout=DEVNULL, 
+            stderr=DEVNULL
+        )
+      #  call(
+      #      "wget https://huggingface.co/manshoety/AD_Stabilized_Motion/resolve/main/mm-Stabilized_high.pth",
+      #      shell=True, 
+      #      stdout=DEVNULL, 
+      #      stderr=DEVNULL
+      #  )
+        # Вовзрат в корневую директорию
+        clear_output()
+        os.chdir("/workspace")
+    else:
+        # Возврат в корневую директорию
+        os.chdir("/workspace")
+
+def InstallWildCards():
+    # Путь к целевой директории
+    target_dir = "/workspace/sd/stable-diffusion-webui/extensions/stable-diffusion-webui-wildcards/wildcards/"
+
+    # Перемещаемся в директорию расширений
+    os.chdir("/workspace/sd/stable-diffusion-webui/extensions")
+
+    if not os.path.exists("stable-diffusion-webui-wildcards"):
+        # Клонирование репозитория, если он ещё не существует
+        call(
+            "git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui-wildcards.git",
+            shell=True,
+        )
+
+        # Перемещаемся в клонированный репозиторий
+        os.chdir("stable-diffusion-webui-wildcards")
+
+        # Скачиваем файл по ID
+        gdown.download(
+            id="1sY9Yv29cCYZuxBvszkmLVgw--aRdwT1P", output="wildcards.zip", quiet=False
+        )
+
+        # Создаем целевую директорию, если она ещё не существует
+        os.makedirs(target_dir, exist_ok=True)
+
+        # Распаковываем архив
+        with zipfile.ZipFile("wildcards.zip", "r") as zip_ref:
+            zip_ref.extractall(target_dir)
+    else:
+        # Если репозиторий уже существует, обновляем его
+        os.chdir("stable-diffusion-webui-wildcards")
+        call(
+            "git reset --hard",
+            shell=True,
+            stdout=open(os.devnull, "w"),
+            stderr=open(os.devnull, "w"),
+        )
+        call(
+            "git pull",
+            shell=True,
+            stdout=open(os.devnull, "w"),
+            stderr=open(os.devnull, "w"),
+        )
+
+    # Возвращаемся в исходную рабочую директорию
+    os.chdir("/workspace")