hz

files_cells/notebooks/en/auto_cleaner_en.ipynb

@@ -23,8 +23,7 @@
         "\n",
         "import os\n",
         "import time\n",
-        "import ipywidgets as widgets\n",
-        "from ipywidgets import Label, Button, VBox, HBox\n",
+        "from ipywidgets import widgets\n",
         "from IPython.display import display, HTML, Javascript\n",
         "\n",
         "\n",
@@ -53,7 +52,7 @@
         "}\n",
         "\n",
         "\"\"\" functions \"\"\"\n",
-        "def clean_directory(directory):\n",
+        "def clean_directory(directory, directory_type):\n",
         "    deleted_files = 0\n",
         "\n",
         "    for root, dirs, files in os.walk(directory):\n",
@@ -62,7 +61,7 @@
         "\n",
         "            if file.endswith(\".txt\"):\n",
         "                continue\n",
-        "            if file.endswith((\".safetensors\", \".pt\", \".png\", \".jpg\", \".jpeg\")):\n",
+        "            if file.endswith((\".safetensors\", \".pt\")) or directory_type == \"Images\":\n",
         "                deleted_files += 1\n",
         "\n",
         "            os.remove(file_path)\n",
@@ -84,7 +83,7 @@
         "\n",
         "    for option in selected_cleaners:\n",
         "        if option in directories:\n",
-        "            deleted_files_dict[option] = clean_directory(directories[option])\n",
+        "            deleted_files_dict[option] = clean_directory(directories[option], option)\n",
         "\n",
         "    output.clear_output()\n",
         "\n",

files_cells/notebooks/en/downloading_en.ipynb

@@ -39,6 +39,7 @@
         "\n",
         "os.chdir(root_path)\n",
         "\n",
+        "\n",
         "# ============ loading settings V4 =============\n",
         "def load_settings(path):\n",
         "    if os.path.exists(path):\n",
@@ -94,11 +95,10 @@
         "        subprocess.run(install_cmd, shell=True, capture_output=True)\n",
         "\n",
         "    # Additional specific packages\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        !curl -s -OL https://github.com/DEX-1101/sd-webui-notebook/raw/main/res/new_tunnel --output-dir {root_path}\n",
         "        !curl -s -Lo /usr/bin/cl https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 && chmod +x /usr/bin/cl\n",
         "        !curl -sLO https://github.com/openziti/zrok/releases/download/v0.4.32/zrok_0.4.32_linux_amd64.tar.gz && tar -xzf zrok_0.4.32_linux_amd64.tar.gz -C /usr/bin && rm -f zrok_0.4.32_linux_amd64.tar.gz\n",
-        "    del cap\n",
         "\n",
         "    clear_output()\n",
         "\n",
@@ -113,23 +113,33 @@
         "\n",
         "# =================== OTHER ====================\n",
         "# Setup Timer\n",
-        "if \"START_COLAB\" in os.environ:\n",
-        "    start_colab = int(os.environ[\"START_COLAB\"])\n",
-        "else:\n",
-        "    start_colab = int(time.time()) - 5\n",
-        "    os.environ[\"START_COLAB\"] = str(start_colab)\n",
-        "\n",
-        "# remove directory func\n",
-        "def _remove_dir(directory_path):\n",
+        "start_colab = int(os.environ.get(\"START_COLAB\", time.time() - 5))\n",
+        "os.environ[\"START_COLAB\"] = str(start_colab)\n",
+        "\n",
+        "def download_cfg_files(file_paths, destination_path):\n",
+        "    base_url = \"https://huggingface.co/NagisaNao/SD-CONFIGS/resolve/main\"\n",
+        "    for filename in file_paths:\n",
+        "        file_name = filename.split('/')[-1]\n",
+        "        !wget -O {destination_path}/{file_name} {base_url}/{filename}\n",
+        "\n",
+        "def cfg_download():\n",
+        "    common_files = [\"styles.csv\"]\n",
+        "    a1111_files = [\"A1111/config.json\", \"A1111/ui-config.json\"]\n",
+        "    forge_files = [\"reForge/config.json\", \"reForge/ui-config.json\"]\n",
+        "\n",
+        "    with capture.capture_output():\n",
+        "        download_cfg_files(common_files, webui_path)\n",
+        "        ui_files = a1111_files if UI == 'A1111' else forge_files\n",
+        "        download_cfg_files(ui_files, webui_path)\n",
+        "\n",
+        "def remove_dir(directory_path):\n",
         "    if directory_path and os.path.exists(directory_path):\n",
         "        try:\n",
         "            shutil.rmtree(directory_path)\n",
         "        except Exception:\n",
         "            !rm -rf {directory_path}\n",
         "\n",
-        "# Save files temporarily\n",
-        "temporarily_dir = f'{root_path}/temp_dir'\n",
-        "\n",
+        "TEMPORARY_DIR = f'{root_path}/temp_dir'\n",
         "def copy_items_with_replace(src_base, dst_base):\n",
         "    items_to_copy = [\n",
         "        'embeddings',\n",
@@ -140,14 +150,13 @@
         "    ]\n",
         "\n",
         "    print(\"⌚ Moving files...\", end='')\n",
-        "    time.sleep(1)\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.exists(src):\n",
         "            if os.path.exists(dst):\n",
-        "                _remove_dir(dst)\n",
+        "                remove_dir(dst)\n",
         "            os.makedirs(os.path.dirname(dst), exist_ok=True)\n",
         "            shutil.move(src, dst)\n",
         "    print(\"\\r🔥 Files moved!\" + \" \"*15)\n",
@@ -166,40 +175,39 @@
         "    start_install = time.time()\n",
         "    print(f\"⌚ Unpacking Stable Diffusion{' (Forge)' if UI == 'Forge' else ''}...\", end='')\n",
         "\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        download_url = \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip\"\n",
         "        if UI == 'Forge':\n",
         "            download_url = \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip\"\n",
         "\n",
-        "        aria2_args = \"--console-log-level=error -c -x 16 -s 16 -k 1M\"\n",
         "        zip_path = f\"{root_path}/repo.zip\"\n",
-        "        !aria2c {aria2_args} {download_url} -d {root_path} -o repo.zip\n",
+        "        !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {download_url} -d {root_path} -o repo.zip\n",
         "        !unzip -q -o {zip_path} -d {webui_path}\n",
         "        !rm -rf {zip_path}\n",
         "\n",
-        "        get_ipython().system(f'echo -n {start_colab} > {webui_path}/static/colabTimer.txt')\n",
-        "    del cap\n",
+        "        !echo -n {start_colab} > {webui_path}/static/colabTimer.txt\n",
         "\n",
         "    install_time = time.time() - start_install\n",
         "    minutes, seconds = divmod(int(install_time), 60)\n",
         "    print(f\"\\r🚀 Unpacking complete! For {minutes:02}:{seconds:02} ⚡\" + \" \"*15)\n",
         "\n",
-        "    if os.path.exists(temporarily_dir):\n",
-        "        copy_items_with_replace(temporarily_dir, webui_path)\n",
-        "        _remove_dir(temporarily_dir)\n",
+        "    if os.path.exists(TEMPORARY_DIR):\n",
+        "        copy_items_with_replace(TEMPORARY_DIR, webui_path)\n",
+        "        remove_dir(TEMPORARY_DIR)\n",
         "\n",
         "# ================= MAIN CODE ==================\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:')\n",
-        "        copy_items_with_replace(webui_path, temporarily_dir)\n",
-        "        _remove_dir(webui_path)\n",
+        "        copy_items_with_replace(webui_path, TEMPORARY_DIR)\n",
+        "        remove_dir(webui_path)\n",
         "        os.environ['SDW_OLD_UI'] = UI\n",
         "        time.sleep(2)\n",
         "        clear_output()\n",
         "\n",
         "if not os.path.exists(webui_path):\n",
         "    unpack_webui()\n",
+        "    cfg_download()\n",
         "else:\n",
         "    print(\"🚀 All unpacked... Skip. ⚡\")\n",
         "    timer_colab = handle_colab_timer(webui_path, start_colab)\n",
@@ -211,7 +219,7 @@
         "if latest_webui or latest_exstensions:\n",
         "    action = \"WebUI and Extensions\" if latest_webui and latest_exstensions else (\"WebUI\" if latest_webui else \"Extensions\")\n",
         "    print(f\"⌚️ Updating {action}...\", end='')\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        !git config --global user.email \"[email protected]\"\n",
         "        !git config --global user.name \"Your Name\"\n",
         "\n",
@@ -224,30 +232,26 @@
         "        ## Update extensions\n",
         "        if latest_exstensions:\n",
         "            !{'for dir in ' + webui_path + '/extensions/*/; do cd \\\"$dir\\\" && git reset --hard && git pull; done'}\n",
-        "    del cap\n",
         "    print(f\"\\r✨ Updating {action} Completed!\")\n",
         "\n",
         "\n",
         "# === FIXING EXTENSIONS ===\n",
         "anxety_repos = \"https://huggingface.co/NagisaNao/fast_repo/resolve/main\"\n",
-        "\n",
-        "with capture.capture_output() as cap:\n",
+        "with capture.capture_output():\n",
         "    # --- Umi-Wildcard ---\n",
         "    !sed -i '521s/open=\\(False\\|True\\)/open=False/' {webui_path}/extensions/Umi-AI-Wildcards/scripts/wildcard_recursive.py  # Closed accordion by default\n",
         "    # --- Encrypt-Image ---\n",
         "    !sed -i '9,37d' {webui_path}/extensions/Encrypt-Image/javascript/encrypt_images_info.js # Removes the weird text in webui\n",
-        "del cap\n",
         "\n",
         "\n",
         "## Version switching\n",
         "if commit_hash:\n",
         "    print('⏳ Time machine activation...', end=\"\")\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        %cd {webui_path}\n",
         "        !git config --global user.email \"[email protected]\"\n",
         "        !git config --global user.name \"Your Name\"\n",
         "        !git reset --hard {commit_hash}\n",
-        "    del cap\n",
         "    print(f\"\\r⌛️ The time machine has been activated! Current commit: \\033[34m{commit_hash}\\033[0m\")\n",
         "\n",
         "\n",
@@ -575,9 +579,8 @@
         "    download(url)\n",
         "    print(\"\\n\\033[33m# =============================== #\\n\\033[0m\")\n",
         "else:\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        download(url)\n",
-        "    del cap\n",
         "\n",
         "print(\"\\r🏁 Download Complete!\" + \" \"*15)\n",
         "\n",
@@ -589,7 +592,7 @@
         "## Install of Custom extensions\n",
         "if len(extension_repo) > 0:\n",
         "    print(\"✨ Installing custom extensions...\", end='')\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        for repo, repo_name in extension_repo:\n",
         "            if not repo_name:\n",
         "                repo_name = repo.split('/')[-1]\n",
@@ -597,7 +600,6 @@
         "                && git clone {repo} {repo_name} \\\n",
         "                && cd {repo_name} \\\n",
         "                && git fetch\n",
-        "    del cap\n",
         "    print(f\"\\r📦 Installed '{len(extension_repo)}', Custom extensions!\")\n",
         "\n",
         "\n",

files_cells/notebooks/en/widgets_en.ipynb

@@ -27,8 +27,7 @@
         "import os\n",
         "import json\n",
         "import time\n",
-        "import ipywidgets as widgets\n",
-        "from ipywidgets import widgets, Layout, Label, Button, VBox, HBox\n",
+        "from ipywidgets import widgets\n",
         "from IPython.display import display, HTML, Javascript, clear_output\n",
         "\n",
         "\n",

files_cells/notebooks/ru/auto_cleaner_ru.ipynb

@@ -28,8 +28,7 @@
         "\n",
         "import os\n",
         "import time\n",
-        "import ipywidgets as widgets\n",
-        "from ipywidgets import Label, Button, VBox, HBox\n",
+        "from ipywidgets import widgets\n",
         "from IPython.display import display, HTML\n",
         "\n",
         "\n",
@@ -58,7 +57,7 @@
         "}\n",
         "\n",
         "\"\"\" functions \"\"\"\n",
-        "def clean_directory(directory):\n",
+        "def clean_directory(directory, directory_type):\n",
         "    deleted_files = 0\n",
         "\n",
         "    for root, dirs, files in os.walk(directory):\n",
@@ -67,7 +66,7 @@
         "\n",
         "            if file.endswith(\".txt\"):\n",
         "                continue\n",
-        "            if file.endswith((\".safetensors\", \".pt\", \".png\", \".jpg\", \".jpeg\")):\n",
+        "            if file.endswith((\".safetensors\", \".pt\")) or directory_type == \"Images\":\n",
         "                deleted_files += 1\n",
         "\n",
         "            os.remove(file_path)\n",
@@ -89,7 +88,7 @@
         "\n",
         "    for option in selected_cleaners:\n",
         "        if option in directories:\n",
-        "            deleted_files_dict[option] = clean_directory(directories[option])\n",
+        "            deleted_files_dict[option] = clean_directory(directories[option], option)\n",
         "\n",
         "    output.clear_output()\n",
         "\n",

files_cells/notebooks/ru/downloading_ru.ipynb

@@ -39,6 +39,7 @@
         "\n",
         "os.chdir(root_path)\n",
         "\n",
+        "\n",
         "# ============ loading settings V4 =============\n",
         "def load_settings(path):\n",
         "    if os.path.exists(path):\n",
@@ -94,11 +95,10 @@
         "        subprocess.run(install_cmd, shell=True, capture_output=True)\n",
         "\n",
         "    # Additional specific packages\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output()\n",
         "        !curl -s -OL https://github.com/DEX-1101/sd-webui-notebook/raw/main/res/new_tunnel --output-dir {root_path}\n",
         "        !curl -s -Lo /usr/bin/cl https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 && chmod +x /usr/bin/cl\n",
         "        !curl -sLO https://github.com/openziti/zrok/releases/download/v0.4.32/zrok_0.4.32_linux_amd64.tar.gz && tar -xzf zrok_0.4.32_linux_amd64.tar.gz -C /usr/bin && rm -f zrok_0.4.32_linux_amd64.tar.gz\n",
-        "    del cap\n",
         "\n",
         "    clear_output()\n",
         "\n",
@@ -113,23 +113,33 @@
         "\n",
         "# =================== OTHER ====================\n",
         "# Setup Timer\n",
-        "if \"START_COLAB\" in os.environ:\n",
-        "    start_colab = int(os.environ[\"START_COLAB\"])\n",
-        "else:\n",
-        "    start_colab = int(time.time()) - 5\n",
-        "    os.environ[\"START_COLAB\"] = str(start_colab)\n",
-        "\n",
-        "# remove directory func\n",
-        "def _remove_dir(directory_path):\n",
+        "start_colab = int(os.environ.get(\"START_COLAB\", time.time() - 5))\n",
+        "os.environ[\"START_COLAB\"] = str(start_colab)\n",
+        "\n",
+        "def download_cfg_files(file_paths, destination_path):\n",
+        "    base_url = \"https://huggingface.co/NagisaNao/SD-CONFIGS/resolve/main\"\n",
+        "    for filename in file_paths:\n",
+        "        file_name = filename.split('/')[-1]\n",
+        "        !wget -O {destination_path}/{file_name} {base_url}/{filename}\n",
+        "\n",
+        "def cfg_download():\n",
+        "    common_files = [\"styles.csv\"]\n",
+        "    a1111_files = [\"A1111/config.json\", \"A1111/ui-config.json\"]\n",
+        "    forge_files = [\"reForge/config.json\", \"reForge/ui-config.json\"]\n",
+        "\n",
+        "    with capture.capture_output():\n",
+        "        download_cfg_files(common_files, webui_path)\n",
+        "        ui_files = a1111_files if UI == 'A1111' else forge_files\n",
+        "        download_cfg_files(ui_files, webui_path)\n",
+        "\n",
+        "def remove_dir(directory_path):\n",
         "    if directory_path and os.path.exists(directory_path):\n",
         "        try:\n",
         "            shutil.rmtree(directory_path)\n",
         "        except Exception:\n",
         "            !rm -rf {directory_path}\n",
         "\n",
-        "# Save files temporarily\n",
-        "temporarily_dir = f'{root_path}/temp_dir'\n",
-        "\n",
+        "TEMPORARY_DIR = f'{root_path}/temp_dir'\n",
         "def copy_items_with_replace(src_base, dst_base):\n",
         "    items_to_copy = [\n",
         "        'embeddings',\n",
@@ -140,14 +150,13 @@
         "    ]\n",
         "\n",
         "    print(\"⌚ Перемещение файлов...\", end='')\n",
-        "    time.sleep(1)\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.exists(src):\n",
         "            if os.path.exists(dst):\n",
-        "                _remove_dir(dst)\n",
+        "                remove_dir(dst)\n",
         "            os.makedirs(os.path.dirname(dst), exist_ok=True)\n",
         "            shutil.move(src, dst)\n",
         "    print(\"\\r🔥 Файлы перемещены!\" + \" \"*15)\n",
@@ -166,40 +175,39 @@
         "    start_install = time.time()\n",
         "    print(f\"⌚ Распаковка Stable Diffusion{' (Forge)' if UI == 'Forge' else ''}...\", end='')\n",
         "\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        download_url = \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip\"\n",
         "        if UI == 'Forge':\n",
         "            download_url = \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip\"\n",
         "\n",
-        "        aria2_args = \"--console-log-level=error -c -x 16 -s 16 -k 1M\"\n",
         "        zip_path = f\"{root_path}/repo.zip\"\n",
-        "        !aria2c {aria2_args} {download_url} -d {root_path} -o repo.zip\n",
+        "        !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {download_url} -d {root_path} -o repo.zip\n",
         "        !unzip -q -o {zip_path} -d {webui_path}\n",
         "        !rm -rf {zip_path}\n",
         "\n",
-        "        get_ipython().system(f'echo -n {start_colab} > {webui_path}/static/colabTimer.txt')\n",
-        "    del cap\n",
+        "        !echo -n {start_colab} > {webui_path}/static/colabTimer.txt\n",
         "\n",
         "    install_time = time.time() - start_install\n",
         "    minutes, seconds = divmod(int(install_time), 60)\n",
         "    print(f\"\\r🚀 Распаковка Завершена! За {minutes:02}:{seconds:02} ⚡\" + \" \"*15)\n",
         "\n",
-        "    if os.path.exists(temporarily_dir):\n",
-        "        copy_items_with_replace(temporarily_dir, webui_path)\n",
-        "        _remove_dir(temporarily_dir)\n",
+        "    if os.path.exists(TEMPORARY_DIR):\n",
+        "        copy_items_with_replace(TEMPORARY_DIR, webui_path)\n",
+        "        remove_dir(TEMPORARY_DIR)\n",
         "\n",
         "# ================= MAIN CODE ==================\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:')\n",
-        "        copy_items_with_replace(webui_path, temporarily_dir)\n",
-        "        _remove_dir(webui_path)\n",
+        "        copy_items_with_replace(webui_path, TEMPORARY_DIR)\n",
+        "        remove_dir(webui_path)\n",
         "        os.environ['SDW_OLD_UI'] = UI\n",
         "        time.sleep(2)\n",
         "        clear_output()\n",
         "\n",
         "if not os.path.exists(webui_path):\n",
         "    unpack_webui()\n",
+        "    cfg_download()\n",
         "else:\n",
         "    print(\"🚀 Все распакованно... Пропуск. ⚡\")\n",
         "    timer_colab = handle_colab_timer(webui_path, start_colab)\n",
@@ -211,7 +219,7 @@
         "if latest_webui or latest_exstensions:\n",
         "    action = \"WebUI и Расширений\" if latest_webui and latest_exstensions else (\"WebUI\" if latest_webui else \"Расширений\")\n",
         "    print(f\"⌚️ Обновление {action}...\", end='')\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        !git config --global user.email \"[email protected]\"\n",
         "        !git config --global user.name \"Your Name\"\n",
         "\n",
@@ -224,30 +232,26 @@
         "        ## Update extensions\n",
         "        if latest_exstensions:\n",
         "            !{'for dir in ' + webui_path + '/extensions/*/; do cd \\\"$dir\\\" && git reset --hard && git pull; done'}\n",
-        "    del cap\n",
         "    print(f\"\\r✨ Обновление {action} Завершено!\")\n",
         "\n",
         "\n",
         "# === FIXING EXTENSIONS ===\n",
         "anxety_repos = \"https://huggingface.co/NagisaNao/fast_repo/resolve/main\"\n",
-        "\n",
-        "with capture.capture_output() as cap:\n",
+        "with capture.capture_output():\n",
         "    # --- Umi-Wildcard ---\n",
         "    !sed -i '521s/open=\\(False\\|True\\)/open=False/' {webui_path}/extensions/Umi-AI-Wildcards/scripts/wildcard_recursive.py  # Closed accordion by default\n",
         "    # --- Encrypt-Image ---\n",
         "    !sed -i '9,37d' {webui_path}/extensions/Encrypt-Image/javascript/encrypt_images_info.js # Removes the weird text in webui\n",
-        "del cap\n",
         "\n",
         "\n",
         "## Version switching\n",
         "if commit_hash:\n",
         "    print('⏳ Активация машины времени...', end=\"\")\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        %cd {webui_path}\n",
         "        !git config --global user.email \"[email protected]\"\n",
         "        !git config --global user.name \"Your Name\"\n",
         "        !git reset --hard {commit_hash}\n",
-        "    del cap\n",
         "    print(f\"\\r⌛️ Машина времени активированна! Текущий коммит: \\033[34m{commit_hash}\\033[0m\")\n",
         "\n",
         "\n",
@@ -575,9 +579,8 @@
         "    download(url)\n",
         "    print(\"\\n\\033[33m# =============================== #\\n\\033[0m\")\n",
         "else:\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        download(url)\n",
-        "    del cap\n",
         "\n",
         "print(\"\\r🏁 Скачивание Завершено!\" + \" \"*15)\n",
         "\n",
@@ -589,7 +592,7 @@
         "## Install of Custom extensions\n",
         "if len(extension_repo) > 0:\n",
         "    print(\"✨ Установка кастомных расширений...\", end='')\n",
-        "    with capture.capture_output() as cap:\n",
+        "    with capture.capture_output():\n",
         "        for repo, repo_name in extension_repo:\n",
         "            if not repo_name:\n",
         "                repo_name = repo.split('/')[-1]\n",
@@ -597,7 +600,6 @@
         "                && git clone {repo} {repo_name} \\\n",
         "                && cd {repo_name} \\\n",
         "                && git fetch\n",
-        "    del cap\n",
         "    print(f\"\\r📦 Установлено '{len(extension_repo)}', Кастомных расширений!\")\n",
         "\n",
         "\n",

files_cells/notebooks/ru/widgets_ru.ipynb

@@ -27,8 +27,7 @@
         "import os\n",
         "import json\n",
         "import time\n",
-        "import ipywidgets as widgets\n",
-        "from ipywidgets import widgets, Layout, Label, Button, VBox, HBox\n",
+        "from ipywidgets import widgets\n",
         "from IPython.display import display, HTML, Javascript, clear_output\n",
         "\n",
         "\n",

files_cells/python/en/auto_cleaner_en.py

@@ -4,8 +4,7 @@ from directory_setup import models_dir, vaes_dir, control_dir, loras_dir, output
 
 import os
 import time
-import ipywidgets as widgets
-from ipywidgets import Label, Button, VBox, HBox
+from ipywidgets import widgets
 from IPython.display import display, HTML, Javascript
 
 
@@ -34,7 +33,7 @@ directories = {
 }
 
 """ functions """
-def clean_directory(directory):
+def clean_directory(directory, directory_type):
     deleted_files = 0
 
     for root, dirs, files in os.walk(directory):
@@ -43,7 +42,7 @@ def clean_directory(directory):
 
             if file.endswith(".txt"):
                 continue
-            if file.endswith((".safetensors", ".pt", ".png", ".jpg", ".jpeg")):
+            if file.endswith((".safetensors", ".pt")) or directory_type == "Images":
                 deleted_files += 1
 
             os.remove(file_path)
@@ -65,7 +64,7 @@ def on_execute_button_press(button):
 
     for option in selected_cleaners:
         if option in directories:
-            deleted_files_dict[option] = clean_directory(directories[option])
+            deleted_files_dict[option] = clean_directory(directories[option], option)
 
     output.clear_output()
 

files_cells/python/en/downloading_en.py

@@ -29,6 +29,7 @@ OLD_UI = os.getenv('SDW_OLD_UI')
 
 os.chdir(root_path)
 
+
 # ============ loading settings V4 =============
 def load_settings(path):
     if os.path.exists(path):
@@ -84,11 +85,10 @@ if not os.path.exists(flag_file):
         subprocess.run(install_cmd, shell=True, capture_output=True)
 
     # Additional specific packages
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         get_ipython().system('curl -s -OL https://github.com/DEX-1101/sd-webui-notebook/raw/main/res/new_tunnel --output-dir {root_path}')
         get_ipython().system('curl -s -Lo /usr/bin/cl https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 && chmod +x /usr/bin/cl')
         get_ipython().system('curl -sLO https://github.com/openziti/zrok/releases/download/v0.4.32/zrok_0.4.32_linux_amd64.tar.gz && tar -xzf zrok_0.4.32_linux_amd64.tar.gz -C /usr/bin && rm -f zrok_0.4.32_linux_amd64.tar.gz')
-    del cap
 
     clear_output()
 
@@ -103,23 +103,33 @@ if not os.path.exists(flag_file):
 
 # =================== OTHER ====================
 # Setup Timer
-if "START_COLAB" in os.environ:
-    start_colab = int(os.environ["START_COLAB"])
-else:
-    start_colab = int(time.time()) - 5
-    os.environ["START_COLAB"] = str(start_colab)
-
-# remove directory func
-def _remove_dir(directory_path):
+start_colab = int(os.environ.get("START_COLAB", time.time() - 5))
+os.environ["START_COLAB"] = str(start_colab)
+
+def download_cfg_files(file_paths, destination_path):
+    base_url = "https://huggingface.co/NagisaNao/SD-CONFIGS/resolve/main"
+    for filename in file_paths:
+        file_name = filename.split('/')[-1]
+        get_ipython().system('wget -O {destination_path}/{file_name} {base_url}/{filename}')
+
+def cfg_download():
+    common_files = ["styles.csv"]
+    a1111_files = ["A1111/config.json", "A1111/ui-config.json"]
+    forge_files = ["reForge/config.json", "reForge/ui-config.json"]
+
+    with capture.capture_output():
+        download_cfg_files(common_files, webui_path)
+        ui_files = a1111_files if UI == 'A1111' else forge_files
+        download_cfg_files(ui_files, webui_path)
+
+def remove_dir(directory_path):
     if directory_path and os.path.exists(directory_path):
         try:
             shutil.rmtree(directory_path)
         except Exception:
             get_ipython().system('rm -rf {directory_path}')
 
-# Save files temporarily
-temporarily_dir = f'{root_path}/temp_dir'
-
+TEMPORARY_DIR = f'{root_path}/temp_dir'
 def copy_items_with_replace(src_base, dst_base):
     items_to_copy = [
         'embeddings',
@@ -130,14 +140,13 @@ def copy_items_with_replace(src_base, dst_base):
     ]
 
     print("⌚ Moving files...", end='')
-    time.sleep(1)
     for item in items_to_copy:
         src = os.path.join(src_base, item)
         dst = os.path.join(dst_base, item)
 
         if os.path.exists(src):
             if os.path.exists(dst):
-                _remove_dir(dst)
+                remove_dir(dst)
             os.makedirs(os.path.dirname(dst), exist_ok=True)
             shutil.move(src, dst)
     print("\r🔥 Files moved!" + " "*15)
@@ -156,40 +165,39 @@ def unpack_webui():
     start_install = time.time()
     print(f"⌚ Unpacking Stable Diffusion{' (Forge)' if UI == 'Forge' else ''}...", end='')
 
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         download_url = "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip"
         if UI == 'Forge':
             download_url = "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip"
 
-        aria2_args = "--console-log-level=error -c -x 16 -s 16 -k 1M"
         zip_path = f"{root_path}/repo.zip"
-        get_ipython().system('aria2c {aria2_args} {download_url} -d {root_path} -o repo.zip')
+        get_ipython().system('aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {download_url} -d {root_path} -o repo.zip')
         get_ipython().system('unzip -q -o {zip_path} -d {webui_path}')
         get_ipython().system('rm -rf {zip_path}')
 
-        get_ipython().system(f'echo -n {start_colab} > {webui_path}/static/colabTimer.txt')
-    del cap
+        get_ipython().system('echo -n {start_colab} > {webui_path}/static/colabTimer.txt')
 
     install_time = time.time() - start_install
     minutes, seconds = divmod(int(install_time), 60)
     print(f"\r🚀 Unpacking complete! For {minutes:02}:{seconds:02} ⚡" + " "*15)
 
-    if os.path.exists(temporarily_dir):
-        copy_items_with_replace(temporarily_dir, webui_path)
-        _remove_dir(temporarily_dir)
+    if os.path.exists(TEMPORARY_DIR):
+        copy_items_with_replace(TEMPORARY_DIR, webui_path)
+        remove_dir(TEMPORARY_DIR)
 
 # ================= MAIN CODE ==================
 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:')
-        copy_items_with_replace(webui_path, temporarily_dir)
-        _remove_dir(webui_path)
+        copy_items_with_replace(webui_path, TEMPORARY_DIR)
+        remove_dir(webui_path)
         os.environ['SDW_OLD_UI'] = UI
         time.sleep(2)
         clear_output()
 
 if not os.path.exists(webui_path):
     unpack_webui()
+    cfg_download()
 else:
     print("🚀 All unpacked... Skip. ⚡")
     timer_colab = handle_colab_timer(webui_path, start_colab)
@@ -201,7 +209,7 @@ else:
 if latest_webui or latest_exstensions:
     action = "WebUI and Extensions" if latest_webui and latest_exstensions else ("WebUI" if latest_webui else "Extensions")
     print(f"⌚️ Updating {action}...", end='')
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         get_ipython().system('git config --global user.email "[email protected]"')
         get_ipython().system('git config --global user.name "Your Name"')
 
@@ -214,30 +222,26 @@ if latest_webui or latest_exstensions:
         ## Update extensions
         if latest_exstensions:
             get_ipython().system('{\'for dir in \' + webui_path + \'/extensions/*/; do cd \\"$dir\\" && git reset --hard && git pull; done\'}')
-    del cap
     print(f"\r✨ Updating {action} Completed!")
 
 
 # === FIXING EXTENSIONS ===
 anxety_repos = "https://huggingface.co/NagisaNao/fast_repo/resolve/main"
-
-with capture.capture_output() as cap:
+with capture.capture_output():
     # --- Umi-Wildcard ---
     get_ipython().system("sed -i '521s/open=\\(False\\|True\\)/open=False/' {webui_path}/extensions/Umi-AI-Wildcards/scripts/wildcard_recursive.py  # Closed accordion by default")
     # --- Encrypt-Image ---
     get_ipython().system("sed -i '9,37d' {webui_path}/extensions/Encrypt-Image/javascript/encrypt_images_info.js # Removes the weird text in webui")
-del cap
 
 
 ## Version switching
 if commit_hash:
     print('⏳ Time machine activation...', end="")
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         get_ipython().run_line_magic('cd', '{webui_path}')
         get_ipython().system('git config --global user.email "[email protected]"')
         get_ipython().system('git config --global user.name "Your Name"')
         get_ipython().system('git reset --hard {commit_hash}')
-    del cap
     print(f"\r⌛️ The time machine has been activated! Current commit: \033[34m{commit_hash}\033[0m")
 
 
@@ -565,9 +569,8 @@ if detailed_download == "on":
     download(url)
     print("\n\033[33m# =============================== #\n\033[0m")
 else:
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         download(url)
-    del cap
 
 print("\r🏁 Download Complete!" + " "*15)
 
@@ -579,12 +582,11 @@ get_ipython().system('find {webui_path} \\( -type d \\( -name ".ipynb_checkpoint
 ## Install of Custom extensions
 if len(extension_repo) > 0:
     print("✨ Installing custom extensions...", end='')
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         for repo, repo_name in extension_repo:
             if not repo_name:
                 repo_name = repo.split('/')[-1]
             get_ipython().system('cd {extensions_dir}                  && git clone {repo} {repo_name}                  && cd {repo_name}                  && git fetch')
-    del cap
     print(f"\r📦 Installed '{len(extension_repo)}', Custom extensions!")
 
 

files_cells/python/en/widgets_en.py

@@ -3,8 +3,7 @@
 import os
 import json
 import time
-import ipywidgets as widgets
-from ipywidgets import widgets, Layout, Label, Button, VBox, HBox
+from ipywidgets import widgets
 from IPython.display import display, HTML, Javascript, clear_output
 
 

files_cells/python/ru/auto_cleaner_ru.py

@@ -4,8 +4,7 @@ from directory_setup import models_dir, vaes_dir, control_dir, loras_dir, output
 
 import os
 import time
-import ipywidgets as widgets
-from ipywidgets import Label, Button, VBox, HBox
+from ipywidgets import widgets
 from IPython.display import display, HTML
 
 
@@ -34,7 +33,7 @@ directories = {
 }
 
 """ functions """
-def clean_directory(directory):
+def clean_directory(directory, directory_type):
     deleted_files = 0
 
     for root, dirs, files in os.walk(directory):
@@ -43,7 +42,7 @@ def clean_directory(directory):
 
             if file.endswith(".txt"):
                 continue
-            if file.endswith((".safetensors", ".pt", ".png", ".jpg", ".jpeg")):
+            if file.endswith((".safetensors", ".pt")) or directory_type == "Images":
                 deleted_files += 1
 
             os.remove(file_path)
@@ -65,7 +64,7 @@ def on_execute_button_press(button):
 
     for option in selected_cleaners:
         if option in directories:
-            deleted_files_dict[option] = clean_directory(directories[option])
+            deleted_files_dict[option] = clean_directory(directories[option], option)
 
     output.clear_output()
 

files_cells/python/ru/downloading_ru.py

@@ -29,6 +29,7 @@ OLD_UI = os.getenv('SDW_OLD_UI')
 
 os.chdir(root_path)
 
+
 # ============ loading settings V4 =============
 def load_settings(path):
     if os.path.exists(path):
@@ -84,11 +85,10 @@ if not os.path.exists(flag_file):
         subprocess.run(install_cmd, shell=True, capture_output=True)
 
     # Additional specific packages
-    with capture.capture_output() as cap:
+    with capture.capture_output()
         get_ipython().system('curl -s -OL https://github.com/DEX-1101/sd-webui-notebook/raw/main/res/new_tunnel --output-dir {root_path}')
         get_ipython().system('curl -s -Lo /usr/bin/cl https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-amd64 && chmod +x /usr/bin/cl')
         get_ipython().system('curl -sLO https://github.com/openziti/zrok/releases/download/v0.4.32/zrok_0.4.32_linux_amd64.tar.gz && tar -xzf zrok_0.4.32_linux_amd64.tar.gz -C /usr/bin && rm -f zrok_0.4.32_linux_amd64.tar.gz')
-    del cap
 
     clear_output()
 
@@ -103,23 +103,33 @@ if not os.path.exists(flag_file):
 
 # =================== OTHER ====================
 # Setup Timer
-if "START_COLAB" in os.environ:
-    start_colab = int(os.environ["START_COLAB"])
-else:
-    start_colab = int(time.time()) - 5
-    os.environ["START_COLAB"] = str(start_colab)
-
-# remove directory func
-def _remove_dir(directory_path):
+start_colab = int(os.environ.get("START_COLAB", time.time() - 5))
+os.environ["START_COLAB"] = str(start_colab)
+
+def download_cfg_files(file_paths, destination_path):
+    base_url = "https://huggingface.co/NagisaNao/SD-CONFIGS/resolve/main"
+    for filename in file_paths:
+        file_name = filename.split('/')[-1]
+        get_ipython().system('wget -O {destination_path}/{file_name} {base_url}/{filename}')
+
+def cfg_download():
+    common_files = ["styles.csv"]
+    a1111_files = ["A1111/config.json", "A1111/ui-config.json"]
+    forge_files = ["reForge/config.json", "reForge/ui-config.json"]
+
+    with capture.capture_output():
+        download_cfg_files(common_files, webui_path)
+        ui_files = a1111_files if UI == 'A1111' else forge_files
+        download_cfg_files(ui_files, webui_path)
+
+def remove_dir(directory_path):
     if directory_path and os.path.exists(directory_path):
         try:
             shutil.rmtree(directory_path)
         except Exception:
             get_ipython().system('rm -rf {directory_path}')
 
-# Save files temporarily
-temporarily_dir = f'{root_path}/temp_dir'
-
+TEMPORARY_DIR = f'{root_path}/temp_dir'
 def copy_items_with_replace(src_base, dst_base):
     items_to_copy = [
         'embeddings',
@@ -130,14 +140,13 @@ def copy_items_with_replace(src_base, dst_base):
     ]
 
     print("⌚ Перемещение файлов...", end='')
-    time.sleep(1)
     for item in items_to_copy:
         src = os.path.join(src_base, item)
         dst = os.path.join(dst_base, item)
 
         if os.path.exists(src):
             if os.path.exists(dst):
-                _remove_dir(dst)
+                remove_dir(dst)
             os.makedirs(os.path.dirname(dst), exist_ok=True)
             shutil.move(src, dst)
     print("\r🔥 Файлы перемещены!" + " "*15)
@@ -156,40 +165,39 @@ def unpack_webui():
     start_install = time.time()
     print(f"⌚ Распаковка Stable Diffusion{' (Forge)' if UI == 'Forge' else ''}...", end='')
 
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         download_url = "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO.zip"
         if UI == 'Forge':
             download_url = "https://huggingface.co/NagisaNao/fast_repo/resolve/main/FULL_REPO_forge.zip"
 
-        aria2_args = "--console-log-level=error -c -x 16 -s 16 -k 1M"
         zip_path = f"{root_path}/repo.zip"
-        get_ipython().system('aria2c {aria2_args} {download_url} -d {root_path} -o repo.zip')
+        get_ipython().system('aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {download_url} -d {root_path} -o repo.zip')
         get_ipython().system('unzip -q -o {zip_path} -d {webui_path}')
         get_ipython().system('rm -rf {zip_path}')
 
-        get_ipython().system(f'echo -n {start_colab} > {webui_path}/static/colabTimer.txt')
-    del cap
+        get_ipython().system('echo -n {start_colab} > {webui_path}/static/colabTimer.txt')
 
     install_time = time.time() - start_install
     minutes, seconds = divmod(int(install_time), 60)
     print(f"\r🚀 Распаковка Завершена! За {minutes:02}:{seconds:02} ⚡" + " "*15)
 
-    if os.path.exists(temporarily_dir):
-        copy_items_with_replace(temporarily_dir, webui_path)
-        _remove_dir(temporarily_dir)
+    if os.path.exists(TEMPORARY_DIR):
+        copy_items_with_replace(TEMPORARY_DIR, webui_path)
+        remove_dir(TEMPORARY_DIR)
 
 # ================= MAIN CODE ==================
 if os.path.exists(webui_path):
     if UI != OLD_UI:
         print(f'Переключение веб-интерфейса с \033[33m{OLD_UI}\033[0m на \033[33m{UI}\033[0m:')
-        copy_items_with_replace(webui_path, temporarily_dir)
-        _remove_dir(webui_path)
+        copy_items_with_replace(webui_path, TEMPORARY_DIR)
+        remove_dir(webui_path)
         os.environ['SDW_OLD_UI'] = UI
         time.sleep(2)
         clear_output()
 
 if not os.path.exists(webui_path):
     unpack_webui()
+    cfg_download()
 else:
     print("🚀 Все распакованно... Пропуск. ⚡")
     timer_colab = handle_colab_timer(webui_path, start_colab)
@@ -201,7 +209,7 @@ else:
 if latest_webui or latest_exstensions:
     action = "WebUI и Расширений" if latest_webui and latest_exstensions else ("WebUI" if latest_webui else "Расширений")
     print(f"⌚️ Обновление {action}...", end='')
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         get_ipython().system('git config --global user.email "[email protected]"')
         get_ipython().system('git config --global user.name "Your Name"')
 
@@ -214,30 +222,26 @@ if latest_webui or latest_exstensions:
         ## Update extensions
         if latest_exstensions:
             get_ipython().system('{\'for dir in \' + webui_path + \'/extensions/*/; do cd \\"$dir\\" && git reset --hard && git pull; done\'}')
-    del cap
     print(f"\r✨ Обновление {action} Завершено!")
 
 
 # === FIXING EXTENSIONS ===
 anxety_repos = "https://huggingface.co/NagisaNao/fast_repo/resolve/main"
-
-with capture.capture_output() as cap:
+with capture.capture_output():
     # --- Umi-Wildcard ---
     get_ipython().system("sed -i '521s/open=\\(False\\|True\\)/open=False/' {webui_path}/extensions/Umi-AI-Wildcards/scripts/wildcard_recursive.py  # Closed accordion by default")
     # --- Encrypt-Image ---
     get_ipython().system("sed -i '9,37d' {webui_path}/extensions/Encrypt-Image/javascript/encrypt_images_info.js # Removes the weird text in webui")
-del cap
 
 
 ## Version switching
 if commit_hash:
     print('⏳ Активация машины времени...', end="")
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         get_ipython().run_line_magic('cd', '{webui_path}')
         get_ipython().system('git config --global user.email "[email protected]"')
         get_ipython().system('git config --global user.name "Your Name"')
         get_ipython().system('git reset --hard {commit_hash}')
-    del cap
     print(f"\r⌛️ Машина времени активированна! Текущий коммит: \033[34m{commit_hash}\033[0m")
 
 
@@ -565,9 +569,8 @@ if detailed_download == "on":
     download(url)
     print("\n\033[33m# =============================== #\n\033[0m")
 else:
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         download(url)
-    del cap
 
 print("\r🏁 Скачивание Завершено!" + " "*15)
 
@@ -579,12 +582,11 @@ get_ipython().system('find {webui_path} \\( -type d \\( -name ".ipynb_checkpoint
 ## Install of Custom extensions
 if len(extension_repo) > 0:
     print("✨ Установка кастомных расширений...", end='')
-    with capture.capture_output() as cap:
+    with capture.capture_output():
         for repo, repo_name in extension_repo:
             if not repo_name:
                 repo_name = repo.split('/')[-1]
             get_ipython().system('cd {extensions_dir}                  && git clone {repo} {repo_name}                  && cd {repo_name}                  && git fetch')
-    del cap
     print(f"\r📦 Установлено '{len(extension_repo)}', Кастомных расширений!")
 
 

files_cells/python/ru/widgets_ru.py

@@ -3,8 +3,7 @@
 import os
 import json
 import time
-import ipywidgets as widgets
-from ipywidgets import widgets, Layout, Label, Button, VBox, HBox
+from ipywidgets import widgets
 from IPython.display import display, HTML, Javascript, clear_output