🐛

CSS/auto_cleaner.css

@@ -0,0 +1,187 @@
+/* General Styles */
+
+hr {
+    border-color: grey;
+    background-color: grey;
+    opacity: 0.25;
+}
+
+.instruction_AC {
+    font-family: cursive;
+    font-size: 18px;
+    color: grey;
+    user-select: none;
+    cursor: default;
+}
+
+
+/* Container style */
+
+.container_AC {
+    position: relative;
+    background-color: #232323;
+    width: 800px;
+    height: auto;
+    padding: 15px;
+    border-radius: 15px;
+    box-shadow: 0 0 50px rgba(0, 0, 0, 0.3);
+    margin-bottom: 5px;
+    overflow: visible;
+}
+
+.container_AC::before {
+    position: absolute;
+    top: 5px;
+    right: 10px;
+    content: "AutoCleanerV3.7";
+    font-weight: bold;
+    font-size: 24px;
+    color: rgba(0, 0, 0, 0.2);
+}
+
+.container_AC::after {
+    position: absolute;
+    top: 30px;
+    right: 10px;
+    content: "ANXETY";
+    font-weight: bold;
+    font-size: 18px;
+    color: rgba(0, 0, 0, 0.2);
+}
+
+.custom-select-multiple_AC select {
+    padding: 10px;
+    font-family: cursive;
+    border: 1px solid #262626 !important;
+    border-radius: 10px;
+    color: white;
+    background-color: #1c1c1c;
+    box-shadow: inset 0 0 5px rgba(0, 0, 0, 0.5);
+}
+
+.output_AC {
+    padding: 10px;
+    height: auto;
+    border: 1px solid #262626;
+    border-radius: 10px;
+    background-color: #1c1c1c;
+    box-shadow: inset 0 0 5px rgba(0, 0, 0, 0.5);
+    width: auto;
+    box-sizing: border-box;
+}
+
+.output_message_AC {
+    font-family: cursive;
+    color: white !important;
+    font-size: 14px;
+    user-select: none;
+    cursor: default
+}
+
+
+.storage_info_AC {
+    padding: 5px 20px;
+    height: auto;
+    border: 1px solid #262626;
+    border-radius: 10px;
+    background-color: #1c1c1c;
+    box-shadow: inset 0 0 5px rgba(0, 0, 0, 0.5);
+    width: auto;
+    font-family: cursive;
+    color: #B2B2B2 !important;
+    font-size: 14px;
+    user-select: none;
+    cursor: default
+}
+
+
+/* Button and storage info layout */
+.lower_information_panel_AC {
+    display: flex;
+    align-items: center;
+    justify-content: space-between;
+}
+
+
+/* Button style */
+
+.button_AC {
+    width: auto;
+    font-family: cursive;
+    color: white !important;
+    font-size: 14px;
+    font-weight: bold;
+    height: 35px;
+    border-radius: 15px;
+    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);
+    background-size: 200% 200%;
+    background-position: left bottom;
+    transition: background 0.5s ease-in-out, transform 0.3s ease;
+}
+
+.button_AC:hover {
+    cursor: pointer;
+    background-size: 200% 200%;
+    background-position: right bottom;
+    transform: translateY(1px);
+}
+
+.button_execute_AC:hover {
+    background-image: radial-gradient(circle at top left, purple 10%, #93ac47 90%);
+}
+
+.button_clear_AC:hover {
+    background-image: radial-gradient(circle at top left, purple 10%, #fc3468 90%);
+}
+
+.button_execute_AC:active,
+.button_clear_AC:active {
+    filter: brightness(0.75);
+}
+
+.jupyter-widgets.lm-Widget:focus {
+    outline: none;
+}
+
+
+/* Animation of elements */
+
+/* Emergence */
+.container_AC {
+    animation-name: slideInTopBlur;
+    animation-duration: 0.7s;
+    animation-fill-mode: forwards;
+}
+
+@keyframes slideInTopBlur {
+    0% {
+        transform: translate3d(0, 50%, 0) scale(0.85) rotate3d(1, 0, 0, -85deg);
+        filter: blur(5px) grayscale(1) brightness(0.5);
+        opacity: 0;
+    }
+    100% {
+        transform: translate3d(0, 0, 0) scale(1) rotate3d(1, 0, 0, 0deg);
+        filter: blur(0) grayscale(0) brightness(1);
+        opacity: 1;
+    }
+}
+
+/* Leaving */
+.container_AC.hide {
+    animation-name: slideOutTopBlur;
+    animation-duration: 0.3s;
+    animation-fill-mode: forwards;
+}
+
+@keyframes slideOutTopBlur {
+    0% {
+        transform: translate3d(0, 0, 0) scale(1);
+        filter: blur(0) grayscale(0) brightness(1);
+        opacity: 1;
+    }
+    100% {
+        transform: translate3d(0, -100%, 0);
+        filter: blur(5px) grayscale(1) brightness(0);
+        opacity: 0;
+    }
+}

CSS/main_widgets.css

@@ -0,0 +1,333 @@
+/* General Styles */
+.header {
+    font-family: cursive;
+    font-size: 20px;
+    font-weight: bold;
+    color: #ff8cee;
+    margin-bottom: 15px;
+    user-select: none;
+    cursor: default;
+    display: inline-block;
+}
+
+hr {
+    border-color: grey;
+    background-color: grey;
+    opacity: 0.25;
+}
+
+a {
+  text-decoration: none;
+  color: inherit;
+}
+
+
+/* Container style */
+
+.container {
+    position: relative;
+    background-color: #232323;
+    width: 1080px;
+    padding: 10px 15px;
+    border-radius: 15px;
+    box-shadow: 0 0 50px rgba(0, 0, 0, 0.3);
+    margin-bottom: 5px;
+    overflow: hidden;
+}
+
+.container::after {
+    position: absolute;
+    top: 5px;
+    right: 10px;
+    content: "ANXETY";
+    font-weight: bold;
+    font-size: 24px;
+    color: rgba(0, 0, 0, 0.2);
+}
+
+/* background img */
+.container::before {
+  content: "";
+  position: absolute;
+  top: 0;
+  left: 0;
+  right: 0;
+  bottom: 0;
+  background-image: var(--img_background);
+  background-size: var(--img_scale);
+  background-repeat: var(--img_mode);
+  opacity: var(--img_opacity);
+  mix-blend-mode: screen;
+  pointer-events: none;
+  filter: blur(var(--img_blur));
+  z-index: -1;
+}
+
+.image_1::before {
+  background-position: var(--image_x) calc(-120px - var(--image_y));
+}
+.image_2::before {
+  background-position: var(--image_x) calc(-290px - var(--image_y));
+}
+.image_3::before {
+  background-position: var(--image_x) calc(-430px - var(--image_y));
+}
+.image_4::before {
+  background-position: var(--image_x) calc(var(--img_height_dif) - var(--image_y));
+}
+
+.container_custom_downlad {
+    height: 55px;
+    transition: all 0.5s;
+}
+
+.container_custom_downlad.expanded {
+    height: 270px;
+}
+
+
+/* Element text style */
+
+.widget-html,
+.widget-button,
+.widget-text label,
+.widget-checkbox label,
+.widget-dropdown label,
+.widget-dropdown select,
+.widget-text input[type="text"] {
+    font-family: cursive;
+    font-size: 14px;
+    color: white !important;
+    user-select: none;
+}
+
+.widget-text input[type="text"]::placeholder {
+    color: grey;
+}
+
+
+/* Input field styles */
+
+.widget-dropdown select,
+.widget-text input[type="text"] {
+    height: 30px;
+    background-color: var(--bg-field-color);
+    border: 1px solid #262626;
+    border-radius: 10px;
+    box-shadow: inset 0 0 5px rgba(0, 0, 0, 0.5);
+    transition: all 0.3s ease-in-out;
+    backdrop-filter: blur(var(--bg-field-blur-level));
+}
+
+.widget-dropdown select:focus,
+.widget-text input[type="text"]:focus {
+    border-color: #006ee5;
+}
+
+.widget-dropdown select:hover,
+.widget-text input[type="text"]:hover {
+    transform: scale(1.003);
+    background-color: var(--bg-field-color-hover);
+    box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.5);
+}
+
+.widget-dropdown option {
+    background-color: #1c1c1c;
+}
+
+
+/* Slider Checkbox style */
+
+.widget-checkbox input[type="checkbox"] {
+    appearance: none;
+    position: relative;
+    top: 4px; /* Why is he taller?! */
+    width: 40px;
+    height: 20px;
+    border: none;
+    border-radius: 10px;
+    background-color: #20b2aa;
+    cursor: pointer;
+    box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.3);
+    transition: background-color 0.3s ease;
+}
+.widget-checkbox input[type="checkbox"]:checked {
+    background-color: #2196F3;
+}
+
+.widget-checkbox input[type="checkbox"]:before {
+    content: '';
+    position: absolute;
+    top: 50%;
+    left: 3px;
+    width: 16px;
+    height: 16px;
+    border-radius: inherit;
+    background-color: white;
+    box-shadow: 0 0 8px rgba(0, 0, 0, 0.3);
+    transform: translateY(-50%);
+    transition: left 0.3s ease;
+}
+.widget-checkbox input[type="checkbox"]:checked:before {
+    left: 21px;
+}
+
+
+/* Button styles */
+
+.button_save {
+    font-size: 15px;
+    font-weight: bold;
+    width: 120px;
+    height: 35px;
+    border-radius: 15px;
+    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);
+    background-size: 200% 200%;
+    background-position: left bottom;
+    transition: background 0.5s ease-in-out, transform 0.3s ease;
+}
+
+.button_save:hover {
+    cursor: pointer;
+    background-image: radial-gradient(circle at top left, purple 10%, #93ac47 90%);
+    background-size: 200% 200%;
+    background-position: right bottom;
+    transform: translateY(1px);
+}
+
+.button_ngrok {
+    font-size: 12px;
+    height: 30px;
+    border-radius: 10px;
+    padding: 1px 12px;
+    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);
+    background-size: 200% 200%;
+    background-position: left bottom;
+    transition: background 0.5s ease-in-out, transform 0.3s ease;
+    white-space: nowrap;
+}
+
+.button_ngrok:hover  {
+    cursor: pointer;
+    background-image: radial-gradient(circle at top left, purple 10%, #1D94BB 90%);
+    background-size: 200% 200%;
+    background-position: right bottom;
+    transform: translateY(1px);
+}
+
+.button_save:active,
+.button_ngrok:active {
+    filter: brightness(0.75) !important;
+}
+
+/* Removes ugly stroke from widget buttons. */
+.jupyter-widgets.lm-Widget:focus {
+    outline: none;
+}
+
+
+/* Popup style of `INFO` window */
+
+.info {
+    position: absolute;
+    top: -5px;
+    right: 95px;
+    color: grey;
+    font-family: cursive;
+    font-size: 14px;
+    font-weight: normal;
+    user-select: none;
+    pointer-events: none;
+    opacity: 0;
+    transition: opacity 0.3s ease-in-out;
+    display: inline-block;
+}
+
+.popup {
+    position: absolute;
+    top: 120px;
+    z-index: 999;
+    width: auto;
+    padding: 10px;
+    text-align: center;
+    background-color: rgba(255, 255, 255, 0.05);
+    backdrop-filter: blur(20px);
+    border: 1px solid rgba(255, 255, 255, 0.45);
+    border-radius: 8px;
+    box-shadow: 0 0 50px rgba(0, 0, 0, 0.5);
+    opacity: 0;
+    color: #fff;
+    font-size: 16px;
+    font-family: cursive;
+    user-select: none;
+    cursor: default;
+    pointer-events: none;
+    transform: rotate(-5deg);
+    transition: top 0.3s ease-in-out, opacity 0.3s ease-in-out, transform 0.3s ease-in-out;
+}
+
+.sample {
+    display: inline-block;
+    margin-top: 25px;
+    padding: 10px 100px;
+    background-color: rgba(255, 255, 255, 0.2);
+    color: #c6e2ff;
+    border: 2px solid rgba(255, 255, 255, 0.2);
+    border-radius: 8px;
+    box-shadow: 0 0 10px rgba(255, 255, 255, 0.2), inset 0 0 25px rgba(255, 255, 255, 0.2);
+}
+
+.info.showed {
+    opacity: 1;
+    pointer-events: auto;
+}
+
+.info:hover + .popup {
+    top: 35px;
+    opacity: 1;
+    pointer-events: initial;
+    transform: rotate(0deg);
+}
+
+
+/* Animation of elements */
+
+.container,
+.button_save {
+    animation-name: showedWidgets;
+    animation-duration: 1s;
+    animation-fill-mode: forwards;
+}
+
+.container.hide,
+.button_save.hide {
+    animation-name: hideWidgets;
+    animation-duration: 0.5s;
+    animation-fill-mode: forwards;
+}
+
+@keyframes showedWidgets {
+    0% {
+        transform: translate3d(-65%, 15%, 0) scale(0) rotate(15deg);
+        filter: blur(25px) grayscale(1) brightness(0.3);
+        opacity: 0;
+    }
+    100% {
+        transform: translate3d(0, 0, 0) scale(1) rotate(0deg);
+        filter: blur(0) grayscale(0) brightness(1);
+        opacity: 1;
+    }
+}
+
+@keyframes hideWidgets {
+    0% {
+        transform: translate3d(0, 0, 0) scale(1) rotate3d(1, 0, 0, 0deg);
+        filter: blur(0) grayscale(0) brightness(1);
+        opacity: 1;
+    }
+    100% {
+        transform: translate3d(0, 5%, 0) scale(0.9) rotate3d(1, 0, 0, 90deg);
+        filter: blur(15px) grayscale(1) brightness(0.5);
+        opacity: 0;
+    }
+}

files_cells/notebooks/en/auto_cleaner_en.ipynb

@@ -0,0 +1,165 @@
+{
+  "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": []
+    }
+  ]
+}

files_cells/notebooks/en/downloading_en.ipynb

@@ -25,21 +25,19 @@
         "from urllib.parse import urlparse, parse_qs\n",
         "\n",
         "\n",
-        "#  ================= DETECT ENV =================\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",
         "\n",
         "env, root_path, free_plan = detect_environment()\n",
         "webui_path = f\"{root_path}/sdw\"\n",
-        "#  ----------------------------------------------\n",
         "\n",
         "\n",
         "# ================ LIBRARIES V2 ================\n",
@@ -90,12 +88,6 @@
         "    time.sleep(2)\n",
         "    clear_output()\n",
         "\n",
-        "## dl special files\n",
-        "with capture.capture_output() as cap:\n",
-        "    !mkdir -p {root_path}/file_cell/special\n",
-        "    !wget https://huggingface.co/NagisaNao/fast_repo/resolve/main/special/dl_display_results.py -O {root_path}/file_cell/special/dl_display_results.py\n",
-        "del cap\n",
-        "\n",
         "\n",
         "# ================= loading settings V4 =================\n",
         "def load_settings(path):\n",
@@ -106,16 +98,17 @@
         "\n",
         "settings = load_settings(f'{root_path}/settings.json')\n",
         "\n",
-        "variables = [\n",
-        "    'Model', 'Model_Num', 'Inpainting_Model',\n",
-        "    'Vae', 'Vae_Num',\n",
-        "    'latest_webui', 'latest_exstensions', 'change_webui', 'detailed_download',\n",
-        "    'controlnet', 'controlnet_Num', 'commit_hash', 'optional_huggingface_token',\n",
+        "VARIABLES = [\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', 'Extensions_url', 'custom_file_urls'\n",
+        "    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url',\n",
+        "    'Extensions_url', 'custom_file_urls'\n",
         "]\n",
         "\n",
-        "locals().update({key: settings.get(key) for key in variables})\n",
+        "locals().update({key: settings.get(key) for key in VARIABLES})\n",
         "\n",
         "\n",
         "# ================= OTHER =================\n",
@@ -251,17 +244,7 @@
         "    ]\n",
         "}\n",
         "\n",
-        "# 1-4 (fp16/cleaned)\n",
         "vae_list = {\n",
-        "    ## vae broke - the author's an asshole\n",
-        "    # \"1.Anime.vae\": [\n",
-        "    #     {\"url\": \"https://civitai.com/api/download/models/131654\", \"name\": \"Anime.vae.safetensors\"},\n",
-        "    #     {\"url\": \"https://civitai.com/api/download/models/131658\", \"name\": \"vae-ft-mse.vae.safetensors\"}\n",
-        "    # ],\n",
-        "    # \"2.Anything.vae\": [{\"url\": \"https://civitai.com/api/download/models/131656\", \"name\": \"Anything.vae.safetensors\"}],\n",
-        "    # \"3.Blessed2.vae\": [{\"url\": \"https://civitai.com/api/download/models/142467\", \"name\": \"Blessed2.vae.safetensors\"}],\n",
-        "    # \"4.ClearVae.vae\": [{\"url\": \"https://civitai.com/api/download/models/133362\", \"name\": \"ClearVae_23.vae.safetensors\"}],\n",
-        "\n",
         "    \"1.Anime.vae\": [{\"url\": \"https://civitai.com/api/download/models/311162\", \"name\": \"vae-ft-mse-840000-ema-pruned.vae.safetensors\"}],\n",
         "    \"2.Anything.vae\": [{\"url\": \"https://civitai.com/api/download/models/119279\", \"name\": \"Anything.vae.safetensors\"}],\n",
         "    \"3.Blessed2.vae\": [{\"url\": \"https://huggingface.co/NoCrypt/blessed_vae/resolve/main/blessed2.vae.pt\", \"name\": \"Blessed2.vae.safetensors\"}],\n",
@@ -339,26 +322,47 @@
         "}\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 = optional_huggingface_token if optional_huggingface_token else \"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\"\n",
+        "hf_token = huggingface_token if huggingface_token else \"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\"\n",
         "user_header = f\"\\\"Authorization: Bearer {hf_token}\\\"\"\n",
         "\n",
+        "''' Formatted Info Output '''\n",
+        "\n",
+        "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",
+        "\n",
+        "def format_output(url, dst_dir, file_name):\n",
+        "    info = f\"[{file_name.split('.')[0]}]\"\n",
+        "    info = center_text(info)\n",
+        "\n",
+        "    print(f\"\\n\\033[32m{'---'*20}]{info}[{'---'*20}\")\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 Image Preview | CivitAi '''\n",
         "\n",
-        "def get_data_from_api(model_id): # get model data\n",
+        "def get_data_from_api(model_id):\n",
+        "    \"\"\"Fetch model data from the API\"\"\"\n",
         "    endpoint_url = f\"https://civitai.com/api/v1/model-versions/{model_id}\"\n",
         "    headers = {\"Content-Type\": \"application/json\"}\n",
         "    try:\n",
         "        response = requests.get(endpoint_url, headers=headers)\n",
-        "        if response.status_code == 200:\n",
-        "            return response.json()\n",
+        "        response.raise_for_status()\n",
+        "        return response.json()\n",
         "    except requests.exceptions.RequestException as e:\n",
         "        print(f\"An error occurred: {e}\")\n",
         "        return None\n",
         "\n",
         "def extract_model_info(data, url):\n",
+        "    \"\"\"Extract model information based on URL\"\"\"\n",
         "    if 'type=' in url:\n",
         "        model_type = parse_qs(urlparse(url).query).get('type', [''])[0]\n",
         "        model_name = data['files'][1]['name']\n",
@@ -375,59 +379,69 @@
         "    return model_type, model_name, image_url\n",
         "\n",
         "def gen_preview_filename(model_name, image_url):\n",
+        "    \"\"\"Generate a preview filename\"\"\"\n",
         "    name = model_name.split('.')\n",
         "    img_exts = image_url.split('.')\n",
-        "    return f\"{name[0]}.preview.{img_exts[-1]}\" # assigning the original image format\n",
+        "    return f\"{name[0]}.preview.{img_exts[-1]}\"\n",
         "\n",
         "''' main download code '''\n",
         "\n",
         "def handle_manual(url):\n",
-        "    original_url = url\n",
-        "    url = url.split(':', 1)[1]\n",
-        "    file_name = re.search(r'\\[(.*?)\\]', url)\n",
-        "    file_name = file_name.group(1) if file_name else None\n",
-        "    if file_name:\n",
-        "        url = re.sub(r'\\[.*?\\]', '', url)\n",
+        "    url_parts = url.split(':', 1)\n",
+        "    prefix = url_parts[0]\n",
+        "    path = url_parts[1]\n",
         "\n",
-        "    for prefix, dir in prefixes.items():\n",
-        "        if original_url.startswith(f\"{prefix}:\"):\n",
-        "            if prefix != \"extension\":\n",
-        "                manual_download(url, dir, file_name=file_name)\n",
-        "            else:\n",
-        "                extension_repo.append((url, file_name))\n",
+        "    file_name_match = re.search(r'\\[(.*?)\\]', path)\n",
+        "    file_name = file_name_match.group(1) if file_name_match else None\n",
+        "    if file_name:\n",
+        "        path = re.sub(r'\\[.*?\\]', '', path)\n",
+        "\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",
+        "            except Exception as e:\n",
+        "                print(f\"Error downloading file: {e}\")\n",
+        "        else:\n",
+        "            extension_repo.append((path, file_name))\n",
         "\n",
         "def manual_download(url, dst_dir, file_name):\n",
         "    aria2_args = '--optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -j5 -x16 -s16 -k1M -c'\n",
         "    basename = url.split(\"/\")[-1] if file_name is None else file_name\n",
         "    header_option = f\"--header={user_header}\"\n",
         "\n",
-        "    # === CivitAi API ===\n",
+        "    # ==== CivitAi API+ ====\n",
         "    support_types = ('Checkpoint', 'Model', 'TextualInversion', 'LORA') # for dl preview image\n",
         "    civitai_token = \"62c0c5956b2f9defbd844d754000180b\"\n",
         "\n",
         "    if 'civitai' in url:\n",
-        "        url = f\"{url}{'&' if '?' in url else '?'}token={civitai_token}\" # some authors are pussies and they need API token...\n",
-        "        model_id = url.split('/')[-1]\n",
-        "        clean_url = url.split('?')[0]\n",
+        "        url = f\"{url}{'&' if '?' in url else '?'}token={civitai_token}\"\n",
+        "        model_id = url.split('/')[-1].split('?')[0]\n",
+        "        clean_url = re.sub(r'[?&]token=[^&]*', '', url) # hide token\n",
         "\n",
         "        data = get_data_from_api(model_id)\n",
         "        if data:\n",
         "            model_type, model_name, image_url = extract_model_info(data, url)\n",
-        "            if model_name and image_url:\n",
-        "                image_file_name = gen_preview_filename(model_name if not file_name else file_name, image_url)\n",
-        "                if any(types in model_type for types in support_types):\n",
-        "                    with capture.capture_output() as cap: # clear shit\n",
-        "                        !aria2c {aria2_args} -d {dst_dir} -o {image_file_name} {image_url}\n",
+        "\n",
+        "            if any(t in model_type for t in support_types):\n",
+        "                if model_name and image_url:\n",
+        "                    image_file_name = gen_preview_filename(model_name if not file_name else file_name, image_url)\n",
+        "                    with capture.capture_output() as cap:\n",
+        "                        !aria2c {aria2_args} -d {dst_dir} -o {image_file_name} '{image_url}'\n",
         "                    del cap\n",
-        "            file_name = file_name or model_name # assigns the original file name if not specified initially\n",
+        "            file_name = file_name or model_name\n",
+        "    else:\n",
+        "        clean_url = url\n",
+        "\n",
+        "    \"\"\" Formatted info output \"\"\"\n",
+        "    model_name_or_basename = file_name if not 'huggingface' in url else basename\n",
+        "    format_output(clean_url or url, dst_dir, model_name_or_basename)\n",
         "\n",
-        "    \"\"\" information output \"\"\"\n",
-        "    # -- wrold's best print info --\n",
-        "    print(f\"\\n\\033[32m{'---'*45}\\n\\033[33mURL: \\033[34m{clean_url if 'civitai' in url else url}\\n\\033[33mSAVE DIR: \\033[34m{dst_dir}\\n\\033[33mFILE NAME: \\033[34m{file_name if not 'huggingface' in url else basename}\\033[0m\")\n",
         "    print(\"\\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 any(types in model_type for types in support_types) and (locals().get('image_file_name') or ''):\n",
+        "    if 'civitai' in url and data and any(t in model_type for t in support_types) and (locals().get('image_file_name') or ''):\n",
         "        print(f\"\\033[32m[Preview DL]:\\033[0m {image_file_name} - {image_url}\\n\")\n",
-        "    # ===================\n",
+        "    # =====================\n",
         "\n",
         "    # -- GDrive --\n",
         "    if 'drive.google' in url:\n",
@@ -471,9 +485,8 @@
         "\n",
         "    unpucking_zip_files()\n",
         "\n",
-        "## unpucking zip files\n",
+        "# unpucking zip files\n",
         "def unpucking_zip_files():\n",
-        "    # directories - above\n",
         "    for directory in directories:\n",
         "        for root, dirs, files in os.walk(directory):\n",
         "            for file in files:\n",
@@ -486,8 +499,6 @@
         "\n",
         "''' submodels - added urls '''\n",
         "\n",
-        "submodels = []\n",
-        "\n",
         "def add_submodels(selection, num_selection, model_dict, dst_dir):\n",
         "    if selection == \"none\":\n",
         "        return []\n",
@@ -499,33 +510,32 @@
         "    else:\n",
         "        selected_models = model_dict[selection]\n",
         "        selected_nums = map(int, num_selection.replace(',', '').split())\n",
-        "\n",
         "        for num in selected_nums:\n",
         "            if 1 <= num <= len(model_dict):\n",
         "                name = list(model_dict)[num - 1]\n",
         "                selected_models.extend(model_dict[name])\n",
         "\n",
         "    unique_models = list({model['name']: model for model in selected_models}.values())\n",
-        "\n",
         "    for model in unique_models:\n",
         "        model['dst_dir'] = dst_dir\n",
         "\n",
         "    return unique_models\n",
         "\n",
-        "submodels += add_submodels(Model, Model_Num, model_list, models_dir)                    # model\n",
-        "submodels += add_submodels(Vae, Vae_Num, vae_list, vaes_dir)                            # vae\n",
-        "submodels += add_submodels(controlnet, controlnet_Num, controlnet_list, control_dir)    # controlnet\n",
+        "def handle_submodels(selection, num_selection, model_dict, dst_dir, url):\n",
+        "    submodels = add_submodels(selection, num_selection, model_dict, dst_dir)\n",
+        "    for submodel in submodels:\n",
+        "        if not inpainting_model and \"inpainting\" in submodel['name']:\n",
+        "            continue\n",
+        "        url += f\"{submodel['url']} {submodel['dst_dir']} {submodel['name']}, \"\n",
+        "    return url\n",
         "\n",
-        "for submodel in submodels:\n",
-        "    if not Inpainting_Model and \"inpainting\" in submodel['name']:\n",
-        "        continue\n",
-        "    url += f\"{submodel['url']} {submodel['dst_dir']} {submodel['name']}, \"\n",
+        "url = handle_submodels(model, model_num, model_list, models_dir, url)\n",
+        "url = handle_submodels(vae, vae_num, vae_list, vaes_dir, url)\n",
+        "url = handle_submodels(controlnet, controlnet_num, controlnet_list, control_dir, url)\n",
         "\n",
         "''' file.txt - added urls '''\n",
         "\n",
-        "unique_urls = []\n",
-        "\n",
-        "def process_file_download(file_url):\n",
+        "def process_file_download(file_url, prefixes, unique_urls):\n",
         "    files_urls = \"\"\n",
         "\n",
         "    if file_url.startswith(\"http\"):\n",
@@ -539,19 +549,23 @@
         "\n",
         "    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",
         "\n",
         "        urls = [url.split('#')[0].strip() for url in line.split(',')]  # filter urls\n",
         "        for url in urls:\n",
-        "            if url.startswith(\"http\") and url not in unique_urls:\n",
+        "            filter_url = url.split('[')[0] # same url filter\n",
+        "\n",
+        "            if url.startswith(\"http\") and filter_url not in unique_urls:\n",
         "                files_urls += f\"{current_tag}:{url}, \"\n",
-        "                unique_urls.append(url)\n",
+        "                unique_urls.add(filter_url)\n",
         "\n",
         "    return files_urls\n",
         "\n",
-        "# fix all possible errors/options and function call\n",
         "file_urls = \"\"\n",
+        "unique_urls = set()\n",
+        "\n",
         "if custom_file_urls:\n",
         "    for custom_file_url in custom_file_urls.replace(',', '').split():\n",
         "        if not custom_file_url.endswith('.txt'):\n",
@@ -561,7 +575,7 @@
         "                custom_file_url = f'{root_path}/{custom_file_url}'\n",
         "\n",
         "        try:\n",
-        "            file_urls += process_file_download(custom_file_url)\n",
+        "            file_urls += process_file_download(custom_file_url, prefixes, unique_urls)\n",
         "        except FileNotFoundError:\n",
         "            pass\n",
         "\n",
@@ -604,40 +618,8 @@
         "## List Models and stuff V2\n",
         "if detailed_download == \"off\":\n",
         "    print(\"\\n\\n\\033[33mIf you don't see any downloaded files, enable the 'Detailed Downloads' feature in the widget.\")\n",
-        "    get_ipython().run_line_magic('run', '{root_path}/file_cell/special/dl_display_results.py') # display widgets result\n",
         "\n",
-        "else:\n",
-        "    if any(not file.endswith('.txt') for file in os.listdir(models_dir)):\n",
-        "        print(\"\\n\\033[33m➤ Models\\033[0m\")\n",
-        "        !find {models_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
-        "    if any(not file.endswith('.txt') for file in os.listdir(vaes_dir)):\n",
-        "        print(\"\\n\\033[33m➤ VAEs\\033[0m\")\n",
-        "        !find {vaes_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
-        "    if any(not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file)) for file in os.listdir(embeddings_dir)):\n",
-        "        print(\"\\n\\033[33m➤ Embeddings\\033[0m\")\n",
-        "        !find {embeddings_dir}/ -mindepth 1 -maxdepth 1 \\( -name '*.pt' -or -name '*.safetensors' \\) -printf '%f\\n'\n",
-        "    if any(not file.endswith('.txt') for file in os.listdir(loras_dir)):\n",
-        "        print(\"\\n\\033[33m➤ LoRAs\\033[0m\")\n",
-        "        !find {loras_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
-        "    print(f\"\\n\\033[33m➤ Extensions\\033[0m\")\n",
-        "    !find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\n'\n",
-        "    if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir)):\n",
-        "        print(\"\\n\\033[33m➤ ControlNet\\033[0m\")\n",
-        "        !find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\([^_]*\\)_fp16\\.safetensors$/\\1/'\n",
-        "\n",
-        "\n",
-        "# === OTHER ===\n",
-        "# Downlaod discord tags UmiWildcards\n",
-        "files_umi = [\n",
-        "    \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/extensions/UmiWildacrd/discord/200_pan_gen.txt\",\n",
-        "    \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/extensions/UmiWildacrd/discord/150_bra_gen.txt\"\n",
-        "]\n",
-        "save_dir_path = f\"{webui_path}/extensions/Umi-AI-Wildcards/wildcards/discord\"\n",
-        "\n",
-        "with capture.capture_output() as cap:\n",
-        "    for file in files_umi:\n",
-        "        !aria2c --optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -j5 -x16 -s16 -k1M -c -d {save_dir_path} {file}\n",
-        "del cap"
+        "%run {root_path}/file_cell/special/dl_display_results.py # display widgets result"
       ]
     }
   ],

files_cells/notebooks/en/widgets_en.ipynb

@@ -16,6 +16,11 @@
   "cells": [
     {
       "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "JKTCrY9LU7Oq"
+      },
+      "outputs": [],
       "source": [
         "##~   WIDGET CODE | BY: ANXETY   ~##\n",
         "\n",
@@ -27,27 +32,24 @@
         "from IPython.display import display, HTML, Javascript, clear_output\n",
         "\n",
         "\n",
-        "#  ================= DETECT ENV =================\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",
         "\n",
         "env, root_path, free_plan = detect_environment()\n",
         "webui_path = f\"{root_path}/sdw\"\n",
-        "\n",
         "!mkdir -p {root_path}\n",
-        "#  ----------------------------------------------\n",
         "\n",
         "\n",
         "# ==================== CSS JS ====================\n",
-        "# custom background images V1.5\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",
@@ -59,8 +61,9 @@
         "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",
+        "\n",
         "url_img = args.image\n",
         "opacity_img = args.opacity\n",
         "blur_img = args.blur\n",
@@ -78,7 +81,6 @@
         "\n",
         "\"\"\" mode img - repeats \"\"\"\n",
         "mode_img = \"repeat\" if not args.mode else \"no-repeat\"\n",
-        "## ---\n",
         "\n",
         "container_background = f'''\n",
         "<style>\n",
@@ -101,345 +103,15 @@
         "'''\n",
         "\n",
         "display(HTML(container_background))\n",
-        "# ---\n",
-        "\n",
-        "CSS = '''\n",
-        "<style>\n",
-        "/* General Styles */\n",
-        ".header {\n",
-        "    font-family: cursive;\n",
-        "    font-size: 20px;\n",
-        "    font-weight: bold;\n",
-        "    color: #ff8cee;\n",
-        "    margin-bottom: 15px;\n",
-        "    user-select: none;\n",
-        "    cursor: default;\n",
-        "    display: inline-block;\n",
-        "}\n",
-        "\n",
-        "hr {\n",
-        "    border-color: grey;\n",
-        "    background-color: grey;\n",
-        "    opacity: 0.25;\n",
-        "}\n",
-        "\n",
-        "a {\n",
-        "  text-decoration: none;\n",
-        "  color: inherit;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Container style */\n",
-        "\n",
-        ".container {\n",
-        "    position: relative;\n",
-        "    background-color: #232323;\n",
-        "    width: 1080px;\n",
-        "    padding: 10px 15px;\n",
-        "    border-radius: 15px;\n",
-        "    box-shadow: 0 0 50px rgba(0, 0, 0, 0.3);\n",
-        "    margin-bottom: 5px;\n",
-        "    overflow: hidden;\n",
-        "}\n",
-        "\n",
-        ".container::after {\n",
-        "    position: absolute;\n",
-        "    top: 5px;\n",
-        "    right: 10px;\n",
-        "    content: \"ANXETY\";\n",
-        "    font-weight: bold;\n",
-        "    font-size: 24px;\n",
-        "    color: rgba(0, 0, 0, 0.2);\n",
-        "}\n",
-        "\n",
-        "/* background img */\n",
-        ".container::before {\n",
-        "    content: \"\";\n",
-        "    position: absolute;\n",
-        "    top: 0;\n",
-        "    left: 0;\n",
-        "    right: 0;\n",
-        "    bottom: 0;\n",
-        "    background-image: var(--img_background);\n",
-        "    background-size: var(--img_scale);\n",
-        "    background-repeat: var(--img_mode);\n",
-        "    opacity: var(--img_opacity);\n",
-        "    mix-blend-mode: screen;\n",
-        "    pointer-events: none;\n",
-        "    filter: blur(var(--img_blur));\n",
-        "    z-index: -1;\n",
-        "}\n",
-        "\n",
-        ".image_1::before {\n",
-        "    background-position: var(--image_x) calc(-120px - var(--image_y));\n",
-        "}\n",
-        ".image_2::before {\n",
-        "    background-position: var(--image_x) calc(-290px - var(--image_y));\n",
-        "}\n",
-        ".image_3::before {\n",
-        "    background-position: var(--image_x) calc(-430px - var(--image_y));\n",
-        "}\n",
-        ".image_4::before {\n",
-        "    background-position: var(--image_x) calc(var(--img_height_dif) - var(--image_y));\n",
-        "}\n",
-        "\n",
-        ".container_custom_downlad {\n",
-        "    height: 55px;\n",
-        "    transition: all 0.5s;\n",
-        "}\n",
-        "\n",
-        ".container_custom_downlad.expanded {\n",
-        "    height: 270px;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Element text style */\n",
-        "\n",
-        ".widget-html,\n",
-        ".widget-button,\n",
-        ".widget-text label,\n",
-        ".widget-checkbox label,\n",
-        ".widget-dropdown label,\n",
-        ".widget-dropdown select,\n",
-        ".widget-text input[type=\"text\"] {\n",
-        "    font-family: cursive;\n",
-        "    font-size: 14px;\n",
-        "    color: white !important;\n",
-        "    user-select: none;\n",
-        "}\n",
-        "\n",
-        ".widget-text input[type=\"text\"]::placeholder {\n",
-        "    color: grey;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Input field styles */\n",
-        "\n",
-        ".widget-dropdown select,\n",
-        ".widget-text input[type=\"text\"] {\n",
-        "    height: 30px;\n",
-        "    background-color: var(--bg-field-color);\n",
-        "    border: 1px solid #262626;\n",
-        "    border-radius: 10px;\n",
-        "    box-shadow: inset 0 0 5px rgba(0, 0, 0, 0.5);\n",
-        "    transition: all 0.3s ease-in-out;\n",
-        "    backdrop-filter: blur(var(--bg-field-blur-level));\n",
-        "}\n",
-        "\n",
-        ".widget-dropdown select:focus,\n",
-        ".widget-text input[type=\"text\"]:focus {\n",
-        "    border-color: #006ee5;\n",
-        "}\n",
-        "\n",
-        ".widget-dropdown select:hover,\n",
-        ".widget-text input[type=\"text\"]:hover {\n",
-        "    transform: scale(1.003);\n",
-        "    background-color: var(--bg-field-color-hover);\n",
-        "    box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.5);\n",
-        "}\n",
-        "\n",
-        ".widget-dropdown option {\n",
-        "    background-color: #1c1c1c;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Slider Checkbox style */\n",
-        "\n",
-        ".widget-checkbox input[type=\"checkbox\"] {\n",
-        "    appearance: none;\n",
-        "    position: relative;\n",
-        "    top: 4px; /* Why is he taller?! */\n",
-        "    width: 40px;\n",
-        "    height: 20px;\n",
-        "    border: none;\n",
-        "    border-radius: 10px;\n",
-        "    background-color: #20b2aa;\n",
-        "    cursor: pointer;\n",
-        "    box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.3);\n",
-        "    transition: background-color 0.3s ease;\n",
-        "}\n",
-        ".widget-checkbox input[type=\"checkbox\"]:checked {\n",
-        "    background-color: #2196F3;\n",
-        "}\n",
-        "\n",
-        ".widget-checkbox input[type=\"checkbox\"]:before {\n",
-        "    content: '';\n",
-        "    position: absolute;\n",
-        "    top: 50%;\n",
-        "    left: 3px;\n",
-        "    width: 16px;\n",
-        "    height: 16px;\n",
-        "    border-radius: inherit;\n",
-        "    background-color: white;\n",
-        "    box-shadow: 0 0 8px rgba(0, 0, 0, 0.3);\n",
-        "    transform: translateY(-50%);\n",
-        "    transition: left 0.3s ease;\n",
-        "}\n",
-        ".widget-checkbox input[type=\"checkbox\"]:checked:before {\n",
-        "    left: 21px;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Button styles */\n",
-        "\n",
-        ".button_save {\n",
-        "    font-size: 15px;\n",
-        "    font-weight: bold;\n",
-        "    width: 120px;\n",
-        "    height: 35px;\n",
-        "    border-radius: 15px;\n",
-        "    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);\n",
-        "    background-size: 200% 200%;\n",
-        "    background-position: left bottom;\n",
-        "    transition: background 0.5s ease-in-out, transform 0.3s ease;\n",
-        "}\n",
-        "\n",
-        ".button_save:hover {\n",
-        "    cursor: pointer;\n",
-        "    background-image: radial-gradient(circle at top left, purple 10%, #93ac47 90%);\n",
-        "    background-size: 200% 200%;\n",
-        "    background-position: right bottom;\n",
-        "    transform: translateY(1px);\n",
-        "}\n",
         "\n",
-        ".button_ngrok {\n",
-        "    font-size: 12px;\n",
-        "    height: 30px;\n",
-        "    border-radius: 10px;\n",
-        "    padding: 1px 12px;\n",
-        "    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);\n",
-        "    background-size: 200% 200%;\n",
-        "    background-position: left bottom;\n",
-        "    transition: background 0.5s ease-in-out, transform 0.3s ease;\n",
-        "    white-space: nowrap;\n",
-        "}\n",
-        "\n",
-        ".button_ngrok:hover  {\n",
-        "    cursor: pointer;\n",
-        "    background-image: radial-gradient(circle at top left, purple 10%, #1D94BB 90%);\n",
-        "    background-size: 200% 200%;\n",
-        "    background-position: right bottom;\n",
-        "    transform: translateY(1px);\n",
-        "}\n",
-        "\n",
-        ".button_save:active,\n",
-        ".button_ngrok:active {\n",
-        "    filter: brightness(0.75) !important;\n",
-        "}\n",
-        "\n",
-        "/* Removes ugly stroke from widget buttons. */\n",
-        ".jupyter-widgets.lm-Widget:focus {\n",
-        "    outline: none;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Popup style of `INFO` window */\n",
-        "\n",
-        ".info {\n",
-        "    position: absolute;\n",
-        "    top: -5px;\n",
-        "    right: 95px;\n",
-        "    color: grey;\n",
-        "    font-family: cursive;\n",
-        "    font-size: 14px;\n",
-        "    font-weight: normal;\n",
-        "    user-select: none;\n",
-        "    pointer-events: none;\n",
-        "    opacity: 0;\n",
-        "    transition: opacity 0.3s ease-in-out;\n",
-        "    display: inline-block;\n",
-        "}\n",
-        "\n",
-        ".popup {\n",
-        "    position: absolute;\n",
-        "    top: 120px;\n",
-        "    z-index: 999;\n",
-        "    width: auto;\n",
-        "    padding: 10px;\n",
-        "    text-align: center;\n",
-        "    background-color: rgba(255, 255, 255, 0.05);\n",
-        "    backdrop-filter: blur(20px);\n",
-        "    border: 1px solid rgba(255, 255, 255, 0.45);\n",
-        "    border-radius: 8px;\n",
-        "    box-shadow: 0 0 50px rgba(0, 0, 0, 0.5);\n",
-        "    opacity: 0;\n",
-        "    color: #fff;\n",
-        "    font-size: 16px;\n",
-        "    font-family: cursive;\n",
-        "    user-select: none;\n",
-        "    cursor: default;\n",
-        "    pointer-events: none;\n",
-        "    transform: rotate(-5deg);\n",
-        "    transition: top 0.3s ease-in-out, opacity 0.3s ease-in-out, transform 0.3s ease-in-out;\n",
-        "}\n",
-        "\n",
-        ".sample {\n",
-        "    display: inline-block;\n",
-        "    margin-top: 25px;\n",
-        "    padding: 10px 100px;\n",
-        "    background-color: rgba(255, 255, 255, 0.2);\n",
-        "    color: #c6e2ff;\n",
-        "    border: 2px solid rgba(255, 255, 255, 0.2);\n",
-        "    border-radius: 8px;\n",
-        "    box-shadow: 0 0 10px rgba(255, 255, 255, 0.2), inset 0 0 25px rgba(255, 255, 255, 0.2);\n",
-        "}\n",
-        "\n",
-        ".info.showed {\n",
-        "    opacity: 1;\n",
-        "    pointer-events: auto;\n",
-        "}\n",
-        "\n",
-        ".info:hover + .popup {\n",
-        "    top: 35px;\n",
-        "    opacity: 1;\n",
-        "    pointer-events: initial;\n",
-        "    transform: rotate(0deg);\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Animation of elements */\n",
-        "\n",
-        ".container,\n",
-        ".button_save {\n",
-        "    animation-name: showedWidgets;\n",
-        "    animation-duration: 1s;\n",
-        "    animation-fill-mode: forwards;\n",
-        "}\n",
-        "\n",
-        ".container.hide,\n",
-        ".button_save.hide {\n",
-        "    animation-name: hideWidgets;\n",
-        "    animation-duration: 0.5s;\n",
-        "    animation-fill-mode: forwards;\n",
-        "}\n",
-        "\n",
-        "@keyframes showedWidgets {\n",
-        "    0% {\n",
-        "        transform: translate3d(-65%, 15%, 0) scale(0) rotate(15deg);\n",
-        "        filter: blur(25px) grayscale(1) brightness(0.3);\n",
-        "        opacity: 0;\n",
-        "    }\n",
-        "    100% {\n",
-        "        transform: translate3d(0, 0, 0) scale(1) rotate(0deg);\n",
-        "        filter: blur(0) grayscale(0) brightness(1);\n",
-        "        opacity: 1;\n",
-        "    }\n",
-        "}\n",
-        "\n",
-        "@keyframes hideWidgets {\n",
-        "    0% {\n",
-        "        transform: translate3d(0, 0, 0) scale(1) rotate3d(1, 0, 0, 0deg);\n",
-        "        filter: blur(0) grayscale(0) brightness(1);\n",
-        "        opacity: 1;\n",
-        "    }\n",
-        "    100% {\n",
-        "        transform: translate3d(0, 5%, 0) scale(0.9) rotate3d(1, 0, 0, 90deg);\n",
-        "        filter: blur(15px) grayscale(1) brightness(0.5);\n",
-        "        opacity: 0;\n",
-        "    }\n",
-        "}\n",
-        "</style>\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",
@@ -451,20 +123,58 @@
         "}\n",
         "</script>\n",
         "'''\n",
+        "display(HTML(JS))\n",
         "\n",
-        "display(HTML(CSS))\n",
-        "# ==================== CSS JS ====================\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",
-        "# ==================== WIDGETS ====================\n",
-        "# --- global widgets ---\n",
-        "style = {'description_width': 'initial'}\n",
-        "layout = widgets.Layout(min_width='1047px')\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",
-        "HR = widgets.HTML('<hr>')\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 = widgets.HTML('<div class=\"header\">Model Selection<div>')\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",
@@ -475,37 +185,37 @@
         "                 '7.Deliberate [Realism] [V6] + INP',\n",
         "                 '8.Meina-Mix [Anime] [V11] + INP',\n",
         "                 '9.Mix-Pro [Anime] [V4] + INP']\n",
-        "# ---\n",
-        "Model_widget = widgets.Dropdown(options=model_options, value='4.Counterfeit [Anime] [V3] + INP', description='Model:', style=style, layout=layout)\n",
-        "Model_Num_widget = widgets.Text(description='Model Number:', placeholder='Enter the model numbers to be downloaded using comma/space.', style=style, layout=layout)\n",
-        "Inpainting_Model_widget = widgets.Checkbox(value=False, description='Inpainting Models', style=style)\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 = widgets.VBox([model_header, Model_widget, Model_Num_widget, Inpainting_Model_widget]).add_class(\"container\").add_class(\"image_1\")\n",
-        "display(all_model_box)\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 = widgets.HTML('<div class=\"header\">VAE Selection</div>')\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 = widgets.Dropdown(options=vae_options, value='3.Blessed2.vae', description='Vae:', style=style, layout=layout)\n",
-        "Vae_Num_widget = widgets.Text(description='Vae Number:', placeholder='Enter the vae numbers to be downloaded using comma/space.', style=style, layout=layout)\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 = widgets.VBox([vae_header, Vae_widget, Vae_Num_widget]).add_class(\"container\").add_class(\"image_2\")\n",
-        "display(all_vae_box)\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 = widgets.HTML('<div class=\"header\">Additional</div>')\n",
-        "latest_webui_widget = widgets.Checkbox(value=True, description='Update WebUI', style=style)\n",
-        "latest_exstensions_widget = widgets.Checkbox(value=True, description='Update Extensions', style=style)\n",
-        "change_webui_widget = widgets.Dropdown(options=['A1111', 'Forge'], value='A1111', description='Change WebUI:', style=style)\n",
-        "detailed_download_widget = widgets.Dropdown(options=['off', 'on'], value='off', description='Detailed Download:', style=style)\n",
-        "latest_changes_widget = HBox([latest_webui_widget, latest_exstensions_widget, change_webui_widget, detailed_download_widget], layout=widgets.Layout(justify_content='space-between'))\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",
@@ -513,33 +223,45 @@
         "                      '8.scribble', '9.segmentation',\n",
         "                      '10.shuffle', '11.tile',\n",
         "                      '12.inpaint', '13.instruct_p2p']\n",
-        "# ---\n",
-        "controlnet_widget = widgets.Dropdown(options=controlnet_options, value='none', description='ControlNet:', style=style, layout=layout)\n",
-        "controlnet_Num_widget = widgets.Text(description='ControlNet Number:', placeholder='Enter the ControlNet model numbers to be downloaded using comma/space.', style=style, layout=layout)\n",
-        "commit_hash_widget = widgets.Text(description='Commit Hash:', style=style, layout=layout)\n",
-        "optional_huggingface_token_widget = widgets.Text(description='HuggingFace Token:', style=style, layout=layout)\n",
-        "ngrok_token_widget = widgets.Text(description='Ngrok Token:', style=style, layout=widgets.Layout(width='1047px'))\n",
-        "ngrock_button = widgets.HTML('<a href=\"https://dashboard.ngrok.com/get-started/your-authtoken\" target=\"_blank\">Get Ngrok Token</a>').add_class(\"button_ngrok\")\n",
-        "ngrok_widget = widgets.HBox([ngrok_token_widget, ngrock_button], style=style, layout=layout)\n",
-        "zrok_token_widget = widgets.Text(description='Zrok Token:', style=style, layout=widgets.Layout(width='1047px'))\n",
-        "zrok_button = widgets.HTML('<a href=\"https://colab.research.google.com/drive/1d2sjWDJi_GYBUavrHSuQyHTDuLy36WpU\" target=\"_blank\">Reg Zrok Token</a>').add_class(\"button_ngrok\")\n",
-        "zrok_widget = widgets.HBox([zrok_token_widget, zrok_button], style=style, layout=layout)\n",
-        "# ---\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 = widgets.Text(description='Arguments:', value=commandline_arguments_options, style=style, layout=layout)\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",
-        "''' Display Additional'''\n",
-        "additional_widget_list = [additional_header, latest_changes_widget, HR, controlnet_widget, controlnet_Num_widget, commit_hash_widget, optional_huggingface_token_widget, ngrok_widget, zrok_widget, HR, commandline_arguments_widget]\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",
-        "    latest_changes_widget.children = [widget for widget in latest_changes_widget.children if widget != change_webui_widget]\n",
-        "# ```\n",
-        "all_additional_box = widgets.VBox(additional_widget_list).add_class(\"container\").add_class(\"image_3\")\n",
-        "display(all_additional_box)\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 = widgets.HTML('''\n",
+        "custom_download_header_popup = factory.create_html('''\n",
         "<style>\n",
         "/* Term Colors */\n",
         ".sample_label {color: #dbafff;}\n",
@@ -564,39 +286,40 @@
         "    </div>\n",
         "</div>\n",
         "''')\n",
-        "# ---\n",
-        "Model_url_widget = widgets.Text(description='Model:', style=style, layout=layout)\n",
-        "Vae_url_widget = widgets.Text(description='Vae:', style=style, layout=layout)\n",
-        "LoRA_url_widget = widgets.Text(description='LoRa:', style=style, layout=layout)\n",
-        "Embedding_url_widget = widgets.Text(description='Embedding:', style=style, layout=layout)\n",
-        "Extensions_url_widget = widgets.Text(description='Extensions:', style=style, layout=layout)\n",
-        "custom_file_urls_widget = widgets.Text(description='File (txt):', style=style, layout=layout)\n",
-        "\n",
-        "''' Display CustomDl'''\n",
-        "all_custom_box = widgets.VBox([\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",
-        "    ]).add_class(\"container\").add_class(\"image_4\").add_class(\"container_custom_downlad\")\n",
-        "display(all_custom_box)\n",
+        "], class_names=[\"container\", \"image_4\", \"container_custom_downlad\"])\n",
+        "factory.display(all_custom_box)\n",
         "\n",
         "# --- Save Button ---\n",
-        "save_button = widgets.Button(description='Save').add_class(\"button_save\")\n",
-        "display(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',\n",
-        "    'latest_webui', 'latest_exstensions', 'change_webui', 'detailed_download',\n",
-        "    'controlnet', 'controlnet_Num', 'commit_hash', 'optional_huggingface_token',\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', 'Extensions_url', 'custom_file_urls'\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",
+        "    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",
@@ -604,28 +327,23 @@
         "    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 save_data(button):\n",
-        "    save_settings()\n",
+        "            for key in SETTINGS_KEYS:\n",
+        "                globals()[f\"{key}_widget\"].value = settings.get(key, \"\")\n",
         "\n",
-        "    # --- uhh - hide... ---\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",
-        "\n",
         "    widgets.Widget.close_all()\n",
         "\n",
-        "settings = load_settings()\n",
+        "def save_data(button):\n",
+        "    save_settings()\n",
+        "    hide_widgets()\n",
+        "\n",
+        "load_settings()\n",
         "save_button.on_click(save_data)"
-      ],
-      "metadata": {
-        "id": "2lJmbqrs3Mu8"
-      },
-      "execution_count": null,
-      "outputs": []
+      ]
     }
   ]
 }

files_cells/notebooks/ru/auto_cleaner_ru.ipynb

@@ -0,0 +1,165 @@
+{
+  "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)"
+      ]
+    }
+  ]
+}

files_cells/notebooks/ru/downloading_ru.ipynb

@@ -25,21 +25,19 @@
         "from urllib.parse import urlparse, parse_qs\n",
         "\n",
         "\n",
-        "#  ================= DETECT ENV =================\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",
         "\n",
         "env, root_path, free_plan = detect_environment()\n",
         "webui_path = f\"{root_path}/sdw\"\n",
-        "#  ----------------------------------------------\n",
         "\n",
         "\n",
         "# ================ LIBRARIES V2 ================\n",
@@ -90,12 +88,6 @@
         "    time.sleep(2)\n",
         "    clear_output()\n",
         "\n",
-        "## dl special files\n",
-        "with capture.capture_output() as cap:\n",
-        "    !mkdir -p {root_path}/file_cell/special\n",
-        "    !wget https://huggingface.co/NagisaNao/fast_repo/resolve/main/special/dl_display_results.py -O {root_path}/file_cell/special/dl_display_results.py\n",
-        "del cap\n",
-        "\n",
         "\n",
         "# ================= loading settings V4 =================\n",
         "def load_settings(path):\n",
@@ -106,16 +98,17 @@
         "\n",
         "settings = load_settings(f'{root_path}/settings.json')\n",
         "\n",
-        "variables = [\n",
-        "    'Model', 'Model_Num', 'Inpainting_Model',\n",
-        "    'Vae', 'Vae_Num',\n",
-        "    'latest_webui', 'latest_exstensions', 'change_webui', 'detailed_download',\n",
-        "    'controlnet', 'controlnet_Num', 'commit_hash', 'optional_huggingface_token',\n",
+        "VARIABLES = [\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', 'Extensions_url', 'custom_file_urls'\n",
+        "    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url',\n",
+        "    'Extensions_url', 'custom_file_urls'\n",
         "]\n",
         "\n",
-        "locals().update({key: settings.get(key) for key in variables})\n",
+        "locals().update({key: settings.get(key) for key in VARIABLES})\n",
         "\n",
         "\n",
         "# ================= OTHER =================\n",
@@ -251,17 +244,7 @@
         "    ]\n",
         "}\n",
         "\n",
-        "# 1-4 (fp16/cleaned)\n",
         "vae_list = {\n",
-        "    ## vae broke - the author's an asshole\n",
-        "    # \"1.Anime.vae\": [\n",
-        "    #     {\"url\": \"https://civitai.com/api/download/models/131654\", \"name\": \"Anime.vae.safetensors\"},\n",
-        "    #     {\"url\": \"https://civitai.com/api/download/models/131658\", \"name\": \"vae-ft-mse.vae.safetensors\"}\n",
-        "    # ],\n",
-        "    # \"2.Anything.vae\": [{\"url\": \"https://civitai.com/api/download/models/131656\", \"name\": \"Anything.vae.safetensors\"}],\n",
-        "    # \"3.Blessed2.vae\": [{\"url\": \"https://civitai.com/api/download/models/142467\", \"name\": \"Blessed2.vae.safetensors\"}],\n",
-        "    # \"4.ClearVae.vae\": [{\"url\": \"https://civitai.com/api/download/models/133362\", \"name\": \"ClearVae_23.vae.safetensors\"}],\n",
-        "\n",
         "    \"1.Anime.vae\": [{\"url\": \"https://civitai.com/api/download/models/311162\", \"name\": \"vae-ft-mse-840000-ema-pruned.vae.safetensors\"}],\n",
         "    \"2.Anything.vae\": [{\"url\": \"https://civitai.com/api/download/models/119279\", \"name\": \"Anything.vae.safetensors\"}],\n",
         "    \"3.Blessed2.vae\": [{\"url\": \"https://huggingface.co/NoCrypt/blessed_vae/resolve/main/blessed2.vae.pt\", \"name\": \"Blessed2.vae.safetensors\"}],\n",
@@ -339,26 +322,47 @@
         "}\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 = optional_huggingface_token if optional_huggingface_token else \"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\"\n",
+        "hf_token = huggingface_token if huggingface_token else \"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\"\n",
         "user_header = f\"\\\"Authorization: Bearer {hf_token}\\\"\"\n",
         "\n",
+        "''' Formatted Info Output '''\n",
+        "\n",
+        "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",
+        "\n",
+        "def format_output(url, dst_dir, file_name):\n",
+        "    info = f\"[{file_name.split('.')[0]}]\"\n",
+        "    info = center_text(info)\n",
+        "\n",
+        "    print(f\"\\n\\033[32m{'---'*20}]{info}[{'---'*20}\")\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 Image Preview | CivitAi '''\n",
         "\n",
-        "def get_data_from_api(model_id): # get model data\n",
+        "def get_data_from_api(model_id):\n",
+        "    \"\"\"Fetch model data from the API\"\"\"\n",
         "    endpoint_url = f\"https://civitai.com/api/v1/model-versions/{model_id}\"\n",
         "    headers = {\"Content-Type\": \"application/json\"}\n",
         "    try:\n",
         "        response = requests.get(endpoint_url, headers=headers)\n",
-        "        if response.status_code == 200:\n",
-        "            return response.json()\n",
+        "        response.raise_for_status()\n",
+        "        return response.json()\n",
         "    except requests.exceptions.RequestException as e:\n",
         "        print(f\"An error occurred: {e}\")\n",
         "        return None\n",
         "\n",
         "def extract_model_info(data, url):\n",
+        "    \"\"\"Extract model information based on URL\"\"\"\n",
         "    if 'type=' in url:\n",
         "        model_type = parse_qs(urlparse(url).query).get('type', [''])[0]\n",
         "        model_name = data['files'][1]['name']\n",
@@ -375,59 +379,69 @@
         "    return model_type, model_name, image_url\n",
         "\n",
         "def gen_preview_filename(model_name, image_url):\n",
+        "    \"\"\"Generate a preview filename\"\"\"\n",
         "    name = model_name.split('.')\n",
         "    img_exts = image_url.split('.')\n",
-        "    return f\"{name[0]}.preview.{img_exts[-1]}\" # assigning the original image format\n",
+        "    return f\"{name[0]}.preview.{img_exts[-1]}\"\n",
         "\n",
         "''' main download code '''\n",
         "\n",
         "def handle_manual(url):\n",
-        "    original_url = url\n",
-        "    url = url.split(':', 1)[1]\n",
-        "    file_name = re.search(r'\\[(.*?)\\]', url)\n",
-        "    file_name = file_name.group(1) if file_name else None\n",
-        "    if file_name:\n",
-        "        url = re.sub(r'\\[.*?\\]', '', url)\n",
+        "    url_parts = url.split(':', 1)\n",
+        "    prefix = url_parts[0]\n",
+        "    path = url_parts[1]\n",
         "\n",
-        "    for prefix, dir in prefixes.items():\n",
-        "        if original_url.startswith(f\"{prefix}:\"):\n",
-        "            if prefix != \"extension\":\n",
-        "                manual_download(url, dir, file_name=file_name)\n",
-        "            else:\n",
-        "                extension_repo.append((url, file_name))\n",
+        "    file_name_match = re.search(r'\\[(.*?)\\]', path)\n",
+        "    file_name = file_name_match.group(1) if file_name_match else None\n",
+        "    if file_name:\n",
+        "        path = re.sub(r'\\[.*?\\]', '', path)\n",
+        "\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",
+        "            except Exception as e:\n",
+        "                print(f\"Error downloading file: {e}\")\n",
+        "        else:\n",
+        "            extension_repo.append((path, file_name))\n",
         "\n",
         "def manual_download(url, dst_dir, file_name):\n",
         "    aria2_args = '--optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -j5 -x16 -s16 -k1M -c'\n",
         "    basename = url.split(\"/\")[-1] if file_name is None else file_name\n",
         "    header_option = f\"--header={user_header}\"\n",
         "\n",
-        "    # === CivitAi API ===\n",
+        "    # ==== CivitAi API+ ====\n",
         "    support_types = ('Checkpoint', 'Model', 'TextualInversion', 'LORA') # for dl preview image\n",
         "    civitai_token = \"62c0c5956b2f9defbd844d754000180b\"\n",
         "\n",
         "    if 'civitai' in url:\n",
-        "        url = f\"{url}{'&' if '?' in url else '?'}token={civitai_token}\" # some authors are pussies and they need API token...\n",
-        "        model_id = url.split('/')[-1]\n",
-        "        clean_url = url.split('?')[0]\n",
+        "        url = f\"{url}{'&' if '?' in url else '?'}token={civitai_token}\"\n",
+        "        model_id = url.split('/')[-1].split('?')[0]\n",
+        "        clean_url = re.sub(r'[?&]token=[^&]*', '', url) # hide token\n",
         "\n",
         "        data = get_data_from_api(model_id)\n",
         "        if data:\n",
         "            model_type, model_name, image_url = extract_model_info(data, url)\n",
-        "            if model_name and image_url:\n",
-        "                image_file_name = gen_preview_filename(model_name if not file_name else file_name, image_url)\n",
-        "                if any(types in model_type for types in support_types):\n",
-        "                    with capture.capture_output() as cap: # clear shit\n",
-        "                        !aria2c {aria2_args} -d {dst_dir} -o {image_file_name} {image_url}\n",
+        "\n",
+        "            if any(t in model_type for t in support_types):\n",
+        "                if model_name and image_url:\n",
+        "                    image_file_name = gen_preview_filename(model_name if not file_name else file_name, image_url)\n",
+        "                    with capture.capture_output() as cap:\n",
+        "                        !aria2c {aria2_args} -d {dst_dir} -o {image_file_name} '{image_url}'\n",
         "                    del cap\n",
-        "            file_name = file_name or model_name # assigns the original file name if not specified initially\n",
+        "            file_name = file_name or model_name\n",
+        "    else:\n",
+        "        clean_url = url\n",
+        "\n",
+        "    \"\"\" Formatted info output \"\"\"\n",
+        "    model_name_or_basename = file_name if not 'huggingface' in url else basename\n",
+        "    format_output(clean_url or url, dst_dir, model_name_or_basename)\n",
         "\n",
-        "    \"\"\" information output \"\"\"\n",
-        "    # -- wrold's best print info --\n",
-        "    print(f\"\\n\\033[32m{'---'*45}\\n\\033[33mURL: \\033[34m{clean_url if 'civitai' in url else url}\\n\\033[33mSAVE DIR: \\033[34m{dst_dir}\\n\\033[33mFILE NAME: \\033[34m{file_name if not 'huggingface' in url else basename}\\033[0m\")\n",
         "    print(\"\\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 any(types in model_type for types in support_types) and (locals().get('image_file_name') or ''):\n",
+        "    if 'civitai' in url and data and any(t in model_type for t in support_types) and (locals().get('image_file_name') or ''):\n",
         "        print(f\"\\033[32m[Preview DL]:\\033[0m {image_file_name} - {image_url}\\n\")\n",
-        "    # ===================\n",
+        "    # =====================\n",
         "\n",
         "    # -- GDrive --\n",
         "    if 'drive.google' in url:\n",
@@ -471,9 +485,8 @@
         "\n",
         "    unpucking_zip_files()\n",
         "\n",
-        "## unpucking zip files\n",
+        "# unpucking zip files\n",
         "def unpucking_zip_files():\n",
-        "    # directories - above\n",
         "    for directory in directories:\n",
         "        for root, dirs, files in os.walk(directory):\n",
         "            for file in files:\n",
@@ -486,8 +499,6 @@
         "\n",
         "''' submodels - added urls '''\n",
         "\n",
-        "submodels = []\n",
-        "\n",
         "def add_submodels(selection, num_selection, model_dict, dst_dir):\n",
         "    if selection == \"none\":\n",
         "        return []\n",
@@ -499,33 +510,32 @@
         "    else:\n",
         "        selected_models = model_dict[selection]\n",
         "        selected_nums = map(int, num_selection.replace(',', '').split())\n",
-        "\n",
         "        for num in selected_nums:\n",
         "            if 1 <= num <= len(model_dict):\n",
         "                name = list(model_dict)[num - 1]\n",
         "                selected_models.extend(model_dict[name])\n",
         "\n",
         "    unique_models = list({model['name']: model for model in selected_models}.values())\n",
-        "\n",
         "    for model in unique_models:\n",
         "        model['dst_dir'] = dst_dir\n",
         "\n",
         "    return unique_models\n",
         "\n",
-        "submodels += add_submodels(Model, Model_Num, model_list, models_dir)                    # model\n",
-        "submodels += add_submodels(Vae, Vae_Num, vae_list, vaes_dir)                            # vae\n",
-        "submodels += add_submodels(controlnet, controlnet_Num, controlnet_list, control_dir)    # controlnet\n",
+        "def handle_submodels(selection, num_selection, model_dict, dst_dir, url):\n",
+        "    submodels = add_submodels(selection, num_selection, model_dict, dst_dir)\n",
+        "    for submodel in submodels:\n",
+        "        if not inpainting_model and \"inpainting\" in submodel['name']:\n",
+        "            continue\n",
+        "        url += f\"{submodel['url']} {submodel['dst_dir']} {submodel['name']}, \"\n",
+        "    return url\n",
         "\n",
-        "for submodel in submodels:\n",
-        "    if not Inpainting_Model and \"inpainting\" in submodel['name']:\n",
-        "        continue\n",
-        "    url += f\"{submodel['url']} {submodel['dst_dir']} {submodel['name']}, \"\n",
+        "url = handle_submodels(model, model_num, model_list, models_dir, url)\n",
+        "url = handle_submodels(vae, vae_num, vae_list, vaes_dir, url)\n",
+        "url = handle_submodels(controlnet, controlnet_num, controlnet_list, control_dir, url)\n",
         "\n",
         "''' file.txt - added urls '''\n",
         "\n",
-        "unique_urls = []\n",
-        "\n",
-        "def process_file_download(file_url):\n",
+        "def process_file_download(file_url, prefixes, unique_urls):\n",
         "    files_urls = \"\"\n",
         "\n",
         "    if file_url.startswith(\"http\"):\n",
@@ -539,19 +549,23 @@
         "\n",
         "    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",
         "\n",
         "        urls = [url.split('#')[0].strip() for url in line.split(',')]  # filter urls\n",
         "        for url in urls:\n",
-        "            if url.startswith(\"http\") and url not in unique_urls:\n",
+        "            filter_url = url.split('[')[0] # same url filter\n",
+        "\n",
+        "            if url.startswith(\"http\") and filter_url not in unique_urls:\n",
         "                files_urls += f\"{current_tag}:{url}, \"\n",
-        "                unique_urls.append(url)\n",
+        "                unique_urls.add(filter_url)\n",
         "\n",
         "    return files_urls\n",
         "\n",
-        "# fix all possible errors/options and function call\n",
         "file_urls = \"\"\n",
+        "unique_urls = set()\n",
+        "\n",
         "if custom_file_urls:\n",
         "    for custom_file_url in custom_file_urls.replace(',', '').split():\n",
         "        if not custom_file_url.endswith('.txt'):\n",
@@ -561,7 +575,7 @@
         "                custom_file_url = f'{root_path}/{custom_file_url}'\n",
         "\n",
         "        try:\n",
-        "            file_urls += process_file_download(custom_file_url)\n",
+        "            file_urls += process_file_download(custom_file_url, prefixes, unique_urls)\n",
         "        except FileNotFoundError:\n",
         "            pass\n",
         "\n",
@@ -604,40 +618,8 @@
         "## List Models and stuff V2\n",
         "if detailed_download == \"off\":\n",
         "    print(\"\\n\\n\\033[33mЕсли вы не видете каких-то скаченных файлов, включите в виджетах функцию 'Подробная Загрузка'.\")\n",
-        "    get_ipython().run_line_magic('run', '{root_path}/file_cell/special/dl_display_results.py') # display widgets result\n",
         "\n",
-        "else:\n",
-        "    if any(not file.endswith('.txt') for file in os.listdir(models_dir)):\n",
-        "        print(\"\\n\\033[33m➤ Models\\033[0m\")\n",
-        "        !find {models_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
-        "    if any(not file.endswith('.txt') for file in os.listdir(vaes_dir)):\n",
-        "        print(\"\\n\\033[33m➤ VAEs\\033[0m\")\n",
-        "        !find {vaes_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
-        "    if any(not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file)) for file in os.listdir(embeddings_dir)):\n",
-        "        print(\"\\n\\033[33m➤ Embeddings\\033[0m\")\n",
-        "        !find {embeddings_dir}/ -mindepth 1 -maxdepth 1 \\( -name '*.pt' -or -name '*.safetensors' \\) -printf '%f\\n'\n",
-        "    if any(not file.endswith('.txt') for file in os.listdir(loras_dir)):\n",
-        "        print(\"\\n\\033[33m➤ LoRAs\\033[0m\")\n",
-        "        !find {loras_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
-        "    print(f\"\\n\\033[33m➤ Extensions\\033[0m\")\n",
-        "    !find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\n'\n",
-        "    if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir)):\n",
-        "        print(\"\\n\\033[33m➤ ControlNet\\033[0m\")\n",
-        "        !find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\([^_]*\\)_fp16\\.safetensors$/\\1/'\n",
-        "\n",
-        "\n",
-        "# === OTHER ===\n",
-        "# Downlaod discord tags UmiWildcards\n",
-        "files_umi = [\n",
-        "    \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/extensions/UmiWildacrd/discord/200_pan_gen.txt\",\n",
-        "    \"https://huggingface.co/NagisaNao/fast_repo/resolve/main/extensions/UmiWildacrd/discord/150_bra_gen.txt\"\n",
-        "]\n",
-        "save_dir_path = f\"{webui_path}/extensions/Umi-AI-Wildcards/wildcards/discord\"\n",
-        "\n",
-        "with capture.capture_output() as cap:\n",
-        "    for file in files_umi:\n",
-        "        !aria2c --optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -j5 -x16 -s16 -k1M -c -d {save_dir_path} {file}\n",
-        "del cap"
+        "%run {root_path}/file_cell/special/dl_display_results.py # display widgets result"
       ]
     }
   ],

files_cells/notebooks/ru/widgets_ru.ipynb

@@ -16,6 +16,11 @@
   "cells": [
     {
       "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "JKTCrY9LU7Oq"
+      },
+      "outputs": [],
       "source": [
         "##~   WIDGET CODE | BY: ANXETY   ~##\n",
         "\n",
@@ -27,27 +32,24 @@
         "from IPython.display import display, HTML, Javascript, clear_output\n",
         "\n",
         "\n",
-        "#  ================= DETECT ENV =================\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",
         "\n",
         "env, root_path, free_plan = detect_environment()\n",
         "webui_path = f\"{root_path}/sdw\"\n",
-        "\n",
         "!mkdir -p {root_path}\n",
-        "#  ----------------------------------------------\n",
         "\n",
         "\n",
         "# ==================== CSS JS ====================\n",
-        "# custom background images V1.5\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",
@@ -59,8 +61,9 @@
         "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",
+        "\n",
         "url_img = args.image\n",
         "opacity_img = args.opacity\n",
         "blur_img = args.blur\n",
@@ -78,7 +81,6 @@
         "\n",
         "\"\"\" mode img - repeats \"\"\"\n",
         "mode_img = \"repeat\" if not args.mode else \"no-repeat\"\n",
-        "## ---\n",
         "\n",
         "container_background = f'''\n",
         "<style>\n",
@@ -101,345 +103,15 @@
         "'''\n",
         "\n",
         "display(HTML(container_background))\n",
-        "# ---\n",
-        "\n",
-        "CSS = '''\n",
-        "<style>\n",
-        "/* General Styles */\n",
-        ".header {\n",
-        "    font-family: cursive;\n",
-        "    font-size: 20px;\n",
-        "    font-weight: bold;\n",
-        "    color: #ff8cee;\n",
-        "    margin-bottom: 15px;\n",
-        "    user-select: none;\n",
-        "    cursor: default;\n",
-        "    display: inline-block;\n",
-        "}\n",
-        "\n",
-        "hr {\n",
-        "    border-color: grey;\n",
-        "    background-color: grey;\n",
-        "    opacity: 0.25;\n",
-        "}\n",
-        "\n",
-        "a {\n",
-        "  text-decoration: none;\n",
-        "  color: inherit;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Container style */\n",
-        "\n",
-        ".container {\n",
-        "    position: relative;\n",
-        "    background-color: #232323;\n",
-        "    width: 1080px;\n",
-        "    padding: 10px 15px;\n",
-        "    border-radius: 15px;\n",
-        "    box-shadow: 0 0 50px rgba(0, 0, 0, 0.3);\n",
-        "    margin-bottom: 5px;\n",
-        "    overflow: hidden;\n",
-        "}\n",
-        "\n",
-        ".container::after {\n",
-        "    position: absolute;\n",
-        "    top: 5px;\n",
-        "    right: 10px;\n",
-        "    content: \"ANXETY\";\n",
-        "    font-weight: bold;\n",
-        "    font-size: 24px;\n",
-        "    color: rgba(0, 0, 0, 0.2);\n",
-        "}\n",
-        "\n",
-        "/* background img */\n",
-        ".container::before {\n",
-        "  content: \"\";\n",
-        "  position: absolute;\n",
-        "  top: 0;\n",
-        "  left: 0;\n",
-        "  right: 0;\n",
-        "  bottom: 0;\n",
-        "  background-image: var(--img_background);\n",
-        "  background-size: var(--img_scale);\n",
-        "  background-repeat: var(--img_mode);\n",
-        "  opacity: var(--img_opacity);\n",
-        "  mix-blend-mode: screen;\n",
-        "  pointer-events: none;\n",
-        "  filter: blur(var(--img_blur));\n",
-        "  z-index: -1;\n",
-        "}\n",
-        "\n",
-        ".image_1::before {\n",
-        "  background-position: var(--image_x) calc(-120px - var(--image_y));\n",
-        "}\n",
-        ".image_2::before {\n",
-        "  background-position: var(--image_x) calc(-290px - var(--image_y));\n",
-        "}\n",
-        ".image_3::before {\n",
-        "  background-position: var(--image_x) calc(-430px - var(--image_y));\n",
-        "}\n",
-        ".image_4::before {\n",
-        "  background-position: var(--image_x) calc(var(--img_height_dif) - var(--image_y));\n",
-        "}\n",
-        "\n",
-        ".container_custom_downlad {\n",
-        "    height: 55px;\n",
-        "    transition: all 0.5s;\n",
-        "}\n",
-        "\n",
-        ".container_custom_downlad.expanded {\n",
-        "    height: 270px;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Element text style */\n",
-        "\n",
-        ".widget-html,\n",
-        ".widget-button,\n",
-        ".widget-text label,\n",
-        ".widget-checkbox label,\n",
-        ".widget-dropdown label,\n",
-        ".widget-dropdown select,\n",
-        ".widget-text input[type=\"text\"] {\n",
-        "    font-family: cursive;\n",
-        "    font-size: 14px;\n",
-        "    color: white !important;\n",
-        "    user-select: none;\n",
-        "}\n",
-        "\n",
-        ".widget-text input[type=\"text\"]::placeholder {\n",
-        "    color: grey;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Input field styles */\n",
-        "\n",
-        ".widget-dropdown select,\n",
-        ".widget-text input[type=\"text\"] {\n",
-        "    height: 30px;\n",
-        "    background-color: var(--bg-field-color);\n",
-        "    border: 1px solid #262626;\n",
-        "    border-radius: 10px;\n",
-        "    box-shadow: inset 0 0 5px rgba(0, 0, 0, 0.5);\n",
-        "    transition: all 0.3s ease-in-out;\n",
-        "    backdrop-filter: blur(var(--bg-field-blur-level));\n",
-        "}\n",
-        "\n",
-        ".widget-dropdown select:focus,\n",
-        ".widget-text input[type=\"text\"]:focus {\n",
-        "    border-color: #006ee5;\n",
-        "}\n",
-        "\n",
-        ".widget-dropdown select:hover,\n",
-        ".widget-text input[type=\"text\"]:hover {\n",
-        "    transform: scale(1.003);\n",
-        "    background-color: var(--bg-field-color-hover);\n",
-        "    box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.5);\n",
-        "}\n",
-        "\n",
-        ".widget-dropdown option {\n",
-        "    background-color: #1c1c1c;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Slider Checkbox style */\n",
-        "\n",
-        ".widget-checkbox input[type=\"checkbox\"] {\n",
-        "    appearance: none;\n",
-        "    position: relative;\n",
-        "    top: 4px; /* Why is he taller?! */\n",
-        "    width: 40px;\n",
-        "    height: 20px;\n",
-        "    border: none;\n",
-        "    border-radius: 10px;\n",
-        "    background-color: #20b2aa;\n",
-        "    cursor: pointer;\n",
-        "    box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.3);\n",
-        "    transition: background-color 0.3s ease;\n",
-        "}\n",
-        ".widget-checkbox input[type=\"checkbox\"]:checked {\n",
-        "    background-color: #2196F3;\n",
-        "}\n",
-        "\n",
-        ".widget-checkbox input[type=\"checkbox\"]:before {\n",
-        "    content: '';\n",
-        "    position: absolute;\n",
-        "    top: 50%;\n",
-        "    left: 3px;\n",
-        "    width: 16px;\n",
-        "    height: 16px;\n",
-        "    border-radius: inherit;\n",
-        "    background-color: white;\n",
-        "    box-shadow: 0 0 8px rgba(0, 0, 0, 0.3);\n",
-        "    transform: translateY(-50%);\n",
-        "    transition: left 0.3s ease;\n",
-        "}\n",
-        ".widget-checkbox input[type=\"checkbox\"]:checked:before {\n",
-        "    left: 21px;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Button styles */\n",
-        "\n",
-        ".button_save {\n",
-        "    font-size: 15px;\n",
-        "    font-weight: bold;\n",
-        "    width: 120px;\n",
-        "    height: 35px;\n",
-        "    border-radius: 15px;\n",
-        "    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);\n",
-        "    background-size: 200% 200%;\n",
-        "    background-position: left bottom;\n",
-        "    transition: background 0.5s ease-in-out, transform 0.3s ease;\n",
-        "}\n",
-        "\n",
-        ".button_save:hover {\n",
-        "    cursor: pointer;\n",
-        "    background-image: radial-gradient(circle at top left, purple 10%, #93ac47 90%);\n",
-        "    background-size: 200% 200%;\n",
-        "    background-position: right bottom;\n",
-        "    transform: translateY(1px);\n",
-        "}\n",
         "\n",
-        ".button_ngrok {\n",
-        "    font-size: 12px;\n",
-        "    height: 30px;\n",
-        "    border-radius: 10px;\n",
-        "    padding: 1px 12px;\n",
-        "    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);\n",
-        "    background-size: 200% 200%;\n",
-        "    background-position: left bottom;\n",
-        "    transition: background 0.5s ease-in-out, transform 0.3s ease;\n",
-        "    white-space: nowrap;\n",
-        "}\n",
-        "\n",
-        ".button_ngrok:hover  {\n",
-        "    cursor: pointer;\n",
-        "    background-image: radial-gradient(circle at top left, purple 10%, #1D94BB 90%);\n",
-        "    background-size: 200% 200%;\n",
-        "    background-position: right bottom;\n",
-        "    transform: translateY(1px);\n",
-        "}\n",
-        "\n",
-        ".button_save:active,\n",
-        ".button_ngrok:active {\n",
-        "    filter: brightness(0.75) !important;\n",
-        "}\n",
-        "\n",
-        "/* Removes ugly stroke from widget buttons. */\n",
-        ".jupyter-widgets.lm-Widget:focus {\n",
-        "    outline: none;\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Popup style of `INFO` window */\n",
-        "\n",
-        ".info {\n",
-        "    position: absolute;\n",
-        "    top: -5px;\n",
-        "    right: 95px;\n",
-        "    color: grey;\n",
-        "    font-family: cursive;\n",
-        "    font-size: 14px;\n",
-        "    font-weight: normal;\n",
-        "    user-select: none;\n",
-        "    pointer-events: none;\n",
-        "    opacity: 0;\n",
-        "    transition: opacity 0.3s ease-in-out;\n",
-        "    display: inline-block;\n",
-        "}\n",
-        "\n",
-        ".popup {\n",
-        "    position: absolute;\n",
-        "    top: 120px;\n",
-        "    z-index: 999;\n",
-        "    width: auto;\n",
-        "    padding: 10px;\n",
-        "    text-align: center;\n",
-        "    background-color: rgba(255, 255, 255, 0.05);\n",
-        "    backdrop-filter: blur(20px);\n",
-        "    border: 1px solid rgba(255, 255, 255, 0.45);\n",
-        "    border-radius: 8px;\n",
-        "    box-shadow: 0 0 50px rgba(0, 0, 0, 0.5);\n",
-        "    opacity: 0;\n",
-        "    color: #fff;\n",
-        "    font-size: 16px;\n",
-        "    font-family: cursive;\n",
-        "    user-select: none;\n",
-        "    cursor: default;\n",
-        "    pointer-events: none;\n",
-        "    transform: rotate(-5deg);\n",
-        "    transition: top 0.3s ease-in-out, opacity 0.3s ease-in-out, transform 0.3s ease-in-out;\n",
-        "}\n",
-        "\n",
-        ".sample {\n",
-        "    display: inline-block;\n",
-        "    margin-top: 25px;\n",
-        "    padding: 10px 100px;\n",
-        "    background-color: rgba(255, 255, 255, 0.2);\n",
-        "    color: #c6e2ff;\n",
-        "    border: 2px solid rgba(255, 255, 255, 0.2);\n",
-        "    border-radius: 8px;\n",
-        "    box-shadow: 0 0 10px rgba(255, 255, 255, 0.2), inset 0 0 25px rgba(255, 255, 255, 0.2);\n",
-        "}\n",
-        "\n",
-        ".info.showed {\n",
-        "    opacity: 1;\n",
-        "    pointer-events: auto;\n",
-        "}\n",
-        "\n",
-        ".info:hover + .popup {\n",
-        "    top: 35px;\n",
-        "    opacity: 1;\n",
-        "    pointer-events: initial;\n",
-        "    transform: rotate(0deg);\n",
-        "}\n",
-        "\n",
-        "\n",
-        "/* Animation of elements */\n",
-        "\n",
-        ".container,\n",
-        ".button_save {\n",
-        "    animation-name: showedWidgets;\n",
-        "    animation-duration: 1s;\n",
-        "    animation-fill-mode: forwards;\n",
-        "}\n",
-        "\n",
-        ".container.hide,\n",
-        ".button_save.hide {\n",
-        "    animation-name: hideWidgets;\n",
-        "    animation-duration: 0.5s;\n",
-        "    animation-fill-mode: forwards;\n",
-        "}\n",
-        "\n",
-        "@keyframes showedWidgets {\n",
-        "    0% {\n",
-        "        transform: translate3d(-65%, 15%, 0) scale(0) rotate(15deg);\n",
-        "        filter: blur(25px) grayscale(1) brightness(0.3);\n",
-        "        opacity: 0;\n",
-        "    }\n",
-        "    100% {\n",
-        "        transform: translate3d(0, 0, 0) scale(1) rotate(0deg);\n",
-        "        filter: blur(0) grayscale(0) brightness(1);\n",
-        "        opacity: 1;\n",
-        "    }\n",
-        "}\n",
-        "\n",
-        "@keyframes hideWidgets {\n",
-        "    0% {\n",
-        "        transform: translate3d(0, 0, 0) scale(1) rotate3d(1, 0, 0, 0deg);\n",
-        "        filter: blur(0) grayscale(0) brightness(1);\n",
-        "        opacity: 1;\n",
-        "    }\n",
-        "    100% {\n",
-        "        transform: translate3d(0, 5%, 0) scale(0.9) rotate3d(1, 0, 0, 90deg);\n",
-        "        filter: blur(15px) grayscale(1) brightness(0.5);\n",
-        "        opacity: 0;\n",
-        "    }\n",
-        "}\n",
-        "</style>\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",
@@ -451,20 +123,58 @@
         "}\n",
         "</script>\n",
         "'''\n",
+        "display(HTML(JS))\n",
         "\n",
-        "display(HTML(CSS))\n",
-        "# ==================== CSS JS ====================\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",
-        "# ==================== WIDGETS ====================\n",
-        "# --- global widgets ---\n",
-        "style = {'description_width': 'initial'}\n",
-        "layout = widgets.Layout(min_width='1047px')\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",
-        "HR = widgets.HTML('<hr>')\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 = widgets.HTML('<div class=\"header\">Выбор Модели<div>')\n",
+        "model_header = factory.create_header('Выбор Модели')\n",
         "model_options = ['none',\n",
         "                 '1.Anime (by XpucT) + INP',\n",
         "                 '2.BluMix [Anime] [V7] + INP',\n",
@@ -475,37 +185,37 @@
         "                 '7.Deliberate [Realism] [V6] + INP',\n",
         "                 '8.Meina-Mix [Anime] [V11] + INP',\n",
         "                 '9.Mix-Pro [Anime] [V4] + INP']\n",
-        "# ---\n",
-        "Model_widget = widgets.Dropdown(options=model_options, value='4.Counterfeit [Anime] [V3] + INP', description='Модель:', style=style, layout=layout)\n",
-        "Model_Num_widget = widgets.Text(description='Номер Модели:', placeholder='Введите номера моделей для скачивания через запятую/пробел.', style=style, layout=layout)\n",
-        "Inpainting_Model_widget = widgets.Checkbox(value=False, description='Inpainting Модели', style=style)\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 = widgets.VBox([model_header, Model_widget, Model_Num_widget, Inpainting_Model_widget]).add_class(\"container\").add_class(\"image_1\")\n",
-        "display(all_model_box)\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 = widgets.HTML('<div class=\"header\">Выбор VAE</div>')\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 = widgets.Dropdown(options=vae_options, value='3.Blessed2.vae', description='Vae:', style=style, layout=layout)\n",
-        "Vae_Num_widget = widgets.Text(description='Номер Vae:', placeholder='Введите номера vae для скачивания через запятую/пробел.', style=style, layout=layout)\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 = widgets.VBox([vae_header, Vae_widget, Vae_Num_widget]).add_class(\"container\").add_class(\"image_2\")\n",
-        "display(all_vae_box)\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 = widgets.HTML('<div class=\"header\">Дополнительно</div>')\n",
-        "latest_webui_widget = widgets.Checkbox(value=True, description='Обновить WebUI', style=style)\n",
-        "latest_exstensions_widget = widgets.Checkbox(value=True, description='Обновить Расширения', style=style)\n",
-        "change_webui_widget = widgets.Dropdown(options=['A1111', 'Forge'], value='A1111', description='Изменить WebUI:', style=style)\n",
-        "detailed_download_widget = widgets.Dropdown(options=['off', 'on'], value='off', description='Подробная Загрузка:', style=style)\n",
-        "latest_changes_widget = HBox([latest_webui_widget, latest_exstensions_widget, change_webui_widget, detailed_download_widget], layout=widgets.Layout(justify_content='space-between'))\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",
@@ -513,33 +223,45 @@
         "                      '8.scribble', '9.segmentation',\n",
         "                      '10.shuffle', '11.tile',\n",
         "                      '12.inpaint', '13.instruct_p2p']\n",
-        "# ---\n",
-        "controlnet_widget = widgets.Dropdown(options=controlnet_options, value='none', description='ControlNet:', style=style, layout=layout)\n",
-        "controlnet_Num_widget = widgets.Text(description='Номер ControlNet:', placeholder='Введите номера моделей ControlNet для скачивания через запятую/пробел.', style=style, layout=layout)\n",
-        "commit_hash_widget = widgets.Text(description='Commit Hash:', style=style, layout=layout)\n",
-        "optional_huggingface_token_widget = widgets.Text(description='Токен HuggingFace:', style=style, layout=layout)\n",
-        "ngrok_token_widget = widgets.Text(description='Токен Ngrok:', style=style, layout=widgets.Layout(width='1047px'))\n",
-        "ngrock_button = widgets.HTML('<a href=\"https://dashboard.ngrok.com/get-started/your-authtoken\" target=\"_blank\">Получить Ngrok Токен</a>').add_class(\"button_ngrok\")\n",
-        "ngrok_widget = widgets.HBox([ngrok_token_widget, ngrock_button], style=style, layout=layout)\n",
-        "zrok_token_widget = widgets.Text(description='Токен Zrok:', style=style, layout=widgets.Layout(width='1047px'))\n",
-        "zrok_button = widgets.HTML('<a href=\"https://colab.research.google.com/drive/1d2sjWDJi_GYBUavrHSuQyHTDuLy36WpU\" target=\"_blank\">Зарегать Zrok Токен</a>').add_class(\"button_ngrok\")\n",
-        "zrok_widget = widgets.HBox([zrok_token_widget, zrok_button], style=style, layout=layout)\n",
-        "# ---\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 = widgets.Text(description='Аргументы:', value=commandline_arguments_options, style=style, layout=layout)\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",
-        "''' Display Additional'''\n",
-        "additional_widget_list = [additional_header, latest_changes_widget, HR, controlnet_widget, controlnet_Num_widget, commit_hash_widget, optional_huggingface_token_widget, ngrok_widget, zrok_widget, HR, commandline_arguments_widget]\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",
-        "    latest_changes_widget.children = [widget for widget in latest_changes_widget.children if widget != change_webui_widget]\n",
-        "# ```\n",
-        "all_additional_box = widgets.VBox(additional_widget_list).add_class(\"container\").add_class(\"image_3\")\n",
-        "display(all_additional_box)\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 = widgets.HTML('''\n",
+        "custom_download_header_popup = factory.create_html('''\n",
         "<style>\n",
         "/* Term Colors */\n",
         ".sample_label {color: #dbafff;}\n",
@@ -564,39 +286,40 @@
         "    </div>\n",
         "</div>\n",
         "''')\n",
-        "# ---\n",
-        "Model_url_widget = widgets.Text(description='Model:', style=style, layout=layout)\n",
-        "Vae_url_widget = widgets.Text(description='Vae:', style=style, layout=layout)\n",
-        "LoRA_url_widget = widgets.Text(description='LoRa:', style=style, layout=layout)\n",
-        "Embedding_url_widget = widgets.Text(description='Embedding:', style=style, layout=layout)\n",
-        "Extensions_url_widget = widgets.Text(description='Extensions:', style=style, layout=layout)\n",
-        "custom_file_urls_widget = widgets.Text(description='Файл (txt):', style=style, layout=layout)\n",
-        "\n",
-        "''' Display CustomDl'''\n",
-        "all_custom_box = widgets.VBox([\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",
-        "    ]).add_class(\"container\").add_class(\"image_4\").add_class(\"container_custom_downlad\")\n",
-        "display(all_custom_box)\n",
+        "], class_names=[\"container\", \"image_4\", \"container_custom_downlad\"])\n",
+        "factory.display(all_custom_box)\n",
         "\n",
         "# --- Save Button ---\n",
-        "save_button = widgets.Button(description='Сохранить').add_class(\"button_save\")\n",
-        "display(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',\n",
-        "    'latest_webui', 'latest_exstensions', 'change_webui', 'detailed_download',\n",
-        "    'controlnet', 'controlnet_Num', 'commit_hash', 'optional_huggingface_token',\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', 'Extensions_url', 'custom_file_urls'\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",
+        "    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",
@@ -604,28 +327,23 @@
         "    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 save_data(button):\n",
-        "    save_settings()\n",
+        "            for key in SETTINGS_KEYS:\n",
+        "                globals()[f\"{key}_widget\"].value = settings.get(key, \"\")\n",
         "\n",
-        "    # --- uhh - hide... ---\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",
-        "\n",
         "    widgets.Widget.close_all()\n",
         "\n",
-        "settings = load_settings()\n",
+        "def save_data(button):\n",
+        "    save_settings()\n",
+        "    hide_widgets()\n",
+        "\n",
+        "load_settings()\n",
         "save_button.on_click(save_data)"
-      ],
-      "metadata": {
-        "id": "2lJmbqrs3Mu8"
-      },
-      "execution_count": null,
-      "outputs": []
+      ]
     }
   ]
 }

files_cells/python/en/auto_cleaner_en.py

@@ -0,0 +1,138 @@
+##~ AutoCleaner V3.6 CODE | BY: ANXETY   ~##
+
+import os
+import time
+import ipywidgets as widgets
+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"
+
+
+# ==================== CSS ====================
+# Main CSS
+css_file_path = f"{root_path}/CSS/auto_cleaner.css"
+with open(css_file_path , "r") as f:
+    CSS_AC = f.read()
+display(HTML(f"<style>{CSS_AC}</style>"))
+
+
+# ================ AutoCleaner function ================
+directories = {
+    "Images": f"{webui_path}/output",
+    "Models": f"{webui_path}/models/Stable-diffusion/",
+    "Vae": f"{webui_path}/models/VAE/",
+    "LoRa": f"{webui_path}/models/Lora/",
+    "ControlNet Models": f"{webui_path}/models/ControlNet/"
+}
+
+""" functions """
+def clean_directory(directory):
+    deleted_files = 0
+    image_dir = directories['Images']
+
+    for root, dirs, files in os.walk(directory):
+        for file in files:
+            file_path = os.path.join(root, file)
+
+            if file.endswith(".txt"):
+                continue
+            if file.endswith((".safetensors", ".pt")) or root == image_dir: # fix for image counter
+                deleted_files += 1
+
+            os.remove(file_path)
+    return deleted_files
+
+def update_memory_info():
+    disk_space = psutil.disk_usage(os.getcwd())
+    total = disk_space.total / (1024 ** 3)
+    used = disk_space.used / (1024 ** 3)
+    free = disk_space.free / (1024 ** 3)
+
+    storage_info.value = f'''
+    <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>
+    '''
+
+def on_execute_button_press(button):
+    selected_cleaners = auto_cleaner_widget.value
+    deleted_files_dict = {}
+
+    for option in selected_cleaners:
+        if option in directories:
+            deleted_files_dict[option] = clean_directory(directories[option])
+
+    output.clear_output()
+
+    with output:
+        for message in generate_messages(deleted_files_dict):
+            message_widget = HTML(f'<p class="output_message_AC">{message}</p>')
+            display(message_widget)
+
+    update_memory_info()
+
+def on_clear_button_press(button):
+    container.add_class("hide")
+    time.sleep(0.5)
+    widgets.Widget.close_all()
+
+def generate_messages(deleted_files_dict):
+    messages = []
+    word_variants = {
+        "Images": "Images",
+        "Models": "Models",
+        "Vae": "Vae",
+        "LoRa": "LoRa",
+        "ControlNet Models": "ControlNet Models"
+    }
+    for key, value in deleted_files_dict.items():
+        object_word = word_variants.get(key)
+        messages.append(f"Deleted {value} {object_word}")
+    return messages
+# ================ AutoCleaner function ================
+
+
+# --- storage memory ---
+import psutil
+disk_space = psutil.disk_usage(os.getcwd())
+total = disk_space.total / (1024 ** 3)
+used = disk_space.used / (1024 ** 3)
+free = disk_space.free / (1024 ** 3)
+
+
+# UI Code
+AutoCleaner_options = AutoCleaner_options = list(directories.keys())
+instruction_label = widgets.HTML('''
+<span class="instruction_AC">Use <span style="color: #B2B2B2;">ctrl</span> or <span style="color: #B2B2B2;">shift</span> for multiple selections.</span>
+''')
+auto_cleaner_widget = widgets.SelectMultiple(options=AutoCleaner_options, layout=widgets.Layout(width='auto')).add_class("custom-select-multiple_AC")
+output = widgets.Output().add_class("output_AC")
+# ---
+execute_button = Button(description='Execute Cleaning').add_class("button_execute_AC").add_class("button_AC")
+execute_button.on_click(on_execute_button_press)
+clear_button = Button(description='Hide Widget').add_class("button_clear_AC").add_class("button_AC")
+clear_button.on_click(on_clear_button_press)
+# ---
+storage_info = widgets.HTML(f'''
+<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>
+''')
+# ---
+buttons = widgets.HBox([execute_button, clear_button])
+lower_information_panel = widgets.HBox([buttons, storage_info]).add_class("lower_information_panel_AC")
+
+container = VBox([instruction_label, widgets.HTML('<hr>'), auto_cleaner_widget, output, widgets.HTML('<hr>'), lower_information_panel]).add_class("container_AC")
+
+display(container)
+

files_cells/python/en/downloading_en.py

@@ -15,21 +15,19 @@ from IPython.display import clear_output
 from urllib.parse import urlparse, parse_qs
 
 
-#  ================= DETECT ENV =================
+# ================= 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"
-#  ----------------------------------------------
 
 
 # ================ LIBRARIES V2 ================
@@ -80,12 +78,6 @@ if not os.path.exists(flag_file):
     time.sleep(2)
     clear_output()
 
-## dl special files
-with capture.capture_output() as cap:
-    get_ipython().system('mkdir -p {root_path}/file_cell/special')
-    get_ipython().system('wget https://huggingface.co/NagisaNao/fast_repo/resolve/main/special/dl_display_results.py -O {root_path}/file_cell/special/dl_display_results.py')
-del cap
-
 
 # ================= loading settings V4 =================
 def load_settings(path):
@@ -96,16 +88,17 @@ def load_settings(path):
 
 settings = load_settings(f'{root_path}/settings.json')
 
-variables = [
-    'Model', 'Model_Num', 'Inpainting_Model',
-    'Vae', 'Vae_Num',
-    'latest_webui', 'latest_exstensions', 'change_webui', 'detailed_download',
-    'controlnet', 'controlnet_Num', 'commit_hash', 'optional_huggingface_token',
+VARIABLES = [
+    'model', 'model_num', 'inpainting_model',
+    'vae', 'vae_num', 'latest_webui', 'latest_exstensions',
+    'change_webui', 'detailed_download', 'controlnet',
+    'controlnet_num', 'commit_hash', 'huggingface_token',
     'ngrok_token', 'zrok_token', 'commandline_arguments',
-    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url', 'Extensions_url', 'custom_file_urls'
+    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url',
+    'Extensions_url', 'custom_file_urls'
 ]
 
-locals().update({key: settings.get(key) for key in variables})
+locals().update({key: settings.get(key) for key in VARIABLES})
 
 
 # ================= OTHER =================
@@ -241,17 +234,7 @@ model_list = {
     ]
 }
 
-# 1-4 (fp16/cleaned)
 vae_list = {
-    ## vae broke - the author's an asshole
-    # "1.Anime.vae": [
-    #     {"url": "https://civitai.com/api/download/models/131654", "name": "Anime.vae.safetensors"},
-    #     {"url": "https://civitai.com/api/download/models/131658", "name": "vae-ft-mse.vae.safetensors"}
-    # ],
-    # "2.Anything.vae": [{"url": "https://civitai.com/api/download/models/131656", "name": "Anything.vae.safetensors"}],
-    # "3.Blessed2.vae": [{"url": "https://civitai.com/api/download/models/142467", "name": "Blessed2.vae.safetensors"}],
-    # "4.ClearVae.vae": [{"url": "https://civitai.com/api/download/models/133362", "name": "ClearVae_23.vae.safetensors"}],
-
     "1.Anime.vae": [{"url": "https://civitai.com/api/download/models/311162", "name": "vae-ft-mse-840000-ema-pruned.vae.safetensors"}],
     "2.Anything.vae": [{"url": "https://civitai.com/api/download/models/119279", "name": "Anything.vae.safetensors"}],
     "3.Blessed2.vae": [{"url": "https://huggingface.co/NoCrypt/blessed_vae/resolve/main/blessed2.vae.pt", "name": "Blessed2.vae.safetensors"}],
@@ -329,26 +312,47 @@ 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 = optional_huggingface_token if optional_huggingface_token else "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"
+hf_token = huggingface_token if huggingface_token else "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"
 user_header = f"\"Authorization: Bearer {hf_token}\""
 
+''' Formatted Info Output '''
+
+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"
+
+def format_output(url, dst_dir, file_name):
+    info = f"[{file_name.split('.')[0]}]"
+    info = center_text(info)
+
+    print(f"\n\033[32m{'---'*20}]{info}[{'---'*20}")
+    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 Image Preview | CivitAi '''
 
-def get_data_from_api(model_id): # get model data
+def get_data_from_api(model_id):
+    """Fetch model data from the API"""
     endpoint_url = f"https://civitai.com/api/v1/model-versions/{model_id}"
     headers = {"Content-Type": "application/json"}
     try:
         response = requests.get(endpoint_url, headers=headers)
-        if response.status_code == 200:
-            return response.json()
+        response.raise_for_status()
+        return response.json()
     except requests.exceptions.RequestException as e:
         print(f"An error occurred: {e}")
         return None
 
 def extract_model_info(data, url):
+    """Extract model information based on URL"""
     if 'type=' in url:
         model_type = parse_qs(urlparse(url).query).get('type', [''])[0]
         model_name = data['files'][1]['name']
@@ -365,59 +369,69 @@ def extract_model_info(data, url):
     return model_type, model_name, image_url
 
 def gen_preview_filename(model_name, image_url):
+    """Generate a preview filename"""
     name = model_name.split('.')
     img_exts = image_url.split('.')
-    return f"{name[0]}.preview.{img_exts[-1]}" # assigning the original image format
+    return f"{name[0]}.preview.{img_exts[-1]}"
 
 ''' main download code '''
 
 def handle_manual(url):
-    original_url = url
-    url = url.split(':', 1)[1]
-    file_name = re.search(r'\[(.*?)\]', url)
-    file_name = file_name.group(1) if file_name else None
-    if file_name:
-        url = re.sub(r'\[.*?\]', '', url)
+    url_parts = url.split(':', 1)
+    prefix = url_parts[0]
+    path = url_parts[1]
 
-    for prefix, dir in prefixes.items():
-        if original_url.startswith(f"{prefix}:"):
-            if prefix != "extension":
-                manual_download(url, dir, file_name=file_name)
-            else:
-                extension_repo.append((url, file_name))
+    file_name_match = re.search(r'\[(.*?)\]', path)
+    file_name = file_name_match.group(1) if file_name_match else None
+    if file_name:
+        path = re.sub(r'\[.*?\]', '', path)
+
+    if prefix in prefixes:
+        dir = prefixes[prefix]
+        if prefix != "extension":
+            try:
+                manual_download(path, dir, file_name=file_name)
+            except Exception as e:
+                print(f"Error downloading file: {e}")
+        else:
+            extension_repo.append((path, file_name))
 
 def manual_download(url, dst_dir, file_name):
     aria2_args = '--optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -j5 -x16 -s16 -k1M -c'
     basename = url.split("/")[-1] if file_name is None else file_name
     header_option = f"--header={user_header}"
 
-    # === CivitAi API ===
+    # ==== CivitAi API+ ====
     support_types = ('Checkpoint', 'Model', 'TextualInversion', 'LORA') # for dl preview image
     civitai_token = "62c0c5956b2f9defbd844d754000180b"
 
     if 'civitai' in url:
-        url = f"{url}{'&' if '?' in url else '?'}token={civitai_token}" # some authors are pussies and they need API token...
-        model_id = url.split('/')[-1]
-        clean_url = url.split('?')[0]
+        url = f"{url}{'&' if '?' in url else '?'}token={civitai_token}"
+        model_id = url.split('/')[-1].split('?')[0]
+        clean_url = re.sub(r'[?&]token=[^&]*', '', url) # hide token
 
         data = get_data_from_api(model_id)
         if data:
             model_type, model_name, image_url = extract_model_info(data, url)
-            if model_name and image_url:
-                image_file_name = gen_preview_filename(model_name if not file_name else file_name, image_url)
-                if any(types in model_type for types in support_types):
-                    with capture.capture_output() as cap: # clear shit
-                        get_ipython().system('aria2c {aria2_args} -d {dst_dir} -o {image_file_name} {image_url}')
+
+            if any(t in model_type for t in support_types):
+                if model_name and image_url:
+                    image_file_name = gen_preview_filename(model_name if not file_name else file_name, image_url)
+                    with capture.capture_output() as cap:
+                        get_ipython().system("aria2c {aria2_args} -d {dst_dir} -o {image_file_name} '{image_url}'")
                     del cap
-            file_name = file_name or model_name # assigns the original file name if not specified initially
+            file_name = file_name or model_name
+    else:
+        clean_url = url
+
+    """ Formatted info output """
+    model_name_or_basename = file_name if not 'huggingface' in url else basename
+    format_output(clean_url or url, dst_dir, model_name_or_basename)
 
-    """ information output """
-    # -- wrold's best print info --
-    print(f"\n\033[32m{'---'*45}\n\033[33mURL: \033[34m{clean_url if 'civitai' in url else url}\n\033[33mSAVE DIR: \033[34m{dst_dir}\n\033[33mFILE NAME: \033[34m{file_name if not 'huggingface' in url else basename}\033[0m")
     print("\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 any(types in model_type for types in support_types) and (locals().get('image_file_name') or ''):
+    if 'civitai' in url and data and any(t in model_type for t in support_types) and (locals().get('image_file_name') or ''):
         print(f"\033[32m[Preview DL]:\033[0m {image_file_name} - {image_url}\n")
-    # ===================
+    # =====================
 
     # -- GDrive --
     if 'drive.google' in url:
@@ -461,9 +475,8 @@ def download(url):
 
     unpucking_zip_files()
 
-## unpucking zip files
+# unpucking zip files
 def unpucking_zip_files():
-    # directories - above
     for directory in directories:
         for root, dirs, files in os.walk(directory):
             for file in files:
@@ -476,8 +489,6 @@ def unpucking_zip_files():
 
 ''' submodels - added urls '''
 
-submodels = []
-
 def add_submodels(selection, num_selection, model_dict, dst_dir):
     if selection == "none":
         return []
@@ -489,33 +500,32 @@ def add_submodels(selection, num_selection, model_dict, dst_dir):
     else:
         selected_models = model_dict[selection]
         selected_nums = map(int, num_selection.replace(',', '').split())
-
         for num in selected_nums:
             if 1 <= num <= len(model_dict):
                 name = list(model_dict)[num - 1]
                 selected_models.extend(model_dict[name])
 
     unique_models = list({model['name']: model for model in selected_models}.values())
-
     for model in unique_models:
         model['dst_dir'] = dst_dir
 
     return unique_models
 
-submodels += add_submodels(Model, Model_Num, model_list, models_dir)                    # model
-submodels += add_submodels(Vae, Vae_Num, vae_list, vaes_dir)                            # vae
-submodels += add_submodels(controlnet, controlnet_Num, controlnet_list, control_dir)    # controlnet
+def handle_submodels(selection, num_selection, model_dict, dst_dir, url):
+    submodels = add_submodels(selection, num_selection, model_dict, dst_dir)
+    for submodel in submodels:
+        if not inpainting_model and "inpainting" in submodel['name']:
+            continue
+        url += f"{submodel['url']} {submodel['dst_dir']} {submodel['name']}, "
+    return url
 
-for submodel in submodels:
-    if not Inpainting_Model and "inpainting" in submodel['name']:
-        continue
-    url += f"{submodel['url']} {submodel['dst_dir']} {submodel['name']}, "
+url = handle_submodels(model, model_num, model_list, models_dir, url)
+url = handle_submodels(vae, vae_num, vae_list, vaes_dir, url)
+url = handle_submodels(controlnet, controlnet_num, controlnet_list, control_dir, url)
 
 ''' file.txt - added urls '''
 
-unique_urls = []
-
-def process_file_download(file_url):
+def process_file_download(file_url, prefixes, unique_urls):
     files_urls = ""
 
     if file_url.startswith("http"):
@@ -529,19 +539,23 @@ def process_file_download(file_url):
 
     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()))
 
         urls = [url.split('#')[0].strip() for url in line.split(',')]  # filter urls
         for url in urls:
-            if url.startswith("http") and url not in unique_urls:
+            filter_url = url.split('[')[0] # same url filter
+
+            if url.startswith("http") and filter_url not in unique_urls:
                 files_urls += f"{current_tag}:{url}, "
-                unique_urls.append(url)
+                unique_urls.add(filter_url)
 
     return files_urls
 
-# fix all possible errors/options and function call
 file_urls = ""
+unique_urls = set()
+
 if custom_file_urls:
     for custom_file_url in custom_file_urls.replace(',', '').split():
         if not custom_file_url.endswith('.txt'):
@@ -551,7 +565,7 @@ if custom_file_urls:
                 custom_file_url = f'{root_path}/{custom_file_url}'
 
         try:
-            file_urls += process_file_download(custom_file_url)
+            file_urls += process_file_download(custom_file_url, prefixes, unique_urls)
         except FileNotFoundError:
             pass
 
@@ -591,38 +605,6 @@ if len(extension_repo) > 0:
 ## List Models and stuff V2
 if detailed_download == "off":
     print("\n\n\033[33mIf you don't see any downloaded files, enable the 'Detailed Downloads' feature in the widget.")
-    get_ipython().run_line_magic('run', '{root_path}/file_cell/special/dl_display_results.py') # display widgets result
 
-else:
-    if any(not file.endswith('.txt') for file in os.listdir(models_dir)):
-        print("\n\033[33m➤ Models\033[0m")
-        get_ipython().system("find {models_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
-    if any(not file.endswith('.txt') for file in os.listdir(vaes_dir)):
-        print("\n\033[33m➤ VAEs\033[0m")
-        get_ipython().system("find {vaes_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
-    if any(not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file)) for file in os.listdir(embeddings_dir)):
-        print("\n\033[33m➤ Embeddings\033[0m")
-        get_ipython().system("find {embeddings_dir}/ -mindepth 1 -maxdepth 1 \\( -name '*.pt' -or -name '*.safetensors' \\) -printf '%f\\n'")
-    if any(not file.endswith('.txt') for file in os.listdir(loras_dir)):
-        print("\n\033[33m➤ LoRAs\033[0m")
-        get_ipython().system("find {loras_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
-    print(f"\n\033[33m➤ Extensions\033[0m")
-    get_ipython().system("find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\n'")
-    if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir)):
-        print("\n\033[33m➤ ControlNet\033[0m")
-        get_ipython().system("find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\([^_]*\\)_fp16\\.safetensors$/\\1/'")
-
-
-# === OTHER ===
-# Downlaod discord tags UmiWildcards
-files_umi = [
-    "https://huggingface.co/NagisaNao/fast_repo/resolve/main/extensions/UmiWildacrd/discord/200_pan_gen.txt",
-    "https://huggingface.co/NagisaNao/fast_repo/resolve/main/extensions/UmiWildacrd/discord/150_bra_gen.txt"
-]
-save_dir_path = f"{webui_path}/extensions/Umi-AI-Wildcards/wildcards/discord"
-
-with capture.capture_output() as cap:
-    for file in files_umi:
-        get_ipython().system('aria2c --optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -j5 -x16 -s16 -k1M -c -d {save_dir_path} {file}')
-del cap
+get_ipython().run_line_magic('run', '{root_path}/file_cell/special/dl_display_results.py # display widgets result')
 

files_cells/python/en/widgets_en.py

@@ -8,27 +8,24 @@ from ipywidgets import widgets, Layout, Label, Button, VBox, HBox
 from IPython.display import display, HTML, Javascript, clear_output
 
 
-#  ================= DETECT ENV =================
+# ================= 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}')
-#  ----------------------------------------------
 
 
 # ==================== CSS JS ====================
-# custom background images V1.5
+##~ custom background images V1.5 ~##
 import argparse
 parser = argparse.ArgumentParser(description='This script processes an background image.')
 parser.add_argument('-i', '--image', type=str, help='URL of the image to process', metavar='')
@@ -40,8 +37,9 @@ parser.add_argument('-s', '--scale', type=int, help='Scale image in %%', metavar
 parser.add_argument('-m', '--mode', action='store_true', help='Removes repetitive image tiles')
 parser.add_argument('-t', '--transparent', action='store_true', help='Makes input/selection fields 35%% more transparent')
 parser.add_argument('-bf', '--blur-fields', type=str, help='Background blur level for input/selection fields', metavar='', default=2)
+
 args = parser.parse_args()
-"""---"""
+
 url_img = args.image
 opacity_img = args.opacity
 blur_img = args.blur
@@ -59,7 +57,6 @@ t_bg_alpha = "1" if not args.transparent else "0.65"
 
 """ mode img - repeats """
 mode_img = "repeat" if not args.mode else "no-repeat"
-## ---
 
 container_background = f'''
 <style>
@@ -82,345 +79,15 @@ container_background = f'''
 '''
 
 display(HTML(container_background))
-# ---
-
-CSS = '''
-<style>
-/* General Styles */
-.header {
-    font-family: cursive;
-    font-size: 20px;
-    font-weight: bold;
-    color: #ff8cee;
-    margin-bottom: 15px;
-    user-select: none;
-    cursor: default;
-    display: inline-block;
-}
-
-hr {
-    border-color: grey;
-    background-color: grey;
-    opacity: 0.25;
-}
-
-a {
-  text-decoration: none;
-  color: inherit;
-}
-
-
-/* Container style */
-
-.container {
-    position: relative;
-    background-color: #232323;
-    width: 1080px;
-    padding: 10px 15px;
-    border-radius: 15px;
-    box-shadow: 0 0 50px rgba(0, 0, 0, 0.3);
-    margin-bottom: 5px;
-    overflow: hidden;
-}
-
-.container::after {
-    position: absolute;
-    top: 5px;
-    right: 10px;
-    content: "ANXETY";
-    font-weight: bold;
-    font-size: 24px;
-    color: rgba(0, 0, 0, 0.2);
-}
-
-/* background img */
-.container::before {
-    content: "";
-    position: absolute;
-    top: 0;
-    left: 0;
-    right: 0;
-    bottom: 0;
-    background-image: var(--img_background);
-    background-size: var(--img_scale);
-    background-repeat: var(--img_mode);
-    opacity: var(--img_opacity);
-    mix-blend-mode: screen;
-    pointer-events: none;
-    filter: blur(var(--img_blur));
-    z-index: -1;
-}
-
-.image_1::before {
-    background-position: var(--image_x) calc(-120px - var(--image_y));
-}
-.image_2::before {
-    background-position: var(--image_x) calc(-290px - var(--image_y));
-}
-.image_3::before {
-    background-position: var(--image_x) calc(-430px - var(--image_y));
-}
-.image_4::before {
-    background-position: var(--image_x) calc(var(--img_height_dif) - var(--image_y));
-}
-
-.container_custom_downlad {
-    height: 55px;
-    transition: all 0.5s;
-}
-
-.container_custom_downlad.expanded {
-    height: 270px;
-}
-
-
-/* Element text style */
-
-.widget-html,
-.widget-button,
-.widget-text label,
-.widget-checkbox label,
-.widget-dropdown label,
-.widget-dropdown select,
-.widget-text input[type="text"] {
-    font-family: cursive;
-    font-size: 14px;
-    color: white !important;
-    user-select: none;
-}
-
-.widget-text input[type="text"]::placeholder {
-    color: grey;
-}
-
-
-/* Input field styles */
-
-.widget-dropdown select,
-.widget-text input[type="text"] {
-    height: 30px;
-    background-color: var(--bg-field-color);
-    border: 1px solid #262626;
-    border-radius: 10px;
-    box-shadow: inset 0 0 5px rgba(0, 0, 0, 0.5);
-    transition: all 0.3s ease-in-out;
-    backdrop-filter: blur(var(--bg-field-blur-level));
-}
-
-.widget-dropdown select:focus,
-.widget-text input[type="text"]:focus {
-    border-color: #006ee5;
-}
-
-.widget-dropdown select:hover,
-.widget-text input[type="text"]:hover {
-    transform: scale(1.003);
-    background-color: var(--bg-field-color-hover);
-    box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.5);
-}
-
-.widget-dropdown option {
-    background-color: #1c1c1c;
-}
-
-
-/* Slider Checkbox style */
-
-.widget-checkbox input[type="checkbox"] {
-    appearance: none;
-    position: relative;
-    top: 4px; /* Why is he taller?! */
-    width: 40px;
-    height: 20px;
-    border: none;
-    border-radius: 10px;
-    background-color: #20b2aa;
-    cursor: pointer;
-    box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.3);
-    transition: background-color 0.3s ease;
-}
-.widget-checkbox input[type="checkbox"]:checked {
-    background-color: #2196F3;
-}
-
-.widget-checkbox input[type="checkbox"]:before {
-    content: '';
-    position: absolute;
-    top: 50%;
-    left: 3px;
-    width: 16px;
-    height: 16px;
-    border-radius: inherit;
-    background-color: white;
-    box-shadow: 0 0 8px rgba(0, 0, 0, 0.3);
-    transform: translateY(-50%);
-    transition: left 0.3s ease;
-}
-.widget-checkbox input[type="checkbox"]:checked:before {
-    left: 21px;
-}
-
-
-/* Button styles */
-
-.button_save {
-    font-size: 15px;
-    font-weight: bold;
-    width: 120px;
-    height: 35px;
-    border-radius: 15px;
-    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);
-    background-size: 200% 200%;
-    background-position: left bottom;
-    transition: background 0.5s ease-in-out, transform 0.3s ease;
-}
-
-.button_save:hover {
-    cursor: pointer;
-    background-image: radial-gradient(circle at top left, purple 10%, #93ac47 90%);
-    background-size: 200% 200%;
-    background-position: right bottom;
-    transform: translateY(1px);
-}
 
-.button_ngrok {
-    font-size: 12px;
-    height: 30px;
-    border-radius: 10px;
-    padding: 1px 12px;
-    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);
-    background-size: 200% 200%;
-    background-position: left bottom;
-    transition: background 0.5s ease-in-out, transform 0.3s ease;
-    white-space: nowrap;
-}
-
-.button_ngrok:hover  {
-    cursor: pointer;
-    background-image: radial-gradient(circle at top left, purple 10%, #1D94BB 90%);
-    background-size: 200% 200%;
-    background-position: right bottom;
-    transform: translateY(1px);
-}
-
-.button_save:active,
-.button_ngrok:active {
-    filter: brightness(0.75) !important;
-}
-
-/* Removes ugly stroke from widget buttons. */
-.jupyter-widgets.lm-Widget:focus {
-    outline: none;
-}
-
-
-/* Popup style of `INFO` window */
-
-.info {
-    position: absolute;
-    top: -5px;
-    right: 95px;
-    color: grey;
-    font-family: cursive;
-    font-size: 14px;
-    font-weight: normal;
-    user-select: none;
-    pointer-events: none;
-    opacity: 0;
-    transition: opacity 0.3s ease-in-out;
-    display: inline-block;
-}
-
-.popup {
-    position: absolute;
-    top: 120px;
-    z-index: 999;
-    width: auto;
-    padding: 10px;
-    text-align: center;
-    background-color: rgba(255, 255, 255, 0.05);
-    backdrop-filter: blur(20px);
-    border: 1px solid rgba(255, 255, 255, 0.45);
-    border-radius: 8px;
-    box-shadow: 0 0 50px rgba(0, 0, 0, 0.5);
-    opacity: 0;
-    color: #fff;
-    font-size: 16px;
-    font-family: cursive;
-    user-select: none;
-    cursor: default;
-    pointer-events: none;
-    transform: rotate(-5deg);
-    transition: top 0.3s ease-in-out, opacity 0.3s ease-in-out, transform 0.3s ease-in-out;
-}
-
-.sample {
-    display: inline-block;
-    margin-top: 25px;
-    padding: 10px 100px;
-    background-color: rgba(255, 255, 255, 0.2);
-    color: #c6e2ff;
-    border: 2px solid rgba(255, 255, 255, 0.2);
-    border-radius: 8px;
-    box-shadow: 0 0 10px rgba(255, 255, 255, 0.2), inset 0 0 25px rgba(255, 255, 255, 0.2);
-}
-
-.info.showed {
-    opacity: 1;
-    pointer-events: auto;
-}
-
-.info:hover + .popup {
-    top: 35px;
-    opacity: 1;
-    pointer-events: initial;
-    transform: rotate(0deg);
-}
-
-
-/* Animation of elements */
-
-.container,
-.button_save {
-    animation-name: showedWidgets;
-    animation-duration: 1s;
-    animation-fill-mode: forwards;
-}
-
-.container.hide,
-.button_save.hide {
-    animation-name: hideWidgets;
-    animation-duration: 0.5s;
-    animation-fill-mode: forwards;
-}
-
-@keyframes showedWidgets {
-    0% {
-        transform: translate3d(-65%, 15%, 0) scale(0) rotate(15deg);
-        filter: blur(25px) grayscale(1) brightness(0.3);
-        opacity: 0;
-    }
-    100% {
-        transform: translate3d(0, 0, 0) scale(1) rotate(0deg);
-        filter: blur(0) grayscale(0) brightness(1);
-        opacity: 1;
-    }
-}
-
-@keyframes hideWidgets {
-    0% {
-        transform: translate3d(0, 0, 0) scale(1) rotate3d(1, 0, 0, 0deg);
-        filter: blur(0) grayscale(0) brightness(1);
-        opacity: 1;
-    }
-    100% {
-        transform: translate3d(0, 5%, 0) scale(0.9) rotate3d(1, 0, 0, 90deg);
-        filter: blur(15px) grayscale(1) brightness(0.5);
-        opacity: 0;
-    }
-}
-</style>
+# Main CSS
+css_file_path = f"{root_path}/CSS/main_widgets.css"
+with open(css_file_path , "r") as f:
+    CSS = f.read()
+display(HTML(f"<style>{CSS}</style>"))
 
+# Main JS
+JS = '''
 <!-- TOGGLE 'CustomDL' SCRIPT -->
 <script>
 function toggleContainer() {
@@ -432,20 +99,58 @@ function toggleContainer() {
 }
 </script>
 '''
+display(HTML(JS))
 
-display(HTML(CSS))
-# ==================== CSS JS ====================
+# ==================== WIDGETS V2 ====================
+HR = widgets.HTML('<hr>')
 
+class WidgetFactory:
+    def __init__(self, style=None, layout=None):
+        self.style = style if style else {'description_width': 'initial'}
+        self.layout = layout if layout else widgets.Layout(max_width='1080px', width='100%')
 
-# ==================== WIDGETS ====================
-# --- global widgets ---
-style = {'description_width': 'initial'}
-layout = widgets.Layout(min_width='1047px')
+    def create_html(self, content, class_name=None):
+        html_widget = widgets.HTML(content)
+        if class_name:
+            html_widget.add_class(class_name)
+        return html_widget
 
-HR = widgets.HTML('<hr>')
+    def create_header(self, name):
+        return widgets.HTML(f'<div class="header">{name}<div>')
+
+    def create_dropdown(self, options, value, description):
+        return widgets.Dropdown(options=options, value=value, description=description, style=self.style, layout=self.layout)
+
+    def create_text(self, description, placeholder='', value=''):
+        return widgets.Text(description=description, placeholder=placeholder, value=value, style=self.style, layout=self.layout)
+
+    def create_checkbox(self, value, description):
+        return widgets.Checkbox(value=value, description=description, style=self.style, layout=self.layout)
+
+    def create_button(self, description, class_name=None):
+        button = widgets.Button(description=description)
+        if class_name:
+            button.add_class(class_name)
+        return button
+
+    def create_hbox(self, children):
+        return widgets.HBox(children)
+
+    def create_vbox(self, children, class_names=None):
+        vbox = widgets.VBox(children)
+        if class_names:
+            for class_name in class_names:
+                vbox.add_class(class_name)
+        return vbox
+
+    def display(self, widget):
+        display(widget)
+
+# Instantiate the factory
+factory = WidgetFactory()
 
 # --- MODEL ---
-model_header = widgets.HTML('<div class="header">Model Selection<div>')
+model_header = factory.create_header('Model Selection')
 model_options = ['none',
                  '1.Anime (by XpucT) + INP',
                  '2.BluMix [Anime] [V7] + INP',
@@ -456,37 +161,37 @@ model_options = ['none',
                  '7.Deliberate [Realism] [V6] + INP',
                  '8.Meina-Mix [Anime] [V11] + INP',
                  '9.Mix-Pro [Anime] [V4] + INP']
-# ---
-Model_widget = widgets.Dropdown(options=model_options, value='4.Counterfeit [Anime] [V3] + INP', description='Model:', style=style, layout=layout)
-Model_Num_widget = widgets.Text(description='Model Number:', placeholder='Enter the model numbers to be downloaded using comma/space.', style=style, layout=layout)
-Inpainting_Model_widget = widgets.Checkbox(value=False, description='Inpainting Models', style=style)
+model_widget = factory.create_dropdown(model_options, '4.Counterfeit [Anime] [V3] + INP', 'Model:')
+model_num_widget = factory.create_text('Model Number:', 'Enter the model numbers to be downloaded using comma/space.')
+inpainting_model_widget = factory.create_checkbox(False, 'Inpainting Models')
 
-''' Display Model'''
-all_model_box = widgets.VBox([model_header, Model_widget, Model_Num_widget, Inpainting_Model_widget]).add_class("container").add_class("image_1")
-display(all_model_box)
+# Display Model
+all_model_box = factory.create_vbox([model_header, model_widget, model_num_widget, inpainting_model_widget], class_names=["container", "image_1"])
+factory.display(all_model_box)
 
 # --- VAE ---
-vae_header = widgets.HTML('<div class="header">VAE Selection</div>')
+vae_header = factory.create_header('VAE Selection')
 vae_options = ['none',
                '1.Anime.vae',
                '2.Anything.vae',
                '3.Blessed2.vae',
                '4.ClearVae.vae',
                '5.WD.vae']
-Vae_widget = widgets.Dropdown(options=vae_options, value='3.Blessed2.vae', description='Vae:', style=style, layout=layout)
-Vae_Num_widget = widgets.Text(description='Vae Number:', placeholder='Enter the vae numbers to be downloaded using comma/space.', style=style, layout=layout)
+vae_widget = factory.create_dropdown(vae_options, '3.Blessed2.vae', 'Vae:')
+vae_num_widget = factory.create_text('Vae Number:', 'Enter the vae numbers to be downloaded using comma/space.')
 
-''' Display Vae'''
-all_vae_box = widgets.VBox([vae_header, Vae_widget, Vae_Num_widget]).add_class("container").add_class("image_2")
-display(all_vae_box)
+# Display Vae
+all_vae_box = factory.create_vbox([vae_header, vae_widget, vae_num_widget], class_names=["container", "image_2"])
+factory.display(all_vae_box)
 
 # --- ADDITIONAL ---
-additional_header = widgets.HTML('<div class="header">Additional</div>')
-latest_webui_widget = widgets.Checkbox(value=True, description='Update WebUI', style=style)
-latest_exstensions_widget = widgets.Checkbox(value=True, description='Update Extensions', style=style)
-change_webui_widget = widgets.Dropdown(options=['A1111', 'Forge'], value='A1111', description='Change WebUI:', style=style)
-detailed_download_widget = widgets.Dropdown(options=['off', 'on'], value='off', description='Detailed Download:', style=style)
-latest_changes_widget = HBox([latest_webui_widget, latest_exstensions_widget, change_webui_widget, detailed_download_widget], layout=widgets.Layout(justify_content='space-between'))
+additional_header = factory.create_header('Additional')
+latest_webui_widget = factory.create_checkbox(True, 'Update WebUI')
+latest_exstensions_widget = factory.create_checkbox(True, 'Update Extensions')
+change_webui_widget = factory.create_dropdown(['A1111', 'Forge'], 'A1111', 'Change WebUI:')
+detailed_download_widget = factory.create_dropdown(['off', 'on'], 'off', 'Detailed Download:')
+choose_changes_widget = factory.create_hbox([latest_webui_widget, latest_exstensions_widget, change_webui_widget, detailed_download_widget])
+
 controlnet_options = ['none', 'ALL', '1.canny',
                       '2.openpose', '3.depth',
                       '4.normal_map', '5.mlsd',
@@ -494,33 +199,45 @@ controlnet_options = ['none', 'ALL', '1.canny',
                       '8.scribble', '9.segmentation',
                       '10.shuffle', '11.tile',
                       '12.inpaint', '13.instruct_p2p']
-# ---
-controlnet_widget = widgets.Dropdown(options=controlnet_options, value='none', description='ControlNet:', style=style, layout=layout)
-controlnet_Num_widget = widgets.Text(description='ControlNet Number:', placeholder='Enter the ControlNet model numbers to be downloaded using comma/space.', style=style, layout=layout)
-commit_hash_widget = widgets.Text(description='Commit Hash:', style=style, layout=layout)
-optional_huggingface_token_widget = widgets.Text(description='HuggingFace Token:', style=style, layout=layout)
-ngrok_token_widget = widgets.Text(description='Ngrok Token:', style=style, layout=widgets.Layout(width='1047px'))
-ngrock_button = widgets.HTML('<a href="https://dashboard.ngrok.com/get-started/your-authtoken" target="_blank">Get Ngrok Token</a>').add_class("button_ngrok")
-ngrok_widget = widgets.HBox([ngrok_token_widget, ngrock_button], style=style, layout=layout)
-zrok_token_widget = widgets.Text(description='Zrok Token:', style=style, layout=widgets.Layout(width='1047px'))
-zrok_button = widgets.HTML('<a href="https://colab.research.google.com/drive/1d2sjWDJi_GYBUavrHSuQyHTDuLy36WpU" target="_blank">Reg Zrok Token</a>').add_class("button_ngrok")
-zrok_widget = widgets.HBox([zrok_token_widget, zrok_button], style=style, layout=layout)
-# ---
+controlnet_widget = factory.create_dropdown(controlnet_options, 'none', 'ControlNet:')
+controlnet_num_widget = factory.create_text('ControlNet Number:', 'Enter the ControlNet model numbers to be downloaded using comma/space.')
+commit_hash_widget = factory.create_text('Commit Hash:')
+huggingface_token_widget = factory.create_text('HuggingFace Token:')
+
+ngrok_token_widget = factory.create_text('Ngrok Token:')
+ngrock_button = factory.create_html('<a href="https://dashboard.ngrok.com/get-started/your-authtoken" target="_blank">Получить Ngrok Токен</a>', class_name="button_ngrok")
+ngrok_widget = factory.create_hbox([ngrok_token_widget, ngrock_button])
+
+zrok_token_widget = factory.create_text('Zrok Token:')
+zrok_button = factory.create_html('<a href="https://colab.research.google.com/drive/1d2sjWDJi_GYBUavrHSuQyHTDuLy36WpU" target="_blank">Зарегать Zrok Токен</a>', class_name="button_ngrok")
+zrok_widget = factory.create_hbox([zrok_token_widget, zrok_button])
+
 commandline_arguments_options = "--listen --enable-insecure-extension-access --theme dark --no-half-vae --disable-console-progressbars --xformers"
-commandline_arguments_widget = widgets.Text(description='Arguments:', value=commandline_arguments_options, style=style, layout=layout)
+commandline_arguments_widget = factory.create_text('Arguments:', value=commandline_arguments_options)
+
+# Display Additional
+additional_widget_list = [additional_header,
+                          choose_changes_widget,
+                          HR,
+                          controlnet_widget,
+                          controlnet_num_widget,
+                          commit_hash_widget,
+                          huggingface_token_widget,
+                          ngrok_widget,
+                          zrok_widget,
+                          HR,
+                          commandline_arguments_widget]
 
-''' Display Additional'''
-additional_widget_list = [additional_header, latest_changes_widget, HR, controlnet_widget, controlnet_Num_widget, commit_hash_widget, optional_huggingface_token_widget, ngrok_widget, zrok_widget, HR, commandline_arguments_widget]
 if free_plan and env == "Google Colab": # remove ngrok from colab
     additional_widget_list.remove(ngrok_widget)
 if os.path.exists(webui_path): # remove selection after selection ;3
-    latest_changes_widget.children = [widget for widget in latest_changes_widget.children if widget != change_webui_widget]
-# ```
-all_additional_box = widgets.VBox(additional_widget_list).add_class("container").add_class("image_3")
-display(all_additional_box)
+    choose_changes_widget.children = [widget for widget in choose_changes_widget.children if widget != change_webui_widget]
+
+all_additional_box = factory.create_vbox(additional_widget_list, class_names=["container", "image_3"])
+factory.display(all_additional_box)
 
 # --- CUSTOM DOWNLOAD ---
-custom_download_header_popup = widgets.HTML('''
+custom_download_header_popup = factory.create_html('''
 <style>
 /* Term Colors */
 .sample_label {color: #dbafff;}
@@ -545,39 +262,40 @@ custom_download_header_popup = widgets.HTML('''
     </div>
 </div>
 ''')
-# ---
-Model_url_widget = widgets.Text(description='Model:', style=style, layout=layout)
-Vae_url_widget = widgets.Text(description='Vae:', style=style, layout=layout)
-LoRA_url_widget = widgets.Text(description='LoRa:', style=style, layout=layout)
-Embedding_url_widget = widgets.Text(description='Embedding:', style=style, layout=layout)
-Extensions_url_widget = widgets.Text(description='Extensions:', style=style, layout=layout)
-custom_file_urls_widget = widgets.Text(description='File (txt):', style=style, layout=layout)
-
-''' Display CustomDl'''
-all_custom_box = widgets.VBox([
+
+Model_url_widget = factory.create_text('Model:')
+Vae_url_widget = factory.create_text('Vae:')
+LoRA_url_widget = factory.create_text('LoRa:')
+Embedding_url_widget = factory.create_text('Embedding:')
+Extensions_url_widget = factory.create_text('Extensions:')
+custom_file_urls_widget = factory.create_text('File (txt):')
+
+# Display CustomDl
+all_custom_box = factory.create_vbox([
     custom_download_header_popup, Model_url_widget, Vae_url_widget, LoRA_url_widget, Embedding_url_widget, Extensions_url_widget, custom_file_urls_widget
-    ]).add_class("container").add_class("image_4").add_class("container_custom_downlad")
-display(all_custom_box)
+], class_names=["container", "image_4", "container_custom_downlad"])
+factory.display(all_custom_box)
 
 # --- Save Button ---
-save_button = widgets.Button(description='Save').add_class("button_save")
-display(save_button)
+save_button = factory.create_button('Save', class_name="button_save")
+factory.display(save_button)
 
 
 # ============ Load / Save - Settings V2 ============
 SETTINGS_FILE = f'{root_path}/settings.json'
 
-settings_keys = [
-    'Model', 'Model_Num', 'Inpainting_Model',
-    'Vae', 'Vae_Num',
-    'latest_webui', 'latest_exstensions', 'change_webui', 'detailed_download',
-    'controlnet', 'controlnet_Num', 'commit_hash', 'optional_huggingface_token',
+SETTINGS_KEYS = [
+    'model', 'model_num', 'inpainting_model',
+    'vae', 'vae_num', 'latest_webui', 'latest_exstensions',
+    'change_webui', 'detailed_download', 'controlnet',
+    'controlnet_num', 'commit_hash', 'huggingface_token',
     'ngrok_token', 'zrok_token', 'commandline_arguments',
-    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url', 'Extensions_url', 'custom_file_urls'
+    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url',
+    'Extensions_url', 'custom_file_urls'
 ]
 
 def save_settings():
-    settings = {key: globals()[f"{key}_widget"].value for key in settings_keys}
+    settings = {key: globals()[f"{key}_widget"].value for key in SETTINGS_KEYS}
     with open(SETTINGS_FILE, 'w') as f:
         json.dump(settings, f, indent=2)
 
@@ -585,20 +303,20 @@ def load_settings():
     if os.path.exists(SETTINGS_FILE):
         with open(SETTINGS_FILE, 'r') as f:
             settings = json.load(f)
-            for key in settings_keys:
-                globals()[f"{key}_widget"].value = settings.get(key)
+            for key in SETTINGS_KEYS:
+                globals()[f"{key}_widget"].value = settings.get(key, "")
 
-def save_data(button):
-    save_settings()
-
-    # --- uhh - hide... ---
+def hide_widgets():
     widgets_list = [all_model_box, all_vae_box, all_additional_box, all_custom_box, save_button]
     for widget in widgets_list:
         widget.add_class("hide")
     time.sleep(0.5)
-
     widgets.Widget.close_all()
 
-settings = load_settings()
+def save_data(button):
+    save_settings()
+    hide_widgets()
+
+load_settings()
 save_button.on_click(save_data)
 

files_cells/python/ru/auto_cleaner_ru.py

@@ -0,0 +1,138 @@
+##~ AutoCleaner V3.7 CODE | BY: ANXETY   ~##
+
+import os
+import time
+import ipywidgets as widgets
+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"
+
+
+# ==================== CSS ====================
+# Main CSS
+css_file_path = f"{root_path}/CSS/auto_cleaner.css"
+with open(css_file_path , "r") as f:
+    CSS_AC = f.read()
+display(HTML(f"<style>{CSS_AC}</style>"))
+
+
+# ================ AutoCleaner function ================
+directories = {
+    "Изображения": f"{webui_path}/output",
+    "Модели": f"{webui_path}/models/Stable-diffusion/",
+    "Vae": f"{webui_path}/models/VAE/",
+    "LoRa": f"{webui_path}/models/Lora/",
+    "ControlNet Модели": f"{webui_path}/models/ControlNet/"
+}
+
+""" functions """
+def clean_directory(directory):
+    deleted_files = 0
+    image_dir = directories['Изображения']
+
+    for root, dirs, files in os.walk(directory):
+        for file in files:
+            file_path = os.path.join(root, file)
+
+            if file.endswith(".txt"):
+                continue
+            if file.endswith((".safetensors", ".pt")) or root == image_dir: # fix for image counter
+                deleted_files += 1
+
+            os.remove(file_path)
+    return deleted_files
+
+def update_memory_info():
+    disk_space = psutil.disk_usage(os.getcwd())
+    total = disk_space.total / (1024 ** 3)
+    used = disk_space.used / (1024 ** 3)
+    free = disk_space.free / (1024 ** 3)
+
+    storage_info.value = f'''
+    <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>
+    '''
+
+def on_execute_button_press(button):
+    selected_cleaners = auto_cleaner_widget.value
+    deleted_files_dict = {}
+
+    for option in selected_cleaners:
+        if option in directories:
+            deleted_files_dict[option] = clean_directory(directories[option])
+
+    output.clear_output()
+
+    with output:
+        for message in generate_messages(deleted_files_dict):
+            message_widget = HTML(f'<p class="output_message_AC">{message}</p>')
+            display(message_widget)
+
+    update_memory_info()
+
+def on_clear_button_press(button):
+    container.add_class("hide")
+    time.sleep(0.5)
+    widgets.Widget.close_all()
+
+def generate_messages(deleted_files_dict):
+    messages = []
+    word_variants = {
+        "Изображения": "Изображений",
+        "Модели": "Моделей",
+        "Vae": "Vae",
+        "LoRa": "LoRa",
+        "ControlNet Модели": "ControlNet Моделей"
+    }
+    for key, value in deleted_files_dict.items():
+        object_word = word_variants.get(key)
+        messages.append(f"Удалено {value} {object_word}")
+    return messages
+
+
+# --- storage memory ---
+import psutil
+disk_space = psutil.disk_usage(os.getcwd())
+total = disk_space.total / (1024 ** 3)
+used = disk_space.used / (1024 ** 3)
+free = disk_space.free / (1024 ** 3)
+
+
+# ================ Widgets ================
+# UI Code
+AutoCleaner_options = AutoCleaner_options = list(directories.keys())
+instruction_label = widgets.HTML('''
+<span class="instruction_AC">Используйте <span style="color: #B2B2B2;">ctrl</span> или <span style="color: #B2B2B2;">shift</span> для множественного выбора.</span>
+''')
+auto_cleaner_widget = widgets.SelectMultiple(options=AutoCleaner_options, layout=widgets.Layout(width="auto")).add_class("custom-select-multiple_AC")
+output = widgets.Output().add_class("output_AC")
+
+execute_button = Button(description='Выполнить Очистку').add_class("button_execute_AC").add_class("button_AC")
+execute_button.on_click(on_execute_button_press)
+clear_button = Button(description='Скрыть Виджет').add_class("button_clear_AC").add_class("button_AC")
+clear_button.on_click(on_clear_button_press)
+
+storage_info = widgets.HTML(f'''
+<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>
+''')
+
+buttons = widgets.HBox([execute_button, clear_button])
+lower_information_panel = widgets.HBox([buttons, storage_info]).add_class("lower_information_panel_AC")
+
+container = VBox([instruction_label, widgets.HTML('<hr>'), auto_cleaner_widget, output, widgets.HTML('<hr>'), lower_information_panel]).add_class("container_AC")
+
+display(container)
+

files_cells/python/ru/downloading_ru.py

@@ -15,21 +15,19 @@ from IPython.display import clear_output
 from urllib.parse import urlparse, parse_qs
 
 
-#  ================= DETECT ENV =================
+# ================= 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"
-#  ----------------------------------------------
 
 
 # ================ LIBRARIES V2 ================
@@ -80,12 +78,6 @@ if not os.path.exists(flag_file):
     time.sleep(2)
     clear_output()
 
-## dl special files
-with capture.capture_output() as cap:
-    get_ipython().system('mkdir -p {root_path}/file_cell/special')
-    get_ipython().system('wget https://huggingface.co/NagisaNao/fast_repo/resolve/main/special/dl_display_results.py -O {root_path}/file_cell/special/dl_display_results.py')
-del cap
-
 
 # ================= loading settings V4 =================
 def load_settings(path):
@@ -96,16 +88,17 @@ def load_settings(path):
 
 settings = load_settings(f'{root_path}/settings.json')
 
-variables = [
-    'Model', 'Model_Num', 'Inpainting_Model',
-    'Vae', 'Vae_Num',
-    'latest_webui', 'latest_exstensions', 'change_webui', 'detailed_download',
-    'controlnet', 'controlnet_Num', 'commit_hash', 'optional_huggingface_token',
+VARIABLES = [
+    'model', 'model_num', 'inpainting_model',
+    'vae', 'vae_num', 'latest_webui', 'latest_exstensions',
+    'change_webui', 'detailed_download', 'controlnet',
+    'controlnet_num', 'commit_hash', 'huggingface_token',
     'ngrok_token', 'zrok_token', 'commandline_arguments',
-    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url', 'Extensions_url', 'custom_file_urls'
+    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url',
+    'Extensions_url', 'custom_file_urls'
 ]
 
-locals().update({key: settings.get(key) for key in variables})
+locals().update({key: settings.get(key) for key in VARIABLES})
 
 
 # ================= OTHER =================
@@ -241,17 +234,7 @@ model_list = {
     ]
 }
 
-# 1-4 (fp16/cleaned)
 vae_list = {
-    ## vae broke - the author's an asshole
-    # "1.Anime.vae": [
-    #     {"url": "https://civitai.com/api/download/models/131654", "name": "Anime.vae.safetensors"},
-    #     {"url": "https://civitai.com/api/download/models/131658", "name": "vae-ft-mse.vae.safetensors"}
-    # ],
-    # "2.Anything.vae": [{"url": "https://civitai.com/api/download/models/131656", "name": "Anything.vae.safetensors"}],
-    # "3.Blessed2.vae": [{"url": "https://civitai.com/api/download/models/142467", "name": "Blessed2.vae.safetensors"}],
-    # "4.ClearVae.vae": [{"url": "https://civitai.com/api/download/models/133362", "name": "ClearVae_23.vae.safetensors"}],
-
     "1.Anime.vae": [{"url": "https://civitai.com/api/download/models/311162", "name": "vae-ft-mse-840000-ema-pruned.vae.safetensors"}],
     "2.Anything.vae": [{"url": "https://civitai.com/api/download/models/119279", "name": "Anything.vae.safetensors"}],
     "3.Blessed2.vae": [{"url": "https://huggingface.co/NoCrypt/blessed_vae/resolve/main/blessed2.vae.pt", "name": "Blessed2.vae.safetensors"}],
@@ -329,26 +312,47 @@ 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 = optional_huggingface_token if optional_huggingface_token else "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"
+hf_token = huggingface_token if huggingface_token else "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"
 user_header = f"\"Authorization: Bearer {hf_token}\""
 
+''' Formatted Info Output '''
+
+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"
+
+def format_output(url, dst_dir, file_name):
+    info = f"[{file_name.split('.')[0]}]"
+    info = center_text(info)
+
+    print(f"\n\033[32m{'---'*20}]{info}[{'---'*20}")
+    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 Image Preview | CivitAi '''
 
-def get_data_from_api(model_id): # get model data
+def get_data_from_api(model_id):
+    """Fetch model data from the API"""
     endpoint_url = f"https://civitai.com/api/v1/model-versions/{model_id}"
     headers = {"Content-Type": "application/json"}
     try:
         response = requests.get(endpoint_url, headers=headers)
-        if response.status_code == 200:
-            return response.json()
+        response.raise_for_status()
+        return response.json()
     except requests.exceptions.RequestException as e:
         print(f"An error occurred: {e}")
         return None
 
 def extract_model_info(data, url):
+    """Extract model information based on URL"""
     if 'type=' in url:
         model_type = parse_qs(urlparse(url).query).get('type', [''])[0]
         model_name = data['files'][1]['name']
@@ -365,59 +369,69 @@ def extract_model_info(data, url):
     return model_type, model_name, image_url
 
 def gen_preview_filename(model_name, image_url):
+    """Generate a preview filename"""
     name = model_name.split('.')
     img_exts = image_url.split('.')
-    return f"{name[0]}.preview.{img_exts[-1]}" # assigning the original image format
+    return f"{name[0]}.preview.{img_exts[-1]}"
 
 ''' main download code '''
 
 def handle_manual(url):
-    original_url = url
-    url = url.split(':', 1)[1]
-    file_name = re.search(r'\[(.*?)\]', url)
-    file_name = file_name.group(1) if file_name else None
-    if file_name:
-        url = re.sub(r'\[.*?\]', '', url)
+    url_parts = url.split(':', 1)
+    prefix = url_parts[0]
+    path = url_parts[1]
 
-    for prefix, dir in prefixes.items():
-        if original_url.startswith(f"{prefix}:"):
-            if prefix != "extension":
-                manual_download(url, dir, file_name=file_name)
-            else:
-                extension_repo.append((url, file_name))
+    file_name_match = re.search(r'\[(.*?)\]', path)
+    file_name = file_name_match.group(1) if file_name_match else None
+    if file_name:
+        path = re.sub(r'\[.*?\]', '', path)
+
+    if prefix in prefixes:
+        dir = prefixes[prefix]
+        if prefix != "extension":
+            try:
+                manual_download(path, dir, file_name=file_name)
+            except Exception as e:
+                print(f"Error downloading file: {e}")
+        else:
+            extension_repo.append((path, file_name))
 
 def manual_download(url, dst_dir, file_name):
     aria2_args = '--optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -j5 -x16 -s16 -k1M -c'
     basename = url.split("/")[-1] if file_name is None else file_name
     header_option = f"--header={user_header}"
 
-    # === CivitAi API ===
+    # ==== CivitAi API+ ====
     support_types = ('Checkpoint', 'Model', 'TextualInversion', 'LORA') # for dl preview image
     civitai_token = "62c0c5956b2f9defbd844d754000180b"
 
     if 'civitai' in url:
-        url = f"{url}{'&' if '?' in url else '?'}token={civitai_token}" # some authors are pussies and they need API token...
-        model_id = url.split('/')[-1]
-        clean_url = url.split('?')[0]
+        url = f"{url}{'&' if '?' in url else '?'}token={civitai_token}"
+        model_id = url.split('/')[-1].split('?')[0]
+        clean_url = re.sub(r'[?&]token=[^&]*', '', url) # hide token
 
         data = get_data_from_api(model_id)
         if data:
             model_type, model_name, image_url = extract_model_info(data, url)
-            if model_name and image_url:
-                image_file_name = gen_preview_filename(model_name if not file_name else file_name, image_url)
-                if any(types in model_type for types in support_types):
-                    with capture.capture_output() as cap: # clear shit
-                        get_ipython().system('aria2c {aria2_args} -d {dst_dir} -o {image_file_name} {image_url}')
+
+            if any(t in model_type for t in support_types):
+                if model_name and image_url:
+                    image_file_name = gen_preview_filename(model_name if not file_name else file_name, image_url)
+                    with capture.capture_output() as cap:
+                        get_ipython().system("aria2c {aria2_args} -d {dst_dir} -o {image_file_name} '{image_url}'")
                     del cap
-            file_name = file_name or model_name # assigns the original file name if not specified initially
+            file_name = file_name or model_name
+    else:
+        clean_url = url
+
+    """ Formatted info output """
+    model_name_or_basename = file_name if not 'huggingface' in url else basename
+    format_output(clean_url or url, dst_dir, model_name_or_basename)
 
-    """ information output """
-    # -- wrold's best print info --
-    print(f"\n\033[32m{'---'*45}\n\033[33mURL: \033[34m{clean_url if 'civitai' in url else url}\n\033[33mSAVE DIR: \033[34m{dst_dir}\n\033[33mFILE NAME: \033[34m{file_name if not 'huggingface' in url else basename}\033[0m")
     print("\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 any(types in model_type for types in support_types) and (locals().get('image_file_name') or ''):
+    if 'civitai' in url and data and any(t in model_type for t in support_types) and (locals().get('image_file_name') or ''):
         print(f"\033[32m[Preview DL]:\033[0m {image_file_name} - {image_url}\n")
-    # ===================
+    # =====================
 
     # -- GDrive --
     if 'drive.google' in url:
@@ -461,9 +475,8 @@ def download(url):
 
     unpucking_zip_files()
 
-## unpucking zip files
+# unpucking zip files
 def unpucking_zip_files():
-    # directories - above
     for directory in directories:
         for root, dirs, files in os.walk(directory):
             for file in files:
@@ -476,8 +489,6 @@ def unpucking_zip_files():
 
 ''' submodels - added urls '''
 
-submodels = []
-
 def add_submodels(selection, num_selection, model_dict, dst_dir):
     if selection == "none":
         return []
@@ -489,33 +500,32 @@ def add_submodels(selection, num_selection, model_dict, dst_dir):
     else:
         selected_models = model_dict[selection]
         selected_nums = map(int, num_selection.replace(',', '').split())
-
         for num in selected_nums:
             if 1 <= num <= len(model_dict):
                 name = list(model_dict)[num - 1]
                 selected_models.extend(model_dict[name])
 
     unique_models = list({model['name']: model for model in selected_models}.values())
-
     for model in unique_models:
         model['dst_dir'] = dst_dir
 
     return unique_models
 
-submodels += add_submodels(Model, Model_Num, model_list, models_dir)                    # model
-submodels += add_submodels(Vae, Vae_Num, vae_list, vaes_dir)                            # vae
-submodels += add_submodels(controlnet, controlnet_Num, controlnet_list, control_dir)    # controlnet
+def handle_submodels(selection, num_selection, model_dict, dst_dir, url):
+    submodels = add_submodels(selection, num_selection, model_dict, dst_dir)
+    for submodel in submodels:
+        if not inpainting_model and "inpainting" in submodel['name']:
+            continue
+        url += f"{submodel['url']} {submodel['dst_dir']} {submodel['name']}, "
+    return url
 
-for submodel in submodels:
-    if not Inpainting_Model and "inpainting" in submodel['name']:
-        continue
-    url += f"{submodel['url']} {submodel['dst_dir']} {submodel['name']}, "
+url = handle_submodels(model, model_num, model_list, models_dir, url)
+url = handle_submodels(vae, vae_num, vae_list, vaes_dir, url)
+url = handle_submodels(controlnet, controlnet_num, controlnet_list, control_dir, url)
 
 ''' file.txt - added urls '''
 
-unique_urls = []
-
-def process_file_download(file_url):
+def process_file_download(file_url, prefixes, unique_urls):
     files_urls = ""
 
     if file_url.startswith("http"):
@@ -529,19 +539,23 @@ def process_file_download(file_url):
 
     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()))
 
         urls = [url.split('#')[0].strip() for url in line.split(',')]  # filter urls
         for url in urls:
-            if url.startswith("http") and url not in unique_urls:
+            filter_url = url.split('[')[0] # same url filter
+
+            if url.startswith("http") and filter_url not in unique_urls:
                 files_urls += f"{current_tag}:{url}, "
-                unique_urls.append(url)
+                unique_urls.add(filter_url)
 
     return files_urls
 
-# fix all possible errors/options and function call
 file_urls = ""
+unique_urls = set()
+
 if custom_file_urls:
     for custom_file_url in custom_file_urls.replace(',', '').split():
         if not custom_file_url.endswith('.txt'):
@@ -551,7 +565,7 @@ if custom_file_urls:
                 custom_file_url = f'{root_path}/{custom_file_url}'
 
         try:
-            file_urls += process_file_download(custom_file_url)
+            file_urls += process_file_download(custom_file_url, prefixes, unique_urls)
         except FileNotFoundError:
             pass
 
@@ -591,38 +605,6 @@ if len(extension_repo) > 0:
 ## List Models and stuff V2
 if detailed_download == "off":
     print("\n\n\033[33mЕсли вы не видете каких-то скаченных файлов, включите в виджетах функцию 'Подробная Загрузка'.")
-    get_ipython().run_line_magic('run', '{root_path}/file_cell/special/dl_display_results.py') # display widgets result
 
-else:
-    if any(not file.endswith('.txt') for file in os.listdir(models_dir)):
-        print("\n\033[33m➤ Models\033[0m")
-        get_ipython().system("find {models_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
-    if any(not file.endswith('.txt') for file in os.listdir(vaes_dir)):
-        print("\n\033[33m➤ VAEs\033[0m")
-        get_ipython().system("find {vaes_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
-    if any(not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file)) for file in os.listdir(embeddings_dir)):
-        print("\n\033[33m➤ Embeddings\033[0m")
-        get_ipython().system("find {embeddings_dir}/ -mindepth 1 -maxdepth 1 \\( -name '*.pt' -or -name '*.safetensors' \\) -printf '%f\\n'")
-    if any(not file.endswith('.txt') for file in os.listdir(loras_dir)):
-        print("\n\033[33m➤ LoRAs\033[0m")
-        get_ipython().system("find {loras_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
-    print(f"\n\033[33m➤ Extensions\033[0m")
-    get_ipython().system("find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\n'")
-    if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir)):
-        print("\n\033[33m➤ ControlNet\033[0m")
-        get_ipython().system("find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\([^_]*\\)_fp16\\.safetensors$/\\1/'")
-
-
-# === OTHER ===
-# Downlaod discord tags UmiWildcards
-files_umi = [
-    "https://huggingface.co/NagisaNao/fast_repo/resolve/main/extensions/UmiWildacrd/discord/200_pan_gen.txt",
-    "https://huggingface.co/NagisaNao/fast_repo/resolve/main/extensions/UmiWildacrd/discord/150_bra_gen.txt"
-]
-save_dir_path = f"{webui_path}/extensions/Umi-AI-Wildcards/wildcards/discord"
-
-with capture.capture_output() as cap:
-    for file in files_umi:
-        get_ipython().system('aria2c --optimize-concurrent-downloads --console-log-level=error --summary-interval=10 -j5 -x16 -s16 -k1M -c -d {save_dir_path} {file}')
-del cap
+get_ipython().run_line_magic('run', '{root_path}/file_cell/special/dl_display_results.py # display widgets result')
 

files_cells/python/ru/widgets_ru.py

@@ -8,27 +8,24 @@ from ipywidgets import widgets, Layout, Label, Button, VBox, HBox
 from IPython.display import display, HTML, Javascript, clear_output
 
 
-#  ================= DETECT ENV =================
+# ================= 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}')
-#  ----------------------------------------------
 
 
 # ==================== CSS JS ====================
-# custom background images V1.5
+##~ custom background images V1.5 ~##
 import argparse
 parser = argparse.ArgumentParser(description='This script processes an background image.')
 parser.add_argument('-i', '--image', type=str, help='URL of the image to process', metavar='')
@@ -40,8 +37,9 @@ parser.add_argument('-s', '--scale', type=int, help='Scale image in %%', metavar
 parser.add_argument('-m', '--mode', action='store_true', help='Removes repetitive image tiles')
 parser.add_argument('-t', '--transparent', action='store_true', help='Makes input/selection fields 35%% more transparent')
 parser.add_argument('-bf', '--blur-fields', type=str, help='Background blur level for input/selection fields', metavar='', default=2)
+
 args = parser.parse_args()
-"""---"""
+
 url_img = args.image
 opacity_img = args.opacity
 blur_img = args.blur
@@ -59,7 +57,6 @@ t_bg_alpha = "1" if not args.transparent else "0.65"
 
 """ mode img - repeats """
 mode_img = "repeat" if not args.mode else "no-repeat"
-## ---
 
 container_background = f'''
 <style>
@@ -82,345 +79,15 @@ container_background = f'''
 '''
 
 display(HTML(container_background))
-# ---
-
-CSS = '''
-<style>
-/* General Styles */
-.header {
-    font-family: cursive;
-    font-size: 20px;
-    font-weight: bold;
-    color: #ff8cee;
-    margin-bottom: 15px;
-    user-select: none;
-    cursor: default;
-    display: inline-block;
-}
-
-hr {
-    border-color: grey;
-    background-color: grey;
-    opacity: 0.25;
-}
-
-a {
-  text-decoration: none;
-  color: inherit;
-}
-
-
-/* Container style */
-
-.container {
-    position: relative;
-    background-color: #232323;
-    width: 1080px;
-    padding: 10px 15px;
-    border-radius: 15px;
-    box-shadow: 0 0 50px rgba(0, 0, 0, 0.3);
-    margin-bottom: 5px;
-    overflow: hidden;
-}
-
-.container::after {
-    position: absolute;
-    top: 5px;
-    right: 10px;
-    content: "ANXETY";
-    font-weight: bold;
-    font-size: 24px;
-    color: rgba(0, 0, 0, 0.2);
-}
-
-/* background img */
-.container::before {
-  content: "";
-  position: absolute;
-  top: 0;
-  left: 0;
-  right: 0;
-  bottom: 0;
-  background-image: var(--img_background);
-  background-size: var(--img_scale);
-  background-repeat: var(--img_mode);
-  opacity: var(--img_opacity);
-  mix-blend-mode: screen;
-  pointer-events: none;
-  filter: blur(var(--img_blur));
-  z-index: -1;
-}
-
-.image_1::before {
-  background-position: var(--image_x) calc(-120px - var(--image_y));
-}
-.image_2::before {
-  background-position: var(--image_x) calc(-290px - var(--image_y));
-}
-.image_3::before {
-  background-position: var(--image_x) calc(-430px - var(--image_y));
-}
-.image_4::before {
-  background-position: var(--image_x) calc(var(--img_height_dif) - var(--image_y));
-}
-
-.container_custom_downlad {
-    height: 55px;
-    transition: all 0.5s;
-}
-
-.container_custom_downlad.expanded {
-    height: 270px;
-}
-
-
-/* Element text style */
-
-.widget-html,
-.widget-button,
-.widget-text label,
-.widget-checkbox label,
-.widget-dropdown label,
-.widget-dropdown select,
-.widget-text input[type="text"] {
-    font-family: cursive;
-    font-size: 14px;
-    color: white !important;
-    user-select: none;
-}
-
-.widget-text input[type="text"]::placeholder {
-    color: grey;
-}
-
-
-/* Input field styles */
-
-.widget-dropdown select,
-.widget-text input[type="text"] {
-    height: 30px;
-    background-color: var(--bg-field-color);
-    border: 1px solid #262626;
-    border-radius: 10px;
-    box-shadow: inset 0 0 5px rgba(0, 0, 0, 0.5);
-    transition: all 0.3s ease-in-out;
-    backdrop-filter: blur(var(--bg-field-blur-level));
-}
-
-.widget-dropdown select:focus,
-.widget-text input[type="text"]:focus {
-    border-color: #006ee5;
-}
-
-.widget-dropdown select:hover,
-.widget-text input[type="text"]:hover {
-    transform: scale(1.003);
-    background-color: var(--bg-field-color-hover);
-    box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.5);
-}
-
-.widget-dropdown option {
-    background-color: #1c1c1c;
-}
-
-
-/* Slider Checkbox style */
-
-.widget-checkbox input[type="checkbox"] {
-    appearance: none;
-    position: relative;
-    top: 4px; /* Why is he taller?! */
-    width: 40px;
-    height: 20px;
-    border: none;
-    border-radius: 10px;
-    background-color: #20b2aa;
-    cursor: pointer;
-    box-shadow: inset 0 0 10px rgba(0, 0, 0, 0.3);
-    transition: background-color 0.3s ease;
-}
-.widget-checkbox input[type="checkbox"]:checked {
-    background-color: #2196F3;
-}
-
-.widget-checkbox input[type="checkbox"]:before {
-    content: '';
-    position: absolute;
-    top: 50%;
-    left: 3px;
-    width: 16px;
-    height: 16px;
-    border-radius: inherit;
-    background-color: white;
-    box-shadow: 0 0 8px rgba(0, 0, 0, 0.3);
-    transform: translateY(-50%);
-    transition: left 0.3s ease;
-}
-.widget-checkbox input[type="checkbox"]:checked:before {
-    left: 21px;
-}
-
-
-/* Button styles */
-
-.button_save {
-    font-size: 15px;
-    font-weight: bold;
-    width: 120px;
-    height: 35px;
-    border-radius: 15px;
-    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);
-    background-size: 200% 200%;
-    background-position: left bottom;
-    transition: background 0.5s ease-in-out, transform 0.3s ease;
-}
-
-.button_save:hover {
-    cursor: pointer;
-    background-image: radial-gradient(circle at top left, purple 10%, #93ac47 90%);
-    background-size: 200% 200%;
-    background-position: right bottom;
-    transform: translateY(1px);
-}
 
-.button_ngrok {
-    font-size: 12px;
-    height: 30px;
-    border-radius: 10px;
-    padding: 1px 12px;
-    background-image: radial-gradient(circle at top left, purple 10%, violet 90%);
-    background-size: 200% 200%;
-    background-position: left bottom;
-    transition: background 0.5s ease-in-out, transform 0.3s ease;
-    white-space: nowrap;
-}
-
-.button_ngrok:hover  {
-    cursor: pointer;
-    background-image: radial-gradient(circle at top left, purple 10%, #1D94BB 90%);
-    background-size: 200% 200%;
-    background-position: right bottom;
-    transform: translateY(1px);
-}
-
-.button_save:active,
-.button_ngrok:active {
-    filter: brightness(0.75) !important;
-}
-
-/* Removes ugly stroke from widget buttons. */
-.jupyter-widgets.lm-Widget:focus {
-    outline: none;
-}
-
-
-/* Popup style of `INFO` window */
-
-.info {
-    position: absolute;
-    top: -5px;
-    right: 95px;
-    color: grey;
-    font-family: cursive;
-    font-size: 14px;
-    font-weight: normal;
-    user-select: none;
-    pointer-events: none;
-    opacity: 0;
-    transition: opacity 0.3s ease-in-out;
-    display: inline-block;
-}
-
-.popup {
-    position: absolute;
-    top: 120px;
-    z-index: 999;
-    width: auto;
-    padding: 10px;
-    text-align: center;
-    background-color: rgba(255, 255, 255, 0.05);
-    backdrop-filter: blur(20px);
-    border: 1px solid rgba(255, 255, 255, 0.45);
-    border-radius: 8px;
-    box-shadow: 0 0 50px rgba(0, 0, 0, 0.5);
-    opacity: 0;
-    color: #fff;
-    font-size: 16px;
-    font-family: cursive;
-    user-select: none;
-    cursor: default;
-    pointer-events: none;
-    transform: rotate(-5deg);
-    transition: top 0.3s ease-in-out, opacity 0.3s ease-in-out, transform 0.3s ease-in-out;
-}
-
-.sample {
-    display: inline-block;
-    margin-top: 25px;
-    padding: 10px 100px;
-    background-color: rgba(255, 255, 255, 0.2);
-    color: #c6e2ff;
-    border: 2px solid rgba(255, 255, 255, 0.2);
-    border-radius: 8px;
-    box-shadow: 0 0 10px rgba(255, 255, 255, 0.2), inset 0 0 25px rgba(255, 255, 255, 0.2);
-}
-
-.info.showed {
-    opacity: 1;
-    pointer-events: auto;
-}
-
-.info:hover + .popup {
-    top: 35px;
-    opacity: 1;
-    pointer-events: initial;
-    transform: rotate(0deg);
-}
-
-
-/* Animation of elements */
-
-.container,
-.button_save {
-    animation-name: showedWidgets;
-    animation-duration: 1s;
-    animation-fill-mode: forwards;
-}
-
-.container.hide,
-.button_save.hide {
-    animation-name: hideWidgets;
-    animation-duration: 0.5s;
-    animation-fill-mode: forwards;
-}
-
-@keyframes showedWidgets {
-    0% {
-        transform: translate3d(-65%, 15%, 0) scale(0) rotate(15deg);
-        filter: blur(25px) grayscale(1) brightness(0.3);
-        opacity: 0;
-    }
-    100% {
-        transform: translate3d(0, 0, 0) scale(1) rotate(0deg);
-        filter: blur(0) grayscale(0) brightness(1);
-        opacity: 1;
-    }
-}
-
-@keyframes hideWidgets {
-    0% {
-        transform: translate3d(0, 0, 0) scale(1) rotate3d(1, 0, 0, 0deg);
-        filter: blur(0) grayscale(0) brightness(1);
-        opacity: 1;
-    }
-    100% {
-        transform: translate3d(0, 5%, 0) scale(0.9) rotate3d(1, 0, 0, 90deg);
-        filter: blur(15px) grayscale(1) brightness(0.5);
-        opacity: 0;
-    }
-}
-</style>
+# Main CSS
+css_file_path = f"{root_path}/CSS/main_widgets.css"
+with open(css_file_path , "r") as f:
+    CSS = f.read()
+display(HTML(f"<style>{CSS}</style>"))
 
+# Main JS
+JS = '''
 <!-- TOGGLE 'CustomDL' SCRIPT -->
 <script>
 function toggleContainer() {
@@ -432,20 +99,58 @@ function toggleContainer() {
 }
 </script>
 '''
+display(HTML(JS))
 
-display(HTML(CSS))
-# ==================== CSS JS ====================
+# ==================== WIDGETS V2 ====================
+HR = widgets.HTML('<hr>')
 
+class WidgetFactory:
+    def __init__(self, style=None, layout=None):
+        self.style = style if style else {'description_width': 'initial'}
+        self.layout = layout if layout else widgets.Layout(max_width='1080px', width='100%')
 
-# ==================== WIDGETS ====================
-# --- global widgets ---
-style = {'description_width': 'initial'}
-layout = widgets.Layout(min_width='1047px')
+    def create_html(self, content, class_name=None):
+        html_widget = widgets.HTML(content)
+        if class_name:
+            html_widget.add_class(class_name)
+        return html_widget
 
-HR = widgets.HTML('<hr>')
+    def create_header(self, name):
+        return widgets.HTML(f'<div class="header">{name}<div>')
+
+    def create_dropdown(self, options, value, description):
+        return widgets.Dropdown(options=options, value=value, description=description, style=self.style, layout=self.layout)
+
+    def create_text(self, description, placeholder='', value=''):
+        return widgets.Text(description=description, placeholder=placeholder, value=value, style=self.style, layout=self.layout)
+
+    def create_checkbox(self, value, description):
+        return widgets.Checkbox(value=value, description=description, style=self.style, layout=self.layout)
+
+    def create_button(self, description, class_name=None):
+        button = widgets.Button(description=description)
+        if class_name:
+            button.add_class(class_name)
+        return button
+
+    def create_hbox(self, children):
+        return widgets.HBox(children)
+
+    def create_vbox(self, children, class_names=None):
+        vbox = widgets.VBox(children)
+        if class_names:
+            for class_name in class_names:
+                vbox.add_class(class_name)
+        return vbox
+
+    def display(self, widget):
+        display(widget)
+
+# Instantiate the factory
+factory = WidgetFactory()
 
 # --- MODEL ---
-model_header = widgets.HTML('<div class="header">Выбор Модели<div>')
+model_header = factory.create_header('Выбор Модели')
 model_options = ['none',
                  '1.Anime (by XpucT) + INP',
                  '2.BluMix [Anime] [V7] + INP',
@@ -456,37 +161,37 @@ model_options = ['none',
                  '7.Deliberate [Realism] [V6] + INP',
                  '8.Meina-Mix [Anime] [V11] + INP',
                  '9.Mix-Pro [Anime] [V4] + INP']
-# ---
-Model_widget = widgets.Dropdown(options=model_options, value='4.Counterfeit [Anime] [V3] + INP', description='Модель:', style=style, layout=layout)
-Model_Num_widget = widgets.Text(description='Номер Модели:', placeholder='Введите номера моделей для скачивания через запятую/пробел.', style=style, layout=layout)
-Inpainting_Model_widget = widgets.Checkbox(value=False, description='Inpainting Модели', style=style)
+model_widget = factory.create_dropdown(model_options, '4.Counterfeit [Anime] [V3] + INP', 'Модель:')
+model_num_widget = factory.create_text('Номер Модели:', 'Введите номера моделей для скачивания через запятую/пробел.')
+inpainting_model_widget = factory.create_checkbox(False, 'Inpainting Модели')
 
-''' Display Model'''
-all_model_box = widgets.VBox([model_header, Model_widget, Model_Num_widget, Inpainting_Model_widget]).add_class("container").add_class("image_1")
-display(all_model_box)
+# Display Model
+all_model_box = factory.create_vbox([model_header, model_widget, model_num_widget, inpainting_model_widget], class_names=["container", "image_1"])
+factory.display(all_model_box)
 
 # --- VAE ---
-vae_header = widgets.HTML('<div class="header">Выбор VAE</div>')
+vae_header = factory.create_header('Выбор VAE')
 vae_options = ['none',
                '1.Anime.vae',
                '2.Anything.vae',
                '3.Blessed2.vae',
                '4.ClearVae.vae',
                '5.WD.vae']
-Vae_widget = widgets.Dropdown(options=vae_options, value='3.Blessed2.vae', description='Vae:', style=style, layout=layout)
-Vae_Num_widget = widgets.Text(description='Номер Vae:', placeholder='Введите номера vae для скачивания через запятую/пробел.', style=style, layout=layout)
+vae_widget = factory.create_dropdown(vae_options, '3.Blessed2.vae', 'Vae:')
+vae_num_widget = factory.create_text('Номер Vae:', 'Введите номера vae для скачивания через запятую/пробел.')
 
-''' Display Vae'''
-all_vae_box = widgets.VBox([vae_header, Vae_widget, Vae_Num_widget]).add_class("container").add_class("image_2")
-display(all_vae_box)
+# Display Vae
+all_vae_box = factory.create_vbox([vae_header, vae_widget, vae_num_widget], class_names=["container", "image_2"])
+factory.display(all_vae_box)
 
 # --- ADDITIONAL ---
-additional_header = widgets.HTML('<div class="header">Дополнительно</div>')
-latest_webui_widget = widgets.Checkbox(value=True, description='Обновить WebUI', style=style)
-latest_exstensions_widget = widgets.Checkbox(value=True, description='Обновить Расширения', style=style)
-change_webui_widget = widgets.Dropdown(options=['A1111', 'Forge'], value='A1111', description='Изменить WebUI:', style=style)
-detailed_download_widget = widgets.Dropdown(options=['off', 'on'], value='off', description='Подробная Загрузка:', style=style)
-latest_changes_widget = HBox([latest_webui_widget, latest_exstensions_widget, change_webui_widget, detailed_download_widget], layout=widgets.Layout(justify_content='space-between'))
+additional_header = factory.create_header('Дополнительно')
+latest_webui_widget = factory.create_checkbox(True, 'Обновить WebUI')
+latest_exstensions_widget = factory.create_checkbox(True, 'Обновить Расширения')
+change_webui_widget = factory.create_dropdown(['A1111', 'Forge'], 'A1111', 'Изменить WebUI:')
+detailed_download_widget = factory.create_dropdown(['off', 'on'], 'off', 'Подробная Загрузка:')
+choose_changes_widget = factory.create_hbox([latest_webui_widget, latest_exstensions_widget, change_webui_widget, detailed_download_widget])
+
 controlnet_options = ['none', 'ALL', '1.canny',
                       '2.openpose', '3.depth',
                       '4.normal_map', '5.mlsd',
@@ -494,33 +199,45 @@ controlnet_options = ['none', 'ALL', '1.canny',
                       '8.scribble', '9.segmentation',
                       '10.shuffle', '11.tile',
                       '12.inpaint', '13.instruct_p2p']
-# ---
-controlnet_widget = widgets.Dropdown(options=controlnet_options, value='none', description='ControlNet:', style=style, layout=layout)
-controlnet_Num_widget = widgets.Text(description='Номер ControlNet:', placeholder='Введите номера моделей ControlNet для скачивания через запятую/пробел.', style=style, layout=layout)
-commit_hash_widget = widgets.Text(description='Commit Hash:', style=style, layout=layout)
-optional_huggingface_token_widget = widgets.Text(description='Токен HuggingFace:', style=style, layout=layout)
-ngrok_token_widget = widgets.Text(description='Токен Ngrok:', style=style, layout=widgets.Layout(width='1047px'))
-ngrock_button = widgets.HTML('<a href="https://dashboard.ngrok.com/get-started/your-authtoken" target="_blank">Получить Ngrok Токен</a>').add_class("button_ngrok")
-ngrok_widget = widgets.HBox([ngrok_token_widget, ngrock_button], style=style, layout=layout)
-zrok_token_widget = widgets.Text(description='Токен Zrok:', style=style, layout=widgets.Layout(width='1047px'))
-zrok_button = widgets.HTML('<a href="https://colab.research.google.com/drive/1d2sjWDJi_GYBUavrHSuQyHTDuLy36WpU" target="_blank">Зарегать Zrok Токен</a>').add_class("button_ngrok")
-zrok_widget = widgets.HBox([zrok_token_widget, zrok_button], style=style, layout=layout)
-# ---
+controlnet_widget = factory.create_dropdown(controlnet_options, 'none', 'ControlNet:')
+controlnet_num_widget = factory.create_text('Номер ControlNet:', 'Введите номера моделей ControlNet для скачивания через запятую/пробел.')
+commit_hash_widget = factory.create_text('Commit Hash:')
+huggingface_token_widget = factory.create_text('Токен HuggingFace:')
+
+ngrok_token_widget = factory.create_text('Токен Ngrok:')
+ngrock_button = factory.create_html('<a href="https://dashboard.ngrok.com/get-started/your-authtoken" target="_blank">Получить Ngrok Токен</a>', class_name="button_ngrok")
+ngrok_widget = factory.create_hbox([ngrok_token_widget, ngrock_button])
+
+zrok_token_widget = factory.create_text('Токен Zrok:')
+zrok_button = factory.create_html('<a href="https://colab.research.google.com/drive/1d2sjWDJi_GYBUavrHSuQyHTDuLy36WpU" target="_blank">Зарегать Zrok Токен</a>', class_name="button_ngrok")
+zrok_widget = factory.create_hbox([zrok_token_widget, zrok_button])
+
 commandline_arguments_options = "--listen --enable-insecure-extension-access --theme dark --no-half-vae --disable-console-progressbars --xformers"
-commandline_arguments_widget = widgets.Text(description='Аргументы:', value=commandline_arguments_options, style=style, layout=layout)
+commandline_arguments_widget = factory.create_text('Аргументы:', value=commandline_arguments_options)
+
+# Display Additional
+additional_widget_list = [additional_header,
+                          choose_changes_widget,
+                          HR,
+                          controlnet_widget,
+                          controlnet_num_widget,
+                          commit_hash_widget,
+                          huggingface_token_widget,
+                          ngrok_widget,
+                          zrok_widget,
+                          HR,
+                          commandline_arguments_widget]
 
-''' Display Additional'''
-additional_widget_list = [additional_header, latest_changes_widget, HR, controlnet_widget, controlnet_Num_widget, commit_hash_widget, optional_huggingface_token_widget, ngrok_widget, zrok_widget, HR, commandline_arguments_widget]
 if free_plan and env == "Google Colab": # remove ngrok from colab
     additional_widget_list.remove(ngrok_widget)
 if os.path.exists(webui_path): # remove selection after selection ;3
-    latest_changes_widget.children = [widget for widget in latest_changes_widget.children if widget != change_webui_widget]
-# ```
-all_additional_box = widgets.VBox(additional_widget_list).add_class("container").add_class("image_3")
-display(all_additional_box)
+    choose_changes_widget.children = [widget for widget in choose_changes_widget.children if widget != change_webui_widget]
+
+all_additional_box = factory.create_vbox(additional_widget_list, class_names=["container", "image_3"])
+factory.display(all_additional_box)
 
 # --- CUSTOM DOWNLOAD ---
-custom_download_header_popup = widgets.HTML('''
+custom_download_header_popup = factory.create_html('''
 <style>
 /* Term Colors */
 .sample_label {color: #dbafff;}
@@ -545,39 +262,40 @@ custom_download_header_popup = widgets.HTML('''
     </div>
 </div>
 ''')
-# ---
-Model_url_widget = widgets.Text(description='Model:', style=style, layout=layout)
-Vae_url_widget = widgets.Text(description='Vae:', style=style, layout=layout)
-LoRA_url_widget = widgets.Text(description='LoRa:', style=style, layout=layout)
-Embedding_url_widget = widgets.Text(description='Embedding:', style=style, layout=layout)
-Extensions_url_widget = widgets.Text(description='Extensions:', style=style, layout=layout)
-custom_file_urls_widget = widgets.Text(description='Файл (txt):', style=style, layout=layout)
-
-''' Display CustomDl'''
-all_custom_box = widgets.VBox([
+
+Model_url_widget = factory.create_text('Model:')
+Vae_url_widget = factory.create_text('Vae:')
+LoRA_url_widget = factory.create_text('LoRa:')
+Embedding_url_widget = factory.create_text('Embedding:')
+Extensions_url_widget = factory.create_text('Extensions:')
+custom_file_urls_widget = factory.create_text('Файл (txt):')
+
+# Display CustomDl
+all_custom_box = factory.create_vbox([
     custom_download_header_popup, Model_url_widget, Vae_url_widget, LoRA_url_widget, Embedding_url_widget, Extensions_url_widget, custom_file_urls_widget
-    ]).add_class("container").add_class("image_4").add_class("container_custom_downlad")
-display(all_custom_box)
+], class_names=["container", "image_4", "container_custom_downlad"])
+factory.display(all_custom_box)
 
 # --- Save Button ---
-save_button = widgets.Button(description='Сохранить').add_class("button_save")
-display(save_button)
+save_button = factory.create_button('Сохранить', class_name="button_save")
+factory.display(save_button)
 
 
 # ============ Load / Save - Settings V2 ============
 SETTINGS_FILE = f'{root_path}/settings.json'
 
-settings_keys = [
-    'Model', 'Model_Num', 'Inpainting_Model',
-    'Vae', 'Vae_Num',
-    'latest_webui', 'latest_exstensions', 'change_webui', 'detailed_download',
-    'controlnet', 'controlnet_Num', 'commit_hash', 'optional_huggingface_token',
+SETTINGS_KEYS = [
+    'model', 'model_num', 'inpainting_model',
+    'vae', 'vae_num', 'latest_webui', 'latest_exstensions',
+    'change_webui', 'detailed_download', 'controlnet',
+    'controlnet_num', 'commit_hash', 'huggingface_token',
     'ngrok_token', 'zrok_token', 'commandline_arguments',
-    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url', 'Extensions_url', 'custom_file_urls'
+    'Model_url', 'Vae_url', 'LoRA_url', 'Embedding_url',
+    'Extensions_url', 'custom_file_urls'
 ]
 
 def save_settings():
-    settings = {key: globals()[f"{key}_widget"].value for key in settings_keys}
+    settings = {key: globals()[f"{key}_widget"].value for key in SETTINGS_KEYS}
     with open(SETTINGS_FILE, 'w') as f:
         json.dump(settings, f, indent=2)
 
@@ -585,20 +303,20 @@ def load_settings():
     if os.path.exists(SETTINGS_FILE):
         with open(SETTINGS_FILE, 'r') as f:
             settings = json.load(f)
-            for key in settings_keys:
-                globals()[f"{key}_widget"].value = settings.get(key)
+            for key in SETTINGS_KEYS:
+                globals()[f"{key}_widget"].value = settings.get(key, "")
 
-def save_data(button):
-    save_settings()
-
-    # --- uhh - hide... ---
+def hide_widgets():
     widgets_list = [all_model_box, all_vae_box, all_additional_box, all_custom_box, save_button]
     for widget in widgets_list:
         widget.add_class("hide")
     time.sleep(0.5)
-
     widgets.Widget.close_all()
 
-settings = load_settings()
+def save_data(button):
+    save_settings()
+    hide_widgets()
+
+load_settings()
 save_button.on_click(save_data)
 

special/dl_display_results.py

@@ -33,7 +33,6 @@ embeddings_dir = f"{webui_path}/embeddings"
 loras_dir = f"{webui_path}/models/Lora"
 extensions_dir = f"{webui_path}/extensions"
 control_dir = f"{webui_path}/models/ControlNet"
-adetailer_dir = f"{webui_path}/models/adetailer"
 
 
 # ==================== CSS ====================
@@ -163,24 +162,23 @@ display(HTML(CSS))
 
 
 # ==================== WIDGETS ====================
+# Constants
 HR = widgets.HTML('<hr>')
-
-header_widget = widgets.HTML(value='''
-<div><span class="header_dl">DOWNLOAD RESULTS</span> <span style="color: grey; opacity: 0.25;">| v0.3</span></div>
-''')
+HEADER_DL = 'DOWNLOAD RESULTS'
+VERSION = 'v0.31'
 
 ## Functions
 def output_container_generator(header, items):
     header_widget = widgets.HTML(value=f'<div class="header_outputs_dl">{header} ➤</div>')
-    content = [widgets.HTML(value=f'<div class="items_dl">{item}</div>') for item in items]
-    container_widget = widgets.VBox([header_widget] + content).add_class("outputs_dl")
+    content_widgets = [widgets.HTML(value=f'<div class="items_dl">{item}</div>') for item in items]
+    container_widget = widgets.VBox([header_widget, *content_widgets]).add_class("outputs_dl")
     return container_widget
 
-def get_files_list(directory, extensions, exclude_dirs=()):
+def get_files_list(directory, extensions):
     return [file for file in os.listdir(directory) if file.endswith(extensions)]
 
 def get_folders_list(directory):
-    return [folder for folder in os.listdir(directory)]
+    return [folder for folder in os.listdir(extensions_dir) if os.path.isdir(os.path.join(extensions_dir, folder))]
 
 def get_controlnets_list(directory, filter_pattern):
     filter_name = re.compile(filter_pattern)
@@ -190,6 +188,10 @@ def get_controlnets_list(directory, filter_pattern):
         if not file.endswith(('.txt', '.yaml'))
     ]
 
+## Widgets
+header_widget = widgets.HTML(value=f'''
+<div><span class="header_dl">{HEADER_DL}</span> <span style="color: grey; opacity: 0.25;">| {VERSION}</span></div>
+''')
 
 # Models
 models_list = get_files_list(models_dir, '.safetensors')
@@ -210,7 +212,6 @@ extensions_widget = output_container_generator('Extensions', extensions_list).ad
 controlnets_list = get_controlnets_list(control_dir, r'^[^_]*_[^_]*_[^_]*_(.*)_fp16\.safetensors')
 controlnets_widget = output_container_generator('ControlNets', controlnets_list)
 
-
 ## Sorting and Output
 widgets_dict = {
     models_widget: models_list,