✨ yay for modularity~

files_cells/notebooks/en/auto_cleaner_en.ipynb

@@ -1,165 +1,157 @@
-{
-  "nbformat": 4,
-  "nbformat_minor": 0,
-  "metadata": {
-    "colab": {
-      "provenance": []
-    },
-    "kernelspec": {
-      "name": "python3",
-      "display_name": "Python 3"
-    },
-    "language_info": {
-      "name": "python"
-    }
-  },
-  "cells": [
-    {
-      "cell_type": "code",
-      "source": [
-        "##~ AutoCleaner V3.6 CODE | BY: ANXETY   ~##\n",
-        "\n",
-        "import os\n",
-        "import time\n",
-        "import ipywidgets as widgets\n",
-        "from ipywidgets import Label, Button, VBox, HBox\n",
-        "from IPython.display import display, HTML, Javascript\n",
-        "\n",
-        "\n",
-        "# ================= DETECT ENV =================\n",
-        "def detect_environment():\n",
-        "    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)\n",
-        "    environments = {\n",
-        "        'COLAB_GPU': ('Google Colab', \"/root\" if free_plan else \"/content\"),\n",
-        "        'KAGGLE_URL_BASE': ('Kaggle', \"/kaggle/working/content\")\n",
-        "    }\n",
-        "    for env_var, (environment, path) in environments.items():\n",
-        "        if env_var in os.environ:\n",
-        "            return environment, path, free_plan\n",
-        "\n",
-        "env, root_path, free_plan = detect_environment()\n",
-        "webui_path = f\"{root_path}/sdw\"\n",
-        "\n",
-        "\n",
-        "# ==================== CSS ====================\n",
-        "# Main CSS\n",
-        "css_file_path = f\"{root_path}/CSS/auto_cleaner.css\"\n",
-        "with open(css_file_path , \"r\") as f:\n",
-        "    CSS_AC = f.read()\n",
-        "display(HTML(f\"<style>{CSS_AC}</style>\"))\n",
-        "\n",
-        "\n",
-        "# ================ AutoCleaner function ================\n",
-        "directories = {\n",
-        "    \"Images\": f\"{webui_path}/output\",\n",
-        "    \"Models\": f\"{webui_path}/models/Stable-diffusion/\",\n",
-        "    \"Vae\": f\"{webui_path}/models/VAE/\",\n",
-        "    \"LoRa\": f\"{webui_path}/models/Lora/\",\n",
-        "    \"ControlNet Models\": f\"{webui_path}/models/ControlNet/\"\n",
-        "}\n",
-        "\n",
-        "\"\"\" functions \"\"\"\n",
-        "def clean_directory(directory):\n",
-        "    deleted_files = 0\n",
-        "    image_dir = directories['Images']\n",
-        "\n",
-        "    for root, dirs, files in os.walk(directory):\n",
-        "        for file in files:\n",
-        "            file_path = os.path.join(root, file)\n",
-        "\n",
-        "            if file.endswith(\".txt\"):\n",
-        "                continue\n",
-        "            if file.endswith((\".safetensors\", \".pt\")) or root == image_dir: # fix for image counter\n",
-        "                deleted_files += 1\n",
-        "\n",
-        "            os.remove(file_path)\n",
-        "    return deleted_files\n",
-        "\n",
-        "def update_memory_info():\n",
-        "    disk_space = psutil.disk_usage(os.getcwd())\n",
-        "    total = disk_space.total / (1024 ** 3)\n",
-        "    used = disk_space.used / (1024 ** 3)\n",
-        "    free = disk_space.free / (1024 ** 3)\n",
-        "\n",
-        "    storage_info.value = f'''\n",
-        "    <div class=\"storage_info_AC\">Total storage: {total:.2f} GB <span style=\"color: #555\">|</span> Used: {used:.2f} GB <span style=\"color: #555\">|</span> Free: {free:.2f} GB</div>\n",
-        "    '''\n",
-        "\n",
-        "def on_execute_button_press(button):\n",
-        "    selected_cleaners = auto_cleaner_widget.value\n",
-        "    deleted_files_dict = {}\n",
-        "\n",
-        "    for option in selected_cleaners:\n",
-        "        if option in directories:\n",
-        "            deleted_files_dict[option] = clean_directory(directories[option])\n",
-        "\n",
-        "    output.clear_output()\n",
-        "\n",
-        "    with output:\n",
-        "        for message in generate_messages(deleted_files_dict):\n",
-        "            message_widget = HTML(f'<p class=\"output_message_AC\">{message}</p>')\n",
-        "            display(message_widget)\n",
-        "\n",
-        "    update_memory_info()\n",
-        "\n",
-        "def on_clear_button_press(button):\n",
-        "    container.add_class(\"hide\")\n",
-        "    time.sleep(0.5)\n",
-        "    widgets.Widget.close_all()\n",
-        "\n",
-        "def generate_messages(deleted_files_dict):\n",
-        "    messages = []\n",
-        "    word_variants = {\n",
-        "        \"Images\": \"Images\",\n",
-        "        \"Models\": \"Models\",\n",
-        "        \"Vae\": \"Vae\",\n",
-        "        \"LoRa\": \"LoRa\",\n",
-        "        \"ControlNet Models\": \"ControlNet Models\"\n",
-        "    }\n",
-        "    for key, value in deleted_files_dict.items():\n",
-        "        object_word = word_variants.get(key)\n",
-        "        messages.append(f\"Deleted {value} {object_word}\")\n",
-        "    return messages\n",
-        "# ================ AutoCleaner function ================\n",
-        "\n",
-        "\n",
-        "# --- storage memory ---\n",
-        "import psutil\n",
-        "disk_space = psutil.disk_usage(os.getcwd())\n",
-        "total = disk_space.total / (1024 ** 3)\n",
-        "used = disk_space.used / (1024 ** 3)\n",
-        "free = disk_space.free / (1024 ** 3)\n",
-        "\n",
-        "\n",
-        "# UI Code\n",
-        "AutoCleaner_options = AutoCleaner_options = list(directories.keys())\n",
-        "instruction_label = widgets.HTML('''\n",
-        "<span class=\"instruction_AC\">Use <span style=\"color: #B2B2B2;\">ctrl</span> or <span style=\"color: #B2B2B2;\">shift</span> for multiple selections.</span>\n",
-        "''')\n",
-        "auto_cleaner_widget = widgets.SelectMultiple(options=AutoCleaner_options, layout=widgets.Layout(width='auto')).add_class(\"custom-select-multiple_AC\")\n",
-        "output = widgets.Output().add_class(\"output_AC\")\n",
-        "# ---\n",
-        "execute_button = Button(description='Execute Cleaning').add_class(\"button_execute_AC\").add_class(\"button_AC\")\n",
-        "execute_button.on_click(on_execute_button_press)\n",
-        "clear_button = Button(description='Hide Widget').add_class(\"button_clear_AC\").add_class(\"button_AC\")\n",
-        "clear_button.on_click(on_clear_button_press)\n",
-        "# ---\n",
-        "storage_info = widgets.HTML(f'''\n",
-        "<div class=\"storage_info_AC\">Total storage: {total:.2f} GB <span style=\"color: #555\">|</span> Used: {used:.2f} GB <span style=\"color: #555\">|</span> Free: {free:.2f} GB</div>\n",
-        "''')\n",
-        "# ---\n",
-        "buttons = widgets.HBox([execute_button, clear_button])\n",
-        "lower_information_panel = widgets.HBox([buttons, storage_info]).add_class(\"lower_information_panel_AC\")\n",
-        "\n",
-        "container = VBox([instruction_label, widgets.HTML('<hr>'), auto_cleaner_widget, output, widgets.HTML('<hr>'), lower_information_panel]).add_class(\"container_AC\")\n",
-        "\n",
-        "display(container)"
-      ],
-      "metadata": {
-        "id": "I22dFg7F2j3G"
-      },
-      "execution_count": null,
-      "outputs": []
-    }
-  ]
+{
+  "nbformat": 4,
+  "nbformat_minor": 0,
+  "metadata": {
+    "colab": {
+      "provenance": []
+    },
+    "kernelspec": {
+      "name": "python3",
+      "display_name": "Python 3"
+    },
+    "language_info": {
+      "name": "python"
+    }
+  },
+  "cells": [
+    {
+      "cell_type": "code",
+      "source": [
+        "##~ AutoCleaner V3.6 CODE | BY: ANXETY   ~##\n",
+        "\n",
+        "import os\n",
+        "import time\n",
+        "import ipywidgets as widgets\n",
+        "from ipywidgets import Label, Button, VBox, HBox\n",
+        "from IPython.display import display, HTML, Javascript\n",
+        "\n",
+        "\n",
+        "# Setup Env\n",
+        "env = os.environ.get('ENV_NAME')\n",
+        "root_path = os.environ.get('ROOT_PATH')\n",
+        "webui_path = os.environ.get('WEBUI_PATH')\n",
+        "free_plan = os.environ.get('FREE_PLAN')\n",
+        "\n",
+        "\n",
+        "# ==================== CSS ====================\n",
+        "# Main CSS\n",
+        "css_file_path = f\"{root_path}/CSS/auto_cleaner.css\"\n",
+        "with open(css_file_path , \"r\") as f:\n",
+        "    CSS_AC = f.read()\n",
+        "display(HTML(f\"<style>{CSS_AC}</style>\"))\n",
+        "\n",
+        "\n",
+        "# ================ AutoCleaner function ================\n",
+        "directories = {\n",
+        "    \"Images\": f\"{webui_path}/output\",\n",
+        "    \"Models\": f\"{webui_path}/models/Stable-diffusion/\",\n",
+        "    \"Vae\": f\"{webui_path}/models/VAE/\",\n",
+        "    \"LoRa\": f\"{webui_path}/models/Lora/\",\n",
+        "    \"ControlNet Models\": f\"{webui_path}/models/ControlNet/\"\n",
+        "}\n",
+        "\n",
+        "\"\"\" functions \"\"\"\n",
+        "def clean_directory(directory):\n",
+        "    deleted_files = 0\n",
+        "    image_dir = directories['Images']\n",
+        "\n",
+        "    for root, dirs, files in os.walk(directory):\n",
+        "        for file in files:\n",
+        "            file_path = os.path.join(root, file)\n",
+        "\n",
+        "            if file.endswith(\".txt\"):\n",
+        "                continue\n",
+        "            if file.endswith((\".safetensors\", \".pt\")) or root == image_dir: # fix for image counter\n",
+        "                deleted_files += 1\n",
+        "\n",
+        "            os.remove(file_path)\n",
+        "    return deleted_files\n",
+        "\n",
+        "def update_memory_info():\n",
+        "    disk_space = psutil.disk_usage(os.getcwd())\n",
+        "    total = disk_space.total / (1024 ** 3)\n",
+        "    used = disk_space.used / (1024 ** 3)\n",
+        "    free = disk_space.free / (1024 ** 3)\n",
+        "\n",
+        "    storage_info.value = f'''\n",
+        "    <div class=\"storage_info_AC\">Total storage: {total:.2f} GB <span style=\"color: #555\">|</span> Used: {used:.2f} GB <span style=\"color: #555\">|</span> Free: {free:.2f} GB</div>\n",
+        "    '''\n",
+        "\n",
+        "def on_execute_button_press(button):\n",
+        "    selected_cleaners = auto_cleaner_widget.value\n",
+        "    deleted_files_dict = {}\n",
+        "\n",
+        "    for option in selected_cleaners:\n",
+        "        if option in directories:\n",
+        "            deleted_files_dict[option] = clean_directory(directories[option])\n",
+        "\n",
+        "    output.clear_output()\n",
+        "\n",
+        "    with output:\n",
+        "        for message in generate_messages(deleted_files_dict):\n",
+        "            message_widget = HTML(f'<p class=\"output_message_AC\">{message}</p>')\n",
+        "            display(message_widget)\n",
+        "\n",
+        "    update_memory_info()\n",
+        "\n",
+        "def on_clear_button_press(button):\n",
+        "    container.add_class(\"hide\")\n",
+        "    time.sleep(0.5)\n",
+        "    widgets.Widget.close_all()\n",
+        "\n",
+        "def generate_messages(deleted_files_dict):\n",
+        "    messages = []\n",
+        "    word_variants = {\n",
+        "        \"Images\": \"Images\",\n",
+        "        \"Models\": \"Models\",\n",
+        "        \"Vae\": \"Vae\",\n",
+        "        \"LoRa\": \"LoRa\",\n",
+        "        \"ControlNet Models\": \"ControlNet Models\"\n",
+        "    }\n",
+        "    for key, value in deleted_files_dict.items():\n",
+        "        object_word = word_variants.get(key)\n",
+        "        messages.append(f\"Deleted {value} {object_word}\")\n",
+        "    return messages\n",
+        "# ================ AutoCleaner function ================\n",
+        "\n",
+        "\n",
+        "# --- storage memory ---\n",
+        "import psutil\n",
+        "disk_space = psutil.disk_usage(os.getcwd())\n",
+        "total = disk_space.total / (1024 ** 3)\n",
+        "used = disk_space.used / (1024 ** 3)\n",
+        "free = disk_space.free / (1024 ** 3)\n",
+        "\n",
+        "\n",
+        "# UI Code\n",
+        "AutoCleaner_options = AutoCleaner_options = list(directories.keys())\n",
+        "instruction_label = widgets.HTML('''\n",
+        "<span class=\"instruction_AC\">Use <span style=\"color: #B2B2B2;\">ctrl</span> or <span style=\"color: #B2B2B2;\">shift</span> for multiple selections.</span>\n",
+        "''')\n",
+        "auto_cleaner_widget = widgets.SelectMultiple(options=AutoCleaner_options, layout=widgets.Layout(width='auto')).add_class(\"custom-select-multiple_AC\")\n",
+        "output = widgets.Output().add_class(\"output_AC\")\n",
+        "# ---\n",
+        "execute_button = Button(description='Execute Cleaning').add_class(\"button_execute_AC\").add_class(\"button_AC\")\n",
+        "execute_button.on_click(on_execute_button_press)\n",
+        "clear_button = Button(description='Hide Widget').add_class(\"button_clear_AC\").add_class(\"button_AC\")\n",
+        "clear_button.on_click(on_clear_button_press)\n",
+        "# ---\n",
+        "storage_info = widgets.HTML(f'''\n",
+        "<div class=\"storage_info_AC\">Total storage: {total:.2f} GB <span style=\"color: #555\">|</span> Used: {used:.2f} GB <span style=\"color: #555\">|</span> Free: {free:.2f} GB</div>\n",
+        "''')\n",
+        "# ---\n",
+        "buttons = widgets.HBox([execute_button, clear_button])\n",
+        "lower_information_panel = widgets.HBox([buttons, storage_info]).add_class(\"lower_information_panel_AC\")\n",
+        "\n",
+        "container = VBox([instruction_label, widgets.HTML('<hr>'), auto_cleaner_widget, output, widgets.HTML('<hr>'), lower_information_panel]).add_class(\"container_AC\")\n",
+        "\n",
+        "display(container)"
+      ],
+      "metadata": {
+        "id": "I22dFg7F2j3G"
+      },
+      "execution_count": null,
+      "outputs": []
+    }
+  ]
 }

files_cells/notebooks/en/downloading_en.ipynb

@@ -10,6 +10,8 @@
       "source": [
         "##~   DOWNLOADING CODE | BY: ANXETY   ~##\n",
         "\n",
+        "from models_data import model_list, vae_list, controlnet_list\n",
+        "\n",
         "import os\n",
         "import re\n",
         "import time\n",
@@ -25,19 +27,11 @@
         "from urllib.parse import urlparse, parse_qs\n",
         "\n",
         "\n",
-        "# ================= DETECT ENV =================\n",
-        "def detect_environment():\n",
-        "    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)\n",
-        "    environments = {\n",
-        "        'COLAB_GPU': ('Google Colab', \"/root\" if free_plan else \"/content\"),\n",
-        "        'KAGGLE_URL_BASE': ('Kaggle', \"/kaggle/working/content\")\n",
-        "    }\n",
-        "    for env_var, (environment, path) in environments.items():\n",
-        "        if env_var in os.environ:\n",
-        "            return environment, path, free_plan\n",
-        "\n",
-        "env, root_path, free_plan = detect_environment()\n",
-        "webui_path = f\"{root_path}/sdw\"\n",
+        "# Setup Env\n",
+        "env = os.environ.get('ENV_NAME')\n",
+        "root_path = os.environ.get('ROOT_PATH')\n",
+        "webui_path = os.environ.get('WEBUI_PATH')\n",
+        "free_plan = os.environ.get('FREE_PLAN')\n",
         "\n",
         "\n",
         "# ================ LIBRARIES V2 ================\n",
@@ -207,112 +201,9 @@
         "\n",
         "## Downloading model and stuff | oh~ Hey! If you're freaked out by that code too, don't worry, me too!\n",
         "print(\"📦 Downloading models and stuff...\", end='')\n",
-        "model_list = {\n",
-        "    \"1.Anime (by XpucT) + INP\": [\n",
-        "        {\"url\": \"https://huggingface.co/XpucT/Anime/resolve/main/Anime_v2.safetensors\", \"name\": \"Anime_V2.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/XpucT/Anime/resolve/main/Anime_v2-inpainting.safetensors\", \"name\": \"Anime_V2-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"2.BluMix [Anime] [V7] + INP\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/361779\", \"name\": \"BluMix_V7.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/363850\", \"name\": \"BluMix_V7-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"3.Cetus-Mix [Anime] [V4] + INP\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/130298\", \"name\": \"CetusMix_V4.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/139882\", \"name\": \"CetusMix_V4-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"4.Counterfeit [Anime] [V3] + INP\": [\n",
-        "        {\"url\": \"https://huggingface.co/gsdf/Counterfeit-V3.0/resolve/main/Counterfeit-V3.0_fix_fp16.safetensors\", \"name\": \"Counterfeit_V3.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/137911\", \"name\": \"Counterfeit_V3-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"5.CuteColor [Anime] [V3]\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/138754\", \"name\": \"CuteColor_V3.safetensors\"}\n",
-        "    ],\n",
-        "    \"6.Dark-Sushi-Mix [Anime]\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/101640\", \"name\": \"DarkSushiMix_2_5D.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/56071\", \"name\": \"DarkSushiMix_colorful.safetensors\"}\n",
-        "    ],\n",
-        "    \"7.Deliberate [Realism] [V6] + INP\": [\n",
-        "        {\"url\": \"https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v6.safetensors\", \"name\": \"Deliberate_V6.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v6-inpainting.safetensors\", \"name\": \"Deliberate_V6-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"8.Meina-Mix [Anime] [V11] + INP\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/119057\", \"name\": \"MeinaMix_V11.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/120702\", \"name\": \"MeinaMix_V11-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"9.Mix-Pro [Anime] [V4] + INP\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/125668\", \"name\": \"MixPro_V4.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/139878\", \"name\": \"MixPro_V4-inpainting.safetensors\"}\n",
-        "    ]\n",
-        "}\n",
-        "\n",
-        "vae_list = {\n",
-        "    \"1.Anime.vae\": [{\"url\": \"https://civitai.com/api/download/models/311162\", \"name\": \"Anime.vae.safetensors\"}],\n",
-        "    \"2.Anything.vae\": [{\"url\": \"https://huggingface.co/NoCrypt/resources/resolve/main/VAE/any.vae.safetensors\", \"name\": \"Anything.vae.safetensors\"}],\n",
-        "    \"3.Blessed2.vae\": [{\"url\": \"https://huggingface.co/NoCrypt/resources/resolve/main/VAE/blessed2.vae.safetensors\", \"name\": \"Blessed2.vae.safetensors\"}],\n",
-        "    \"4.ClearVae.vae\": [{\"url\": \"https://civitai.com/api/download/models/88156\", \"name\": \"ClearVae_23.vae.safetensors\"}],\n",
-        "    \"5.WD.vae\": [{\"url\": \"https://huggingface.co/NoCrypt/resources/resolve/main/VAE/wd.vae.safetensors\", \"name\": \"WD.vae.safetensors\"}]\n",
-        "}\n",
-        "\n",
-        "controlnet_list = {\n",
-        "    \"1.canny\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_canny_fp16.safetensors\", \"name\": \"control_v11p_sd15_canny_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_canny_fp16.yaml\", \"name\": \"control_v11p_sd15_canny_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"2.openpose\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose_fp16.safetensors\", \"name\": \"control_v11p_sd15_openpose_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_openpose_fp16.yaml\", \"name\": \"control_v11p_sd15_openpose_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"3.depth\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors\", \"name\": \"control_v11f1p_sd15_depth_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11f1p_sd15_depth_fp16.yaml\", \"name\": \"control_v11f1p_sd15_depth_fp16.yaml\"},\n",
-        "        {\"url\": \"https://huggingface.co/NagisaNao/models/resolve/main/ControlNet_v11/control_v11p_sd15_depth_anything_fp16.safetensors\", \"name\": \"control_v11p_sd15_depth_anything_fp16.safetensors\"}\n",
-        "    ],\n",
-        "    \"4.normal_map\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors\", \"name\": \"control_v11p_sd15_normalbae_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_normalbae_fp16.yaml\", \"name\": \"control_v11p_sd15_normalbae_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"5.mlsd\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors\", \"name\": \"control_v11p_sd15_mlsd_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_mlsd_fp16.yaml\", \"name\": \"control_v11p_sd15_mlsd_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"6.lineart\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_lineart_fp16.safetensors\", \"name\": \"control_v11p_sd15_lineart_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors\", \"name\": \"control_v11p_sd15s2_lineart_anime_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_lineart_fp16.yaml\", \"name\": \"control_v11p_sd15_lineart_fp16.yaml\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15s2_lineart_anime_fp16.yaml\", \"name\": \"control_v11p_sd15s2_lineart_anime_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"7.soft_edge\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_softedge_fp16.safetensors\", \"name\": \"control_v11p_sd15_softedge_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_softedge_fp16.yaml\", \"name\": \"control_v11p_sd15_softedge_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"8.scribble\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_scribble_fp16.safetensors\", \"name\": \"control_v11p_sd15_scribble_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_scribble_fp16.yaml\", \"name\": \"control_v11p_sd15_scribble_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"9.segmentation\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_seg_fp16.safetensors\", \"name\": \"control_v11p_sd15_seg_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_seg_fp16.yaml\", \"name\": \"control_v11p_sd15_seg_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"10.shuffle\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_shuffle_fp16.safetensors\", \"name\": \"control_v11e_sd15_shuffle_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_shuffle_fp16.yaml\", \"name\": \"control_v11e_sd15_shuffle_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"11.tile\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11f1e_sd15_tile_fp16.safetensors\", \"name\": \"control_v11f1e_sd15_tile_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11f1e_sd15_tile_fp16.yaml\", \"name\": \"control_v11f1e_sd15_tile_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"12.inpaint\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors\", \"name\": \"control_v11p_sd15_inpaint_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_inpaint_fp16.yaml\", \"name\": \"control_v11p_sd15_inpaint_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"13.instruct_p2p\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_ip2p_fp16.safetensors\", \"name\": \"control_v11e_sd15_ip2p_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_ip2p_fp16.yaml\", \"name\": \"control_v11e_sd15_ip2p_fp16.yaml\"}\n",
-        "    ]\n",
-        "}\n",
         "\n",
         "url = \"\"\n",
-        "prefixes = {\n",
+        "PREFIXES = {\n",
         "    \"model\": models_dir,\n",
         "    \"vae\": vaes_dir,\n",
         "    \"lora\": loras_dir,\n",
@@ -324,7 +215,7 @@
         "}\n",
         "\n",
         "extension_repo = []\n",
-        "directories = [value for key, value in prefixes.items()] # for unpucking zip files\n",
+        "directories = [value for key, value in PREFIXES.items()] # for unpucking zip files\n",
         "!mkdir -p {\" \".join(directories)}\n",
         "\n",
         "hf_token = huggingface_token if huggingface_token else \"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\"\n",
@@ -335,137 +226,123 @@
         "from math import floor\n",
         "\n",
         "def center_text(text, terminal_width=45):\n",
-        "    text_length = len(text)\n",
-        "    left_padding = floor((terminal_width - text_length) / 2)\n",
-        "    right_padding = terminal_width - text_length - left_padding\n",
-        "    return f\"\\033[1m\\033[36m{' ' * left_padding}{text}{' ' * right_padding}\\033[0m\\033[32m\"\n",
+        "    padding = (terminal_width - len(text)) // 2\n",
+        "    return f\"\\033[1m\\033[36m{' ' * padding}{text}{' ' * padding}\\033[0m\\033[32m\"\n",
         "\n",
         "def format_output(url, dst_dir, file_name):\n",
-        "    info = f\"[{file_name.split('.')[0]}]\"\n",
-        "    info = center_text(info)\n",
+        "    info = center_text(f\"[{file_name.split('.')[0]}]\")\n",
+        "    separation_line = '\\033[32m' + '---' * 20\n",
         "\n",
-        "    print(f\"\\n\\033[32m{'---'*20}]{info}[{'---'*20}\")\n",
+        "    print(f\"\\n{separation_line}{info}{separation_line}\")\n",
         "    print(f\"\\033[33mURL: \\033[34m{url}\")\n",
         "    print(f\"\\033[33mSAVE DIR: \\033[34m{dst_dir}\")\n",
         "    print(f\"\\033[33mFILE NAME: \\033[34m{file_name}\\033[0m\")\n",
         "\n",
         "''' GET CivitAi API - DATA '''\n",
         "\n",
-        "def strip_(url, file_name=None):\n",
-        "    if 'github.com' in url:\n",
-        "        if '/blob/' in url:\n",
-        "            url = url.replace('/blob/', '/raw/')\n",
-        "\n",
-        "    elif \"civitai.com\" in url:\n",
-        "        return CivitAi_API(url, file_name)\n",
-        "\n",
-        "    elif \"huggingface.co\" in url:\n",
-        "        if '/blob/' in url:\n",
-        "            url = url.replace('/blob/', '/resolve/')\n",
-        "        if '?' in url:\n",
-        "            url = url.split('?')[0]\n",
-        "\n",
-        "    return url\n",
-        "\n",
         "def CivitAi_API(url, file_name=None):\n",
-        "    support_types = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')\n",
-        "    civitai_token = \"62c0c5956b2f9defbd844d754000180b\"\n",
+        "    SUPPORT_TYPES = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')\n",
+        "    CIVITAI_TOKEN = \"62c0c5956b2f9defbd844d754000180b\"\n",
         "\n",
-        "    if '?token=' in url:\n",
-        "        url = url.split('?token=')[0]\n",
-        "    if '?type=' in url:\n",
-        "        url = url.replace('?type=', f'?token={civitai_token}&type=')\n",
-        "    else:\n",
-        "        url = f\"{url}?token={civitai_token}\"\n",
-        "\n",
-        "    # Determine model or version id\n",
-        "    if \"civitai.com/models/\" in url:\n",
-        "        if '?modelVersionId=' in url:\n",
-        "            version_id = url.split('?modelVersionId=')[1]\n",
-        "            response = requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\")\n",
-        "            # print(f\"end - https://civitai.com/api/v1/model-versions/{version_id}\")\n",
+        "    url = url.split('?token=')[0] if '?token=' in url else url\n",
+        "    url = url.replace('?type=', f'?token={CIVITAI_TOKEN}&type=') if '?type=' in url else f\"{url}?token={CIVITAI_TOKEN}\"\n",
+        "\n",
+        "    def get_model_data(url):\n",
+        "        if \"civitai.com/models/\" in url:\n",
+        "            if '?modelVersionId=' in url:\n",
+        "                version_id = url.split('?modelVersionId=')[1]\n",
+        "                return requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\").json()\n",
+        "            else:\n",
+        "                model_id = url.split('/models/')[1].split('/')[0]\n",
+        "                return requests.get(f\"https://civitai.com/api/v1/models/{model_id}\").json()\n",
         "        else:\n",
-        "            model_id = url.split('/models/')[1].split('/')[0]\n",
-        "            response = requests.get(f\"https://civitai.com/api/v1/models/{model_id}\")\n",
-        "            # print(f\"end - https://civitai.com/api/v1/models/{model_id}\")\n",
-        "    else:\n",
-        "        version_id = url.split('/models/')[1].split('/')[0]\n",
-        "        response = requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\")\n",
-        "        # print(f\"end - https://civitai.com/api/v1/model-versions/{version_id}\")\n",
+        "            version_id = url.split('/models/')[1].split('/')[0]\n",
+        "            return requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\").json()\n",
         "\n",
-        "    data = response.json()\n",
+        "    data = get_model_data(url)\n",
         "\n",
-        "    if response.status_code != 200:\n",
+        "    if not data:\n",
         "        return None, None, None, None, None, None, None\n",
         "\n",
-        "    # Define model type and name\n",
-        "    if \"civitai.com/models/\" in url:\n",
-        "        if '?modelVersionId=' in url:\n",
-        "            model_type = data['model']['type']\n",
-        "            model_name = data['files'][0]['name']\n",
+        "    def extract_model_info(url, data):\n",
+        "        if \"civitai.com/models/\" in url:\n",
+        "            if '?modelVersionId=' in url:\n",
+        "                model_type = data['model']['type']\n",
+        "                model_name = data['files'][0]['name']\n",
+        "            else:\n",
+        "                model_type = data['type']\n",
+        "                model_name = data['modelVersions'][0]['files'][0]['name']\n",
+        "        elif 'type=' in url:\n",
+        "            model_type = parse_qs(urlparse(url).query).get('type', [''])[0]\n",
+        "            if 'model' in model_type.lower():\n",
+        "                model_name = data['files'][0]['name']\n",
+        "            else:\n",
+        "                model_name = data['files'][1]['name']\n",
         "        else:\n",
-        "            model_type = data['type']\n",
-        "            model_name = data['modelVersions'][0]['files'][0]['name']\n",
-        "    elif 'type=' in url:\n",
-        "        model_type = parse_qs(urlparse(url).query).get('type', [''])[0]\n",
-        "        if 'model' in model_type.lower():\n",
+        "            model_type = data['model']['type']\n",
         "            model_name = data['files'][0]['name']\n",
-        "        else:\n",
-        "            model_name = data['files'][1]['name']\n",
-        "    else:\n",
-        "        model_type = data['model']['type']\n",
-        "        model_name = data['files'][0]['name']\n",
+        "        return model_type, model_name\n",
         "\n",
+        "    model_type, model_name = extract_model_info(url, data)\n",
         "    model_name = file_name or model_name\n",
         "\n",
-        "    # Determine DownloadUrl\n",
-        "    if \"civitai.com/models/\" in url:\n",
-        "        if '?modelVersionId=' in url:\n",
-        "            download_url = data.get('downloadUrl')\n",
-        "        else:\n",
-        "            download_url = data[\"modelVersions\"][0].get(\"downloadUrl\", \"\")\n",
-        "    elif 'type=' in url:\n",
-        "        if any(t.lower() in model_type.lower() for t in support_types):\n",
-        "            download_url = data['files'][0]['downloadUrl']\n",
+        "    def get_download_url(url, data, model_type):\n",
+        "        if \"civitai.com/models/\" in url:\n",
+        "            if '?modelVersionId=' in url:\n",
+        "                return data.get('downloadUrl')\n",
+        "            else:\n",
+        "                return data[\"modelVersions\"][0].get(\"downloadUrl\", \"\")\n",
+        "        elif 'type=' in url:\n",
+        "            if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):\n",
+        "                return data['files'][0]['downloadUrl']\n",
+        "            else:\n",
+        "                return data['files'][1]['downloadUrl']\n",
         "        else:\n",
-        "            download_url = data['files'][1]['downloadUrl']\n",
-        "    else:\n",
-        "        download_url = data.get('downloadUrl')\n",
+        "            return data.get('downloadUrl')\n",
         "\n",
-        "    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url) # hide token\n",
+        "    download_url = get_download_url(url, data, model_type)\n",
+        "    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url)\n",
         "\n",
-        "    # Find a safe image: level less than 4 | Kaggle\n",
-        "    image_url, image_name = None, None\n",
-        "    if any(t in model_type for t in support_types):\n",
-        "        try:\n",
-        "            images = data.get('images') or data['modelVersions'][0].get('images', [])\n",
-        "            if env == 'Kaggle':\n",
-        "                image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)\n",
-        "            else:\n",
-        "                image_url = images[0]['url'] if images else None\n",
-        "        except KeyError:\n",
-        "            pass\n",
+        "    def get_image_info(data, model_type, model_name):\n",
+        "        image_url, image_name = None, None\n",
+        "        if any(t in model_type for t in SUPPORT_TYPES):\n",
+        "            try:\n",
+        "                images = data.get('images') or data['modelVersions'][0].get('images', [])\n",
+        "                if env == 'Kaggle':\n",
+        "                    image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)\n",
+        "                else:\n",
+        "                    image_url = images[0]['url'] if images else None\n",
+        "            except KeyError:\n",
+        "                pass\n",
+        "\n",
+        "            image_name = f\"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}\" if image_url else None\n",
+        "        return image_url, image_name\n",
         "\n",
-        "        # Generate a name to save the image\n",
-        "        image_name = f\"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}\" if image_url else None\n",
+        "    image_url, image_name = get_image_info(data, model_type, model_name)\n",
         "\n",
-        "    return f\"{download_url}{'&' if '?' in download_url else '?'}token={civitai_token}\", clean_url, model_type, model_name, image_url, image_name, data\n",
+        "    return f\"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}\", clean_url, model_type, model_name, image_url, image_name, data\n",
         "\n",
         "''' Main Download Code '''\n",
         "\n",
+        "def strip_(url):\n",
+        "    if 'github.com' in url:\n",
+        "        return url.replace('/blob/', '/raw/')\n",
+        "    elif \"huggingface.co\" in url:\n",
+        "        url = url.replace('/blob/', '/resolve/')\n",
+        "        return url.split('?')[0] if '?' in url else url\n",
+        "    return url\n",
+        "\n",
         "def download(url):\n",
         "    links_and_paths = [link_or_path.strip() for link_or_path in url.split(',') if link_or_path.strip()]\n",
         "\n",
         "    for link_or_path in links_and_paths:\n",
-        "        if any(link_or_path.lower().startswith(prefix) for prefix in prefixes):\n",
+        "        if any(link_or_path.lower().startswith(prefix) for prefix in PREFIXES):\n",
         "            handle_manual(link_or_path)\n",
         "        else:\n",
         "            url, dst_dir, file_name = link_or_path.split()\n",
         "            manual_download(url, dst_dir, file_name)\n",
         "\n",
-        "    unpack_zip_files()\n",
-        "\n",
-        "def unpack_zip_files():\n",
+        "    # Unpuck ZIPs Files\n",
         "    for directory in directories:\n",
         "        for root, _, files in os.walk(directory):\n",
         "            for file in files:\n",
@@ -485,8 +362,8 @@
         "    if file_name:\n",
         "        path = re.sub(r'\\[.*?\\]', '', path)\n",
         "\n",
-        "    if prefix in prefixes:\n",
-        "        dir = prefixes[prefix]\n",
+        "    if prefix in PREFIXES:\n",
+        "        dir = PREFIXES[prefix]\n",
         "        if prefix != \"extension\":\n",
         "            try:\n",
         "                manual_download(path, dir, file_name=file_name)\n",
@@ -500,60 +377,52 @@
         "    aria2c_header = \"--header='User-Agent: Mozilla/5.0' --allow-overwrite=true\"\n",
         "    aria2_args = \"--optimize-concurrent-downloads --console-log-level=error --summary-interval=10 --stderr=true -c -x16 -s16 -k1M -j5\"\n",
         "\n",
-        "    if 'github.com' in url:\n",
-        "        url = strip_(url)\n",
-        "\n",
-        "    # -- CivitAi APi+ V2 --\n",
-        "    elif 'civitai' in url:\n",
-        "        url, clean_url, model_type, file_name, image_url, image_name, data = strip_(url, file_name)\n",
+        "    clean_url = strip_(url)\n",
         "\n",
+        "    if 'civitai' in url:\n",
+        "        url, clean_url, model_type, file_name, image_url, image_name, data = CivitAi_API(url, file_name)\n",
         "        if image_url and image_name:\n",
-        "            with capture.capture_output() as cap:\n",
-        "                !aria2c {aria2_args} -d {dst_dir} -o '{image_name}' '{image_url}'\n",
-        "            del cap\n",
+        "            command = [\"aria2c\"] + aria2_args.split() + [\"-d\", dst_dir, \"-o\", image_name, image_url]\n",
+        "            subprocess.run(command, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)\n",
         "\n",
-        "    elif \"huggingface.co\" in url:\n",
-        "        clean_url = strip_(url)\n",
-        "        basename = clean_url.split(\"/\")[-1] if file_name is None else file_name\n",
+        "    elif 'github' in url or \"huggingface.co\" in url:\n",
+        "        basename = url.split(\"/\")[-1] if file_name is None else file_name\n",
         "\n",
         "    \"\"\" Formatted info output \"\"\"\n",
-        "    model_name_or_basename = file_name if not 'huggingface' in url else basename\n",
+        "    model_name_or_basename = file_name if file_name else basename\n",
         "    format_output(clean_url or url, dst_dir, model_name_or_basename)\n",
         "\n",
-        "    # ## -- for my tests --\n",
         "    # print(url, dst_dir, model_name_or_basename)\n",
-        "    print(f\"\\033[31m[Data Info]:\\033[0m Failed to retrieve data from the API.\\n\") if 'civitai' in url and not data else None\n",
-        "    if 'civitai' in url and data and image_name:\n",
-        "        print(f\"\\033[32m[Preview DL]:\\033[0m {image_name} - {image_url}\\n\")\n",
+        "    if 'civitai' in url:\n",
+        "        if not data:\n",
+        "            print(\"\\033[31m[Data Info]:\\033[0m Failed to retrieve data from the API.\\n\")\n",
+        "        if data and image_name:\n",
+        "            print(f\"\\033[32m[Preview DL]:\\033[0m {image_name} - {image_url}\\n\")\n",
+        "\n",
         "    # =====================\n",
+        "    def run_aria2c(url, dst_dir, file_name=None, args=\"\", header=\"\"):\n",
+        "        out = f\"-o '{file_name}'\" if file_name else \"\"\n",
+        "        !aria2c {header} {args} -d {dst_dir} {out} '{url}'\n",
         "\n",
-        "    # -- Git Hub --\n",
-        "    if 'github.com' in url or 'githubusercontent.com' in url:\n",
-        "        !aria2c {aria2_args} -d {dst_dir} -o '{basename}' '{url}'\n",
-        "\n",
-        "    # -- GDrive --\n",
-        "    elif 'drive.google' in url:\n",
-        "        try:\n",
-        "            have_drive_link\n",
-        "        except:\n",
-        "            !pip install -q gdown==5.2.0 > /dev/null\n",
-        "            have_drive_link = True\n",
+        "    # -- Google Drive --\n",
+        "    if 'drive.google' in url:\n",
+        "        if not globals().get('have_drive_link', False):\n",
+        "            os.system(\"pip install -U gdown > /dev/null\")\n",
+        "            globals()['have_drive_link'] = True\n",
         "\n",
         "        if 'folders' in url:\n",
-        "            !gdown --folder \"{url}\" -O {dst_dir} --fuzzy -c\n",
+        "            os.system(f\"gdown --folder \\\"{url}\\\" -O {dst_dir} --fuzzy -c\")\n",
         "        else:\n",
-        "            if file_name:\n",
-        "               !gdown \"{url}\" -O {dst_dir}/{file_name} --fuzzy -c\n",
-        "            else:\n",
-        "               !gdown \"{url}\" -O {dst_dir} --fuzzy -c\n",
+        "            out_path = f\"{dst_dir}/{file_name}\" if file_name else dst_dir\n",
+        "            os.system(f\"gdown \\\"{url}\\\" -O {out_path} --fuzzy -c\")\n",
         "\n",
-        "    # -- Hugging Face --\n",
-        "    elif 'huggingface' in url:\n",
-        "        !aria2c {header_option} {aria2_args} -d {dst_dir} -o '{basename}' '{url}'\n",
+        "    # -- GitHub or Hugging Face --\n",
+        "    elif 'github' in url or 'huggingface' in url:\n",
+        "        run_aria2c(clean_url, dst_dir, basename, aria2_args, header_option if 'huggingface' in url else '')\n",
         "\n",
-        "    # -- Other --\n",
+        "    # -- Other HTTP/Sources --\n",
         "    elif 'http' in url:\n",
-        "        !aria2c {aria2c_header} {aria2_args} -d {dst_dir} -o \"{file_name if file_name else ''}\" '{url}'\n",
+        "        run_aria2c(url, dst_dir, file_name, aria2_args, aria2c_header)\n",
         "\n",
         "''' SubModels - Added URLs '''\n",
         "\n",
@@ -593,7 +462,7 @@
         "\n",
         "''' file.txt - added urls '''\n",
         "\n",
-        "def process_file_download(file_url, prefixes, unique_urls):\n",
+        "def process_file_download(file_url, PREFIXES, unique_urls):\n",
         "    files_urls = \"\"\n",
         "\n",
         "    if file_url.startswith(\"http\"):\n",
@@ -608,8 +477,8 @@
         "    current_tag = None\n",
         "    for line in lines:\n",
         "        line = line.strip()\n",
-        "        if any(f'# {tag}' in line.lower() for tag in prefixes):\n",
-        "            current_tag = next((tag for tag in prefixes if tag in line.lower()))\n",
+        "        if any(f'# {tag}' in line.lower() for tag in PREFIXES):\n",
+        "            current_tag = next((tag for tag in PREFIXES if tag in line.lower()))\n",
         "\n",
         "        urls = [url.split('#')[0].strip() for url in line.split(',')]  # filter urls\n",
         "        for url in urls:\n",
@@ -633,13 +502,13 @@
         "                custom_file_url = f'{root_path}/{custom_file_url}'\n",
         "\n",
         "        try:\n",
-        "            file_urls += process_file_download(custom_file_url, prefixes, unique_urls)\n",
+        "            file_urls += process_file_download(custom_file_url, PREFIXES, unique_urls)\n",
         "        except FileNotFoundError:\n",
         "            pass\n",
         "\n",
         "# url prefixing\n",
         "urls = (Model_url, Vae_url, LoRA_url, Embedding_url, Extensions_url)\n",
-        "prefixed_urls = (f\"{prefix}:{url}\" for prefix, url in zip(prefixes.keys(), urls) if url for url in url.replace(',', '').split())\n",
+        "prefixed_urls = (f\"{prefix}:{url}\" for prefix, url in zip(PREFIXES.keys(), urls) if url for url in url.replace(',', '').split())\n",
         "url += \", \".join(prefixed_urls) + \", \" + file_urls\n",
         "\n",
         "if detailed_download == \"on\":\n",

files_cells/notebooks/en/launch_en.ipynb

@@ -33,21 +33,13 @@
         "from datetime import timedelta\n",
         "from IPython.display import clear_output\n",
         "\n",
-        "#  ================= DETECT ENV =================\n",
-        "def detect_environment():\n",
-        "    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)\n",
-        "    environments = {\n",
-        "        'COLAB_GPU': ('Google Colab', \"/root\" if free_plan else \"/content\"),\n",
-        "        'KAGGLE_URL_BASE': ('Kaggle', \"/kaggle/working/content\")\n",
-        "    }\n",
-        "\n",
-        "    for env_var, (environment, path) in environments.items():\n",
-        "        if env_var in os.environ:\n",
-        "            return environment, path, free_plan\n",
-        "    return 'Unknown', '/unknown/path', free_plan\n",
-        "\n",
-        "env, root_path, free_plan = detect_environment()\n",
-        "webui_path = f\"{root_path}/sdw\"\n",
+        "\n",
+        "# Setup Env\n",
+        "env = os.environ.get('ENV_NAME')\n",
+        "root_path = os.environ.get('ROOT_PATH')\n",
+        "webui_path = os.environ.get('WEBUI_PATH')\n",
+        "free_plan = os.environ.get('FREE_PLAN')\n",
+        "\n",
         "\n",
         "def load_settings():\n",
         "    SETTINGS_FILE = f'{root_path}/settings.json'\n",
@@ -62,6 +54,7 @@
         "commandline_arguments = settings.get('commandline_arguments', \"\")\n",
         "change_webui = settings.get('change_webui', \"\")\n",
         "\n",
+        "\n",
         "# ======================== TUNNEL V2 ========================\n",
         "print('Please Wait...')\n",
         "\n",
@@ -95,6 +88,7 @@
         "\n",
         "clear_output()\n",
         "\n",
+        "\n",
         "# =============== Automatic Fixing Path V3 ===============\n",
         "paths_to_check = {\n",
         "    \"tagger_hf_cache_dir\": f\"{webui_path}/models/interrogators/\",\n",

files_cells/notebooks/en/widgets_en.ipynb

@@ -1,349 +1,340 @@
-{
-  "nbformat": 4,
-  "nbformat_minor": 0,
-  "metadata": {
-    "colab": {
-      "provenance": []
-    },
-    "kernelspec": {
-      "name": "python3",
-      "display_name": "Python 3"
-    },
-    "language_info": {
-      "name": "python"
-    }
-  },
-  "cells": [
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "JKTCrY9LU7Oq"
-      },
-      "outputs": [],
-      "source": [
-        "##~   WIDGET CODE | BY: ANXETY   ~##\n",
-        "\n",
-        "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 IPython.display import display, HTML, Javascript, clear_output\n",
-        "\n",
-        "\n",
-        "# ================= DETECT ENV =================\n",
-        "def detect_environment():\n",
-        "    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)\n",
-        "    environments = {\n",
-        "        'COLAB_GPU': ('Google Colab', \"/root\" if free_plan else \"/content\"),\n",
-        "        'KAGGLE_URL_BASE': ('Kaggle', \"/kaggle/working/content\")\n",
-        "    }\n",
-        "    for env_var, (environment, path) in environments.items():\n",
-        "        if env_var in os.environ:\n",
-        "            return environment, path, free_plan\n",
-        "\n",
-        "env, root_path, free_plan = detect_environment()\n",
-        "webui_path = f\"{root_path}/sdw\"\n",
-        "!mkdir -p {root_path}\n",
-        "\n",
-        "\n",
-        "# ==================== CSS JS ====================\n",
-        "##~ custom background images V1.5 ~##\n",
-        "import argparse\n",
-        "parser = argparse.ArgumentParser(description='This script processes an background image.')\n",
-        "parser.add_argument('-i', '--image', type=str, help='URL of the image to process', metavar='')\n",
-        "parser.add_argument('-o', '--opacity', type=float, help='Opacity level for the image, between 0 and 1', metavar='', default=0.3)\n",
-        "parser.add_argument('-b', '--blur', type=str, help='Blur level for the image', metavar='', default=0)\n",
-        "parser.add_argument('-y', type=int, help='Y coordinate for the image in px', metavar='', default=0)\n",
-        "parser.add_argument('-x', type=int, help='X coordinate for the image in px', metavar='', default=0)\n",
-        "parser.add_argument('-s', '--scale', type=int, help='Scale image in %%', metavar='', default=100)\n",
-        "parser.add_argument('-m', '--mode', action='store_true', help='Removes repetitive image tiles')\n",
-        "parser.add_argument('-t', '--transparent', action='store_true', help='Makes input/selection fields 35%% more transparent')\n",
-        "parser.add_argument('-bf', '--blur-fields', type=str, help='Background blur level for input/selection fields', metavar='', default=2)\n",
-        "\n",
-        "args = parser.parse_args()\n",
-        "\n",
-        "url_img = args.image\n",
-        "opacity_img = args.opacity\n",
-        "blur_img = args.blur\n",
-        "y_img = args.y\n",
-        "x_img = args.x\n",
-        "scale_img = args.scale\n",
-        "blur_fields = args.blur_fields\n",
-        "\n",
-        "## ---\n",
-        "\"\"\" WTF KAGGLE - WHAT THE FUCK IS THE DIFFERENCE OF 35 PIXELS!?!?!? \"\"\"\n",
-        "fix_heigh_img = \"-810px\" if env == \"Kaggle\" else \"-775px\"\n",
-        "\n",
-        "\"\"\" transperent fields \"\"\"\n",
-        "t_bg_alpha = \"1\" if not args.transparent else \"0.65\"\n",
-        "\n",
-        "\"\"\" mode img - repeats \"\"\"\n",
-        "mode_img = \"repeat\" if not args.mode else \"no-repeat\"\n",
-        "\n",
-        "container_background = f'''\n",
-        "<style>\n",
-        ":root {{\n",
-        "  /* for background container*/\n",
-        "  --img_background: url({url_img});\n",
-        "  --img_opacity: {opacity_img};\n",
-        "  --img_blur: {blur_img}px;\n",
-        "  --image_y: {y_img}px;\n",
-        "  --image_x: {x_img}px;\n",
-        "  --img_scale: {scale_img}%;\n",
-        "  --img_mode: {mode_img};\n",
-        "  --img_height_dif: {fix_heigh_img};\n",
-        "\n",
-        "  /* for fields */\n",
-        "  --bg-field-color: rgba(28, 28, 28, {t_bg_alpha}); /* -> #1c1c1c */\n",
-        "  --bg-field-color-hover: rgba(38, 38, 38, {t_bg_alpha}); /* -> #262626; */\n",
-        "  --bg-field-blur-level: {blur_fields}px;\n",
-        "}}\n",
-        "'''\n",
-        "\n",
-        "display(HTML(container_background))\n",
-        "\n",
-        "# Main CSS\n",
-        "css_file_path = f\"{root_path}/CSS/main_widgets.css\"\n",
-        "with open(css_file_path , \"r\") as f:\n",
-        "    CSS = f.read()\n",
-        "display(HTML(f\"<style>{CSS}</style>\"))\n",
-        "\n",
-        "# Main JS\n",
-        "JS = '''\n",
-        "<!-- TOGGLE 'CustomDL' SCRIPT -->\n",
-        "<script>\n",
-        "function toggleContainer() {\n",
-        "    let downloadContainer = document.querySelector('.container_custom_downlad');\n",
-        "    let info = document.querySelector('.info');\n",
-        "\n",
-        "    downloadContainer.classList.toggle('expanded');\n",
-        "    info.classList.toggle('showed');\n",
-        "}\n",
-        "</script>\n",
-        "'''\n",
-        "display(HTML(JS))\n",
-        "\n",
-        "# ==================== WIDGETS V2 ====================\n",
-        "HR = widgets.HTML('<hr>')\n",
-        "\n",
-        "class WidgetFactory:\n",
-        "    def __init__(self, style=None, layout=None):\n",
-        "        self.style = style if style else {'description_width': 'initial'}\n",
-        "        self.layout = layout if layout else widgets.Layout(max_width='1080px', width='100%')\n",
-        "\n",
-        "    def create_html(self, content, class_name=None):\n",
-        "        html_widget = widgets.HTML(content)\n",
-        "        if class_name:\n",
-        "            html_widget.add_class(class_name)\n",
-        "        return html_widget\n",
-        "\n",
-        "    def create_header(self, name):\n",
-        "        return widgets.HTML(f'<div class=\"header\">{name}<div>')\n",
-        "\n",
-        "    def create_dropdown(self, options, value, description):\n",
-        "        return widgets.Dropdown(options=options, value=value, description=description, style=self.style, layout=self.layout)\n",
-        "\n",
-        "    def create_text(self, description, placeholder='', value=''):\n",
-        "        return widgets.Text(description=description, placeholder=placeholder, value=value, style=self.style, layout=self.layout)\n",
-        "\n",
-        "    def create_checkbox(self, value, description):\n",
-        "        return widgets.Checkbox(value=value, description=description, style=self.style, layout=self.layout)\n",
-        "\n",
-        "    def create_button(self, description, class_name=None):\n",
-        "        button = widgets.Button(description=description)\n",
-        "        if class_name:\n",
-        "            button.add_class(class_name)\n",
-        "        return button\n",
-        "\n",
-        "    def create_hbox(self, children):\n",
-        "        return widgets.HBox(children)\n",
-        "\n",
-        "    def create_vbox(self, children, class_names=None):\n",
-        "        vbox = widgets.VBox(children)\n",
-        "        if class_names:\n",
-        "            for class_name in class_names:\n",
-        "                vbox.add_class(class_name)\n",
-        "        return vbox\n",
-        "\n",
-        "    def display(self, widget):\n",
-        "        display(widget)\n",
-        "\n",
-        "# Instantiate the factory\n",
-        "factory = WidgetFactory()\n",
-        "\n",
-        "# --- MODEL ---\n",
-        "model_header = factory.create_header('Model Selection')\n",
-        "model_options = ['none',\n",
-        "                 '1.Anime (by XpucT) + INP',\n",
-        "                 '2.BluMix [Anime] [V7] + INP',\n",
-        "                 '3.Cetus-Mix [Anime] [V4] + INP',\n",
-        "                 '4.Counterfeit [Anime] [V3] + INP',\n",
-        "                 '5.CuteColor [Anime] [V3]',\n",
-        "                 '6.Dark-Sushi-Mix [Anime]',\n",
-        "                 '7.Deliberate [Realism] [V6] + INP',\n",
-        "                 '8.Meina-Mix [Anime] [V11] + INP',\n",
-        "                 '9.Mix-Pro [Anime] [V4] + INP']\n",
-        "model_widget = factory.create_dropdown(model_options, '4.Counterfeit [Anime] [V3] + INP', 'Model:')\n",
-        "model_num_widget = factory.create_text('Model Number:', 'Enter the model numbers to be downloaded using comma/space.')\n",
-        "inpainting_model_widget = factory.create_checkbox(False, 'Inpainting Models')\n",
-        "\n",
-        "# Display Model\n",
-        "all_model_box = factory.create_vbox([model_header, model_widget, model_num_widget, inpainting_model_widget], class_names=[\"container\", \"image_1\"])\n",
-        "factory.display(all_model_box)\n",
-        "\n",
-        "# --- VAE ---\n",
-        "vae_header = factory.create_header('VAE Selection')\n",
-        "vae_options = ['none',\n",
-        "               '1.Anime.vae',\n",
-        "               '2.Anything.vae',\n",
-        "               '3.Blessed2.vae',\n",
-        "               '4.ClearVae.vae',\n",
-        "               '5.WD.vae']\n",
-        "vae_widget = factory.create_dropdown(vae_options, '3.Blessed2.vae', 'Vae:')\n",
-        "vae_num_widget = factory.create_text('Vae Number:', 'Enter the vae numbers to be downloaded using comma/space.')\n",
-        "\n",
-        "# Display Vae\n",
-        "all_vae_box = factory.create_vbox([vae_header, vae_widget, vae_num_widget], class_names=[\"container\", \"image_2\"])\n",
-        "factory.display(all_vae_box)\n",
-        "\n",
-        "# --- ADDITIONAL ---\n",
-        "additional_header = factory.create_header('Additional')\n",
-        "latest_webui_widget = factory.create_checkbox(True, 'Update WebUI')\n",
-        "latest_exstensions_widget = factory.create_checkbox(True, 'Update Extensions')\n",
-        "change_webui_widget = factory.create_dropdown(['A1111', 'Forge'], 'A1111', 'Change WebUI:')\n",
-        "detailed_download_widget = factory.create_dropdown(['off', 'on'], 'off', 'Detailed Download:')\n",
-        "choose_changes_widget = factory.create_hbox([latest_webui_widget, latest_exstensions_widget, change_webui_widget, detailed_download_widget])\n",
-        "\n",
-        "controlnet_options = ['none', 'ALL', '1.canny',\n",
-        "                      '2.openpose', '3.depth',\n",
-        "                      '4.normal_map', '5.mlsd',\n",
-        "                      '6.lineart', '7.soft_edge',\n",
-        "                      '8.scribble', '9.segmentation',\n",
-        "                      '10.shuffle', '11.tile',\n",
-        "                      '12.inpaint', '13.instruct_p2p']\n",
-        "controlnet_widget = factory.create_dropdown(controlnet_options, 'none', 'ControlNet:')\n",
-        "controlnet_num_widget = factory.create_text('ControlNet Number:', 'Enter the ControlNet model numbers to be downloaded using comma/space.')\n",
-        "commit_hash_widget = factory.create_text('Commit Hash:')\n",
-        "huggingface_token_widget = factory.create_text('HuggingFace Token:')\n",
-        "\n",
-        "ngrok_token_widget = factory.create_text('Ngrok Token:')\n",
-        "ngrock_button = factory.create_html('<a href=\"https://dashboard.ngrok.com/get-started/your-authtoken\" target=\"_blank\">Получить Ngrok Токен</a>', class_name=\"button_ngrok\")\n",
-        "ngrok_widget = factory.create_hbox([ngrok_token_widget, ngrock_button])\n",
-        "\n",
-        "zrok_token_widget = factory.create_text('Zrok Token:')\n",
-        "zrok_button = factory.create_html('<a href=\"https://colab.research.google.com/drive/1d2sjWDJi_GYBUavrHSuQyHTDuLy36WpU\" target=\"_blank\">Зарегать Zrok Токен</a>', class_name=\"button_ngrok\")\n",
-        "zrok_widget = factory.create_hbox([zrok_token_widget, zrok_button])\n",
-        "\n",
-        "commandline_arguments_options = \"--listen --enable-insecure-extension-access --theme dark --no-half-vae --disable-console-progressbars --xformers\"\n",
-        "commandline_arguments_widget = factory.create_text('Arguments:', value=commandline_arguments_options)\n",
-        "\n",
-        "# Display Additional\n",
-        "additional_widget_list = [additional_header,\n",
-        "                          choose_changes_widget,\n",
-        "                          HR,\n",
-        "                          controlnet_widget,\n",
-        "                          controlnet_num_widget,\n",
-        "                          commit_hash_widget,\n",
-        "                          huggingface_token_widget,\n",
-        "                          ngrok_widget,\n",
-        "                          zrok_widget,\n",
-        "                          HR,\n",
-        "                          commandline_arguments_widget]\n",
-        "\n",
-        "if free_plan and env == \"Google Colab\": # remove ngrok from colab\n",
-        "    additional_widget_list.remove(ngrok_widget)\n",
-        "if os.path.exists(webui_path): # remove selection after selection ;3\n",
-        "    choose_changes_widget.children = [widget for widget in choose_changes_widget.children if widget != change_webui_widget]\n",
-        "\n",
-        "all_additional_box = factory.create_vbox(additional_widget_list, class_names=[\"container\", \"image_3\"])\n",
-        "factory.display(all_additional_box)\n",
-        "\n",
-        "# --- CUSTOM DOWNLOAD ---\n",
-        "custom_download_header_popup = factory.create_html('''\n",
-        "<style>\n",
-        "/* Term Colors */\n",
-        ".sample_label {color: #dbafff;}\n",
-        ".braces {color: #ffff00;}\n",
-        ".extension {color: #eb934b;}\n",
-        ".file_name {color: #ffffd8;}\n",
-        "</style>\n",
-        "\n",
-        "<div class=\"header\" style=\"cursor: pointer;\" onclick=\"toggleContainer()\">Custom Download</div>\n",
-        "<!-- PopUp Window -->\n",
-        "<div class=\"info\">INFO</div>\n",
-        "<div class=\"popup\">\n",
-        "    Separate multiple URLs with a comma/space. For a <span class=\"file_name\">custom name</span> file/extension, specify it with <span class=\"braces\">[]</span>\n",
-        "    after the URL without spaces.\n",
-        "    <span style=\"color: #ff9999\">For files, be sure to specify</span> - <span class=\"extension\">Filename Extension.</span>\n",
-        "    <div class=\"sample\">\n",
-        "        <span class=\"sample_label\">Example for File:</span>\n",
-        "        https://civitai.com/api/download/models/229782<span class=\"braces\">[</span><span class=\"file_name\">Detailer</span><span class=\"extension\">.safetensors</span><span class=\"braces\">]</span>\n",
-        "        <br>\n",
-        "        <span class=\"sample_label\">Example for Extension:</span>\n",
-        "        https://github.com/hako-mikan/sd-webui-regional-prompter<span class=\"braces\">[</span><span class=\"file_name\">Regional-Prompter</span><span class=\"braces\">]</span>\n",
-        "    </div>\n",
-        "</div>\n",
-        "''')\n",
-        "\n",
-        "Model_url_widget = factory.create_text('Model:')\n",
-        "Vae_url_widget = factory.create_text('Vae:')\n",
-        "LoRA_url_widget = factory.create_text('LoRa:')\n",
-        "Embedding_url_widget = factory.create_text('Embedding:')\n",
-        "Extensions_url_widget = factory.create_text('Extensions:')\n",
-        "custom_file_urls_widget = factory.create_text('File (txt):')\n",
-        "\n",
-        "# Display CustomDl\n",
-        "all_custom_box = factory.create_vbox([\n",
-        "    custom_download_header_popup, Model_url_widget, Vae_url_widget, LoRA_url_widget, Embedding_url_widget, Extensions_url_widget, custom_file_urls_widget\n",
-        "], class_names=[\"container\", \"image_4\", \"container_custom_downlad\"])\n",
-        "factory.display(all_custom_box)\n",
-        "\n",
-        "# --- Save Button ---\n",
-        "save_button = factory.create_button('Save', class_name=\"button_save\")\n",
-        "factory.display(save_button)\n",
-        "\n",
-        "\n",
-        "# ============ Load / Save - Settings V2 ============\n",
-        "SETTINGS_FILE = f'{root_path}/settings.json'\n",
-        "\n",
-        "SETTINGS_KEYS = [\n",
-        "    'model', 'model_num', 'inpainting_model',\n",
-        "    'vae', 'vae_num', 'latest_webui', 'latest_exstensions',\n",
-        "    'change_webui', 'detailed_download', 'controlnet',\n",
-        "    'controlnet_num', 'commit_hash', 'huggingface_token',\n",
-        "    'ngrok_token', 'zrok_token', 'commandline_arguments',\n",
-        "    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url',\n",
-        "    'Extensions_url', 'custom_file_urls'\n",
-        "]\n",
-        "\n",
-        "def save_settings():\n",
-        "    settings = {key: globals()[f\"{key}_widget\"].value for key in SETTINGS_KEYS}\n",
-        "    with open(SETTINGS_FILE, 'w') as f:\n",
-        "        json.dump(settings, f, indent=2)\n",
-        "\n",
-        "def load_settings():\n",
-        "    if os.path.exists(SETTINGS_FILE):\n",
-        "        with open(SETTINGS_FILE, 'r') as f:\n",
-        "            settings = json.load(f)\n",
-        "            for key in SETTINGS_KEYS:\n",
-        "                globals()[f\"{key}_widget\"].value = settings.get(key, \"\")\n",
-        "\n",
-        "def hide_widgets():\n",
-        "    widgets_list = [all_model_box, all_vae_box, all_additional_box, all_custom_box, save_button]\n",
-        "    for widget in widgets_list:\n",
-        "        widget.add_class(\"hide\")\n",
-        "    time.sleep(0.5)\n",
-        "    widgets.Widget.close_all()\n",
-        "\n",
-        "def save_data(button):\n",
-        "    save_settings()\n",
-        "    hide_widgets()\n",
-        "\n",
-        "load_settings()\n",
-        "save_button.on_click(save_data)"
-      ]
-    }
-  ]
+{
+  "nbformat": 4,
+  "nbformat_minor": 0,
+  "metadata": {
+    "colab": {
+      "provenance": []
+    },
+    "kernelspec": {
+      "name": "python3",
+      "display_name": "Python 3"
+    },
+    "language_info": {
+      "name": "python"
+    }
+  },
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "JKTCrY9LU7Oq"
+      },
+      "outputs": [],
+      "source": [
+        "##~   WIDGET CODE | BY: ANXETY   ~##\n",
+        "\n",
+        "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 IPython.display import display, HTML, Javascript, clear_output\n",
+        "\n",
+        "\n",
+        "# Setup Env\n",
+        "env = os.environ.get('ENV_NAME')\n",
+        "root_path = os.environ.get('ROOT_PATH')\n",
+        "webui_path = os.environ.get('WEBUI_PATH')\n",
+        "free_plan = os.environ.get('FREE_PLAN')\n",
+        "\n",
+        "\n",
+        "# ==================== CSS JS ====================\n",
+        "##~ custom background images V1.5 ~##\n",
+        "import argparse\n",
+        "parser = argparse.ArgumentParser(description='This script processes an background image.')\n",
+        "parser.add_argument('-i', '--image', type=str, help='URL of the image to process', metavar='')\n",
+        "parser.add_argument('-o', '--opacity', type=float, help='Opacity level for the image, between 0 and 1', metavar='', default=0.3)\n",
+        "parser.add_argument('-b', '--blur', type=str, help='Blur level for the image', metavar='', default=0)\n",
+        "parser.add_argument('-y', type=int, help='Y coordinate for the image in px', metavar='', default=0)\n",
+        "parser.add_argument('-x', type=int, help='X coordinate for the image in px', metavar='', default=0)\n",
+        "parser.add_argument('-s', '--scale', type=int, help='Scale image in %%', metavar='', default=100)\n",
+        "parser.add_argument('-m', '--mode', action='store_true', help='Removes repetitive image tiles')\n",
+        "parser.add_argument('-t', '--transparent', action='store_true', help='Makes input/selection fields 35%% more transparent')\n",
+        "parser.add_argument('-bf', '--blur-fields', type=str, help='Background blur level for input/selection fields', metavar='', default=2)\n",
+        "\n",
+        "args = parser.parse_args()\n",
+        "\n",
+        "url_img = args.image\n",
+        "opacity_img = args.opacity\n",
+        "blur_img = args.blur\n",
+        "y_img = args.y\n",
+        "x_img = args.x\n",
+        "scale_img = args.scale\n",
+        "blur_fields = args.blur_fields\n",
+        "\n",
+        "## ---\n",
+        "\"\"\" WTF KAGGLE - WHAT THE FUCK IS THE DIFFERENCE OF 35 PIXELS!?!?!? \"\"\"\n",
+        "fix_heigh_img = \"-810px\" if env == \"Kaggle\" else \"-775px\"\n",
+        "\n",
+        "\"\"\" transperent fields \"\"\"\n",
+        "t_bg_alpha = \"1\" if not args.transparent else \"0.65\"\n",
+        "\n",
+        "\"\"\" mode img - repeats \"\"\"\n",
+        "mode_img = \"repeat\" if not args.mode else \"no-repeat\"\n",
+        "\n",
+        "container_background = f'''\n",
+        "<style>\n",
+        ":root {{\n",
+        "  /* for background container*/\n",
+        "  --img_background: url({url_img});\n",
+        "  --img_opacity: {opacity_img};\n",
+        "  --img_blur: {blur_img}px;\n",
+        "  --image_y: {y_img}px;\n",
+        "  --image_x: {x_img}px;\n",
+        "  --img_scale: {scale_img}%;\n",
+        "  --img_mode: {mode_img};\n",
+        "  --img_height_dif: {fix_heigh_img};\n",
+        "\n",
+        "  /* for fields */\n",
+        "  --bg-field-color: rgba(28, 28, 28, {t_bg_alpha}); /* -> #1c1c1c */\n",
+        "  --bg-field-color-hover: rgba(38, 38, 38, {t_bg_alpha}); /* -> #262626; */\n",
+        "  --bg-field-blur-level: {blur_fields}px;\n",
+        "}}\n",
+        "'''\n",
+        "\n",
+        "display(HTML(container_background))\n",
+        "\n",
+        "# Main CSS\n",
+        "css_file_path = f\"{root_path}/CSS/main_widgets.css\"\n",
+        "with open(css_file_path , \"r\") as f:\n",
+        "    CSS = f.read()\n",
+        "display(HTML(f\"<style>{CSS}</style>\"))\n",
+        "\n",
+        "# Main JS\n",
+        "JS = '''\n",
+        "<!-- TOGGLE 'CustomDL' SCRIPT -->\n",
+        "<script>\n",
+        "function toggleContainer() {\n",
+        "    let downloadContainer = document.querySelector('.container_custom_downlad');\n",
+        "    let info = document.querySelector('.info');\n",
+        "\n",
+        "    downloadContainer.classList.toggle('expanded');\n",
+        "    info.classList.toggle('showed');\n",
+        "}\n",
+        "</script>\n",
+        "'''\n",
+        "display(HTML(JS))\n",
+        "\n",
+        "# ==================== WIDGETS V2 ====================\n",
+        "HR = widgets.HTML('<hr>')\n",
+        "\n",
+        "class WidgetFactory:\n",
+        "    def __init__(self, style=None, layout=None):\n",
+        "        self.style = style if style else {'description_width': 'initial'}\n",
+        "        self.layout = layout if layout else widgets.Layout(max_width='1080px', width='100%')\n",
+        "\n",
+        "    def create_html(self, content, class_name=None):\n",
+        "        html_widget = widgets.HTML(content)\n",
+        "        if class_name:\n",
+        "            html_widget.add_class(class_name)\n",
+        "        return html_widget\n",
+        "\n",
+        "    def create_header(self, name):\n",
+        "        return widgets.HTML(f'<div class=\"header\">{name}<div>')\n",
+        "\n",
+        "    def create_dropdown(self, options, value, description):\n",
+        "        return widgets.Dropdown(options=options, value=value, description=description, style=self.style, layout=self.layout)\n",
+        "\n",
+        "    def create_text(self, description, placeholder='', value=''):\n",
+        "        return widgets.Text(description=description, placeholder=placeholder, value=value, style=self.style, layout=self.layout)\n",
+        "\n",
+        "    def create_checkbox(self, value, description):\n",
+        "        return widgets.Checkbox(value=value, description=description, style=self.style, layout=self.layout)\n",
+        "\n",
+        "    def create_button(self, description, class_name=None):\n",
+        "        button = widgets.Button(description=description)\n",
+        "        if class_name:\n",
+        "            button.add_class(class_name)\n",
+        "        return button\n",
+        "\n",
+        "    def create_hbox(self, children):\n",
+        "        return widgets.HBox(children)\n",
+        "\n",
+        "    def create_vbox(self, children, class_names=None):\n",
+        "        vbox = widgets.VBox(children)\n",
+        "        if class_names:\n",
+        "            for class_name in class_names:\n",
+        "                vbox.add_class(class_name)\n",
+        "        return vbox\n",
+        "\n",
+        "    def display(self, widget):\n",
+        "        display(widget)\n",
+        "\n",
+        "# Instantiate the factory\n",
+        "factory = WidgetFactory()\n",
+        "\n",
+        "# --- MODEL ---\n",
+        "model_header = factory.create_header('Model Selection')\n",
+        "model_options = ['none',\n",
+        "                 '1.Anime (by XpucT) + INP',\n",
+        "                 '2.BluMix [Anime] [V7] + INP',\n",
+        "                 '3.Cetus-Mix [Anime] [V4] + INP',\n",
+        "                 '4.Counterfeit [Anime] [V3] + INP',\n",
+        "                 '5.CuteColor [Anime] [V3]',\n",
+        "                 '6.Dark-Sushi-Mix [Anime]',\n",
+        "                 '7.Deliberate [Realism] [V6] + INP',\n",
+        "                 '8.Meina-Mix [Anime] [V11] + INP',\n",
+        "                 '9.Mix-Pro [Anime] [V4] + INP']\n",
+        "model_widget = factory.create_dropdown(model_options, '4.Counterfeit [Anime] [V3] + INP', 'Model:')\n",
+        "model_num_widget = factory.create_text('Model Number:', 'Enter the model numbers to be downloaded using comma/space.')\n",
+        "inpainting_model_widget = factory.create_checkbox(False, 'Inpainting Models')\n",
+        "\n",
+        "# Display Model\n",
+        "all_model_box = factory.create_vbox([model_header, model_widget, model_num_widget, inpainting_model_widget], class_names=[\"container\", \"image_1\"])\n",
+        "factory.display(all_model_box)\n",
+        "\n",
+        "# --- VAE ---\n",
+        "vae_header = factory.create_header('VAE Selection')\n",
+        "vae_options = ['none',\n",
+        "               '1.Anime.vae',\n",
+        "               '2.Anything.vae',\n",
+        "               '3.Blessed2.vae',\n",
+        "               '4.ClearVae.vae',\n",
+        "               '5.WD.vae']\n",
+        "vae_widget = factory.create_dropdown(vae_options, '3.Blessed2.vae', 'Vae:')\n",
+        "vae_num_widget = factory.create_text('Vae Number:', 'Enter the vae numbers to be downloaded using comma/space.')\n",
+        "\n",
+        "# Display Vae\n",
+        "all_vae_box = factory.create_vbox([vae_header, vae_widget, vae_num_widget], class_names=[\"container\", \"image_2\"])\n",
+        "factory.display(all_vae_box)\n",
+        "\n",
+        "# --- ADDITIONAL ---\n",
+        "additional_header = factory.create_header('Additional')\n",
+        "latest_webui_widget = factory.create_checkbox(True, 'Update WebUI')\n",
+        "latest_exstensions_widget = factory.create_checkbox(True, 'Update Extensions')\n",
+        "change_webui_widget = factory.create_dropdown(['A1111', 'Forge'], 'A1111', 'Change WebUI:')\n",
+        "detailed_download_widget = factory.create_dropdown(['off', 'on'], 'off', 'Detailed Download:')\n",
+        "choose_changes_widget = factory.create_hbox([latest_webui_widget, latest_exstensions_widget, change_webui_widget, detailed_download_widget])\n",
+        "\n",
+        "controlnet_options = ['none', 'ALL', '1.canny',\n",
+        "                      '2.openpose', '3.depth',\n",
+        "                      '4.normal_map', '5.mlsd',\n",
+        "                      '6.lineart', '7.soft_edge',\n",
+        "                      '8.scribble', '9.segmentation',\n",
+        "                      '10.shuffle', '11.tile',\n",
+        "                      '12.inpaint', '13.instruct_p2p']\n",
+        "controlnet_widget = factory.create_dropdown(controlnet_options, 'none', 'ControlNet:')\n",
+        "controlnet_num_widget = factory.create_text('ControlNet Number:', 'Enter the ControlNet model numbers to be downloaded using comma/space.')\n",
+        "commit_hash_widget = factory.create_text('Commit Hash:')\n",
+        "huggingface_token_widget = factory.create_text('HuggingFace Token:')\n",
+        "\n",
+        "ngrok_token_widget = factory.create_text('Ngrok Token:')\n",
+        "ngrock_button = factory.create_html('<a href=\"https://dashboard.ngrok.com/get-started/your-authtoken\" target=\"_blank\">Получить Ngrok Токен</a>', class_name=\"button_ngrok\")\n",
+        "ngrok_widget = factory.create_hbox([ngrok_token_widget, ngrock_button])\n",
+        "\n",
+        "zrok_token_widget = factory.create_text('Zrok Token:')\n",
+        "zrok_button = factory.create_html('<a href=\"https://colab.research.google.com/drive/1d2sjWDJi_GYBUavrHSuQyHTDuLy36WpU\" target=\"_blank\">Зарегать Zrok Токен</a>', class_name=\"button_ngrok\")\n",
+        "zrok_widget = factory.create_hbox([zrok_token_widget, zrok_button])\n",
+        "\n",
+        "commandline_arguments_options = \"--listen --enable-insecure-extension-access --theme dark --no-half-vae --disable-console-progressbars --xformers\"\n",
+        "commandline_arguments_widget = factory.create_text('Arguments:', value=commandline_arguments_options)\n",
+        "\n",
+        "# Display Additional\n",
+        "additional_widget_list = [additional_header,\n",
+        "                          choose_changes_widget,\n",
+        "                          HR,\n",
+        "                          controlnet_widget,\n",
+        "                          controlnet_num_widget,\n",
+        "                          commit_hash_widget,\n",
+        "                          huggingface_token_widget,\n",
+        "                          ngrok_widget,\n",
+        "                          zrok_widget,\n",
+        "                          HR,\n",
+        "                          commandline_arguments_widget]\n",
+        "\n",
+        "if free_plan and env == \"Google Colab\": # remove ngrok from colab\n",
+        "    additional_widget_list.remove(ngrok_widget)\n",
+        "if os.path.exists(webui_path): # remove selection after selection ;3\n",
+        "    choose_changes_widget.children = [widget for widget in choose_changes_widget.children if widget != change_webui_widget]\n",
+        "\n",
+        "all_additional_box = factory.create_vbox(additional_widget_list, class_names=[\"container\", \"image_3\"])\n",
+        "factory.display(all_additional_box)\n",
+        "\n",
+        "# --- CUSTOM DOWNLOAD ---\n",
+        "custom_download_header_popup = factory.create_html('''\n",
+        "<style>\n",
+        "/* Term Colors */\n",
+        ".sample_label {color: #dbafff;}\n",
+        ".braces {color: #ffff00;}\n",
+        ".extension {color: #eb934b;}\n",
+        ".file_name {color: #ffffd8;}\n",
+        "</style>\n",
+        "\n",
+        "<div class=\"header\" style=\"cursor: pointer;\" onclick=\"toggleContainer()\">Custom Download</div>\n",
+        "<!-- PopUp Window -->\n",
+        "<div class=\"info\">INFO</div>\n",
+        "<div class=\"popup\">\n",
+        "    Separate multiple URLs with a comma/space. For a <span class=\"file_name\">custom name</span> file/extension, specify it with <span class=\"braces\">[]</span>\n",
+        "    after the URL without spaces.\n",
+        "    <span style=\"color: #ff9999\">For files, be sure to specify</span> - <span class=\"extension\">Filename Extension.</span>\n",
+        "    <div class=\"sample\">\n",
+        "        <span class=\"sample_label\">Example for File:</span>\n",
+        "        https://civitai.com/api/download/models/229782<span class=\"braces\">[</span><span class=\"file_name\">Detailer</span><span class=\"extension\">.safetensors</span><span class=\"braces\">]</span>\n",
+        "        <br>\n",
+        "        <span class=\"sample_label\">Example for Extension:</span>\n",
+        "        https://github.com/hako-mikan/sd-webui-regional-prompter<span class=\"braces\">[</span><span class=\"file_name\">Regional-Prompter</span><span class=\"braces\">]</span>\n",
+        "    </div>\n",
+        "</div>\n",
+        "''')\n",
+        "\n",
+        "Model_url_widget = factory.create_text('Model:')\n",
+        "Vae_url_widget = factory.create_text('Vae:')\n",
+        "LoRA_url_widget = factory.create_text('LoRa:')\n",
+        "Embedding_url_widget = factory.create_text('Embedding:')\n",
+        "Extensions_url_widget = factory.create_text('Extensions:')\n",
+        "custom_file_urls_widget = factory.create_text('File (txt):')\n",
+        "\n",
+        "# Display CustomDl\n",
+        "all_custom_box = factory.create_vbox([\n",
+        "    custom_download_header_popup, Model_url_widget, Vae_url_widget, LoRA_url_widget, Embedding_url_widget, Extensions_url_widget, custom_file_urls_widget\n",
+        "], class_names=[\"container\", \"image_4\", \"container_custom_downlad\"])\n",
+        "factory.display(all_custom_box)\n",
+        "\n",
+        "# --- Save Button ---\n",
+        "save_button = factory.create_button('Save', class_name=\"button_save\")\n",
+        "factory.display(save_button)\n",
+        "\n",
+        "\n",
+        "# ============ Load / Save - Settings V2 ============\n",
+        "SETTINGS_FILE = f'{root_path}/settings.json'\n",
+        "\n",
+        "SETTINGS_KEYS = [\n",
+        "    'model', 'model_num', 'inpainting_model',\n",
+        "    'vae', 'vae_num', 'latest_webui', 'latest_exstensions',\n",
+        "    'change_webui', 'detailed_download', 'controlnet',\n",
+        "    'controlnet_num', 'commit_hash', 'huggingface_token',\n",
+        "    'ngrok_token', 'zrok_token', 'commandline_arguments',\n",
+        "    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url',\n",
+        "    'Extensions_url', 'custom_file_urls'\n",
+        "]\n",
+        "\n",
+        "def save_settings():\n",
+        "    settings = {key: globals()[f\"{key}_widget\"].value for key in SETTINGS_KEYS}\n",
+        "    with open(SETTINGS_FILE, 'w') as f:\n",
+        "        json.dump(settings, f, indent=2)\n",
+        "\n",
+        "def load_settings():\n",
+        "    if os.path.exists(SETTINGS_FILE):\n",
+        "        with open(SETTINGS_FILE, 'r') as f:\n",
+        "            settings = json.load(f)\n",
+        "            for key in SETTINGS_KEYS:\n",
+        "                globals()[f\"{key}_widget\"].value = settings.get(key, \"\")\n",
+        "\n",
+        "def hide_widgets():\n",
+        "    widgets_list = [all_model_box, all_vae_box, all_additional_box, all_custom_box, save_button]\n",
+        "    for widget in widgets_list:\n",
+        "        widget.add_class(\"hide\")\n",
+        "    time.sleep(0.5)\n",
+        "    widgets.Widget.close_all()\n",
+        "\n",
+        "def save_data(button):\n",
+        "    save_settings()\n",
+        "    hide_widgets()\n",
+        "\n",
+        "load_settings()\n",
+        "save_button.on_click(save_data)"
+      ]
+    }
+  ]
 }

files_cells/notebooks/ru/auto_cleaner_ru.ipynb

@@ -1,165 +1,157 @@
-{
-  "nbformat": 4,
-  "nbformat_minor": 0,
-  "metadata": {
-    "colab": {
-      "provenance": []
-    },
-    "kernelspec": {
-      "name": "python3",
-      "display_name": "Python 3"
-    },
-    "language_info": {
-      "name": "python"
-    }
-  },
-  "cells": [
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "JKTCrY9LU7Oq"
-      },
-      "outputs": [],
-      "source": [
-        "##~ AutoCleaner V3.7 CODE | BY: ANXETY   ~##\n",
-        "\n",
-        "import os\n",
-        "import time\n",
-        "import ipywidgets as widgets\n",
-        "from ipywidgets import Label, Button, VBox, HBox\n",
-        "from IPython.display import display, HTML\n",
-        "\n",
-        "\n",
-        "# ================= DETECT ENV =================\n",
-        "def detect_environment():\n",
-        "    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)\n",
-        "    environments = {\n",
-        "        'COLAB_GPU': ('Google Colab', \"/root\" if free_plan else \"/content\"),\n",
-        "        'KAGGLE_URL_BASE': ('Kaggle', \"/kaggle/working/content\")\n",
-        "    }\n",
-        "    for env_var, (environment, path) in environments.items():\n",
-        "        if env_var in os.environ:\n",
-        "            return environment, path, free_plan\n",
-        "\n",
-        "env, root_path, free_plan = detect_environment()\n",
-        "webui_path = f\"{root_path}/sdw\"\n",
-        "\n",
-        "\n",
-        "# ==================== CSS ====================\n",
-        "# Main CSS\n",
-        "css_file_path = f\"{root_path}/CSS/auto_cleaner.css\"\n",
-        "with open(css_file_path , \"r\") as f:\n",
-        "    CSS_AC = f.read()\n",
-        "display(HTML(f\"<style>{CSS_AC}</style>\"))\n",
-        "\n",
-        "\n",
-        "# ================ AutoCleaner function ================\n",
-        "directories = {\n",
-        "    \"Изображения\": f\"{webui_path}/output\",\n",
-        "    \"Модели\": f\"{webui_path}/models/Stable-diffusion/\",\n",
-        "    \"Vae\": f\"{webui_path}/models/VAE/\",\n",
-        "    \"LoRa\": f\"{webui_path}/models/Lora/\",\n",
-        "    \"ControlNet Модели\": f\"{webui_path}/models/ControlNet/\"\n",
-        "}\n",
-        "\n",
-        "\"\"\" functions \"\"\"\n",
-        "def clean_directory(directory):\n",
-        "    deleted_files = 0\n",
-        "    image_dir = directories['Изображения']\n",
-        "\n",
-        "    for root, dirs, files in os.walk(directory):\n",
-        "        for file in files:\n",
-        "            file_path = os.path.join(root, file)\n",
-        "\n",
-        "            if file.endswith(\".txt\"):\n",
-        "                continue\n",
-        "            if file.endswith((\".safetensors\", \".pt\")) or root == image_dir: # fix for image counter\n",
-        "                deleted_files += 1\n",
-        "\n",
-        "            os.remove(file_path)\n",
-        "    return deleted_files\n",
-        "\n",
-        "def update_memory_info():\n",
-        "    disk_space = psutil.disk_usage(os.getcwd())\n",
-        "    total = disk_space.total / (1024 ** 3)\n",
-        "    used = disk_space.used / (1024 ** 3)\n",
-        "    free = disk_space.free / (1024 ** 3)\n",
-        "\n",
-        "    storage_info.value = f'''\n",
-        "    <div class=\"storage_info_AC\">Всего: {total:.2f} GB <span style=\"color: #555\">|</span> Используется: {used:.2f} GB <span style=\"color: #555\">|</span> Свободно: {free:.2f} GB</div>\n",
-        "    '''\n",
-        "\n",
-        "def on_execute_button_press(button):\n",
-        "    selected_cleaners = auto_cleaner_widget.value\n",
-        "    deleted_files_dict = {}\n",
-        "\n",
-        "    for option in selected_cleaners:\n",
-        "        if option in directories:\n",
-        "            deleted_files_dict[option] = clean_directory(directories[option])\n",
-        "\n",
-        "    output.clear_output()\n",
-        "\n",
-        "    with output:\n",
-        "        for message in generate_messages(deleted_files_dict):\n",
-        "            message_widget = HTML(f'<p class=\"output_message_AC\">{message}</p>')\n",
-        "            display(message_widget)\n",
-        "\n",
-        "    update_memory_info()\n",
-        "\n",
-        "def on_clear_button_press(button):\n",
-        "    container.add_class(\"hide\")\n",
-        "    time.sleep(0.5)\n",
-        "    widgets.Widget.close_all()\n",
-        "\n",
-        "def generate_messages(deleted_files_dict):\n",
-        "    messages = []\n",
-        "    word_variants = {\n",
-        "        \"Изображения\": \"Изображений\",\n",
-        "        \"Модели\": \"Моделей\",\n",
-        "        \"Vae\": \"Vae\",\n",
-        "        \"LoRa\": \"LoRa\",\n",
-        "        \"ControlNet Модели\": \"ControlNet Моделей\"\n",
-        "    }\n",
-        "    for key, value in deleted_files_dict.items():\n",
-        "        object_word = word_variants.get(key)\n",
-        "        messages.append(f\"Удалено {value} {object_word}\")\n",
-        "    return messages\n",
-        "\n",
-        "\n",
-        "# --- storage memory ---\n",
-        "import psutil\n",
-        "disk_space = psutil.disk_usage(os.getcwd())\n",
-        "total = disk_space.total / (1024 ** 3)\n",
-        "used = disk_space.used / (1024 ** 3)\n",
-        "free = disk_space.free / (1024 ** 3)\n",
-        "\n",
-        "\n",
-        "# ================ Widgets ================\n",
-        "# UI Code\n",
-        "AutoCleaner_options = AutoCleaner_options = list(directories.keys())\n",
-        "instruction_label = widgets.HTML('''\n",
-        "<span class=\"instruction_AC\">Используйте <span style=\"color: #B2B2B2;\">ctrl</span> или <span style=\"color: #B2B2B2;\">shift</span> для множественного выбора.</span>\n",
-        "''')\n",
-        "auto_cleaner_widget = widgets.SelectMultiple(options=AutoCleaner_options, layout=widgets.Layout(width=\"auto\")).add_class(\"custom-select-multiple_AC\")\n",
-        "output = widgets.Output().add_class(\"output_AC\")\n",
-        "\n",
-        "execute_button = Button(description='Выполнить Очистку').add_class(\"button_execute_AC\").add_class(\"button_AC\")\n",
-        "execute_button.on_click(on_execute_button_press)\n",
-        "clear_button = Button(description='Скрыть Виджет').add_class(\"button_clear_AC\").add_class(\"button_AC\")\n",
-        "clear_button.on_click(on_clear_button_press)\n",
-        "\n",
-        "storage_info = widgets.HTML(f'''\n",
-        "<div class=\"storage_info_AC\">Всего: {total:.2f} GB <span style=\"color: #555\">|</span> Используется: {used:.2f} GB <span style=\"color: #555\">|</span> Свободно: {free:.2f} GB</div>\n",
-        "''')\n",
-        "\n",
-        "buttons = widgets.HBox([execute_button, clear_button])\n",
-        "lower_information_panel = widgets.HBox([buttons, storage_info]).add_class(\"lower_information_panel_AC\")\n",
-        "\n",
-        "container = VBox([instruction_label, widgets.HTML('<hr>'), auto_cleaner_widget, output, widgets.HTML('<hr>'), lower_information_panel]).add_class(\"container_AC\")\n",
-        "\n",
-        "display(container)"
-      ]
-    }
-  ]
+{
+  "nbformat": 4,
+  "nbformat_minor": 0,
+  "metadata": {
+    "colab": {
+      "provenance": []
+    },
+    "kernelspec": {
+      "name": "python3",
+      "display_name": "Python 3"
+    },
+    "language_info": {
+      "name": "python"
+    }
+  },
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "JKTCrY9LU7Oq"
+      },
+      "outputs": [],
+      "source": [
+        "##~ AutoCleaner V3.7 CODE | BY: ANXETY   ~##\n",
+        "\n",
+        "import os\n",
+        "import time\n",
+        "import ipywidgets as widgets\n",
+        "from ipywidgets import Label, Button, VBox, HBox\n",
+        "from IPython.display import display, HTML\n",
+        "\n",
+        "\n",
+        "# Setup Env\n",
+        "env = os.environ.get('ENV_NAME')\n",
+        "root_path = os.environ.get('ROOT_PATH')\n",
+        "webui_path = os.environ.get('WEBUI_PATH')\n",
+        "free_plan = os.environ.get('FREE_PLAN')\n",
+        "\n",
+        "\n",
+        "# ==================== CSS ====================\n",
+        "# Main CSS\n",
+        "css_file_path = f\"{root_path}/CSS/auto_cleaner.css\"\n",
+        "with open(css_file_path , \"r\") as f:\n",
+        "    CSS_AC = f.read()\n",
+        "display(HTML(f\"<style>{CSS_AC}</style>\"))\n",
+        "\n",
+        "\n",
+        "# ================ AutoCleaner function ================\n",
+        "directories = {\n",
+        "    \"Изображения\": f\"{webui_path}/output\",\n",
+        "    \"Модели\": f\"{webui_path}/models/Stable-diffusion/\",\n",
+        "    \"Vae\": f\"{webui_path}/models/VAE/\",\n",
+        "    \"LoRa\": f\"{webui_path}/models/Lora/\",\n",
+        "    \"ControlNet Модели\": f\"{webui_path}/models/ControlNet/\"\n",
+        "}\n",
+        "\n",
+        "\"\"\" functions \"\"\"\n",
+        "def clean_directory(directory):\n",
+        "    deleted_files = 0\n",
+        "    image_dir = directories['Изображения']\n",
+        "\n",
+        "    for root, dirs, files in os.walk(directory):\n",
+        "        for file in files:\n",
+        "            file_path = os.path.join(root, file)\n",
+        "\n",
+        "            if file.endswith(\".txt\"):\n",
+        "                continue\n",
+        "            if file.endswith((\".safetensors\", \".pt\")) or root == image_dir: # fix for image counter\n",
+        "                deleted_files += 1\n",
+        "\n",
+        "            os.remove(file_path)\n",
+        "    return deleted_files\n",
+        "\n",
+        "def update_memory_info():\n",
+        "    disk_space = psutil.disk_usage(os.getcwd())\n",
+        "    total = disk_space.total / (1024 ** 3)\n",
+        "    used = disk_space.used / (1024 ** 3)\n",
+        "    free = disk_space.free / (1024 ** 3)\n",
+        "\n",
+        "    storage_info.value = f'''\n",
+        "    <div class=\"storage_info_AC\">Всего: {total:.2f} GB <span style=\"color: #555\">|</span> Используется: {used:.2f} GB <span style=\"color: #555\">|</span> Свободно: {free:.2f} GB</div>\n",
+        "    '''\n",
+        "\n",
+        "def on_execute_button_press(button):\n",
+        "    selected_cleaners = auto_cleaner_widget.value\n",
+        "    deleted_files_dict = {}\n",
+        "\n",
+        "    for option in selected_cleaners:\n",
+        "        if option in directories:\n",
+        "            deleted_files_dict[option] = clean_directory(directories[option])\n",
+        "\n",
+        "    output.clear_output()\n",
+        "\n",
+        "    with output:\n",
+        "        for message in generate_messages(deleted_files_dict):\n",
+        "            message_widget = HTML(f'<p class=\"output_message_AC\">{message}</p>')\n",
+        "            display(message_widget)\n",
+        "\n",
+        "    update_memory_info()\n",
+        "\n",
+        "def on_clear_button_press(button):\n",
+        "    container.add_class(\"hide\")\n",
+        "    time.sleep(0.5)\n",
+        "    widgets.Widget.close_all()\n",
+        "\n",
+        "def generate_messages(deleted_files_dict):\n",
+        "    messages = []\n",
+        "    word_variants = {\n",
+        "        \"Изображения\": \"Изображений\",\n",
+        "        \"Модели\": \"Моделей\",\n",
+        "        \"Vae\": \"Vae\",\n",
+        "        \"LoRa\": \"LoRa\",\n",
+        "        \"ControlNet Модели\": \"ControlNet Моделей\"\n",
+        "    }\n",
+        "    for key, value in deleted_files_dict.items():\n",
+        "        object_word = word_variants.get(key)\n",
+        "        messages.append(f\"Удалено {value} {object_word}\")\n",
+        "    return messages\n",
+        "\n",
+        "\n",
+        "# --- storage memory ---\n",
+        "import psutil\n",
+        "disk_space = psutil.disk_usage(os.getcwd())\n",
+        "total = disk_space.total / (1024 ** 3)\n",
+        "used = disk_space.used / (1024 ** 3)\n",
+        "free = disk_space.free / (1024 ** 3)\n",
+        "\n",
+        "\n",
+        "# ================ Widgets ================\n",
+        "# UI Code\n",
+        "AutoCleaner_options = AutoCleaner_options = list(directories.keys())\n",
+        "instruction_label = widgets.HTML('''\n",
+        "<span class=\"instruction_AC\">Используйте <span style=\"color: #B2B2B2;\">ctrl</span> или <span style=\"color: #B2B2B2;\">shift</span> для множественного выбора.</span>\n",
+        "''')\n",
+        "auto_cleaner_widget = widgets.SelectMultiple(options=AutoCleaner_options, layout=widgets.Layout(width=\"auto\")).add_class(\"custom-select-multiple_AC\")\n",
+        "output = widgets.Output().add_class(\"output_AC\")\n",
+        "\n",
+        "execute_button = Button(description='Выполнить Очистку').add_class(\"button_execute_AC\").add_class(\"button_AC\")\n",
+        "execute_button.on_click(on_execute_button_press)\n",
+        "clear_button = Button(description='Скрыть Виджет').add_class(\"button_clear_AC\").add_class(\"button_AC\")\n",
+        "clear_button.on_click(on_clear_button_press)\n",
+        "\n",
+        "storage_info = widgets.HTML(f'''\n",
+        "<div class=\"storage_info_AC\">Всего: {total:.2f} GB <span style=\"color: #555\">|</span> Используется: {used:.2f} GB <span style=\"color: #555\">|</span> Свободно: {free:.2f} GB</div>\n",
+        "''')\n",
+        "\n",
+        "buttons = widgets.HBox([execute_button, clear_button])\n",
+        "lower_information_panel = widgets.HBox([buttons, storage_info]).add_class(\"lower_information_panel_AC\")\n",
+        "\n",
+        "container = VBox([instruction_label, widgets.HTML('<hr>'), auto_cleaner_widget, output, widgets.HTML('<hr>'), lower_information_panel]).add_class(\"container_AC\")\n",
+        "\n",
+        "display(container)"
+      ]
+    }
+  ]
 }

files_cells/notebooks/ru/downloading_ru.ipynb

@@ -10,6 +10,8 @@
       "source": [
         "##~   DOWNLOADING CODE | BY: ANXETY   ~##\n",
         "\n",
+        "from models_data import model_list, vae_list, controlnet_list\n",
+        "\n",
         "import os\n",
         "import re\n",
         "import time\n",
@@ -25,19 +27,11 @@
         "from urllib.parse import urlparse, parse_qs\n",
         "\n",
         "\n",
-        "# ================= DETECT ENV =================\n",
-        "def detect_environment():\n",
-        "    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)\n",
-        "    environments = {\n",
-        "        'COLAB_GPU': ('Google Colab', \"/root\" if free_plan else \"/content\"),\n",
-        "        'KAGGLE_URL_BASE': ('Kaggle', \"/kaggle/working/content\")\n",
-        "    }\n",
-        "    for env_var, (environment, path) in environments.items():\n",
-        "        if env_var in os.environ:\n",
-        "            return environment, path, free_plan\n",
-        "\n",
-        "env, root_path, free_plan = detect_environment()\n",
-        "webui_path = f\"{root_path}/sdw\"\n",
+        "# Setup Env\n",
+        "env = os.environ.get('ENV_NAME')\n",
+        "root_path = os.environ.get('ROOT_PATH')\n",
+        "webui_path = os.environ.get('WEBUI_PATH')\n",
+        "free_plan = os.environ.get('FREE_PLAN')\n",
         "\n",
         "\n",
         "# ================ LIBRARIES V2 ================\n",
@@ -207,112 +201,9 @@
         "\n",
         "## Downloading model and stuff | oh~ Hey! If you're freaked out by that code too, don't worry, me too!\n",
         "print(\"📦 Скачивание моделей и прочего...\", end='')\n",
-        "model_list = {\n",
-        "    \"1.Anime (by XpucT) + INP\": [\n",
-        "        {\"url\": \"https://huggingface.co/XpucT/Anime/resolve/main/Anime_v2.safetensors\", \"name\": \"Anime_V2.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/XpucT/Anime/resolve/main/Anime_v2-inpainting.safetensors\", \"name\": \"Anime_V2-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"2.BluMix [Anime] [V7] + INP\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/361779\", \"name\": \"BluMix_V7.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/363850\", \"name\": \"BluMix_V7-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"3.Cetus-Mix [Anime] [V4] + INP\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/130298\", \"name\": \"CetusMix_V4.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/139882\", \"name\": \"CetusMix_V4-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"4.Counterfeit [Anime] [V3] + INP\": [\n",
-        "        {\"url\": \"https://huggingface.co/gsdf/Counterfeit-V3.0/resolve/main/Counterfeit-V3.0_fix_fp16.safetensors\", \"name\": \"Counterfeit_V3.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/137911\", \"name\": \"Counterfeit_V3-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"5.CuteColor [Anime] [V3]\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/138754\", \"name\": \"CuteColor_V3.safetensors\"}\n",
-        "    ],\n",
-        "    \"6.Dark-Sushi-Mix [Anime]\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/101640\", \"name\": \"DarkSushiMix_2_5D.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/56071\", \"name\": \"DarkSushiMix_colorful.safetensors\"}\n",
-        "    ],\n",
-        "    \"7.Deliberate [Realism] [V6] + INP\": [\n",
-        "        {\"url\": \"https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v6.safetensors\", \"name\": \"Deliberate_V6.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v6-inpainting.safetensors\", \"name\": \"Deliberate_V6-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"8.Meina-Mix [Anime] [V11] + INP\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/119057\", \"name\": \"MeinaMix_V11.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/120702\", \"name\": \"MeinaMix_V11-inpainting.safetensors\"}\n",
-        "    ],\n",
-        "    \"9.Mix-Pro [Anime] [V4] + INP\": [\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/125668\", \"name\": \"MixPro_V4.safetensors\"},\n",
-        "        {\"url\": \"https://civitai.com/api/download/models/139878\", \"name\": \"MixPro_V4-inpainting.safetensors\"}\n",
-        "    ]\n",
-        "}\n",
-        "\n",
-        "vae_list = {\n",
-        "    \"1.Anime.vae\": [{\"url\": \"https://civitai.com/api/download/models/311162\", \"name\": \"Anime.vae.safetensors\"}],\n",
-        "    \"2.Anything.vae\": [{\"url\": \"https://huggingface.co/NoCrypt/resources/resolve/main/VAE/any.vae.safetensors\", \"name\": \"Anything.vae.safetensors\"}],\n",
-        "    \"3.Blessed2.vae\": [{\"url\": \"https://huggingface.co/NoCrypt/resources/resolve/main/VAE/blessed2.vae.safetensors\", \"name\": \"Blessed2.vae.safetensors\"}],\n",
-        "    \"4.ClearVae.vae\": [{\"url\": \"https://civitai.com/api/download/models/88156\", \"name\": \"ClearVae_23.vae.safetensors\"}],\n",
-        "    \"5.WD.vae\": [{\"url\": \"https://huggingface.co/NoCrypt/resources/resolve/main/VAE/wd.vae.safetensors\", \"name\": \"WD.vae.safetensors\"}]\n",
-        "}\n",
-        "\n",
-        "controlnet_list = {\n",
-        "    \"1.canny\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_canny_fp16.safetensors\", \"name\": \"control_v11p_sd15_canny_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_canny_fp16.yaml\", \"name\": \"control_v11p_sd15_canny_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"2.openpose\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose_fp16.safetensors\", \"name\": \"control_v11p_sd15_openpose_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_openpose_fp16.yaml\", \"name\": \"control_v11p_sd15_openpose_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"3.depth\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors\", \"name\": \"control_v11f1p_sd15_depth_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11f1p_sd15_depth_fp16.yaml\", \"name\": \"control_v11f1p_sd15_depth_fp16.yaml\"},\n",
-        "        {\"url\": \"https://huggingface.co/NagisaNao/models/resolve/main/ControlNet_v11/control_v11p_sd15_depth_anything_fp16.safetensors\", \"name\": \"control_v11p_sd15_depth_anything_fp16.safetensors\"}\n",
-        "    ],\n",
-        "    \"4.normal_map\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors\", \"name\": \"control_v11p_sd15_normalbae_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_normalbae_fp16.yaml\", \"name\": \"control_v11p_sd15_normalbae_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"5.mlsd\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors\", \"name\": \"control_v11p_sd15_mlsd_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_mlsd_fp16.yaml\", \"name\": \"control_v11p_sd15_mlsd_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"6.lineart\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_lineart_fp16.safetensors\", \"name\": \"control_v11p_sd15_lineart_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors\", \"name\": \"control_v11p_sd15s2_lineart_anime_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_lineart_fp16.yaml\", \"name\": \"control_v11p_sd15_lineart_fp16.yaml\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15s2_lineart_anime_fp16.yaml\", \"name\": \"control_v11p_sd15s2_lineart_anime_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"7.soft_edge\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_softedge_fp16.safetensors\", \"name\": \"control_v11p_sd15_softedge_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_softedge_fp16.yaml\", \"name\": \"control_v11p_sd15_softedge_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"8.scribble\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_scribble_fp16.safetensors\", \"name\": \"control_v11p_sd15_scribble_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_scribble_fp16.yaml\", \"name\": \"control_v11p_sd15_scribble_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"9.segmentation\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_seg_fp16.safetensors\", \"name\": \"control_v11p_sd15_seg_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_seg_fp16.yaml\", \"name\": \"control_v11p_sd15_seg_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"10.shuffle\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_shuffle_fp16.safetensors\", \"name\": \"control_v11e_sd15_shuffle_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_shuffle_fp16.yaml\", \"name\": \"control_v11e_sd15_shuffle_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"11.tile\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11f1e_sd15_tile_fp16.safetensors\", \"name\": \"control_v11f1e_sd15_tile_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11f1e_sd15_tile_fp16.yaml\", \"name\": \"control_v11f1e_sd15_tile_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"12.inpaint\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors\", \"name\": \"control_v11p_sd15_inpaint_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_inpaint_fp16.yaml\", \"name\": \"control_v11p_sd15_inpaint_fp16.yaml\"}\n",
-        "    ],\n",
-        "    \"13.instruct_p2p\": [\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_ip2p_fp16.safetensors\", \"name\": \"control_v11e_sd15_ip2p_fp16.safetensors\"},\n",
-        "        {\"url\": \"https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_ip2p_fp16.yaml\", \"name\": \"control_v11e_sd15_ip2p_fp16.yaml\"}\n",
-        "    ]\n",
-        "}\n",
         "\n",
         "url = \"\"\n",
-        "prefixes = {\n",
+        "PREFIXES = {\n",
         "    \"model\": models_dir,\n",
         "    \"vae\": vaes_dir,\n",
         "    \"lora\": loras_dir,\n",
@@ -324,7 +215,7 @@
         "}\n",
         "\n",
         "extension_repo = []\n",
-        "directories = [value for key, value in prefixes.items()] # for unpucking zip files\n",
+        "directories = [value for key, value in PREFIXES.items()] # for unpucking zip files\n",
         "!mkdir -p {\" \".join(directories)}\n",
         "\n",
         "hf_token = huggingface_token if huggingface_token else \"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\"\n",
@@ -335,137 +226,123 @@
         "from math import floor\n",
         "\n",
         "def center_text(text, terminal_width=45):\n",
-        "    text_length = len(text)\n",
-        "    left_padding = floor((terminal_width - text_length) / 2)\n",
-        "    right_padding = terminal_width - text_length - left_padding\n",
-        "    return f\"\\033[1m\\033[36m{' ' * left_padding}{text}{' ' * right_padding}\\033[0m\\033[32m\"\n",
+        "    padding = (terminal_width - len(text)) // 2\n",
+        "    return f\"\\033[1m\\033[36m{' ' * padding}{text}{' ' * padding}\\033[0m\\033[32m\"\n",
         "\n",
         "def format_output(url, dst_dir, file_name):\n",
-        "    info = f\"[{file_name.split('.')[0]}]\"\n",
-        "    info = center_text(info)\n",
+        "    info = center_text(f\"[{file_name.split('.')[0]}]\")\n",
+        "    separation_line = '\\033[32m' + '---' * 20\n",
         "\n",
-        "    print(f\"\\n\\033[32m{'---'*20}]{info}[{'---'*20}\")\n",
+        "    print(f\"\\n{separation_line}{info}{separation_line}\")\n",
         "    print(f\"\\033[33mURL: \\033[34m{url}\")\n",
         "    print(f\"\\033[33mSAVE DIR: \\033[34m{dst_dir}\")\n",
         "    print(f\"\\033[33mFILE NAME: \\033[34m{file_name}\\033[0m\")\n",
         "\n",
         "''' GET CivitAi API - DATA '''\n",
         "\n",
-        "def strip_(url, file_name=None):\n",
-        "    if 'github.com' in url:\n",
-        "        if '/blob/' in url:\n",
-        "            url = url.replace('/blob/', '/raw/')\n",
-        "\n",
-        "    elif \"civitai.com\" in url:\n",
-        "        return CivitAi_API(url, file_name)\n",
-        "\n",
-        "    elif \"huggingface.co\" in url:\n",
-        "        if '/blob/' in url:\n",
-        "            url = url.replace('/blob/', '/resolve/')\n",
-        "        if '?' in url:\n",
-        "            url = url.split('?')[0]\n",
-        "\n",
-        "    return url\n",
-        "\n",
         "def CivitAi_API(url, file_name=None):\n",
-        "    support_types = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')\n",
-        "    civitai_token = \"62c0c5956b2f9defbd844d754000180b\"\n",
+        "    SUPPORT_TYPES = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')\n",
+        "    CIVITAI_TOKEN = \"62c0c5956b2f9defbd844d754000180b\"\n",
         "\n",
-        "    if '?token=' in url:\n",
-        "        url = url.split('?token=')[0]\n",
-        "    if '?type=' in url:\n",
-        "        url = url.replace('?type=', f'?token={civitai_token}&type=')\n",
-        "    else:\n",
-        "        url = f\"{url}?token={civitai_token}\"\n",
-        "\n",
-        "    # Determine model or version id\n",
-        "    if \"civitai.com/models/\" in url:\n",
-        "        if '?modelVersionId=' in url:\n",
-        "            version_id = url.split('?modelVersionId=')[1]\n",
-        "            response = requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\")\n",
-        "            # print(f\"end - https://civitai.com/api/v1/model-versions/{version_id}\")\n",
+        "    url = url.split('?token=')[0] if '?token=' in url else url\n",
+        "    url = url.replace('?type=', f'?token={CIVITAI_TOKEN}&type=') if '?type=' in url else f\"{url}?token={CIVITAI_TOKEN}\"\n",
+        "\n",
+        "    def get_model_data(url):\n",
+        "        if \"civitai.com/models/\" in url:\n",
+        "            if '?modelVersionId=' in url:\n",
+        "                version_id = url.split('?modelVersionId=')[1]\n",
+        "                return requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\").json()\n",
+        "            else:\n",
+        "                model_id = url.split('/models/')[1].split('/')[0]\n",
+        "                return requests.get(f\"https://civitai.com/api/v1/models/{model_id}\").json()\n",
         "        else:\n",
-        "            model_id = url.split('/models/')[1].split('/')[0]\n",
-        "            response = requests.get(f\"https://civitai.com/api/v1/models/{model_id}\")\n",
-        "            # print(f\"end - https://civitai.com/api/v1/models/{model_id}\")\n",
-        "    else:\n",
-        "        version_id = url.split('/models/')[1].split('/')[0]\n",
-        "        response = requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\")\n",
-        "        # print(f\"end - https://civitai.com/api/v1/model-versions/{version_id}\")\n",
+        "            version_id = url.split('/models/')[1].split('/')[0]\n",
+        "            return requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\").json()\n",
         "\n",
-        "    data = response.json()\n",
+        "    data = get_model_data(url)\n",
         "\n",
-        "    if response.status_code != 200:\n",
+        "    if not data:\n",
         "        return None, None, None, None, None, None, None\n",
         "\n",
-        "    # Define model type and name\n",
-        "    if \"civitai.com/models/\" in url:\n",
-        "        if '?modelVersionId=' in url:\n",
-        "            model_type = data['model']['type']\n",
-        "            model_name = data['files'][0]['name']\n",
+        "    def extract_model_info(url, data):\n",
+        "        if \"civitai.com/models/\" in url:\n",
+        "            if '?modelVersionId=' in url:\n",
+        "                model_type = data['model']['type']\n",
+        "                model_name = data['files'][0]['name']\n",
+        "            else:\n",
+        "                model_type = data['type']\n",
+        "                model_name = data['modelVersions'][0]['files'][0]['name']\n",
+        "        elif 'type=' in url:\n",
+        "            model_type = parse_qs(urlparse(url).query).get('type', [''])[0]\n",
+        "            if 'model' in model_type.lower():\n",
+        "                model_name = data['files'][0]['name']\n",
+        "            else:\n",
+        "                model_name = data['files'][1]['name']\n",
         "        else:\n",
-        "            model_type = data['type']\n",
-        "            model_name = data['modelVersions'][0]['files'][0]['name']\n",
-        "    elif 'type=' in url:\n",
-        "        model_type = parse_qs(urlparse(url).query).get('type', [''])[0]\n",
-        "        if 'model' in model_type.lower():\n",
+        "            model_type = data['model']['type']\n",
         "            model_name = data['files'][0]['name']\n",
-        "        else:\n",
-        "            model_name = data['files'][1]['name']\n",
-        "    else:\n",
-        "        model_type = data['model']['type']\n",
-        "        model_name = data['files'][0]['name']\n",
+        "        return model_type, model_name\n",
         "\n",
+        "    model_type, model_name = extract_model_info(url, data)\n",
         "    model_name = file_name or model_name\n",
         "\n",
-        "    # Determine DownloadUrl\n",
-        "    if \"civitai.com/models/\" in url:\n",
-        "        if '?modelVersionId=' in url:\n",
-        "            download_url = data.get('downloadUrl')\n",
-        "        else:\n",
-        "            download_url = data[\"modelVersions\"][0].get(\"downloadUrl\", \"\")\n",
-        "    elif 'type=' in url:\n",
-        "        if any(t.lower() in model_type.lower() for t in support_types):\n",
-        "            download_url = data['files'][0]['downloadUrl']\n",
+        "    def get_download_url(url, data, model_type):\n",
+        "        if \"civitai.com/models/\" in url:\n",
+        "            if '?modelVersionId=' in url:\n",
+        "                return data.get('downloadUrl')\n",
+        "            else:\n",
+        "                return data[\"modelVersions\"][0].get(\"downloadUrl\", \"\")\n",
+        "        elif 'type=' in url:\n",
+        "            if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):\n",
+        "                return data['files'][0]['downloadUrl']\n",
+        "            else:\n",
+        "                return data['files'][1]['downloadUrl']\n",
         "        else:\n",
-        "            download_url = data['files'][1]['downloadUrl']\n",
-        "    else:\n",
-        "        download_url = data.get('downloadUrl')\n",
+        "            return data.get('downloadUrl')\n",
         "\n",
-        "    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url) # hide token\n",
+        "    download_url = get_download_url(url, data, model_type)\n",
+        "    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url)\n",
         "\n",
-        "    # Find a safe image: level less than 4 | Kaggle\n",
-        "    image_url, image_name = None, None\n",
-        "    if any(t in model_type for t in support_types):\n",
-        "        try:\n",
-        "            images = data.get('images') or data['modelVersions'][0].get('images', [])\n",
-        "            if env == 'Kaggle':\n",
-        "                image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)\n",
-        "            else:\n",
-        "                image_url = images[0]['url'] if images else None\n",
-        "        except KeyError:\n",
-        "            pass\n",
+        "    def get_image_info(data, model_type, model_name):\n",
+        "        image_url, image_name = None, None\n",
+        "        if any(t in model_type for t in SUPPORT_TYPES):\n",
+        "            try:\n",
+        "                images = data.get('images') or data['modelVersions'][0].get('images', [])\n",
+        "                if env == 'Kaggle':\n",
+        "                    image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)\n",
+        "                else:\n",
+        "                    image_url = images[0]['url'] if images else None\n",
+        "            except KeyError:\n",
+        "                pass\n",
+        "\n",
+        "            image_name = f\"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}\" if image_url else None\n",
+        "        return image_url, image_name\n",
         "\n",
-        "        # Generate a name to save the image\n",
-        "        image_name = f\"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}\" if image_url else None\n",
+        "    image_url, image_name = get_image_info(data, model_type, model_name)\n",
         "\n",
-        "    return f\"{download_url}{'&' if '?' in download_url else '?'}token={civitai_token}\", clean_url, model_type, model_name, image_url, image_name, data\n",
+        "    return f\"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}\", clean_url, model_type, model_name, image_url, image_name, data\n",
         "\n",
         "''' Main Download Code '''\n",
         "\n",
+        "def strip_(url):\n",
+        "    if 'github.com' in url:\n",
+        "        return url.replace('/blob/', '/raw/')\n",
+        "    elif \"huggingface.co\" in url:\n",
+        "        url = url.replace('/blob/', '/resolve/')\n",
+        "        return url.split('?')[0] if '?' in url else url\n",
+        "    return url\n",
+        "\n",
         "def download(url):\n",
         "    links_and_paths = [link_or_path.strip() for link_or_path in url.split(',') if link_or_path.strip()]\n",
         "\n",
         "    for link_or_path in links_and_paths:\n",
-        "        if any(link_or_path.lower().startswith(prefix) for prefix in prefixes):\n",
+        "        if any(link_or_path.lower().startswith(prefix) for prefix in PREFIXES):\n",
         "            handle_manual(link_or_path)\n",
         "        else:\n",
         "            url, dst_dir, file_name = link_or_path.split()\n",
         "            manual_download(url, dst_dir, file_name)\n",
         "\n",
-        "    unpack_zip_files()\n",
-        "\n",
-        "def unpack_zip_files():\n",
+        "    # Unpuck ZIPs Files\n",
         "    for directory in directories:\n",
         "        for root, _, files in os.walk(directory):\n",
         "            for file in files:\n",
@@ -485,8 +362,8 @@
         "    if file_name:\n",
         "        path = re.sub(r'\\[.*?\\]', '', path)\n",
         "\n",
-        "    if prefix in prefixes:\n",
-        "        dir = prefixes[prefix]\n",
+        "    if prefix in PREFIXES:\n",
+        "        dir = PREFIXES[prefix]\n",
         "        if prefix != \"extension\":\n",
         "            try:\n",
         "                manual_download(path, dir, file_name=file_name)\n",
@@ -500,60 +377,52 @@
         "    aria2c_header = \"--header='User-Agent: Mozilla/5.0' --allow-overwrite=true\"\n",
         "    aria2_args = \"--optimize-concurrent-downloads --console-log-level=error --summary-interval=10 --stderr=true -c -x16 -s16 -k1M -j5\"\n",
         "\n",
-        "    if 'github.com' in url:\n",
-        "        url = strip_(url)\n",
-        "\n",
-        "    # -- CivitAi APi+ V2 --\n",
-        "    elif 'civitai' in url:\n",
-        "        url, clean_url, model_type, file_name, image_url, image_name, data = strip_(url, file_name)\n",
+        "    clean_url = strip_(url)\n",
         "\n",
+        "    if 'civitai' in url:\n",
+        "        url, clean_url, model_type, file_name, image_url, image_name, data = CivitAi_API(url, file_name)\n",
         "        if image_url and image_name:\n",
-        "            with capture.capture_output() as cap:\n",
-        "                !aria2c {aria2_args} -d {dst_dir} -o '{image_name}' '{image_url}'\n",
-        "            del cap\n",
+        "            command = [\"aria2c\"] + aria2_args.split() + [\"-d\", dst_dir, \"-o\", image_name, image_url]\n",
+        "            subprocess.run(command, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)\n",
         "\n",
-        "    elif \"huggingface.co\" in url:\n",
-        "        clean_url = strip_(url)\n",
-        "        basename = clean_url.split(\"/\")[-1] if file_name is None else file_name\n",
+        "    elif 'github' in url or \"huggingface.co\" in url:\n",
+        "        basename = url.split(\"/\")[-1] if file_name is None else file_name\n",
         "\n",
         "    \"\"\" Formatted info output \"\"\"\n",
-        "    model_name_or_basename = file_name if not 'huggingface' in url else basename\n",
+        "    model_name_or_basename = file_name if file_name else basename\n",
         "    format_output(clean_url or url, dst_dir, model_name_or_basename)\n",
         "\n",
-        "    # ## -- for my tests --\n",
         "    # print(url, dst_dir, model_name_or_basename)\n",
-        "    print(f\"\\033[31m[Data Info]:\\033[0m Failed to retrieve data from the API.\\n\") if 'civitai' in url and not data else None\n",
-        "    if 'civitai' in url and data and image_name:\n",
-        "        print(f\"\\033[32m[Preview DL]:\\033[0m {image_name} - {image_url}\\n\")\n",
+        "    if 'civitai' in url:\n",
+        "        if not data:\n",
+        "            print(\"\\033[31m[Data Info]:\\033[0m Failed to retrieve data from the API.\\n\")\n",
+        "        if data and image_name:\n",
+        "            print(f\"\\033[32m[Preview DL]:\\033[0m {image_name} - {image_url}\\n\")\n",
+        "\n",
         "    # =====================\n",
+        "    def run_aria2c(url, dst_dir, file_name=None, args=\"\", header=\"\"):\n",
+        "        out = f\"-o '{file_name}'\" if file_name else \"\"\n",
+        "        !aria2c {header} {args} -d {dst_dir} {out} '{url}'\n",
         "\n",
-        "    # -- Git Hub --\n",
-        "    if 'github.com' in url or 'githubusercontent.com' in url:\n",
-        "        !aria2c {aria2_args} -d {dst_dir} -o '{basename}' '{url}'\n",
-        "\n",
-        "    # -- GDrive --\n",
-        "    elif 'drive.google' in url:\n",
-        "        try:\n",
-        "            have_drive_link\n",
-        "        except:\n",
-        "            !pip install -q gdown==5.2.0 > /dev/null\n",
-        "            have_drive_link = True\n",
+        "    # -- Google Drive --\n",
+        "    if 'drive.google' in url:\n",
+        "        if not globals().get('have_drive_link', False):\n",
+        "            os.system(\"pip install -U gdown > /dev/null\")\n",
+        "            globals()['have_drive_link'] = True\n",
         "\n",
         "        if 'folders' in url:\n",
-        "            !gdown --folder \"{url}\" -O {dst_dir} --fuzzy -c\n",
+        "            os.system(f\"gdown --folder \\\"{url}\\\" -O {dst_dir} --fuzzy -c\")\n",
         "        else:\n",
-        "            if file_name:\n",
-        "               !gdown \"{url}\" -O {dst_dir}/{file_name} --fuzzy -c\n",
-        "            else:\n",
-        "               !gdown \"{url}\" -O {dst_dir} --fuzzy -c\n",
+        "            out_path = f\"{dst_dir}/{file_name}\" if file_name else dst_dir\n",
+        "            os.system(f\"gdown \\\"{url}\\\" -O {out_path} --fuzzy -c\")\n",
         "\n",
-        "    # -- Hugging Face --\n",
-        "    elif 'huggingface' in url:\n",
-        "        !aria2c {header_option} {aria2_args} -d {dst_dir} -o '{basename}' '{url}'\n",
+        "    # -- GitHub or Hugging Face --\n",
+        "    elif 'github' in url or 'huggingface' in url:\n",
+        "        run_aria2c(clean_url, dst_dir, basename, aria2_args, header_option if 'huggingface' in url else '')\n",
         "\n",
-        "    # -- Other --\n",
+        "    # -- Other HTTP/Sources --\n",
         "    elif 'http' in url:\n",
-        "        !aria2c {aria2c_header} {aria2_args} -d {dst_dir} -o \"{file_name if file_name else ''}\" '{url}'\n",
+        "        run_aria2c(url, dst_dir, file_name, aria2_args, aria2c_header)\n",
         "\n",
         "''' SubModels - Added URLs '''\n",
         "\n",
@@ -593,7 +462,7 @@
         "\n",
         "''' file.txt - added urls '''\n",
         "\n",
-        "def process_file_download(file_url, prefixes, unique_urls):\n",
+        "def process_file_download(file_url, PREFIXES, unique_urls):\n",
         "    files_urls = \"\"\n",
         "\n",
         "    if file_url.startswith(\"http\"):\n",
@@ -608,8 +477,8 @@
         "    current_tag = None\n",
         "    for line in lines:\n",
         "        line = line.strip()\n",
-        "        if any(f'# {tag}' in line.lower() for tag in prefixes):\n",
-        "            current_tag = next((tag for tag in prefixes if tag in line.lower()))\n",
+        "        if any(f'# {tag}' in line.lower() for tag in PREFIXES):\n",
+        "            current_tag = next((tag for tag in PREFIXES if tag in line.lower()))\n",
         "\n",
         "        urls = [url.split('#')[0].strip() for url in line.split(',')]  # filter urls\n",
         "        for url in urls:\n",
@@ -633,13 +502,13 @@
         "                custom_file_url = f'{root_path}/{custom_file_url}'\n",
         "\n",
         "        try:\n",
-        "            file_urls += process_file_download(custom_file_url, prefixes, unique_urls)\n",
+        "            file_urls += process_file_download(custom_file_url, PREFIXES, unique_urls)\n",
         "        except FileNotFoundError:\n",
         "            pass\n",
         "\n",
         "# url prefixing\n",
         "urls = (Model_url, Vae_url, LoRA_url, Embedding_url, Extensions_url)\n",
-        "prefixed_urls = (f\"{prefix}:{url}\" for prefix, url in zip(prefixes.keys(), urls) if url for url in url.replace(',', '').split())\n",
+        "prefixed_urls = (f\"{prefix}:{url}\" for prefix, url in zip(PREFIXES.keys(), urls) if url for url in url.replace(',', '').split())\n",
         "url += \", \".join(prefixed_urls) + \", \" + file_urls\n",
         "\n",
         "if detailed_download == \"on\":\n",

files_cells/notebooks/ru/launch_ru.ipynb

@@ -33,21 +33,13 @@
         "from datetime import timedelta\n",
         "from IPython.display import clear_output\n",
         "\n",
-        "#  ================= DETECT ENV =================\n",
-        "def detect_environment():\n",
-        "    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)\n",
-        "    environments = {\n",
-        "        'COLAB_GPU': ('Google Colab', \"/root\" if free_plan else \"/content\"),\n",
-        "        'KAGGLE_URL_BASE': ('Kaggle', \"/kaggle/working/content\")\n",
-        "    }\n",
-        "\n",
-        "    for env_var, (environment, path) in environments.items():\n",
-        "        if env_var in os.environ:\n",
-        "            return environment, path, free_plan\n",
-        "    return 'Unknown', '/unknown/path', free_plan\n",
-        "\n",
-        "env, root_path, free_plan = detect_environment()\n",
-        "webui_path = f\"{root_path}/sdw\"\n",
+        "\n",
+        "# Setup Env\n",
+        "env = os.environ.get('ENV_NAME')\n",
+        "root_path = os.environ.get('ROOT_PATH')\n",
+        "webui_path = os.environ.get('WEBUI_PATH')\n",
+        "free_plan = os.environ.get('FREE_PLAN')\n",
+        "\n",
         "\n",
         "def load_settings():\n",
         "    SETTINGS_FILE = f'{root_path}/settings.json'\n",
@@ -62,6 +54,7 @@
         "commandline_arguments = settings.get('commandline_arguments', \"\")\n",
         "change_webui = settings.get('change_webui', \"\")\n",
         "\n",
+        "\n",
         "# ======================== TUNNEL V2 ========================\n",
         "print('Please Wait...')\n",
         "\n",
@@ -95,6 +88,7 @@
         "\n",
         "clear_output()\n",
         "\n",
+        "\n",
         "# =============== Automatic Fixing Path V3 ===============\n",
         "paths_to_check = {\n",
         "    \"tagger_hf_cache_dir\": f\"{webui_path}/models/interrogators/\",\n",

files_cells/notebooks/ru/widgets_ru.ipynb

@@ -1,349 +1,340 @@
-{
-  "nbformat": 4,
-  "nbformat_minor": 0,
-  "metadata": {
-    "colab": {
-      "provenance": []
-    },
-    "kernelspec": {
-      "name": "python3",
-      "display_name": "Python 3"
-    },
-    "language_info": {
-      "name": "python"
-    }
-  },
-  "cells": [
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "JKTCrY9LU7Oq"
-      },
-      "outputs": [],
-      "source": [
-        "##~   WIDGET CODE | BY: ANXETY   ~##\n",
-        "\n",
-        "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 IPython.display import display, HTML, Javascript, clear_output\n",
-        "\n",
-        "\n",
-        "# ================= DETECT ENV =================\n",
-        "def detect_environment():\n",
-        "    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)\n",
-        "    environments = {\n",
-        "        'COLAB_GPU': ('Google Colab', \"/root\" if free_plan else \"/content\"),\n",
-        "        'KAGGLE_URL_BASE': ('Kaggle', \"/kaggle/working/content\")\n",
-        "    }\n",
-        "    for env_var, (environment, path) in environments.items():\n",
-        "        if env_var in os.environ:\n",
-        "            return environment, path, free_plan\n",
-        "\n",
-        "env, root_path, free_plan = detect_environment()\n",
-        "webui_path = f\"{root_path}/sdw\"\n",
-        "!mkdir -p {root_path}\n",
-        "\n",
-        "\n",
-        "# ==================== CSS JS ====================\n",
-        "##~ custom background images V1.5 ~##\n",
-        "import argparse\n",
-        "parser = argparse.ArgumentParser(description='This script processes an background image.')\n",
-        "parser.add_argument('-i', '--image', type=str, help='URL of the image to process', metavar='')\n",
-        "parser.add_argument('-o', '--opacity', type=float, help='Opacity level for the image, between 0 and 1', metavar='', default=0.3)\n",
-        "parser.add_argument('-b', '--blur', type=str, help='Blur level for the image', metavar='', default=0)\n",
-        "parser.add_argument('-y', type=int, help='Y coordinate for the image in px', metavar='', default=0)\n",
-        "parser.add_argument('-x', type=int, help='X coordinate for the image in px', metavar='', default=0)\n",
-        "parser.add_argument('-s', '--scale', type=int, help='Scale image in %%', metavar='', default=100)\n",
-        "parser.add_argument('-m', '--mode', action='store_true', help='Removes repetitive image tiles')\n",
-        "parser.add_argument('-t', '--transparent', action='store_true', help='Makes input/selection fields 35%% more transparent')\n",
-        "parser.add_argument('-bf', '--blur-fields', type=str, help='Background blur level for input/selection fields', metavar='', default=2)\n",
-        "\n",
-        "args = parser.parse_args()\n",
-        "\n",
-        "url_img = args.image\n",
-        "opacity_img = args.opacity\n",
-        "blur_img = args.blur\n",
-        "y_img = args.y\n",
-        "x_img = args.x\n",
-        "scale_img = args.scale\n",
-        "blur_fields = args.blur_fields\n",
-        "\n",
-        "## ---\n",
-        "\"\"\" WTF KAGGLE - WHAT THE FUCK IS THE DIFFERENCE OF 35 PIXELS!?!?!? \"\"\"\n",
-        "fix_heigh_img = \"-810px\" if env == \"Kaggle\" else \"-775px\"\n",
-        "\n",
-        "\"\"\" transperent fields \"\"\"\n",
-        "t_bg_alpha = \"1\" if not args.transparent else \"0.65\"\n",
-        "\n",
-        "\"\"\" mode img - repeats \"\"\"\n",
-        "mode_img = \"repeat\" if not args.mode else \"no-repeat\"\n",
-        "\n",
-        "container_background = f'''\n",
-        "<style>\n",
-        ":root {{\n",
-        "  /* for background container*/\n",
-        "  --img_background: url({url_img});\n",
-        "  --img_opacity: {opacity_img};\n",
-        "  --img_blur: {blur_img}px;\n",
-        "  --image_y: {y_img}px;\n",
-        "  --image_x: {x_img}px;\n",
-        "  --img_scale: {scale_img}%;\n",
-        "  --img_mode: {mode_img};\n",
-        "  --img_height_dif: {fix_heigh_img};\n",
-        "\n",
-        "  /* for fields */\n",
-        "  --bg-field-color: rgba(28, 28, 28, {t_bg_alpha}); /* -> #1c1c1c */\n",
-        "  --bg-field-color-hover: rgba(38, 38, 38, {t_bg_alpha}); /* -> #262626; */\n",
-        "  --bg-field-blur-level: {blur_fields}px;\n",
-        "}}\n",
-        "'''\n",
-        "\n",
-        "display(HTML(container_background))\n",
-        "\n",
-        "# Main CSS\n",
-        "css_file_path = f\"{root_path}/CSS/main_widgets.css\"\n",
-        "with open(css_file_path , \"r\") as f:\n",
-        "    CSS = f.read()\n",
-        "display(HTML(f\"<style>{CSS}</style>\"))\n",
-        "\n",
-        "# Main JS\n",
-        "JS = '''\n",
-        "<!-- TOGGLE 'CustomDL' SCRIPT -->\n",
-        "<script>\n",
-        "function toggleContainer() {\n",
-        "    let downloadContainer = document.querySelector('.container_custom_downlad');\n",
-        "    let info = document.querySelector('.info');\n",
-        "\n",
-        "    downloadContainer.classList.toggle('expanded');\n",
-        "    info.classList.toggle('showed');\n",
-        "}\n",
-        "</script>\n",
-        "'''\n",
-        "display(HTML(JS))\n",
-        "\n",
-        "# ==================== WIDGETS V2 ====================\n",
-        "HR = widgets.HTML('<hr>')\n",
-        "\n",
-        "class WidgetFactory:\n",
-        "    def __init__(self, style=None, layout=None):\n",
-        "        self.style = style if style else {'description_width': 'initial'}\n",
-        "        self.layout = layout if layout else widgets.Layout(max_width='1080px', width='100%')\n",
-        "\n",
-        "    def create_html(self, content, class_name=None):\n",
-        "        html_widget = widgets.HTML(content)\n",
-        "        if class_name:\n",
-        "            html_widget.add_class(class_name)\n",
-        "        return html_widget\n",
-        "\n",
-        "    def create_header(self, name):\n",
-        "        return widgets.HTML(f'<div class=\"header\">{name}<div>')\n",
-        "\n",
-        "    def create_dropdown(self, options, value, description):\n",
-        "        return widgets.Dropdown(options=options, value=value, description=description, style=self.style, layout=self.layout)\n",
-        "\n",
-        "    def create_text(self, description, placeholder='', value=''):\n",
-        "        return widgets.Text(description=description, placeholder=placeholder, value=value, style=self.style, layout=self.layout)\n",
-        "\n",
-        "    def create_checkbox(self, value, description):\n",
-        "        return widgets.Checkbox(value=value, description=description, style=self.style, layout=self.layout)\n",
-        "\n",
-        "    def create_button(self, description, class_name=None):\n",
-        "        button = widgets.Button(description=description)\n",
-        "        if class_name:\n",
-        "            button.add_class(class_name)\n",
-        "        return button\n",
-        "\n",
-        "    def create_hbox(self, children):\n",
-        "        return widgets.HBox(children)\n",
-        "\n",
-        "    def create_vbox(self, children, class_names=None):\n",
-        "        vbox = widgets.VBox(children)\n",
-        "        if class_names:\n",
-        "            for class_name in class_names:\n",
-        "                vbox.add_class(class_name)\n",
-        "        return vbox\n",
-        "\n",
-        "    def display(self, widget):\n",
-        "        display(widget)\n",
-        "\n",
-        "# Instantiate the factory\n",
-        "factory = WidgetFactory()\n",
-        "\n",
-        "# --- MODEL ---\n",
-        "model_header = factory.create_header('Выбор Модели')\n",
-        "model_options = ['none',\n",
-        "                 '1.Anime (by XpucT) + INP',\n",
-        "                 '2.BluMix [Anime] [V7] + INP',\n",
-        "                 '3.Cetus-Mix [Anime] [V4] + INP',\n",
-        "                 '4.Counterfeit [Anime] [V3] + INP',\n",
-        "                 '5.CuteColor [Anime] [V3]',\n",
-        "                 '6.Dark-Sushi-Mix [Anime]',\n",
-        "                 '7.Deliberate [Realism] [V6] + INP',\n",
-        "                 '8.Meina-Mix [Anime] [V11] + INP',\n",
-        "                 '9.Mix-Pro [Anime] [V4] + INP']\n",
-        "model_widget = factory.create_dropdown(model_options, '4.Counterfeit [Anime] [V3] + INP', 'Модель:')\n",
-        "model_num_widget = factory.create_text('Номер Модели:', 'Введите номера моделей для скачивания через запятую/пробел.')\n",
-        "inpainting_model_widget = factory.create_checkbox(False, 'Inpainting Модели')\n",
-        "\n",
-        "# Display Model\n",
-        "all_model_box = factory.create_vbox([model_header, model_widget, model_num_widget, inpainting_model_widget], class_names=[\"container\", \"image_1\"])\n",
-        "factory.display(all_model_box)\n",
-        "\n",
-        "# --- VAE ---\n",
-        "vae_header = factory.create_header('Выбор VAE')\n",
-        "vae_options = ['none',\n",
-        "               '1.Anime.vae',\n",
-        "               '2.Anything.vae',\n",
-        "               '3.Blessed2.vae',\n",
-        "               '4.ClearVae.vae',\n",
-        "               '5.WD.vae']\n",
-        "vae_widget = factory.create_dropdown(vae_options, '3.Blessed2.vae', 'Vae:')\n",
-        "vae_num_widget = factory.create_text('Номер Vae:', 'Введите номера vae для скачивания через запятую/пробел.')\n",
-        "\n",
-        "# Display Vae\n",
-        "all_vae_box = factory.create_vbox([vae_header, vae_widget, vae_num_widget], class_names=[\"container\", \"image_2\"])\n",
-        "factory.display(all_vae_box)\n",
-        "\n",
-        "# --- ADDITIONAL ---\n",
-        "additional_header = factory.create_header('Дополнительно')\n",
-        "latest_webui_widget = factory.create_checkbox(True, 'Обновить WebUI')\n",
-        "latest_exstensions_widget = factory.create_checkbox(True, 'Обновить Расширения')\n",
-        "change_webui_widget = factory.create_dropdown(['A1111', 'Forge'], 'A1111', 'Изменить WebUI:')\n",
-        "detailed_download_widget = factory.create_dropdown(['off', 'on'], 'off', 'Подробная Загрузка:')\n",
-        "choose_changes_widget = factory.create_hbox([latest_webui_widget, latest_exstensions_widget, change_webui_widget, detailed_download_widget])\n",
-        "\n",
-        "controlnet_options = ['none', 'ALL', '1.canny',\n",
-        "                      '2.openpose', '3.depth',\n",
-        "                      '4.normal_map', '5.mlsd',\n",
-        "                      '6.lineart', '7.soft_edge',\n",
-        "                      '8.scribble', '9.segmentation',\n",
-        "                      '10.shuffle', '11.tile',\n",
-        "                      '12.inpaint', '13.instruct_p2p']\n",
-        "controlnet_widget = factory.create_dropdown(controlnet_options, 'none', 'ControlNet:')\n",
-        "controlnet_num_widget = factory.create_text('Номер ControlNet:', 'Введите номера моделей ControlNet для скачивания через запятую/пробел.')\n",
-        "commit_hash_widget = factory.create_text('Commit Hash:')\n",
-        "huggingface_token_widget = factory.create_text('Токен HuggingFace:')\n",
-        "\n",
-        "ngrok_token_widget = factory.create_text('Токен Ngrok:')\n",
-        "ngrock_button = factory.create_html('<a href=\"https://dashboard.ngrok.com/get-started/your-authtoken\" target=\"_blank\">Получить Ngrok Токен</a>', class_name=\"button_ngrok\")\n",
-        "ngrok_widget = factory.create_hbox([ngrok_token_widget, ngrock_button])\n",
-        "\n",
-        "zrok_token_widget = factory.create_text('Токен Zrok:')\n",
-        "zrok_button = factory.create_html('<a href=\"https://colab.research.google.com/drive/1d2sjWDJi_GYBUavrHSuQyHTDuLy36WpU\" target=\"_blank\">Зарегать Zrok Токен</a>', class_name=\"button_ngrok\")\n",
-        "zrok_widget = factory.create_hbox([zrok_token_widget, zrok_button])\n",
-        "\n",
-        "commandline_arguments_options = \"--listen --enable-insecure-extension-access --theme dark --no-half-vae --disable-console-progressbars --xformers\"\n",
-        "commandline_arguments_widget = factory.create_text('Аргументы:', value=commandline_arguments_options)\n",
-        "\n",
-        "# Display Additional\n",
-        "additional_widget_list = [additional_header,\n",
-        "                          choose_changes_widget,\n",
-        "                          HR,\n",
-        "                          controlnet_widget,\n",
-        "                          controlnet_num_widget,\n",
-        "                          commit_hash_widget,\n",
-        "                          huggingface_token_widget,\n",
-        "                          ngrok_widget,\n",
-        "                          zrok_widget,\n",
-        "                          HR,\n",
-        "                          commandline_arguments_widget]\n",
-        "\n",
-        "if free_plan and env == \"Google Colab\": # remove ngrok from colab\n",
-        "    additional_widget_list.remove(ngrok_widget)\n",
-        "if os.path.exists(webui_path): # remove selection after selection ;3\n",
-        "    choose_changes_widget.children = [widget for widget in choose_changes_widget.children if widget != change_webui_widget]\n",
-        "\n",
-        "all_additional_box = factory.create_vbox(additional_widget_list, class_names=[\"container\", \"image_3\"])\n",
-        "factory.display(all_additional_box)\n",
-        "\n",
-        "# --- CUSTOM DOWNLOAD ---\n",
-        "custom_download_header_popup = factory.create_html('''\n",
-        "<style>\n",
-        "/* Term Colors */\n",
-        ".sample_label {color: #dbafff;}\n",
-        ".braces {color: #ffff00;}\n",
-        ".extension {color: #eb934b;}\n",
-        ".file_name {color: #ffffd8;}\n",
-        "</style>\n",
-        "\n",
-        "<div class=\"header\" style=\"cursor: pointer;\" onclick=\"toggleContainer()\">Кастомная Загрузка</div>\n",
-        "<!-- PopUp Window -->\n",
-        "<div class=\"info\" id=\"info_dl\">INFO</div>\n",
-        "<div class=\"popup\">\n",
-        "    Разделите несколько URL-адресов запятой/пробелом. Для <span class=\"file_name\">пользовательского имени</span> файла/расширения укажите его через <span class=\"braces\">[]</span>\n",
-        "    после URL без пробелов.\n",
-        "    <span style=\"color: #ff9999\">Для файлов обязательно укажите</span> - <span class=\"extension\">Расширение Файла.</span>\n",
-        "    <div class=\"sample\">\n",
-        "        <span class=\"sample_label\">Пример для Файла:</span>\n",
-        "        https://civitai.com/api/download/models/229782<span class=\"braces\">[</span><span class=\"file_name\">Detailer</span><span class=\"extension\">.safetensors</span><span class=\"braces\">]</span>\n",
-        "        <br>\n",
-        "        <span class=\"sample_label\">Пример для Расширения:</span>\n",
-        "        https://github.com/hako-mikan/sd-webui-regional-prompter<span class=\"braces\">[</span><span class=\"file_name\">Regional-Prompter</span><span class=\"braces\">]</span>\n",
-        "    </div>\n",
-        "</div>\n",
-        "''')\n",
-        "\n",
-        "Model_url_widget = factory.create_text('Model:')\n",
-        "Vae_url_widget = factory.create_text('Vae:')\n",
-        "LoRA_url_widget = factory.create_text('LoRa:')\n",
-        "Embedding_url_widget = factory.create_text('Embedding:')\n",
-        "Extensions_url_widget = factory.create_text('Extensions:')\n",
-        "custom_file_urls_widget = factory.create_text('Файл (txt):')\n",
-        "\n",
-        "# Display CustomDl\n",
-        "all_custom_box = factory.create_vbox([\n",
-        "    custom_download_header_popup, Model_url_widget, Vae_url_widget, LoRA_url_widget, Embedding_url_widget, Extensions_url_widget, custom_file_urls_widget\n",
-        "], class_names=[\"container\", \"image_4\", \"container_custom_downlad\"])\n",
-        "factory.display(all_custom_box)\n",
-        "\n",
-        "# --- Save Button ---\n",
-        "save_button = factory.create_button('Сохранить', class_name=\"button_save\")\n",
-        "factory.display(save_button)\n",
-        "\n",
-        "\n",
-        "# ============ Load / Save - Settings V2 ============\n",
-        "SETTINGS_FILE = f'{root_path}/settings.json'\n",
-        "\n",
-        "SETTINGS_KEYS = [\n",
-        "    'model', 'model_num', 'inpainting_model',\n",
-        "    'vae', 'vae_num', 'latest_webui', 'latest_exstensions',\n",
-        "    'change_webui', 'detailed_download', 'controlnet',\n",
-        "    'controlnet_num', 'commit_hash', 'huggingface_token',\n",
-        "    'ngrok_token', 'zrok_token', 'commandline_arguments',\n",
-        "    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url',\n",
-        "    'Extensions_url', 'custom_file_urls'\n",
-        "]\n",
-        "\n",
-        "def save_settings():\n",
-        "    settings = {key: globals()[f\"{key}_widget\"].value for key in SETTINGS_KEYS}\n",
-        "    with open(SETTINGS_FILE, 'w') as f:\n",
-        "        json.dump(settings, f, indent=2)\n",
-        "\n",
-        "def load_settings():\n",
-        "    if os.path.exists(SETTINGS_FILE):\n",
-        "        with open(SETTINGS_FILE, 'r') as f:\n",
-        "            settings = json.load(f)\n",
-        "            for key in SETTINGS_KEYS:\n",
-        "                globals()[f\"{key}_widget\"].value = settings.get(key, \"\")\n",
-        "\n",
-        "def hide_widgets():\n",
-        "    widgets_list = [all_model_box, all_vae_box, all_additional_box, all_custom_box, save_button]\n",
-        "    for widget in widgets_list:\n",
-        "        widget.add_class(\"hide\")\n",
-        "    time.sleep(0.5)\n",
-        "    widgets.Widget.close_all()\n",
-        "\n",
-        "def save_data(button):\n",
-        "    save_settings()\n",
-        "    hide_widgets()\n",
-        "\n",
-        "load_settings()\n",
-        "save_button.on_click(save_data)"
-      ]
-    }
-  ]
+{
+  "nbformat": 4,
+  "nbformat_minor": 0,
+  "metadata": {
+    "colab": {
+      "provenance": []
+    },
+    "kernelspec": {
+      "name": "python3",
+      "display_name": "Python 3"
+    },
+    "language_info": {
+      "name": "python"
+    }
+  },
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "JKTCrY9LU7Oq"
+      },
+      "outputs": [],
+      "source": [
+        "##~   WIDGET CODE | BY: ANXETY   ~##\n",
+        "\n",
+        "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 IPython.display import display, HTML, Javascript, clear_output\n",
+        "\n",
+        "\n",
+        "# Setup Env\n",
+        "env = os.environ.get('ENV_NAME')\n",
+        "root_path = os.environ.get('ROOT_PATH')\n",
+        "webui_path = os.environ.get('WEBUI_PATH')\n",
+        "free_plan = os.environ.get('FREE_PLAN')\n",
+        "\n",
+        "\n",
+        "# ==================== CSS JS ====================\n",
+        "##~ custom background images V1.5 ~##\n",
+        "import argparse\n",
+        "parser = argparse.ArgumentParser(description='This script processes an background image.')\n",
+        "parser.add_argument('-i', '--image', type=str, help='URL of the image to process', metavar='')\n",
+        "parser.add_argument('-o', '--opacity', type=float, help='Opacity level for the image, between 0 and 1', metavar='', default=0.3)\n",
+        "parser.add_argument('-b', '--blur', type=str, help='Blur level for the image', metavar='', default=0)\n",
+        "parser.add_argument('-y', type=int, help='Y coordinate for the image in px', metavar='', default=0)\n",
+        "parser.add_argument('-x', type=int, help='X coordinate for the image in px', metavar='', default=0)\n",
+        "parser.add_argument('-s', '--scale', type=int, help='Scale image in %%', metavar='', default=100)\n",
+        "parser.add_argument('-m', '--mode', action='store_true', help='Removes repetitive image tiles')\n",
+        "parser.add_argument('-t', '--transparent', action='store_true', help='Makes input/selection fields 35%% more transparent')\n",
+        "parser.add_argument('-bf', '--blur-fields', type=str, help='Background blur level for input/selection fields', metavar='', default=2)\n",
+        "\n",
+        "args = parser.parse_args()\n",
+        "\n",
+        "url_img = args.image\n",
+        "opacity_img = args.opacity\n",
+        "blur_img = args.blur\n",
+        "y_img = args.y\n",
+        "x_img = args.x\n",
+        "scale_img = args.scale\n",
+        "blur_fields = args.blur_fields\n",
+        "\n",
+        "## ---\n",
+        "\"\"\" WTF KAGGLE - WHAT THE FUCK IS THE DIFFERENCE OF 35 PIXELS!?!?!? \"\"\"\n",
+        "fix_heigh_img = \"-810px\" if env == \"Kaggle\" else \"-775px\"\n",
+        "\n",
+        "\"\"\" transperent fields \"\"\"\n",
+        "t_bg_alpha = \"1\" if not args.transparent else \"0.65\"\n",
+        "\n",
+        "\"\"\" mode img - repeats \"\"\"\n",
+        "mode_img = \"repeat\" if not args.mode else \"no-repeat\"\n",
+        "\n",
+        "container_background = f'''\n",
+        "<style>\n",
+        ":root {{\n",
+        "  /* for background container*/\n",
+        "  --img_background: url({url_img});\n",
+        "  --img_opacity: {opacity_img};\n",
+        "  --img_blur: {blur_img}px;\n",
+        "  --image_y: {y_img}px;\n",
+        "  --image_x: {x_img}px;\n",
+        "  --img_scale: {scale_img}%;\n",
+        "  --img_mode: {mode_img};\n",
+        "  --img_height_dif: {fix_heigh_img};\n",
+        "\n",
+        "  /* for fields */\n",
+        "  --bg-field-color: rgba(28, 28, 28, {t_bg_alpha}); /* -> #1c1c1c */\n",
+        "  --bg-field-color-hover: rgba(38, 38, 38, {t_bg_alpha}); /* -> #262626; */\n",
+        "  --bg-field-blur-level: {blur_fields}px;\n",
+        "}}\n",
+        "'''\n",
+        "\n",
+        "display(HTML(container_background))\n",
+        "\n",
+        "# Main CSS\n",
+        "css_file_path = f\"{root_path}/CSS/main_widgets.css\"\n",
+        "with open(css_file_path , \"r\") as f:\n",
+        "    CSS = f.read()\n",
+        "display(HTML(f\"<style>{CSS}</style>\"))\n",
+        "\n",
+        "# Main JS\n",
+        "JS = '''\n",
+        "<!-- TOGGLE 'CustomDL' SCRIPT -->\n",
+        "<script>\n",
+        "function toggleContainer() {\n",
+        "    let downloadContainer = document.querySelector('.container_custom_downlad');\n",
+        "    let info = document.querySelector('.info');\n",
+        "\n",
+        "    downloadContainer.classList.toggle('expanded');\n",
+        "    info.classList.toggle('showed');\n",
+        "}\n",
+        "</script>\n",
+        "'''\n",
+        "display(HTML(JS))\n",
+        "\n",
+        "# ==================== WIDGETS V2 ====================\n",
+        "HR = widgets.HTML('<hr>')\n",
+        "\n",
+        "class WidgetFactory:\n",
+        "    def __init__(self, style=None, layout=None):\n",
+        "        self.style = style if style else {'description_width': 'initial'}\n",
+        "        self.layout = layout if layout else widgets.Layout(max_width='1080px', width='100%')\n",
+        "\n",
+        "    def create_html(self, content, class_name=None):\n",
+        "        html_widget = widgets.HTML(content)\n",
+        "        if class_name:\n",
+        "            html_widget.add_class(class_name)\n",
+        "        return html_widget\n",
+        "\n",
+        "    def create_header(self, name):\n",
+        "        return widgets.HTML(f'<div class=\"header\">{name}<div>')\n",
+        "\n",
+        "    def create_dropdown(self, options, value, description):\n",
+        "        return widgets.Dropdown(options=options, value=value, description=description, style=self.style, layout=self.layout)\n",
+        "\n",
+        "    def create_text(self, description, placeholder='', value=''):\n",
+        "        return widgets.Text(description=description, placeholder=placeholder, value=value, style=self.style, layout=self.layout)\n",
+        "\n",
+        "    def create_checkbox(self, value, description):\n",
+        "        return widgets.Checkbox(value=value, description=description, style=self.style, layout=self.layout)\n",
+        "\n",
+        "    def create_button(self, description, class_name=None):\n",
+        "        button = widgets.Button(description=description)\n",
+        "        if class_name:\n",
+        "            button.add_class(class_name)\n",
+        "        return button\n",
+        "\n",
+        "    def create_hbox(self, children):\n",
+        "        return widgets.HBox(children)\n",
+        "\n",
+        "    def create_vbox(self, children, class_names=None):\n",
+        "        vbox = widgets.VBox(children)\n",
+        "        if class_names:\n",
+        "            for class_name in class_names:\n",
+        "                vbox.add_class(class_name)\n",
+        "        return vbox\n",
+        "\n",
+        "    def display(self, widget):\n",
+        "        display(widget)\n",
+        "\n",
+        "# Instantiate the factory\n",
+        "factory = WidgetFactory()\n",
+        "\n",
+        "# --- MODEL ---\n",
+        "model_header = factory.create_header('Выбор Модели')\n",
+        "model_options = ['none',\n",
+        "                 '1.Anime (by XpucT) + INP',\n",
+        "                 '2.BluMix [Anime] [V7] + INP',\n",
+        "                 '3.Cetus-Mix [Anime] [V4] + INP',\n",
+        "                 '4.Counterfeit [Anime] [V3] + INP',\n",
+        "                 '5.CuteColor [Anime] [V3]',\n",
+        "                 '6.Dark-Sushi-Mix [Anime]',\n",
+        "                 '7.Deliberate [Realism] [V6] + INP',\n",
+        "                 '8.Meina-Mix [Anime] [V11] + INP',\n",
+        "                 '9.Mix-Pro [Anime] [V4] + INP']\n",
+        "model_widget = factory.create_dropdown(model_options, '4.Counterfeit [Anime] [V3] + INP', 'Модель:')\n",
+        "model_num_widget = factory.create_text('Номер Модели:', 'Введите номера моделей для скачивания через запятую/пробел.')\n",
+        "inpainting_model_widget = factory.create_checkbox(False, 'Inpainting Модели')\n",
+        "\n",
+        "# Display Model\n",
+        "all_model_box = factory.create_vbox([model_header, model_widget, model_num_widget, inpainting_model_widget], class_names=[\"container\", \"image_1\"])\n",
+        "factory.display(all_model_box)\n",
+        "\n",
+        "# --- VAE ---\n",
+        "vae_header = factory.create_header('Выбор VAE')\n",
+        "vae_options = ['none',\n",
+        "               '1.Anime.vae',\n",
+        "               '2.Anything.vae',\n",
+        "               '3.Blessed2.vae',\n",
+        "               '4.ClearVae.vae',\n",
+        "               '5.WD.vae']\n",
+        "vae_widget = factory.create_dropdown(vae_options, '3.Blessed2.vae', 'Vae:')\n",
+        "vae_num_widget = factory.create_text('Номер Vae:', 'Введите номера vae для скачивания через запятую/пробел.')\n",
+        "\n",
+        "# Display Vae\n",
+        "all_vae_box = factory.create_vbox([vae_header, vae_widget, vae_num_widget], class_names=[\"container\", \"image_2\"])\n",
+        "factory.display(all_vae_box)\n",
+        "\n",
+        "# --- ADDITIONAL ---\n",
+        "additional_header = factory.create_header('Дополнительно')\n",
+        "latest_webui_widget = factory.create_checkbox(True, 'Обновить WebUI')\n",
+        "latest_exstensions_widget = factory.create_checkbox(True, 'Обновить Расширения')\n",
+        "change_webui_widget = factory.create_dropdown(['A1111', 'Forge'], 'A1111', 'Изменить WebUI:')\n",
+        "detailed_download_widget = factory.create_dropdown(['off', 'on'], 'off', 'Подробная Загрузка:')\n",
+        "choose_changes_widget = factory.create_hbox([latest_webui_widget, latest_exstensions_widget, change_webui_widget, detailed_download_widget])\n",
+        "\n",
+        "controlnet_options = ['none', 'ALL', '1.canny',\n",
+        "                      '2.openpose', '3.depth',\n",
+        "                      '4.normal_map', '5.mlsd',\n",
+        "                      '6.lineart', '7.soft_edge',\n",
+        "                      '8.scribble', '9.segmentation',\n",
+        "                      '10.shuffle', '11.tile',\n",
+        "                      '12.inpaint', '13.instruct_p2p']\n",
+        "controlnet_widget = factory.create_dropdown(controlnet_options, 'none', 'ControlNet:')\n",
+        "controlnet_num_widget = factory.create_text('Номер ControlNet:', 'Введите номера моделей ControlNet для скачивания через запятую/пробел.')\n",
+        "commit_hash_widget = factory.create_text('Commit Hash:')\n",
+        "huggingface_token_widget = factory.create_text('Токен HuggingFace:')\n",
+        "\n",
+        "ngrok_token_widget = factory.create_text('Токен Ngrok:')\n",
+        "ngrock_button = factory.create_html('<a href=\"https://dashboard.ngrok.com/get-started/your-authtoken\" target=\"_blank\">Получить Ngrok Токен</a>', class_name=\"button_ngrok\")\n",
+        "ngrok_widget = factory.create_hbox([ngrok_token_widget, ngrock_button])\n",
+        "\n",
+        "zrok_token_widget = factory.create_text('Токен Zrok:')\n",
+        "zrok_button = factory.create_html('<a href=\"https://colab.research.google.com/drive/1d2sjWDJi_GYBUavrHSuQyHTDuLy36WpU\" target=\"_blank\">Зарегать Zrok Токен</a>', class_name=\"button_ngrok\")\n",
+        "zrok_widget = factory.create_hbox([zrok_token_widget, zrok_button])\n",
+        "\n",
+        "commandline_arguments_options = \"--listen --enable-insecure-extension-access --theme dark --no-half-vae --disable-console-progressbars --xformers\"\n",
+        "commandline_arguments_widget = factory.create_text('Аргументы:', value=commandline_arguments_options)\n",
+        "\n",
+        "# Display Additional\n",
+        "additional_widget_list = [additional_header,\n",
+        "                          choose_changes_widget,\n",
+        "                          HR,\n",
+        "                          controlnet_widget,\n",
+        "                          controlnet_num_widget,\n",
+        "                          commit_hash_widget,\n",
+        "                          huggingface_token_widget,\n",
+        "                          ngrok_widget,\n",
+        "                          zrok_widget,\n",
+        "                          HR,\n",
+        "                          commandline_arguments_widget]\n",
+        "\n",
+        "if free_plan and env == \"Google Colab\": # remove ngrok from colab\n",
+        "    additional_widget_list.remove(ngrok_widget)\n",
+        "if os.path.exists(webui_path): # remove selection after selection ;3\n",
+        "    choose_changes_widget.children = [widget for widget in choose_changes_widget.children if widget != change_webui_widget]\n",
+        "\n",
+        "all_additional_box = factory.create_vbox(additional_widget_list, class_names=[\"container\", \"image_3\"])\n",
+        "factory.display(all_additional_box)\n",
+        "\n",
+        "# --- CUSTOM DOWNLOAD ---\n",
+        "custom_download_header_popup = factory.create_html('''\n",
+        "<style>\n",
+        "/* Term Colors */\n",
+        ".sample_label {color: #dbafff;}\n",
+        ".braces {color: #ffff00;}\n",
+        ".extension {color: #eb934b;}\n",
+        ".file_name {color: #ffffd8;}\n",
+        "</style>\n",
+        "\n",
+        "<div class=\"header\" style=\"cursor: pointer;\" onclick=\"toggleContainer()\">Кастомная Загрузка</div>\n",
+        "<!-- PopUp Window -->\n",
+        "<div class=\"info\" id=\"info_dl\">INFO</div>\n",
+        "<div class=\"popup\">\n",
+        "    Разделите несколько URL-адресов запятой/пробелом. Для <span class=\"file_name\">пользовательского имени</span> файла/расширения укажите его через <span class=\"braces\">[]</span>\n",
+        "    после URL без пробелов.\n",
+        "    <span style=\"color: #ff9999\">Для файлов обязательно укажите</span> - <span class=\"extension\">Расширение Файла.</span>\n",
+        "    <div class=\"sample\">\n",
+        "        <span class=\"sample_label\">Пример для Файла:</span>\n",
+        "        https://civitai.com/api/download/models/229782<span class=\"braces\">[</span><span class=\"file_name\">Detailer</span><span class=\"extension\">.safetensors</span><span class=\"braces\">]</span>\n",
+        "        <br>\n",
+        "        <span class=\"sample_label\">Пример для Расширения:</span>\n",
+        "        https://github.com/hako-mikan/sd-webui-regional-prompter<span class=\"braces\">[</span><span class=\"file_name\">Regional-Prompter</span><span class=\"braces\">]</span>\n",
+        "    </div>\n",
+        "</div>\n",
+        "''')\n",
+        "\n",
+        "Model_url_widget = factory.create_text('Model:')\n",
+        "Vae_url_widget = factory.create_text('Vae:')\n",
+        "LoRA_url_widget = factory.create_text('LoRa:')\n",
+        "Embedding_url_widget = factory.create_text('Embedding:')\n",
+        "Extensions_url_widget = factory.create_text('Extensions:')\n",
+        "custom_file_urls_widget = factory.create_text('Файл (txt):')\n",
+        "\n",
+        "# Display CustomDl\n",
+        "all_custom_box = factory.create_vbox([\n",
+        "    custom_download_header_popup, Model_url_widget, Vae_url_widget, LoRA_url_widget, Embedding_url_widget, Extensions_url_widget, custom_file_urls_widget\n",
+        "], class_names=[\"container\", \"image_4\", \"container_custom_downlad\"])\n",
+        "factory.display(all_custom_box)\n",
+        "\n",
+        "# --- Save Button ---\n",
+        "save_button = factory.create_button('Сохранить', class_name=\"button_save\")\n",
+        "factory.display(save_button)\n",
+        "\n",
+        "\n",
+        "# ============ Load / Save - Settings V2 ============\n",
+        "SETTINGS_FILE = f'{root_path}/settings.json'\n",
+        "\n",
+        "SETTINGS_KEYS = [\n",
+        "    'model', 'model_num', 'inpainting_model',\n",
+        "    'vae', 'vae_num', 'latest_webui', 'latest_exstensions',\n",
+        "    'change_webui', 'detailed_download', 'controlnet',\n",
+        "    'controlnet_num', 'commit_hash', 'huggingface_token',\n",
+        "    'ngrok_token', 'zrok_token', 'commandline_arguments',\n",
+        "    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url',\n",
+        "    'Extensions_url', 'custom_file_urls'\n",
+        "]\n",
+        "\n",
+        "def save_settings():\n",
+        "    settings = {key: globals()[f\"{key}_widget\"].value for key in SETTINGS_KEYS}\n",
+        "    with open(SETTINGS_FILE, 'w') as f:\n",
+        "        json.dump(settings, f, indent=2)\n",
+        "\n",
+        "def load_settings():\n",
+        "    if os.path.exists(SETTINGS_FILE):\n",
+        "        with open(SETTINGS_FILE, 'r') as f:\n",
+        "            settings = json.load(f)\n",
+        "            for key in SETTINGS_KEYS:\n",
+        "                globals()[f\"{key}_widget\"].value = settings.get(key, \"\")\n",
+        "\n",
+        "def hide_widgets():\n",
+        "    widgets_list = [all_model_box, all_vae_box, all_additional_box, all_custom_box, save_button]\n",
+        "    for widget in widgets_list:\n",
+        "        widget.add_class(\"hide\")\n",
+        "    time.sleep(0.5)\n",
+        "    widgets.Widget.close_all()\n",
+        "\n",
+        "def save_data(button):\n",
+        "    save_settings()\n",
+        "    hide_widgets()\n",
+        "\n",
+        "load_settings()\n",
+        "save_button.on_click(save_data)"
+      ]
+    }
+  ]
 }

files_cells/python/en/auto_cleaner_en.py

@@ -7,19 +7,11 @@ from ipywidgets import Label, Button, VBox, HBox
 from IPython.display import display, HTML, Javascript
 
 
-# ================= DETECT ENV =================
-def detect_environment():
-    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
-    environments = {
-        'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
-        'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content")
-    }
-    for env_var, (environment, path) in environments.items():
-        if env_var in os.environ:
-            return environment, path, free_plan
-
-env, root_path, free_plan = detect_environment()
-webui_path = f"{root_path}/sdw"
+# Setup Env
+env = os.environ.get('ENV_NAME')
+root_path = os.environ.get('ROOT_PATH')
+webui_path = os.environ.get('WEBUI_PATH')
+free_plan = os.environ.get('FREE_PLAN')
 
 
 # ==================== CSS ====================

files_cells/python/en/downloading_en.py

@@ -1,5 +1,7 @@
 ##~   DOWNLOADING CODE | BY: ANXETY   ~##
 
+from models_data import model_list, vae_list, controlnet_list
+
 import os
 import re
 import time
@@ -15,19 +17,11 @@ from IPython.display import clear_output
 from urllib.parse import urlparse, parse_qs
 
 
-# ================= DETECT ENV =================
-def detect_environment():
-    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
-    environments = {
-        'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
-        'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content")
-    }
-    for env_var, (environment, path) in environments.items():
-        if env_var in os.environ:
-            return environment, path, free_plan
-
-env, root_path, free_plan = detect_environment()
-webui_path = f"{root_path}/sdw"
+# Setup Env
+env = os.environ.get('ENV_NAME')
+root_path = os.environ.get('ROOT_PATH')
+webui_path = os.environ.get('WEBUI_PATH')
+free_plan = os.environ.get('FREE_PLAN')
 
 
 # ================ LIBRARIES V2 ================
@@ -197,112 +191,9 @@ if commit_hash:
 
 ## Downloading model and stuff | oh~ Hey! If you're freaked out by that code too, don't worry, me too!
 print("📦 Downloading models and stuff...", end='')
-model_list = {
-    "1.Anime (by XpucT) + INP": [
-        {"url": "https://huggingface.co/XpucT/Anime/resolve/main/Anime_v2.safetensors", "name": "Anime_V2.safetensors"},
-        {"url": "https://huggingface.co/XpucT/Anime/resolve/main/Anime_v2-inpainting.safetensors", "name": "Anime_V2-inpainting.safetensors"}
-    ],
-    "2.BluMix [Anime] [V7] + INP": [
-        {"url": "https://civitai.com/api/download/models/361779", "name": "BluMix_V7.safetensors"},
-        {"url": "https://civitai.com/api/download/models/363850", "name": "BluMix_V7-inpainting.safetensors"}
-    ],
-    "3.Cetus-Mix [Anime] [V4] + INP": [
-        {"url": "https://civitai.com/api/download/models/130298", "name": "CetusMix_V4.safetensors"},
-        {"url": "https://civitai.com/api/download/models/139882", "name": "CetusMix_V4-inpainting.safetensors"}
-    ],
-    "4.Counterfeit [Anime] [V3] + INP": [
-        {"url": "https://huggingface.co/gsdf/Counterfeit-V3.0/resolve/main/Counterfeit-V3.0_fix_fp16.safetensors", "name": "Counterfeit_V3.safetensors"},
-        {"url": "https://civitai.com/api/download/models/137911", "name": "Counterfeit_V3-inpainting.safetensors"}
-    ],
-    "5.CuteColor [Anime] [V3]": [
-        {"url": "https://civitai.com/api/download/models/138754", "name": "CuteColor_V3.safetensors"}
-    ],
-    "6.Dark-Sushi-Mix [Anime]": [
-        {"url": "https://civitai.com/api/download/models/101640", "name": "DarkSushiMix_2_5D.safetensors"},
-        {"url": "https://civitai.com/api/download/models/56071", "name": "DarkSushiMix_colorful.safetensors"}
-    ],
-    "7.Deliberate [Realism] [V6] + INP": [
-        {"url": "https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v6.safetensors", "name": "Deliberate_V6.safetensors"},
-        {"url": "https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v6-inpainting.safetensors", "name": "Deliberate_V6-inpainting.safetensors"}
-    ],
-    "8.Meina-Mix [Anime] [V11] + INP": [
-        {"url": "https://civitai.com/api/download/models/119057", "name": "MeinaMix_V11.safetensors"},
-        {"url": "https://civitai.com/api/download/models/120702", "name": "MeinaMix_V11-inpainting.safetensors"}
-    ],
-    "9.Mix-Pro [Anime] [V4] + INP": [
-        {"url": "https://civitai.com/api/download/models/125668", "name": "MixPro_V4.safetensors"},
-        {"url": "https://civitai.com/api/download/models/139878", "name": "MixPro_V4-inpainting.safetensors"}
-    ]
-}
-
-vae_list = {
-    "1.Anime.vae": [{"url": "https://civitai.com/api/download/models/311162", "name": "Anime.vae.safetensors"}],
-    "2.Anything.vae": [{"url": "https://huggingface.co/NoCrypt/resources/resolve/main/VAE/any.vae.safetensors", "name": "Anything.vae.safetensors"}],
-    "3.Blessed2.vae": [{"url": "https://huggingface.co/NoCrypt/resources/resolve/main/VAE/blessed2.vae.safetensors", "name": "Blessed2.vae.safetensors"}],
-    "4.ClearVae.vae": [{"url": "https://civitai.com/api/download/models/88156", "name": "ClearVae_23.vae.safetensors"}],
-    "5.WD.vae": [{"url": "https://huggingface.co/NoCrypt/resources/resolve/main/VAE/wd.vae.safetensors", "name": "WD.vae.safetensors"}]
-}
-
-controlnet_list = {
-    "1.canny": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_canny_fp16.safetensors", "name": "control_v11p_sd15_canny_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_canny_fp16.yaml", "name": "control_v11p_sd15_canny_fp16.yaml"}
-    ],
-    "2.openpose": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose_fp16.safetensors", "name": "control_v11p_sd15_openpose_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_openpose_fp16.yaml", "name": "control_v11p_sd15_openpose_fp16.yaml"}
-    ],
-    "3.depth": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors", "name": "control_v11f1p_sd15_depth_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11f1p_sd15_depth_fp16.yaml", "name": "control_v11f1p_sd15_depth_fp16.yaml"},
-        {"url": "https://huggingface.co/NagisaNao/models/resolve/main/ControlNet_v11/control_v11p_sd15_depth_anything_fp16.safetensors", "name": "control_v11p_sd15_depth_anything_fp16.safetensors"}
-    ],
-    "4.normal_map": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors", "name": "control_v11p_sd15_normalbae_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_normalbae_fp16.yaml", "name": "control_v11p_sd15_normalbae_fp16.yaml"}
-    ],
-    "5.mlsd": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors", "name": "control_v11p_sd15_mlsd_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_mlsd_fp16.yaml", "name": "control_v11p_sd15_mlsd_fp16.yaml"}
-    ],
-    "6.lineart": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_lineart_fp16.safetensors", "name": "control_v11p_sd15_lineart_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors", "name": "control_v11p_sd15s2_lineart_anime_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_lineart_fp16.yaml", "name": "control_v11p_sd15_lineart_fp16.yaml"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15s2_lineart_anime_fp16.yaml", "name": "control_v11p_sd15s2_lineart_anime_fp16.yaml"}
-    ],
-    "7.soft_edge": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_softedge_fp16.safetensors", "name": "control_v11p_sd15_softedge_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_softedge_fp16.yaml", "name": "control_v11p_sd15_softedge_fp16.yaml"}
-    ],
-    "8.scribble": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_scribble_fp16.safetensors", "name": "control_v11p_sd15_scribble_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_scribble_fp16.yaml", "name": "control_v11p_sd15_scribble_fp16.yaml"}
-    ],
-    "9.segmentation": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_seg_fp16.safetensors", "name": "control_v11p_sd15_seg_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_seg_fp16.yaml", "name": "control_v11p_sd15_seg_fp16.yaml"}
-    ],
-    "10.shuffle": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_shuffle_fp16.safetensors", "name": "control_v11e_sd15_shuffle_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_shuffle_fp16.yaml", "name": "control_v11e_sd15_shuffle_fp16.yaml"}
-    ],
-    "11.tile": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11f1e_sd15_tile_fp16.safetensors", "name": "control_v11f1e_sd15_tile_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11f1e_sd15_tile_fp16.yaml", "name": "control_v11f1e_sd15_tile_fp16.yaml"}
-    ],
-    "12.inpaint": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors", "name": "control_v11p_sd15_inpaint_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_inpaint_fp16.yaml", "name": "control_v11p_sd15_inpaint_fp16.yaml"}
-    ],
-    "13.instruct_p2p": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_ip2p_fp16.safetensors", "name": "control_v11e_sd15_ip2p_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_ip2p_fp16.yaml", "name": "control_v11e_sd15_ip2p_fp16.yaml"}
-    ]
-}
 
 url = ""
-prefixes = {
+PREFIXES = {
     "model": models_dir,
     "vae": vaes_dir,
     "lora": loras_dir,
@@ -314,7 +205,7 @@ prefixes = {
 }
 
 extension_repo = []
-directories = [value for key, value in prefixes.items()] # for unpucking zip files
+directories = [value for key, value in PREFIXES.items()] # for unpucking zip files
 get_ipython().system('mkdir -p {" ".join(directories)}')
 
 hf_token = huggingface_token if huggingface_token else "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"
@@ -325,137 +216,123 @@ user_header = f"\"Authorization: Bearer {hf_token}\""
 from math import floor
 
 def center_text(text, terminal_width=45):
-    text_length = len(text)
-    left_padding = floor((terminal_width - text_length) / 2)
-    right_padding = terminal_width - text_length - left_padding
-    return f"\033[1m\033[36m{' ' * left_padding}{text}{' ' * right_padding}\033[0m\033[32m"
+    padding = (terminal_width - len(text)) // 2
+    return f"\033[1m\033[36m{' ' * padding}{text}{' ' * padding}\033[0m\033[32m"
 
 def format_output(url, dst_dir, file_name):
-    info = f"[{file_name.split('.')[0]}]"
-    info = center_text(info)
+    info = center_text(f"[{file_name.split('.')[0]}]")
+    separation_line = '\033[32m' + '---' * 20
 
-    print(f"\n\033[32m{'---'*20}]{info}[{'---'*20}")
+    print(f"\n{separation_line}{info}{separation_line}")
     print(f"\033[33mURL: \033[34m{url}")
     print(f"\033[33mSAVE DIR: \033[34m{dst_dir}")
     print(f"\033[33mFILE NAME: \033[34m{file_name}\033[0m")
 
 ''' GET CivitAi API - DATA '''
 
-def strip_(url, file_name=None):
-    if 'github.com' in url:
-        if '/blob/' in url:
-            url = url.replace('/blob/', '/raw/')
-
-    elif "civitai.com" in url:
-        return CivitAi_API(url, file_name)
-
-    elif "huggingface.co" in url:
-        if '/blob/' in url:
-            url = url.replace('/blob/', '/resolve/')
-        if '?' in url:
-            url = url.split('?')[0]
-
-    return url
-
 def CivitAi_API(url, file_name=None):
-    support_types = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')
-    civitai_token = "62c0c5956b2f9defbd844d754000180b"
+    SUPPORT_TYPES = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')
+    CIVITAI_TOKEN = "62c0c5956b2f9defbd844d754000180b"
 
-    if '?token=' in url:
-        url = url.split('?token=')[0]
-    if '?type=' in url:
-        url = url.replace('?type=', f'?token={civitai_token}&type=')
-    else:
-        url = f"{url}?token={civitai_token}"
-
-    # Determine model or version id
-    if "civitai.com/models/" in url:
-        if '?modelVersionId=' in url:
-            version_id = url.split('?modelVersionId=')[1]
-            response = requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}")
-            # print(f"end - https://civitai.com/api/v1/model-versions/{version_id}")
+    url = url.split('?token=')[0] if '?token=' in url else url
+    url = url.replace('?type=', f'?token={CIVITAI_TOKEN}&type=') if '?type=' in url else f"{url}?token={CIVITAI_TOKEN}"
+
+    def get_model_data(url):
+        if "civitai.com/models/" in url:
+            if '?modelVersionId=' in url:
+                version_id = url.split('?modelVersionId=')[1]
+                return requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}").json()
+            else:
+                model_id = url.split('/models/')[1].split('/')[0]
+                return requests.get(f"https://civitai.com/api/v1/models/{model_id}").json()
         else:
-            model_id = url.split('/models/')[1].split('/')[0]
-            response = requests.get(f"https://civitai.com/api/v1/models/{model_id}")
-            # print(f"end - https://civitai.com/api/v1/models/{model_id}")
-    else:
-        version_id = url.split('/models/')[1].split('/')[0]
-        response = requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}")
-        # print(f"end - https://civitai.com/api/v1/model-versions/{version_id}")
+            version_id = url.split('/models/')[1].split('/')[0]
+            return requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}").json()
 
-    data = response.json()
+    data = get_model_data(url)
 
-    if response.status_code != 200:
+    if not data:
         return None, None, None, None, None, None, None
 
-    # Define model type and name
-    if "civitai.com/models/" in url:
-        if '?modelVersionId=' in url:
-            model_type = data['model']['type']
-            model_name = data['files'][0]['name']
+    def extract_model_info(url, data):
+        if "civitai.com/models/" in url:
+            if '?modelVersionId=' in url:
+                model_type = data['model']['type']
+                model_name = data['files'][0]['name']
+            else:
+                model_type = data['type']
+                model_name = data['modelVersions'][0]['files'][0]['name']
+        elif 'type=' in url:
+            model_type = parse_qs(urlparse(url).query).get('type', [''])[0]
+            if 'model' in model_type.lower():
+                model_name = data['files'][0]['name']
+            else:
+                model_name = data['files'][1]['name']
         else:
-            model_type = data['type']
-            model_name = data['modelVersions'][0]['files'][0]['name']
-    elif 'type=' in url:
-        model_type = parse_qs(urlparse(url).query).get('type', [''])[0]
-        if 'model' in model_type.lower():
+            model_type = data['model']['type']
             model_name = data['files'][0]['name']
-        else:
-            model_name = data['files'][1]['name']
-    else:
-        model_type = data['model']['type']
-        model_name = data['files'][0]['name']
+        return model_type, model_name
 
+    model_type, model_name = extract_model_info(url, data)
     model_name = file_name or model_name
 
-    # Determine DownloadUrl
-    if "civitai.com/models/" in url:
-        if '?modelVersionId=' in url:
-            download_url = data.get('downloadUrl')
-        else:
-            download_url = data["modelVersions"][0].get("downloadUrl", "")
-    elif 'type=' in url:
-        if any(t.lower() in model_type.lower() for t in support_types):
-            download_url = data['files'][0]['downloadUrl']
+    def get_download_url(url, data, model_type):
+        if "civitai.com/models/" in url:
+            if '?modelVersionId=' in url:
+                return data.get('downloadUrl')
+            else:
+                return data["modelVersions"][0].get("downloadUrl", "")
+        elif 'type=' in url:
+            if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):
+                return data['files'][0]['downloadUrl']
+            else:
+                return data['files'][1]['downloadUrl']
         else:
-            download_url = data['files'][1]['downloadUrl']
-    else:
-        download_url = data.get('downloadUrl')
+            return data.get('downloadUrl')
 
-    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url) # hide token
+    download_url = get_download_url(url, data, model_type)
+    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url)
 
-    # Find a safe image: level less than 4 | Kaggle
-    image_url, image_name = None, None
-    if any(t in model_type for t in support_types):
-        try:
-            images = data.get('images') or data['modelVersions'][0].get('images', [])
-            if env == 'Kaggle':
-                image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)
-            else:
-                image_url = images[0]['url'] if images else None
-        except KeyError:
-            pass
+    def get_image_info(data, model_type, model_name):
+        image_url, image_name = None, None
+        if any(t in model_type for t in SUPPORT_TYPES):
+            try:
+                images = data.get('images') or data['modelVersions'][0].get('images', [])
+                if env == 'Kaggle':
+                    image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)
+                else:
+                    image_url = images[0]['url'] if images else None
+            except KeyError:
+                pass
+
+            image_name = f"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}" if image_url else None
+        return image_url, image_name
 
-        # Generate a name to save the image
-        image_name = f"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}" if image_url else None
+    image_url, image_name = get_image_info(data, model_type, model_name)
 
-    return f"{download_url}{'&' if '?' in download_url else '?'}token={civitai_token}", clean_url, model_type, model_name, image_url, image_name, data
+    return f"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}", clean_url, model_type, model_name, image_url, image_name, data
 
 ''' Main Download Code '''
 
+def strip_(url):
+    if 'github.com' in url:
+        return url.replace('/blob/', '/raw/')
+    elif "huggingface.co" in url:
+        url = url.replace('/blob/', '/resolve/')
+        return url.split('?')[0] if '?' in url else url
+    return url
+
 def download(url):
     links_and_paths = [link_or_path.strip() for link_or_path in url.split(',') if link_or_path.strip()]
 
     for link_or_path in links_and_paths:
-        if any(link_or_path.lower().startswith(prefix) for prefix in prefixes):
+        if any(link_or_path.lower().startswith(prefix) for prefix in PREFIXES):
             handle_manual(link_or_path)
         else:
             url, dst_dir, file_name = link_or_path.split()
             manual_download(url, dst_dir, file_name)
 
-    unpack_zip_files()
-
-def unpack_zip_files():
+    # Unpuck ZIPs Files
     for directory in directories:
         for root, _, files in os.walk(directory):
             for file in files:
@@ -475,8 +352,8 @@ def handle_manual(url):
     if file_name:
         path = re.sub(r'\[.*?\]', '', path)
 
-    if prefix in prefixes:
-        dir = prefixes[prefix]
+    if prefix in PREFIXES:
+        dir = PREFIXES[prefix]
         if prefix != "extension":
             try:
                 manual_download(path, dir, file_name=file_name)
@@ -490,60 +367,52 @@ def manual_download(url, dst_dir, file_name):
     aria2c_header = "--header='User-Agent: Mozilla/5.0' --allow-overwrite=true"
     aria2_args = "--optimize-concurrent-downloads --console-log-level=error --summary-interval=10 --stderr=true -c -x16 -s16 -k1M -j5"
 
-    if 'github.com' in url:
-        url = strip_(url)
-
-    # -- CivitAi APi+ V2 --
-    elif 'civitai' in url:
-        url, clean_url, model_type, file_name, image_url, image_name, data = strip_(url, file_name)
+    clean_url = strip_(url)
 
+    if 'civitai' in url:
+        url, clean_url, model_type, file_name, image_url, image_name, data = CivitAi_API(url, file_name)
         if image_url and image_name:
-            with capture.capture_output() as cap:
-                get_ipython().system("aria2c {aria2_args} -d {dst_dir} -o '{image_name}' '{image_url}'")
-            del cap
+            command = ["aria2c"] + aria2_args.split() + ["-d", dst_dir, "-o", image_name, image_url]
+            subprocess.run(command, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
 
-    elif "huggingface.co" in url:
-        clean_url = strip_(url)
-        basename = clean_url.split("/")[-1] if file_name is None else file_name
+    elif 'github' in url or "huggingface.co" in url:
+        basename = url.split("/")[-1] if file_name is None else file_name
 
     """ Formatted info output """
-    model_name_or_basename = file_name if not 'huggingface' in url else basename
+    model_name_or_basename = file_name if file_name else basename
     format_output(clean_url or url, dst_dir, model_name_or_basename)
 
-    # ## -- for my tests --
     # print(url, dst_dir, model_name_or_basename)
-    print(f"\033[31m[Data Info]:\033[0m Failed to retrieve data from the API.\n") if 'civitai' in url and not data else None
-    if 'civitai' in url and data and image_name:
-        print(f"\033[32m[Preview DL]:\033[0m {image_name} - {image_url}\n")
+    if 'civitai' in url:
+        if not data:
+            print("\033[31m[Data Info]:\033[0m Failed to retrieve data from the API.\n")
+        if data and image_name:
+            print(f"\033[32m[Preview DL]:\033[0m {image_name} - {image_url}\n")
+
     # =====================
+    def run_aria2c(url, dst_dir, file_name=None, args="", header=""):
+        out = f"-o '{file_name}'" if file_name else ""
+        get_ipython().system("aria2c {header} {args} -d {dst_dir} {out} '{url}'")
 
-    # -- Git Hub --
-    if 'github.com' in url or 'githubusercontent.com' in url:
-        get_ipython().system("aria2c {aria2_args} -d {dst_dir} -o '{basename}' '{url}'")
-
-    # -- GDrive --
-    elif 'drive.google' in url:
-        try:
-            have_drive_link
-        except:
-            get_ipython().system('pip install -q gdown==5.2.0 > /dev/null')
-            have_drive_link = True
+    # -- Google Drive --
+    if 'drive.google' in url:
+        if not globals().get('have_drive_link', False):
+            os.system("pip install -U gdown > /dev/null")
+            globals()['have_drive_link'] = True
 
         if 'folders' in url:
-            get_ipython().system('gdown --folder "{url}" -O {dst_dir} --fuzzy -c')
+            os.system(f"gdown --folder \"{url}\" -O {dst_dir} --fuzzy -c")
         else:
-            if file_name:
-               get_ipython().system('gdown "{url}" -O {dst_dir}/{file_name} --fuzzy -c')
-            else:
-               get_ipython().system('gdown "{url}" -O {dst_dir} --fuzzy -c')
+            out_path = f"{dst_dir}/{file_name}" if file_name else dst_dir
+            os.system(f"gdown \"{url}\" -O {out_path} --fuzzy -c")
 
-    # -- Hugging Face --
-    elif 'huggingface' in url:
-        get_ipython().system("aria2c {header_option} {aria2_args} -d {dst_dir} -o '{basename}' '{url}'")
+    # -- GitHub or Hugging Face --
+    elif 'github' in url or 'huggingface' in url:
+        run_aria2c(clean_url, dst_dir, basename, aria2_args, header_option if 'huggingface' in url else '')
 
-    # -- Other --
+    # -- Other HTTP/Sources --
     elif 'http' in url:
-        get_ipython().system('aria2c {aria2c_header} {aria2_args} -d {dst_dir} -o "{file_name if file_name else \'\'}" \'{url}\'')
+        run_aria2c(url, dst_dir, file_name, aria2_args, aria2c_header)
 
 ''' SubModels - Added URLs '''
 
@@ -583,7 +452,7 @@ url = handle_submodels(controlnet, controlnet_num, controlnet_list, control_dir,
 
 ''' file.txt - added urls '''
 
-def process_file_download(file_url, prefixes, unique_urls):
+def process_file_download(file_url, PREFIXES, unique_urls):
     files_urls = ""
 
     if file_url.startswith("http"):
@@ -598,8 +467,8 @@ def process_file_download(file_url, prefixes, unique_urls):
     current_tag = None
     for line in lines:
         line = line.strip()
-        if any(f'# {tag}' in line.lower() for tag in prefixes):
-            current_tag = next((tag for tag in prefixes if tag in line.lower()))
+        if any(f'# {tag}' in line.lower() for tag in PREFIXES):
+            current_tag = next((tag for tag in PREFIXES if tag in line.lower()))
 
         urls = [url.split('#')[0].strip() for url in line.split(',')]  # filter urls
         for url in urls:
@@ -623,13 +492,13 @@ if custom_file_urls:
                 custom_file_url = f'{root_path}/{custom_file_url}'
 
         try:
-            file_urls += process_file_download(custom_file_url, prefixes, unique_urls)
+            file_urls += process_file_download(custom_file_url, PREFIXES, unique_urls)
         except FileNotFoundError:
             pass
 
 # url prefixing
 urls = (Model_url, Vae_url, LoRA_url, Embedding_url, Extensions_url)
-prefixed_urls = (f"{prefix}:{url}" for prefix, url in zip(prefixes.keys(), urls) if url for url in url.replace(',', '').split())
+prefixed_urls = (f"{prefix}:{url}" for prefix, url in zip(PREFIXES.keys(), urls) if url for url in url.replace(',', '').split())
 url += ", ".join(prefixed_urls) + ", " + file_urls
 
 if detailed_download == "on":

files_cells/python/en/launch_en.py

@@ -9,21 +9,13 @@ import cloudpickle as pickle
 from datetime import timedelta
 from IPython.display import clear_output
 
-#  ================= DETECT ENV =================
-def detect_environment():
-    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
-    environments = {
-        'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
-        'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content")
-    }
-
-    for env_var, (environment, path) in environments.items():
-        if env_var in os.environ:
-            return environment, path, free_plan
-    return 'Unknown', '/unknown/path', free_plan
-
-env, root_path, free_plan = detect_environment()
-webui_path = f"{root_path}/sdw"
+
+# Setup Env
+env = os.environ.get('ENV_NAME')
+root_path = os.environ.get('ROOT_PATH')
+webui_path = os.environ.get('WEBUI_PATH')
+free_plan = os.environ.get('FREE_PLAN')
+
 
 def load_settings():
     SETTINGS_FILE = f'{root_path}/settings.json'
@@ -38,6 +30,7 @@ zrok_token = settings.get('zrok_token', "")
 commandline_arguments = settings.get('commandline_arguments', "")
 change_webui = settings.get('change_webui', "")
 
+
 # ======================== TUNNEL V2 ========================
 print('Please Wait...')
 
@@ -71,6 +64,7 @@ if zrok_token:
 
 clear_output()
 
+
 # =============== Automatic Fixing Path V3 ===============
 paths_to_check = {
     "tagger_hf_cache_dir": f"{webui_path}/models/interrogators/",

files_cells/python/en/widgets_en.py

@@ -8,20 +8,11 @@ from ipywidgets import widgets, Layout, Label, Button, VBox, HBox
 from IPython.display import display, HTML, Javascript, clear_output
 
 
-# ================= DETECT ENV =================
-def detect_environment():
-    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
-    environments = {
-        'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
-        'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content")
-    }
-    for env_var, (environment, path) in environments.items():
-        if env_var in os.environ:
-            return environment, path, free_plan
-
-env, root_path, free_plan = detect_environment()
-webui_path = f"{root_path}/sdw"
-get_ipython().system('mkdir -p {root_path}')
+# Setup Env
+env = os.environ.get('ENV_NAME')
+root_path = os.environ.get('ROOT_PATH')
+webui_path = os.environ.get('WEBUI_PATH')
+free_plan = os.environ.get('FREE_PLAN')
 
 
 # ==================== CSS JS ====================

files_cells/python/ru/auto_cleaner_ru.py

@@ -7,19 +7,11 @@ from ipywidgets import Label, Button, VBox, HBox
 from IPython.display import display, HTML
 
 
-# ================= DETECT ENV =================
-def detect_environment():
-    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
-    environments = {
-        'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
-        'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content")
-    }
-    for env_var, (environment, path) in environments.items():
-        if env_var in os.environ:
-            return environment, path, free_plan
-
-env, root_path, free_plan = detect_environment()
-webui_path = f"{root_path}/sdw"
+# Setup Env
+env = os.environ.get('ENV_NAME')
+root_path = os.environ.get('ROOT_PATH')
+webui_path = os.environ.get('WEBUI_PATH')
+free_plan = os.environ.get('FREE_PLAN')
 
 
 # ==================== CSS ====================

files_cells/python/ru/downloading_ru.py

@@ -1,5 +1,7 @@
 ##~   DOWNLOADING CODE | BY: ANXETY   ~##
 
+from models_data import model_list, vae_list, controlnet_list
+
 import os
 import re
 import time
@@ -15,19 +17,11 @@ from IPython.display import clear_output
 from urllib.parse import urlparse, parse_qs
 
 
-# ================= DETECT ENV =================
-def detect_environment():
-    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
-    environments = {
-        'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
-        'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content")
-    }
-    for env_var, (environment, path) in environments.items():
-        if env_var in os.environ:
-            return environment, path, free_plan
-
-env, root_path, free_plan = detect_environment()
-webui_path = f"{root_path}/sdw"
+# Setup Env
+env = os.environ.get('ENV_NAME')
+root_path = os.environ.get('ROOT_PATH')
+webui_path = os.environ.get('WEBUI_PATH')
+free_plan = os.environ.get('FREE_PLAN')
 
 
 # ================ LIBRARIES V2 ================
@@ -197,112 +191,9 @@ if commit_hash:
 
 ## Downloading model and stuff | oh~ Hey! If you're freaked out by that code too, don't worry, me too!
 print("📦 Скачивание моделей и прочего...", end='')
-model_list = {
-    "1.Anime (by XpucT) + INP": [
-        {"url": "https://huggingface.co/XpucT/Anime/resolve/main/Anime_v2.safetensors", "name": "Anime_V2.safetensors"},
-        {"url": "https://huggingface.co/XpucT/Anime/resolve/main/Anime_v2-inpainting.safetensors", "name": "Anime_V2-inpainting.safetensors"}
-    ],
-    "2.BluMix [Anime] [V7] + INP": [
-        {"url": "https://civitai.com/api/download/models/361779", "name": "BluMix_V7.safetensors"},
-        {"url": "https://civitai.com/api/download/models/363850", "name": "BluMix_V7-inpainting.safetensors"}
-    ],
-    "3.Cetus-Mix [Anime] [V4] + INP": [
-        {"url": "https://civitai.com/api/download/models/130298", "name": "CetusMix_V4.safetensors"},
-        {"url": "https://civitai.com/api/download/models/139882", "name": "CetusMix_V4-inpainting.safetensors"}
-    ],
-    "4.Counterfeit [Anime] [V3] + INP": [
-        {"url": "https://huggingface.co/gsdf/Counterfeit-V3.0/resolve/main/Counterfeit-V3.0_fix_fp16.safetensors", "name": "Counterfeit_V3.safetensors"},
-        {"url": "https://civitai.com/api/download/models/137911", "name": "Counterfeit_V3-inpainting.safetensors"}
-    ],
-    "5.CuteColor [Anime] [V3]": [
-        {"url": "https://civitai.com/api/download/models/138754", "name": "CuteColor_V3.safetensors"}
-    ],
-    "6.Dark-Sushi-Mix [Anime]": [
-        {"url": "https://civitai.com/api/download/models/101640", "name": "DarkSushiMix_2_5D.safetensors"},
-        {"url": "https://civitai.com/api/download/models/56071", "name": "DarkSushiMix_colorful.safetensors"}
-    ],
-    "7.Deliberate [Realism] [V6] + INP": [
-        {"url": "https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v6.safetensors", "name": "Deliberate_V6.safetensors"},
-        {"url": "https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v6-inpainting.safetensors", "name": "Deliberate_V6-inpainting.safetensors"}
-    ],
-    "8.Meina-Mix [Anime] [V11] + INP": [
-        {"url": "https://civitai.com/api/download/models/119057", "name": "MeinaMix_V11.safetensors"},
-        {"url": "https://civitai.com/api/download/models/120702", "name": "MeinaMix_V11-inpainting.safetensors"}
-    ],
-    "9.Mix-Pro [Anime] [V4] + INP": [
-        {"url": "https://civitai.com/api/download/models/125668", "name": "MixPro_V4.safetensors"},
-        {"url": "https://civitai.com/api/download/models/139878", "name": "MixPro_V4-inpainting.safetensors"}
-    ]
-}
-
-vae_list = {
-    "1.Anime.vae": [{"url": "https://civitai.com/api/download/models/311162", "name": "Anime.vae.safetensors"}],
-    "2.Anything.vae": [{"url": "https://huggingface.co/NoCrypt/resources/resolve/main/VAE/any.vae.safetensors", "name": "Anything.vae.safetensors"}],
-    "3.Blessed2.vae": [{"url": "https://huggingface.co/NoCrypt/resources/resolve/main/VAE/blessed2.vae.safetensors", "name": "Blessed2.vae.safetensors"}],
-    "4.ClearVae.vae": [{"url": "https://civitai.com/api/download/models/88156", "name": "ClearVae_23.vae.safetensors"}],
-    "5.WD.vae": [{"url": "https://huggingface.co/NoCrypt/resources/resolve/main/VAE/wd.vae.safetensors", "name": "WD.vae.safetensors"}]
-}
-
-controlnet_list = {
-    "1.canny": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_canny_fp16.safetensors", "name": "control_v11p_sd15_canny_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_canny_fp16.yaml", "name": "control_v11p_sd15_canny_fp16.yaml"}
-    ],
-    "2.openpose": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose_fp16.safetensors", "name": "control_v11p_sd15_openpose_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_openpose_fp16.yaml", "name": "control_v11p_sd15_openpose_fp16.yaml"}
-    ],
-    "3.depth": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors", "name": "control_v11f1p_sd15_depth_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11f1p_sd15_depth_fp16.yaml", "name": "control_v11f1p_sd15_depth_fp16.yaml"},
-        {"url": "https://huggingface.co/NagisaNao/models/resolve/main/ControlNet_v11/control_v11p_sd15_depth_anything_fp16.safetensors", "name": "control_v11p_sd15_depth_anything_fp16.safetensors"}
-    ],
-    "4.normal_map": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors", "name": "control_v11p_sd15_normalbae_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_normalbae_fp16.yaml", "name": "control_v11p_sd15_normalbae_fp16.yaml"}
-    ],
-    "5.mlsd": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors", "name": "control_v11p_sd15_mlsd_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_mlsd_fp16.yaml", "name": "control_v11p_sd15_mlsd_fp16.yaml"}
-    ],
-    "6.lineart": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_lineart_fp16.safetensors", "name": "control_v11p_sd15_lineart_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors", "name": "control_v11p_sd15s2_lineart_anime_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_lineart_fp16.yaml", "name": "control_v11p_sd15_lineart_fp16.yaml"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15s2_lineart_anime_fp16.yaml", "name": "control_v11p_sd15s2_lineart_anime_fp16.yaml"}
-    ],
-    "7.soft_edge": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_softedge_fp16.safetensors", "name": "control_v11p_sd15_softedge_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_softedge_fp16.yaml", "name": "control_v11p_sd15_softedge_fp16.yaml"}
-    ],
-    "8.scribble": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_scribble_fp16.safetensors", "name": "control_v11p_sd15_scribble_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_scribble_fp16.yaml", "name": "control_v11p_sd15_scribble_fp16.yaml"}
-    ],
-    "9.segmentation": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_seg_fp16.safetensors", "name": "control_v11p_sd15_seg_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_seg_fp16.yaml", "name": "control_v11p_sd15_seg_fp16.yaml"}
-    ],
-    "10.shuffle": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_shuffle_fp16.safetensors", "name": "control_v11e_sd15_shuffle_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_shuffle_fp16.yaml", "name": "control_v11e_sd15_shuffle_fp16.yaml"}
-    ],
-    "11.tile": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11f1e_sd15_tile_fp16.safetensors", "name": "control_v11f1e_sd15_tile_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11f1e_sd15_tile_fp16.yaml", "name": "control_v11f1e_sd15_tile_fp16.yaml"}
-    ],
-    "12.inpaint": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors", "name": "control_v11p_sd15_inpaint_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_inpaint_fp16.yaml", "name": "control_v11p_sd15_inpaint_fp16.yaml"}
-    ],
-    "13.instruct_p2p": [
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_ip2p_fp16.safetensors", "name": "control_v11e_sd15_ip2p_fp16.safetensors"},
-        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_ip2p_fp16.yaml", "name": "control_v11e_sd15_ip2p_fp16.yaml"}
-    ]
-}
 
 url = ""
-prefixes = {
+PREFIXES = {
     "model": models_dir,
     "vae": vaes_dir,
     "lora": loras_dir,
@@ -314,7 +205,7 @@ prefixes = {
 }
 
 extension_repo = []
-directories = [value for key, value in prefixes.items()] # for unpucking zip files
+directories = [value for key, value in PREFIXES.items()] # for unpucking zip files
 get_ipython().system('mkdir -p {" ".join(directories)}')
 
 hf_token = huggingface_token if huggingface_token else "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"
@@ -325,137 +216,123 @@ user_header = f"\"Authorization: Bearer {hf_token}\""
 from math import floor
 
 def center_text(text, terminal_width=45):
-    text_length = len(text)
-    left_padding = floor((terminal_width - text_length) / 2)
-    right_padding = terminal_width - text_length - left_padding
-    return f"\033[1m\033[36m{' ' * left_padding}{text}{' ' * right_padding}\033[0m\033[32m"
+    padding = (terminal_width - len(text)) // 2
+    return f"\033[1m\033[36m{' ' * padding}{text}{' ' * padding}\033[0m\033[32m"
 
 def format_output(url, dst_dir, file_name):
-    info = f"[{file_name.split('.')[0]}]"
-    info = center_text(info)
+    info = center_text(f"[{file_name.split('.')[0]}]")
+    separation_line = '\033[32m' + '---' * 20
 
-    print(f"\n\033[32m{'---'*20}]{info}[{'---'*20}")
+    print(f"\n{separation_line}{info}{separation_line}")
     print(f"\033[33mURL: \033[34m{url}")
     print(f"\033[33mSAVE DIR: \033[34m{dst_dir}")
     print(f"\033[33mFILE NAME: \033[34m{file_name}\033[0m")
 
 ''' GET CivitAi API - DATA '''
 
-def strip_(url, file_name=None):
-    if 'github.com' in url:
-        if '/blob/' in url:
-            url = url.replace('/blob/', '/raw/')
-
-    elif "civitai.com" in url:
-        return CivitAi_API(url, file_name)
-
-    elif "huggingface.co" in url:
-        if '/blob/' in url:
-            url = url.replace('/blob/', '/resolve/')
-        if '?' in url:
-            url = url.split('?')[0]
-
-    return url
-
 def CivitAi_API(url, file_name=None):
-    support_types = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')
-    civitai_token = "62c0c5956b2f9defbd844d754000180b"
+    SUPPORT_TYPES = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')
+    CIVITAI_TOKEN = "62c0c5956b2f9defbd844d754000180b"
 
-    if '?token=' in url:
-        url = url.split('?token=')[0]
-    if '?type=' in url:
-        url = url.replace('?type=', f'?token={civitai_token}&type=')
-    else:
-        url = f"{url}?token={civitai_token}"
-
-    # Determine model or version id
-    if "civitai.com/models/" in url:
-        if '?modelVersionId=' in url:
-            version_id = url.split('?modelVersionId=')[1]
-            response = requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}")
-            # print(f"end - https://civitai.com/api/v1/model-versions/{version_id}")
+    url = url.split('?token=')[0] if '?token=' in url else url
+    url = url.replace('?type=', f'?token={CIVITAI_TOKEN}&type=') if '?type=' in url else f"{url}?token={CIVITAI_TOKEN}"
+
+    def get_model_data(url):
+        if "civitai.com/models/" in url:
+            if '?modelVersionId=' in url:
+                version_id = url.split('?modelVersionId=')[1]
+                return requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}").json()
+            else:
+                model_id = url.split('/models/')[1].split('/')[0]
+                return requests.get(f"https://civitai.com/api/v1/models/{model_id}").json()
         else:
-            model_id = url.split('/models/')[1].split('/')[0]
-            response = requests.get(f"https://civitai.com/api/v1/models/{model_id}")
-            # print(f"end - https://civitai.com/api/v1/models/{model_id}")
-    else:
-        version_id = url.split('/models/')[1].split('/')[0]
-        response = requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}")
-        # print(f"end - https://civitai.com/api/v1/model-versions/{version_id}")
+            version_id = url.split('/models/')[1].split('/')[0]
+            return requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}").json()
 
-    data = response.json()
+    data = get_model_data(url)
 
-    if response.status_code != 200:
+    if not data:
         return None, None, None, None, None, None, None
 
-    # Define model type and name
-    if "civitai.com/models/" in url:
-        if '?modelVersionId=' in url:
-            model_type = data['model']['type']
-            model_name = data['files'][0]['name']
+    def extract_model_info(url, data):
+        if "civitai.com/models/" in url:
+            if '?modelVersionId=' in url:
+                model_type = data['model']['type']
+                model_name = data['files'][0]['name']
+            else:
+                model_type = data['type']
+                model_name = data['modelVersions'][0]['files'][0]['name']
+        elif 'type=' in url:
+            model_type = parse_qs(urlparse(url).query).get('type', [''])[0]
+            if 'model' in model_type.lower():
+                model_name = data['files'][0]['name']
+            else:
+                model_name = data['files'][1]['name']
         else:
-            model_type = data['type']
-            model_name = data['modelVersions'][0]['files'][0]['name']
-    elif 'type=' in url:
-        model_type = parse_qs(urlparse(url).query).get('type', [''])[0]
-        if 'model' in model_type.lower():
+            model_type = data['model']['type']
             model_name = data['files'][0]['name']
-        else:
-            model_name = data['files'][1]['name']
-    else:
-        model_type = data['model']['type']
-        model_name = data['files'][0]['name']
+        return model_type, model_name
 
+    model_type, model_name = extract_model_info(url, data)
     model_name = file_name or model_name
 
-    # Determine DownloadUrl
-    if "civitai.com/models/" in url:
-        if '?modelVersionId=' in url:
-            download_url = data.get('downloadUrl')
-        else:
-            download_url = data["modelVersions"][0].get("downloadUrl", "")
-    elif 'type=' in url:
-        if any(t.lower() in model_type.lower() for t in support_types):
-            download_url = data['files'][0]['downloadUrl']
+    def get_download_url(url, data, model_type):
+        if "civitai.com/models/" in url:
+            if '?modelVersionId=' in url:
+                return data.get('downloadUrl')
+            else:
+                return data["modelVersions"][0].get("downloadUrl", "")
+        elif 'type=' in url:
+            if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):
+                return data['files'][0]['downloadUrl']
+            else:
+                return data['files'][1]['downloadUrl']
         else:
-            download_url = data['files'][1]['downloadUrl']
-    else:
-        download_url = data.get('downloadUrl')
+            return data.get('downloadUrl')
 
-    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url) # hide token
+    download_url = get_download_url(url, data, model_type)
+    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url)
 
-    # Find a safe image: level less than 4 | Kaggle
-    image_url, image_name = None, None
-    if any(t in model_type for t in support_types):
-        try:
-            images = data.get('images') or data['modelVersions'][0].get('images', [])
-            if env == 'Kaggle':
-                image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)
-            else:
-                image_url = images[0]['url'] if images else None
-        except KeyError:
-            pass
+    def get_image_info(data, model_type, model_name):
+        image_url, image_name = None, None
+        if any(t in model_type for t in SUPPORT_TYPES):
+            try:
+                images = data.get('images') or data['modelVersions'][0].get('images', [])
+                if env == 'Kaggle':
+                    image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)
+                else:
+                    image_url = images[0]['url'] if images else None
+            except KeyError:
+                pass
+
+            image_name = f"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}" if image_url else None
+        return image_url, image_name
 
-        # Generate a name to save the image
-        image_name = f"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}" if image_url else None
+    image_url, image_name = get_image_info(data, model_type, model_name)
 
-    return f"{download_url}{'&' if '?' in download_url else '?'}token={civitai_token}", clean_url, model_type, model_name, image_url, image_name, data
+    return f"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}", clean_url, model_type, model_name, image_url, image_name, data
 
 ''' Main Download Code '''
 
+def strip_(url):
+    if 'github.com' in url:
+        return url.replace('/blob/', '/raw/')
+    elif "huggingface.co" in url:
+        url = url.replace('/blob/', '/resolve/')
+        return url.split('?')[0] if '?' in url else url
+    return url
+
 def download(url):
     links_and_paths = [link_or_path.strip() for link_or_path in url.split(',') if link_or_path.strip()]
 
     for link_or_path in links_and_paths:
-        if any(link_or_path.lower().startswith(prefix) for prefix in prefixes):
+        if any(link_or_path.lower().startswith(prefix) for prefix in PREFIXES):
             handle_manual(link_or_path)
         else:
             url, dst_dir, file_name = link_or_path.split()
             manual_download(url, dst_dir, file_name)
 
-    unpack_zip_files()
-
-def unpack_zip_files():
+    # Unpuck ZIPs Files
     for directory in directories:
         for root, _, files in os.walk(directory):
             for file in files:
@@ -475,8 +352,8 @@ def handle_manual(url):
     if file_name:
         path = re.sub(r'\[.*?\]', '', path)
 
-    if prefix in prefixes:
-        dir = prefixes[prefix]
+    if prefix in PREFIXES:
+        dir = PREFIXES[prefix]
         if prefix != "extension":
             try:
                 manual_download(path, dir, file_name=file_name)
@@ -490,60 +367,52 @@ def manual_download(url, dst_dir, file_name):
     aria2c_header = "--header='User-Agent: Mozilla/5.0' --allow-overwrite=true"
     aria2_args = "--optimize-concurrent-downloads --console-log-level=error --summary-interval=10 --stderr=true -c -x16 -s16 -k1M -j5"
 
-    if 'github.com' in url:
-        url = strip_(url)
-
-    # -- CivitAi APi+ V2 --
-    elif 'civitai' in url:
-        url, clean_url, model_type, file_name, image_url, image_name, data = strip_(url, file_name)
+    clean_url = strip_(url)
 
+    if 'civitai' in url:
+        url, clean_url, model_type, file_name, image_url, image_name, data = CivitAi_API(url, file_name)
         if image_url and image_name:
-            with capture.capture_output() as cap:
-                get_ipython().system("aria2c {aria2_args} -d {dst_dir} -o '{image_name}' '{image_url}'")
-            del cap
+            command = ["aria2c"] + aria2_args.split() + ["-d", dst_dir, "-o", image_name, image_url]
+            subprocess.run(command, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
 
-    elif "huggingface.co" in url:
-        clean_url = strip_(url)
-        basename = clean_url.split("/")[-1] if file_name is None else file_name
+    elif 'github' in url or "huggingface.co" in url:
+        basename = url.split("/")[-1] if file_name is None else file_name
 
     """ Formatted info output """
-    model_name_or_basename = file_name if not 'huggingface' in url else basename
+    model_name_or_basename = file_name if file_name else basename
     format_output(clean_url or url, dst_dir, model_name_or_basename)
 
-    # ## -- for my tests --
     # print(url, dst_dir, model_name_or_basename)
-    print(f"\033[31m[Data Info]:\033[0m Failed to retrieve data from the API.\n") if 'civitai' in url and not data else None
-    if 'civitai' in url and data and image_name:
-        print(f"\033[32m[Preview DL]:\033[0m {image_name} - {image_url}\n")
+    if 'civitai' in url:
+        if not data:
+            print("\033[31m[Data Info]:\033[0m Failed to retrieve data from the API.\n")
+        if data and image_name:
+            print(f"\033[32m[Preview DL]:\033[0m {image_name} - {image_url}\n")
+
     # =====================
+    def run_aria2c(url, dst_dir, file_name=None, args="", header=""):
+        out = f"-o '{file_name}'" if file_name else ""
+        get_ipython().system("aria2c {header} {args} -d {dst_dir} {out} '{url}'")
 
-    # -- Git Hub --
-    if 'github.com' in url or 'githubusercontent.com' in url:
-        get_ipython().system("aria2c {aria2_args} -d {dst_dir} -o '{basename}' '{url}'")
-
-    # -- GDrive --
-    elif 'drive.google' in url:
-        try:
-            have_drive_link
-        except:
-            get_ipython().system('pip install -q gdown==5.2.0 > /dev/null')
-            have_drive_link = True
+    # -- Google Drive --
+    if 'drive.google' in url:
+        if not globals().get('have_drive_link', False):
+            os.system("pip install -U gdown > /dev/null")
+            globals()['have_drive_link'] = True
 
         if 'folders' in url:
-            get_ipython().system('gdown --folder "{url}" -O {dst_dir} --fuzzy -c')
+            os.system(f"gdown --folder \"{url}\" -O {dst_dir} --fuzzy -c")
         else:
-            if file_name:
-               get_ipython().system('gdown "{url}" -O {dst_dir}/{file_name} --fuzzy -c')
-            else:
-               get_ipython().system('gdown "{url}" -O {dst_dir} --fuzzy -c')
+            out_path = f"{dst_dir}/{file_name}" if file_name else dst_dir
+            os.system(f"gdown \"{url}\" -O {out_path} --fuzzy -c")
 
-    # -- Hugging Face --
-    elif 'huggingface' in url:
-        get_ipython().system("aria2c {header_option} {aria2_args} -d {dst_dir} -o '{basename}' '{url}'")
+    # -- GitHub or Hugging Face --
+    elif 'github' in url or 'huggingface' in url:
+        run_aria2c(clean_url, dst_dir, basename, aria2_args, header_option if 'huggingface' in url else '')
 
-    # -- Other --
+    # -- Other HTTP/Sources --
     elif 'http' in url:
-        get_ipython().system('aria2c {aria2c_header} {aria2_args} -d {dst_dir} -o "{file_name if file_name else \'\'}" \'{url}\'')
+        run_aria2c(url, dst_dir, file_name, aria2_args, aria2c_header)
 
 ''' SubModels - Added URLs '''
 
@@ -583,7 +452,7 @@ url = handle_submodels(controlnet, controlnet_num, controlnet_list, control_dir,
 
 ''' file.txt - added urls '''
 
-def process_file_download(file_url, prefixes, unique_urls):
+def process_file_download(file_url, PREFIXES, unique_urls):
     files_urls = ""
 
     if file_url.startswith("http"):
@@ -598,8 +467,8 @@ def process_file_download(file_url, prefixes, unique_urls):
     current_tag = None
     for line in lines:
         line = line.strip()
-        if any(f'# {tag}' in line.lower() for tag in prefixes):
-            current_tag = next((tag for tag in prefixes if tag in line.lower()))
+        if any(f'# {tag}' in line.lower() for tag in PREFIXES):
+            current_tag = next((tag for tag in PREFIXES if tag in line.lower()))
 
         urls = [url.split('#')[0].strip() for url in line.split(',')]  # filter urls
         for url in urls:
@@ -623,13 +492,13 @@ if custom_file_urls:
                 custom_file_url = f'{root_path}/{custom_file_url}'
 
         try:
-            file_urls += process_file_download(custom_file_url, prefixes, unique_urls)
+            file_urls += process_file_download(custom_file_url, PREFIXES, unique_urls)
         except FileNotFoundError:
             pass
 
 # url prefixing
 urls = (Model_url, Vae_url, LoRA_url, Embedding_url, Extensions_url)
-prefixed_urls = (f"{prefix}:{url}" for prefix, url in zip(prefixes.keys(), urls) if url for url in url.replace(',', '').split())
+prefixed_urls = (f"{prefix}:{url}" for prefix, url in zip(PREFIXES.keys(), urls) if url for url in url.replace(',', '').split())
 url += ", ".join(prefixed_urls) + ", " + file_urls
 
 if detailed_download == "on":

files_cells/python/ru/launch_ru.py

@@ -9,21 +9,13 @@ import cloudpickle as pickle
 from datetime import timedelta
 from IPython.display import clear_output
 
-#  ================= DETECT ENV =================
-def detect_environment():
-    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
-    environments = {
-        'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
-        'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content")
-    }
-
-    for env_var, (environment, path) in environments.items():
-        if env_var in os.environ:
-            return environment, path, free_plan
-    return 'Unknown', '/unknown/path', free_plan
-
-env, root_path, free_plan = detect_environment()
-webui_path = f"{root_path}/sdw"
+
+# Setup Env
+env = os.environ.get('ENV_NAME')
+root_path = os.environ.get('ROOT_PATH')
+webui_path = os.environ.get('WEBUI_PATH')
+free_plan = os.environ.get('FREE_PLAN')
+
 
 def load_settings():
     SETTINGS_FILE = f'{root_path}/settings.json'
@@ -38,6 +30,7 @@ zrok_token = settings.get('zrok_token', "")
 commandline_arguments = settings.get('commandline_arguments', "")
 change_webui = settings.get('change_webui', "")
 
+
 # ======================== TUNNEL V2 ========================
 print('Please Wait...')
 
@@ -71,6 +64,7 @@ if zrok_token:
 
 clear_output()
 
+
 # =============== Automatic Fixing Path V3 ===============
 paths_to_check = {
     "tagger_hf_cache_dir": f"{webui_path}/models/interrogators/",

files_cells/python/ru/widgets_ru.py

@@ -8,20 +8,11 @@ from ipywidgets import widgets, Layout, Label, Button, VBox, HBox
 from IPython.display import display, HTML, Javascript, clear_output
 
 
-# ================= DETECT ENV =================
-def detect_environment():
-    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
-    environments = {
-        'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
-        'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content")
-    }
-    for env_var, (environment, path) in environments.items():
-        if env_var in os.environ:
-            return environment, path, free_plan
-
-env, root_path, free_plan = detect_environment()
-webui_path = f"{root_path}/sdw"
-get_ipython().system('mkdir -p {root_path}')
+# Setup Env
+env = os.environ.get('ENV_NAME')
+root_path = os.environ.get('ROOT_PATH')
+webui_path = os.environ.get('WEBUI_PATH')
+free_plan = os.environ.get('FREE_PLAN')
 
 
 # ==================== CSS JS ====================

modules/models_data.py

@@ -0,0 +1,103 @@
+model_list = {
+    "1.Anime (by XpucT) + INP": [
+        {"url": "https://huggingface.co/XpucT/Anime/resolve/main/Anime_v2.safetensors", "name": "Anime_V2.safetensors"},
+        {"url": "https://huggingface.co/XpucT/Anime/resolve/main/Anime_v2-inpainting.safetensors", "name": "Anime_V2-inpainting.safetensors"}
+    ],
+    "2.BluMix [Anime] [V7] + INP": [
+        {"url": "https://civitai.com/api/download/models/361779", "name": "BluMix_V7.safetensors"},
+        {"url": "https://civitai.com/api/download/models/363850", "name": "BluMix_V7-inpainting.safetensors"}
+    ],
+    "3.Cetus-Mix [Anime] [V4] + INP": [
+        {"url": "https://civitai.com/api/download/models/130298", "name": "CetusMix_V4.safetensors"},
+        {"url": "https://civitai.com/api/download/models/139882", "name": "CetusMix_V4-inpainting.safetensors"}
+    ],
+    "4.Counterfeit [Anime] [V3] + INP": [
+        {"url": "https://huggingface.co/gsdf/Counterfeit-V3.0/resolve/main/Counterfeit-V3.0_fix_fp16.safetensors", "name": "Counterfeit_V3.safetensors"},
+        {"url": "https://civitai.com/api/download/models/137911", "name": "Counterfeit_V3-inpainting.safetensors"}
+    ],
+    "5.CuteColor [Anime] [V3]": [
+        {"url": "https://civitai.com/api/download/models/138754", "name": "CuteColor_V3.safetensors"}
+    ],
+    "6.Dark-Sushi-Mix [Anime]": [
+        {"url": "https://civitai.com/api/download/models/101640", "name": "DarkSushiMix_2_5D.safetensors"},
+        {"url": "https://civitai.com/api/download/models/56071", "name": "DarkSushiMix_colorful.safetensors"}
+    ],
+    "7.Deliberate [Realism] [V6] + INP": [
+        {"url": "https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v6.safetensors", "name": "Deliberate_V6.safetensors"},
+        {"url": "https://huggingface.co/XpucT/Deliberate/resolve/main/Deliberate_v6-inpainting.safetensors", "name": "Deliberate_V6-inpainting.safetensors"}
+    ],
+    "8.Meina-Mix [Anime] [V11] + INP": [
+        {"url": "https://civitai.com/api/download/models/119057", "name": "MeinaMix_V11.safetensors"},
+        {"url": "https://civitai.com/api/download/models/120702", "name": "MeinaMix_V11-inpainting.safetensors"}
+    ],
+    "9.Mix-Pro [Anime] [V4] + INP": [
+        {"url": "https://civitai.com/api/download/models/125668", "name": "MixPro_V4.safetensors"},
+        {"url": "https://civitai.com/api/download/models/139878", "name": "MixPro_V4-inpainting.safetensors"}
+    ]
+}
+
+vae_list = {
+    "1.Anime.vae": [{"url": "https://civitai.com/api/download/models/311162", "name": "Anime.vae.safetensors"}],
+    "2.Anything.vae": [{"url": "https://huggingface.co/NoCrypt/resources/resolve/main/VAE/any.vae.safetensors", "name": "Anything.vae.safetensors"}],
+    "3.Blessed2.vae": [{"url": "https://huggingface.co/NoCrypt/resources/resolve/main/VAE/blessed2.vae.safetensors", "name": "Blessed2.vae.safetensors"}],
+    "4.ClearVae.vae": [{"url": "https://civitai.com/api/download/models/88156", "name": "ClearVae_23.vae.safetensors"}],
+    "5.WD.vae": [{"url": "https://huggingface.co/NoCrypt/resources/resolve/main/VAE/wd.vae.safetensors", "name": "WD.vae.safetensors"}]
+}
+
+controlnet_list = {
+    "1.canny": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_canny_fp16.safetensors", "name": "control_v11p_sd15_canny_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_canny_fp16.yaml", "name": "control_v11p_sd15_canny_fp16.yaml"}
+    ],
+    "2.openpose": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose_fp16.safetensors", "name": "control_v11p_sd15_openpose_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_openpose_fp16.yaml", "name": "control_v11p_sd15_openpose_fp16.yaml"}
+    ],
+    "3.depth": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors", "name": "control_v11f1p_sd15_depth_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11f1p_sd15_depth_fp16.yaml", "name": "control_v11f1p_sd15_depth_fp16.yaml"},
+        {"url": "https://huggingface.co/NagisaNao/models/resolve/main/ControlNet_v11/control_v11p_sd15_depth_anything_fp16.safetensors", "name": "control_v11p_sd15_depth_anything_fp16.safetensors"}
+    ],
+    "4.normal_map": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors", "name": "control_v11p_sd15_normalbae_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_normalbae_fp16.yaml", "name": "control_v11p_sd15_normalbae_fp16.yaml"}
+    ],
+    "5.mlsd": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors", "name": "control_v11p_sd15_mlsd_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_mlsd_fp16.yaml", "name": "control_v11p_sd15_mlsd_fp16.yaml"}
+    ],
+    "6.lineart": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_lineart_fp16.safetensors", "name": "control_v11p_sd15_lineart_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors", "name": "control_v11p_sd15s2_lineart_anime_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_lineart_fp16.yaml", "name": "control_v11p_sd15_lineart_fp16.yaml"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15s2_lineart_anime_fp16.yaml", "name": "control_v11p_sd15s2_lineart_anime_fp16.yaml"}
+    ],
+    "7.soft_edge": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_softedge_fp16.safetensors", "name": "control_v11p_sd15_softedge_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_softedge_fp16.yaml", "name": "control_v11p_sd15_softedge_fp16.yaml"}
+    ],
+    "8.scribble": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_scribble_fp16.safetensors", "name": "control_v11p_sd15_scribble_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_scribble_fp16.yaml", "name": "control_v11p_sd15_scribble_fp16.yaml"}
+    ],
+    "9.segmentation": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_seg_fp16.safetensors", "name": "control_v11p_sd15_seg_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_seg_fp16.yaml", "name": "control_v11p_sd15_seg_fp16.yaml"}
+    ],
+    "10.shuffle": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_shuffle_fp16.safetensors", "name": "control_v11e_sd15_shuffle_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_shuffle_fp16.yaml", "name": "control_v11e_sd15_shuffle_fp16.yaml"}
+    ],
+    "11.tile": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11f1e_sd15_tile_fp16.safetensors", "name": "control_v11f1e_sd15_tile_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11f1e_sd15_tile_fp16.yaml", "name": "control_v11f1e_sd15_tile_fp16.yaml"}
+    ],
+    "12.inpaint": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors", "name": "control_v11p_sd15_inpaint_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11p_sd15_inpaint_fp16.yaml", "name": "control_v11p_sd15_inpaint_fp16.yaml"}
+    ],
+    "13.instruct_p2p": [
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/resolve/main/control_v11e_sd15_ip2p_fp16.safetensors", "name": "control_v11e_sd15_ip2p_fp16.safetensors"},
+        {"url": "https://huggingface.co/ckpt/ControlNet-v1-1/raw/main/control_v11e_sd15_ip2p_fp16.yaml", "name": "control_v11e_sd15_ip2p_fp16.yaml"}
+    ]
+}

modules/setup_en.py

@@ -0,0 +1,88 @@
+import os
+import sys
+import shutil
+import argparse
+import importlib
+import subprocess
+
+def parse_args():
+    parser = argparse.ArgumentParser(description='Script configuration')
+    parser.add_argument('--lang', type=str, default='en', help='Код языка, по умолчанию "en"')
+    parser.add_argument('--repo', type=str, required=True, help='Repository Name')
+    return parser.parse_args()
+
+def detect_environment():
+    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
+    environments = {
+        'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
+        'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content")
+    }
+    for env_var, (environment, path) in environments.items():
+        if env_var in os.environ:
+            return environment, path, free_plan
+    print("\033[31mError: an unsupported runtime environment was detected.\n\033[34mSupported environments:\033[0m Google Colab, Kaggle")
+    return None, None, None
+
+def setup_module_folder(root_path):
+    modules_folder = os.path.join(root_path, "modules")
+    os.makedirs(modules_folder, exist_ok=True)
+    if modules_folder not in sys.path:
+        sys.path.append(modules_folder)
+
+def clear_module_cache(modules_folder):
+    for module_name in list(sys.modules.keys()):
+        module = sys.modules[module_name]
+        if hasattr(module, '__file__') and module.__file__ and module.__file__.startswith(modules_folder):
+            del sys.modules[module_name]
+    
+    importlib.invalidate_caches()
+
+def download_files(root_path, lang, repo):
+    print("Please wait for the files to download... 👀", end='', flush=True)
+    files_dict = {
+        'CSS': {'CSS': ['main_widgets.css', 'auto_cleaner.css']},
+        'file_cell': {f'files_cells/python/{lang}': [f'widgets_{lang}.py', f'downloading_{lang}.py', f'launch_{lang}.py', f'auto_cleaner_{lang}.py']},
+        'file_cell/special': {f'special': ['dl_display_results.py']},
+        'modules': {f'modules': ['models_data.py']}
+    }
+    for folder, contents in files_dict.items():
+        folder_path = os.path.join(root_path, folder)
+        if os.path.exists(folder_path):
+            shutil.rmtree(folder_path)
+        os.makedirs(folder_path)
+        for path_url, files in contents.items():
+            for file in files:
+                file_url = f"https://huggingface.co/NagisaNao/{repo}/resolve/main/{path_url}/{file}"
+                file_path = os.path.join(folder_path, file)
+                os.system(f'wget -q {file_url} -O {file_path}')
+    print("\rDone! Now you can run the cells below. ☄️" + " "*30)
+
+def main():
+    args = parse_args()
+    lang = args.lang
+    repo = args.repo
+
+    env, root_path, free_plan = detect_environment()
+
+    if env and root_path:
+        webui_path = f"{root_path}/sdw"
+        download_files(root_path, lang, repo)
+        clear_module_cache(os.path.join(root_path, "modules"))
+        setup_module_folder(root_path)
+        
+        # Set global environment variable
+        os.environ['ENV_NAME'] = env
+        os.environ['ROOT_PATH'] = root_path
+        os.environ['WEBUI_PATH'] = webui_path
+        os.environ['FREE_PLAN'] = 'True' if free_plan else 'False'
+
+        print(f"Runtime environment: \033[33m{env}\033[0m")
+        if env == "Google Colab":
+            print(f"Colab Pro subscription: \033[34m{not free_plan}\033[0m")
+            print(f"File location: \033[32m{root_path}\033[0m")
+            
+        if repo != 'fast_repo':
+            print('\n\033[31mWARNING: Test mode is used, there may be errors in use!\033[0m')
+
+if __name__ == "__main__":
+    main()

modules/setup_ru.py

@@ -0,0 +1,88 @@
+import os
+import sys
+import shutil
+import argparse
+import importlib
+import subprocess
+
+def parse_args():
+    parser = argparse.ArgumentParser(description='Script configuration')
+    parser.add_argument('--lang', type=str, default='en', help='Код языка, по умолчанию "en"')
+    parser.add_argument('--repo', type=str, required=True, help='Название репозитория')
+    return parser.parse_args()
+
+def detect_environment():
+    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
+    environments = {
+        'COLAB_GPU': ('Google Colab', "/root" if free_plan else "/content"),
+        'KAGGLE_URL_BASE': ('Kaggle', "/kaggle/working/content")
+    }
+    for env_var, (environment, path) in environments.items():
+        if env_var in os.environ:
+            return environment, path, free_plan
+    print("\033[31mОшибка: обнаружена неизвестная среда выполнения.\n\033[34mПоддерживаемые среды:\033[0m Google Colab, Kaggle.")
+    return None, None, None
+
+def setup_module_folder(root_path):
+    modules_folder = os.path.join(root_path, "modules")
+    os.makedirs(modules_folder, exist_ok=True)
+    if modules_folder not in sys.path:
+        sys.path.append(modules_folder)
+
+def clear_module_cache(modules_folder):
+    for module_name in list(sys.modules.keys()):
+        module = sys.modules[module_name]
+        if hasattr(module, '__file__') and module.__file__ and module.__file__.startswith(modules_folder):
+            del sys.modules[module_name]
+    
+    importlib.invalidate_caches()
+
+def download_files(root_path, lang, repo):
+    print("Пожалуйста, дождитесь загрузки файлов... 👀", end='', flush=True)
+    files_dict = {
+        'CSS': {'CSS': ['main_widgets.css', 'auto_cleaner.css']},
+        'file_cell': {f'files_cells/python/{lang}': [f'widgets_{lang}.py', f'downloading_{lang}.py', f'launch_{lang}.py', f'auto_cleaner_{lang}.py']},
+        'file_cell/special': {f'special': ['dl_display_results.py']},
+        'modules': {f'modules': ['models_data.py']}
+    }
+    for folder, contents in files_dict.items():
+        folder_path = os.path.join(root_path, folder)
+        if os.path.exists(folder_path):
+            shutil.rmtree(folder_path)
+        os.makedirs(folder_path)
+        for path_url, files in contents.items():
+            for file in files:
+                file_url = f"https://huggingface.co/NagisaNao/{repo}/resolve/main/{path_url}/{file}"
+                file_path = os.path.join(folder_path, file)
+                os.system(f'wget -q {file_url} -O {file_path}')
+    print(f"\rГотово! Теперь вы можете запустить ячейки ниже. ☄️" + " "*30)
+
+def main():
+    args = parse_args()
+    lang = args.lang
+    repo = args.repo
+
+    env, root_path, free_plan = detect_environment()
+
+    if env and root_path:
+        webui_path = f"{root_path}/sdw"
+        download_files(root_path, lang, repo)
+        clear_module_cache(os.path.join(root_path, "modules"))
+        setup_module_folder(root_path)
+        
+        # Set global environment variable
+        os.environ['ENV_NAME'] = env
+        os.environ['ROOT_PATH'] = root_path
+        os.environ['WEBUI_PATH'] = webui_path
+        os.environ['FREE_PLAN'] = 'True' if free_plan else 'False'
+
+        print(f"Среда выполнения: \033[33m{env}\033[0m")
+        if env == "Google Colab":
+            print(f"Подписка Colab Pro: \033[34m{not free_plan}\033[0m")
+            print(f"Расположение файлов: \033[32m{root_path}\033[0m")
+            
+        if repo != 'fast_repo':
+            print('\n\033[31mВНИМАНИЕ: Используется тестовый режим, возможны ошибки при использовании!\033[0m')
+
+if __name__ == "__main__":
+    main()