🚧 Saving files when switching WebUIs

files_cells/notebooks/en/downloading_en.ipynb

@@ -111,18 +111,47 @@
         "\n",
         "\n",
         "# ================= OTHER =================\n",
+        "# Setup Timer\n",
         "try:\n",
         "    start_colab\n",
         "except:\n",
         "    start_colab = int(time.time())-5\n",
         "\n",
+        "# Save files temporarily\n",
+        "temporarily_dir = f'{root_path}/temp_dir'\n",
+        "\n",
+        "def copy_items_with_replace(src_base, dst_base):\n",
+        "    items_to_copy = [\n",
+        "        'embeddings',\n",
+        "        'models/Stable-diffusion',\n",
+        "        'models/VAE',\n",
+        "        'models/Lora'\n",
+        "    ]\n",
+        "\n",
+        "    print(\"- There's a file move in progress... (This can take a minute or more)\", end='')\n",
+        "    for item in items_to_copy:\n",
+        "        src = os.path.join(src_base, item)\n",
+        "        dst = os.path.join(dst_base, item)\n",
+        "\n",
+        "        if os.path.isdir(src):\n",
+        "            if os.path.exists(dst):\n",
+        "                shutil.rmtree(dst)\n",
+        "            shutil.move(src, dst)\n",
+        "        elif os.path.isfile(src):\n",
+        "            dst_dir = os.path.dirname(dst)\n",
+        "            os.makedirs(dst_dir, exist_ok=True)\n",
+        "            shutil.move(src, dst)\n",
+        "    print(\"\\r🔥 The files have been moved!\")\n",
+        "\n",
         "\n",
         "# ================= MAIN CODE =================\n",
         "# WebUI switch\n",
         "if os.path.exists(webui_path):\n",
         "    if UI != OLD_UI:\n",
         "        print(f'Switching the WebUI from \\033[33m{OLD_UI}\\033[0m to \\033[33m{UI}\\033[0m...', end='')\n",
+        "        copy_items_with_replace(webui_path, temporarily_dir)\n",
         "        shutil.rmtree(webui_path)\n",
+        "\n",
         "        # Updating the environment variable after a change\n",
         "        os.environ['SDW_OLD_UI'] = UI\n",
         "        print('\\rDone!')\n",
@@ -132,15 +161,16 @@
         "# Unpacking the WebUI\n",
         "if not os.path.exists(webui_path):\n",
         "    start_install = int(time.time())\n",
-        "    print(\"⌚ Unpacking Stable Diffusion...\" if UI != 'Forge' else \"⌚ Unpacking Stable Diffusion (Forge)...\", end='')\n",
+        "    print(\"⌚ Unpacking Stable Diffusion...\" if change_webui != 'Forge' else \"⌚ Unpacking Stable Diffusion (Forge)...\", end='')\n",
         "    with capture.capture_output() as cap:\n",
         "        aria2_command = \"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M\"\n",
-        "        url = \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip\" if UI != 'Forge' else \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip\"\n",
+        "        url = \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip\" if change_webui != 'Forge' else \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip\"\n",
         "        !{aria2_command} {url} -o repo.zip\n",
         "\n",
         "        !unzip -q -o repo.zip -d {webui_path}\n",
         "        !rm -rf repo.zip\n",
         "\n",
+        "        # other\n",
         "        %cd {root_path}\n",
         "        os.environ[\"SAFETENSORS_FAST_GPU\"]='1'\n",
         "        os.environ[\"CUDA_MODULE_LOADING\"]=\"LAZY\"\n",
@@ -151,6 +181,11 @@
         "    del cap\n",
         "    install_time = timedelta(seconds=time.time()-start_install)\n",
         "    print(\"\\r🚀 Unpacking is complete! For\",\"%02d:%02d:%02d ⚡\\n\" % (install_time.seconds / 3600, (install_time.seconds / 60) % 60, install_time.seconds % 60), end='', flush=True)\n",
+        "\n",
+        "    # Move temporary files\n",
+        "    if os.path.exists(temporarily_dir):\n",
+        "        copy_items_with_replace(temporarily_dir, webui_path)\n",
+        "        shutil.rmtree(temporarily_dir)\n",
         "else:\n",
         "    print(\"🚀 All unpacked... Skip. ⚡\")\n",
         "    start_colab = float(open(f'{webui_path}/static/colabTimer.txt', 'r').read())\n",

files_cells/notebooks/ru/downloading_ru.ipynb

@@ -110,19 +110,48 @@
         "locals().update({key: settings.get(key) for key in VARIABLES})\n",
         "\n",
         "\n",
-        "# ================= OTHER =================\n",
+        "# =================== OTHER ===================\n",
+        "# Setup Timer\n",
         "try:\n",
         "    start_colab\n",
         "except:\n",
         "    start_colab = int(time.time())-5\n",
         "\n",
+        "# Save files temporarily\n",
+        "temporarily_dir = f'{root_path}/temp_dir'\n",
+        "\n",
+        "def copy_items_with_replace(src_base, dst_base):\n",
+        "    items_to_copy = [\n",
+        "        'embeddings',\n",
+        "        'models/Stable-diffusion',\n",
+        "        'models/VAE',\n",
+        "        'models/Lora'\n",
+        "    ]\n",
+        "\n",
+        "    print(\"- Идет перемещение файлов... (Это может занять минуту или больше)\", end='')\n",
+        "    for item in items_to_copy:\n",
+        "        src = os.path.join(src_base, item)\n",
+        "        dst = os.path.join(dst_base, item)\n",
+        "\n",
+        "        if os.path.isdir(src):\n",
+        "            if os.path.exists(dst):\n",
+        "                shutil.rmtree(dst)\n",
+        "            shutil.move(src, dst)\n",
+        "        elif os.path.isfile(src):\n",
+        "            dst_dir = os.path.dirname(dst)\n",
+        "            os.makedirs(dst_dir, exist_ok=True)\n",
+        "            shutil.move(src, dst)\n",
+        "    print(\"\\r🔥 Файлы были перемещены!\")\n",
+        "\n",
         "\n",
         "# ================= MAIN CODE =================\n",
         "# WebUI switch\n",
         "if os.path.exists(webui_path):\n",
         "    if UI != OLD_UI:\n",
         "        print(f'Переключение веб-интерфейса с \\033[33m{OLD_UI}\\033[0m на \\033[33m{UI}\\033[0m...', end='')\n",
+        "        copy_items_with_replace(webui_path, temporarily_dir)\n",
         "        shutil.rmtree(webui_path)\n",
+        "\n",
         "        # Updating the environment variable after a change\n",
         "        os.environ['SDW_OLD_UI'] = UI\n",
         "        print('\\rГотово!')\n",
@@ -132,16 +161,17 @@
         "# Unpacking the WebUI\n",
         "if not os.path.exists(webui_path):\n",
         "    start_install = int(time.time())\n",
-        "    print(\"⌚ Распаковка Stable Diffusion...\" if UI != 'Forge' else \"⌚ Распаковка Stable Diffusion (Forge)...\", end='')\n",
+        "    print(\"⌚ Распаковка Stable Diffusion...\" if change_webui != 'Forge' else \"⌚ Распаковка Stable Diffusion (Forge)...\", end='')\n",
         "    with capture.capture_output() as cap:\n",
         "        aria2_command = \"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M\"\n",
-        "        url = \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip\" if UI != 'Forge' else \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip\"\n",
+        "        url = \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip\" if change_webui != 'Forge' else \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip\"\n",
         "        !{aria2_command} {url} -o repo.zip\n",
         "\n",
         "        !unzip -q -o repo.zip -d {webui_path}\n",
         "        !rm -rf repo.zip\n",
         "\n",
-        "        %cd {root_path}\n",
+        "        # other\n",
+        "        get_ipython().run_line_magic('cd', '{root_path}')\n",
         "        os.environ[\"SAFETENSORS_FAST_GPU\"]='1'\n",
         "        os.environ[\"CUDA_MODULE_LOADING\"]=\"LAZY\"\n",
         "        os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"3\"\n",
@@ -151,6 +181,11 @@
         "    del cap\n",
         "    install_time = timedelta(seconds=time.time()-start_install)\n",
         "    print(\"\\r🚀 Распаковка Завершена! За\",\"%02d:%02d:%02d ⚡\\n\" % (install_time.seconds / 3600, (install_time.seconds / 60) % 60, install_time.seconds % 60), end='', flush=True)\n",
+        "\n",
+        "    # Move temporary files\n",
+        "    if os.path.exists(temporarily_dir):\n",
+        "        copy_items_with_replace(temporarily_dir, webui_path)\n",
+        "        shutil.rmtree(temporarily_dir)\n",
         "else:\n",
         "    print(\"🚀 Все распакованно... Пропуск. ⚡\")\n",
         "    start_colab = float(open(f'{webui_path}/static/colabTimer.txt', 'r').read())\n",

files_cells/python/en/downloading_en.py

@@ -101,18 +101,47 @@ locals().update({key: settings.get(key) for key in VARIABLES})
 
 
 # ================= OTHER =================
+# Setup Timer
 try:
     start_colab
 except:
     start_colab = int(time.time())-5
 
+# Save files temporarily
+temporarily_dir = f'{root_path}/temp_dir'
+
+def copy_items_with_replace(src_base, dst_base):
+    items_to_copy = [
+        'embeddings',
+        'models/Stable-diffusion',
+        'models/VAE',
+        'models/Lora'
+    ]
+
+    print("- There's a file move in progress... (This can take a minute or more)", end='')
+    for item in items_to_copy:
+        src = os.path.join(src_base, item)
+        dst = os.path.join(dst_base, item)
+
+        if os.path.isdir(src):
+            if os.path.exists(dst):
+                shutil.rmtree(dst)
+            shutil.move(src, dst)
+        elif os.path.isfile(src):
+            dst_dir = os.path.dirname(dst)
+            os.makedirs(dst_dir, exist_ok=True)
+            shutil.move(src, dst)
+    print("\r🔥 The files have been moved!")
+
 
 # ================= MAIN CODE =================
 # WebUI switch
 if os.path.exists(webui_path):
     if UI != OLD_UI:
         print(f'Switching the WebUI from \033[33m{OLD_UI}\033[0m to \033[33m{UI}\033[0m...', end='')
+        copy_items_with_replace(webui_path, temporarily_dir)
         shutil.rmtree(webui_path)
+
         # Updating the environment variable after a change
         os.environ['SDW_OLD_UI'] = UI
         print('\rDone!')
@@ -122,15 +151,16 @@ if os.path.exists(webui_path):
 # Unpacking the WebUI
 if not os.path.exists(webui_path):
     start_install = int(time.time())
-    print("⌚ Unpacking Stable Diffusion..." if UI != 'Forge' else "⌚ Unpacking Stable Diffusion (Forge)...", end='')
+    print("⌚ Unpacking Stable Diffusion..." if change_webui != 'Forge' else "⌚ Unpacking Stable Diffusion (Forge)...", end='')
     with capture.capture_output() as cap:
         aria2_command = "aria2c --console-log-level=error -c -x 16 -s 16 -k 1M"
-        url = "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip" if UI != 'Forge' else "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip"
+        url = "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip" if change_webui != 'Forge' else "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip"
         get_ipython().system('{aria2_command} {url} -o repo.zip')
 
         get_ipython().system('unzip -q -o repo.zip -d {webui_path}')
         get_ipython().system('rm -rf repo.zip')
 
+        # other
         get_ipython().run_line_magic('cd', '{root_path}')
         os.environ["SAFETENSORS_FAST_GPU"]='1'
         os.environ["CUDA_MODULE_LOADING"]="LAZY"
@@ -141,6 +171,11 @@ if not os.path.exists(webui_path):
     del cap
     install_time = timedelta(seconds=time.time()-start_install)
     print("\r🚀 Unpacking is complete! For","%02d:%02d:%02d ⚡\n" % (install_time.seconds / 3600, (install_time.seconds / 60) % 60, install_time.seconds % 60), end='', flush=True)
+
+    # Move temporary files
+    if os.path.exists(temporarily_dir):
+        copy_items_with_replace(temporarily_dir, webui_path)
+        shutil.rmtree(temporarily_dir)
 else:
     print("🚀 All unpacked... Skip. ⚡")
     start_colab = float(open(f'{webui_path}/static/colabTimer.txt', 'r').read())

files_cells/python/ru/downloading_ru.py

@@ -100,19 +100,48 @@ VARIABLES = [
 locals().update({key: settings.get(key) for key in VARIABLES})
 
 
-# ================= OTHER =================
+# =================== OTHER ===================
+# Setup Timer
 try:
     start_colab
 except:
     start_colab = int(time.time())-5
 
+# Save files temporarily
+temporarily_dir = f'{root_path}/temp_dir'
+
+def copy_items_with_replace(src_base, dst_base):
+    items_to_copy = [
+        'embeddings',
+        'models/Stable-diffusion',
+        'models/VAE',
+        'models/Lora'
+    ]
+
+    print("- Идет перемещение файлов... (Это может занять минуту или больше)", end='')
+    for item in items_to_copy:
+        src = os.path.join(src_base, item)
+        dst = os.path.join(dst_base, item)
+
+        if os.path.isdir(src):
+            if os.path.exists(dst):
+                shutil.rmtree(dst)
+            shutil.move(src, dst)
+        elif os.path.isfile(src):
+            dst_dir = os.path.dirname(dst)
+            os.makedirs(dst_dir, exist_ok=True)
+            shutil.move(src, dst)
+    print("\r🔥 Файлы были перемещены!")
+
 
 # ================= MAIN CODE =================
 # WebUI switch
 if os.path.exists(webui_path):
     if UI != OLD_UI:
         print(f'Переключение веб-интерфейса с \033[33m{OLD_UI}\033[0m на \033[33m{UI}\033[0m...', end='')
+        copy_items_with_replace(webui_path, temporarily_dir)
         shutil.rmtree(webui_path)
+
         # Updating the environment variable after a change
         os.environ['SDW_OLD_UI'] = UI
         print('\rГотово!')
@@ -122,15 +151,16 @@ if os.path.exists(webui_path):
 # Unpacking the WebUI
 if not os.path.exists(webui_path):
     start_install = int(time.time())
-    print("⌚ Распаковка Stable Diffusion..." if UI != 'Forge' else "⌚ Распаковка Stable Diffusion (Forge)...", end='')
+    print("⌚ Распаковка Stable Diffusion..." if change_webui != 'Forge' else "⌚ Распаковка Stable Diffusion (Forge)...", end='')
     with capture.capture_output() as cap:
         aria2_command = "aria2c --console-log-level=error -c -x 16 -s 16 -k 1M"
-        url = "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip" if UI != 'Forge' else "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip"
+        url = "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip" if change_webui != 'Forge' else "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip"
         get_ipython().system('{aria2_command} {url} -o repo.zip')
 
         get_ipython().system('unzip -q -o repo.zip -d {webui_path}')
         get_ipython().system('rm -rf repo.zip')
 
+        # other
         get_ipython().run_line_magic('cd', '{root_path}')
         os.environ["SAFETENSORS_FAST_GPU"]='1'
         os.environ["CUDA_MODULE_LOADING"]="LAZY"
@@ -141,6 +171,11 @@ if not os.path.exists(webui_path):
     del cap
     install_time = timedelta(seconds=time.time()-start_install)
     print("\r🚀 Распаковка Завершена! За","%02d:%02d:%02d ⚡\n" % (install_time.seconds / 3600, (install_time.seconds / 60) % 60, install_time.seconds % 60), end='', flush=True)
+
+    # Move temporary files
+    if os.path.exists(temporarily_dir):
+        copy_items_with_replace(temporarily_dir, webui_path)
+        shutil.rmtree(temporarily_dir)
 else:
     print("🚀 Все распакованно... Пропуск. ⚡")
     start_colab = float(open(f'{webui_path}/static/colabTimer.txt', 'r').read())