...

files_cells/notebooks/en/downloading_en.ipynb

@@ -34,6 +34,9 @@
         "webui_path = os.getenv('WEBUI_PATH')\n",
         "free_plan = os.getenv('FREE_PLAN')\n",
         "\n",
+        "OLD_UI = os.getenv('SDW_OLD_UI')\n",
+        "UI = os.getenv('SDW_UI')\n",
+        "\n",
         "\n",
         "# ================ LIBRARIES V2 ================\n",
         "flag_file = f\"{root_path}/libraries_installed.txt\"\n",
@@ -115,6 +118,18 @@
         "\n",
         "\n",
         "# ================= MAIN CODE =================\n",
+        "# WebUI switch\n",
+        "if os.path.exists(webui_path):\n",
+        "    if UI != OLD_UI:\n",
+        "        print(f'Switching the WebUI from \\033[33m{OLD_UI}\\033[0m to \\033[33m{UI}\\033[0m...', end='')\n",
+        "        shutil.rmtree(webui_path)\n",
+        "        # Updating the environment variable after a change\n",
+        "        os.environ['SDW_OLD_UI'] = UI\n",
+        "        print('\\rDone!')\n",
+        "        time.sleep(2)\n",
+        "        clear_output()\n",
+        "\n",
+        "# Unpacking the WebUI\n",
         "if not os.path.exists(webui_path):\n",
         "    start_install = int(time.time())\n",
         "    print(\"⌚ Unpacking Stable Diffusion...\" if change_webui != 'Forge' else \"⌚ Unpacking Stable Diffusion (Forge)...\", end='')\n",
@@ -221,7 +236,7 @@
         "    padding = (terminal_width - len(text)) // 2\n",
         "    return f\"\\033[1m\\033[36m{' ' * padding}{text}{' ' * padding}\\033[0m\\033[32m\"\n",
         "\n",
-        "def format_output(url, dst_dir, file_name):\n",
+        "def format_output(url, dst_dir, file_name, image_name=None, image_url=None):\n",
         "    info = center_text(f\"[{file_name.split('.')[0]}]\")\n",
         "    separation_line = '\\033[32m' + '---' * 20\n",
         "\n",
@@ -230,89 +245,81 @@
         "    print(f\"\\033[33mSAVE DIR: \\033[34m{dst_dir}\")\n",
         "    print(f\"\\033[33mFILE NAME: \\033[34m{file_name}\\033[0m\")\n",
         "\n",
+        "    if 'civitai' in url and image_url:\n",
+        "        print(f\"\\033[32m[Preview DL]:\\033[0m {image_name} - {image_url}\\n\")\n",
+        "\n",
         "''' GET CivitAi API - DATA '''\n",
         "\n",
         "def CivitAi_API(url, file_name=None):\n",
-        "    SUPPORT_TYPES = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')\n",
+        "    SUPPORT_TYPES = ('Checkpoint', 'TextualInversion', 'LORA')\n",
         "    CIVITAI_TOKEN = \"62c0c5956b2f9defbd844d754000180b\"\n",
         "\n",
         "    url = url.split('?token=')[0] if '?token=' in url else url\n",
         "    url = url.replace('?type=', f'?token={CIVITAI_TOKEN}&type=') if '?type=' in url else f\"{url}?token={CIVITAI_TOKEN}\"\n",
         "\n",
         "    def get_model_data(url):\n",
-        "        if \"civitai.com/models/\" in url:\n",
-        "            if '?modelVersionId=' in url:\n",
-        "                version_id = url.split('?modelVersionId=')[1]\n",
-        "                return requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\").json()\n",
+        "        base_url = \"https://civitai.com/api/v1\"\n",
+        "        try:\n",
+        "            if \"civitai.com/models/\" in url:\n",
+        "                if '?modelVersionId=' in url:\n",
+        "                    version_id = url.split('?modelVersionId=')[1]\n",
+        "                else:\n",
+        "                    model_id = url.split('/models/')[1].split('/')[0]\n",
+        "                    model_data = requests.get(f\"{base_url}/models/{model_id}\").json()\n",
+        "                    version_id = model_data['modelVersions'][0].get('id')\n",
         "            else:\n",
-        "                model_id = url.split('/models/')[1].split('/')[0]\n",
-        "                return requests.get(f\"https://civitai.com/api/v1/models/{model_id}\").json()\n",
-        "        else:\n",
-        "            version_id = url.split('/models/')[1].split('/')[0]\n",
-        "            return requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\").json()\n",
+        "                version_id = url.split('/models/')[1].split('/')[0]\n",
+        "\n",
+        "            return requests.get(f\"{base_url}/model-versions/{version_id}\").json()\n",
+        "        except (KeyError, IndexError, requests.RequestException) as e:\n",
+        "            return None\n",
         "\n",
         "    data = get_model_data(url)\n",
         "\n",
         "    if not data:\n",
-        "        return None, None, None, None, None, None, None\n",
+        "        print(\"\\033[31m[Data Info]:\\033[0m Failed to retrieve data from the API.\\n\")\n",
+        "        return 'None', None, None, None, None, None, None\n",
         "\n",
         "    def extract_model_info(url, data):\n",
-        "        if \"civitai.com/models/\" in url:\n",
-        "            if '?modelVersionId=' in url:\n",
-        "                model_type = data['model']['type']\n",
-        "                model_name = data['files'][0]['name']\n",
-        "            else:\n",
-        "                model_type = data['type']\n",
-        "                model_name = data['modelVersions'][0]['files'][0]['name']\n",
-        "        elif 'type=' in url:\n",
-        "            model_type = parse_qs(urlparse(url).query).get('type', [''])[0]\n",
-        "            if 'model' in model_type.lower():\n",
-        "                model_name = data['files'][0]['name']\n",
-        "            else:\n",
+        "        model_type = data['model']['type']\n",
+        "        model_name = data['files'][0]['name']\n",
+        "\n",
+        "        if 'type=' in url:\n",
+        "            url_model_type = parse_qs(urlparse(url).query).get('type', [''])[0].lower()\n",
+        "            if 'vae' in url_model_type:\n",
+        "                model_type = data['files'][1]['type']\n",
         "                model_name = data['files'][1]['name']\n",
-        "        else:\n",
-        "            model_type = data['model']['type']\n",
-        "            model_name = data['files'][0]['name']\n",
+        "\n",
         "        return model_type, model_name\n",
         "\n",
         "    model_type, model_name = extract_model_info(url, data)\n",
         "    model_name = file_name or model_name\n",
         "\n",
-        "    def get_download_url(url, data, model_type):\n",
-        "        if \"civitai.com/models/\" in url:\n",
-        "            if '?modelVersionId=' in url:\n",
-        "                return data.get('downloadUrl')\n",
-        "            else:\n",
-        "                return data[\"modelVersions\"][0].get(\"downloadUrl\", \"\")\n",
-        "        elif 'type=' in url:\n",
-        "            if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):\n",
-        "                return data['files'][0]['downloadUrl']\n",
-        "            else:\n",
-        "                return data['files'][1]['downloadUrl']\n",
+        "    def get_download_url(data, model_type):\n",
+        "        if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):\n",
+        "            return data['files'][0]['downloadUrl']\n",
         "        else:\n",
-        "            return data.get('downloadUrl')\n",
+        "            return data['files'][1]['downloadUrl']\n",
         "\n",
-        "    download_url = get_download_url(url, data, model_type)\n",
-        "    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url)\n",
+        "    download_url = get_download_url(data, model_type)\n",
         "\n",
         "    def get_image_info(data, model_type, model_name):\n",
-        "        image_url, image_name = None, None\n",
-        "        if any(t in model_type for t in SUPPORT_TYPES):\n",
-        "            try:\n",
-        "                images = data.get('images') or data['modelVersions'][0].get('images', [])\n",
-        "                if env == 'Kaggle':\n",
-        "                    image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)\n",
-        "                else:\n",
-        "                    image_url = images[0]['url'] if images else None\n",
-        "            except KeyError:\n",
-        "                pass\n",
+        "        if not any(t in model_type for t in SUPPORT_TYPES):\n",
+        "            return None, None\n",
+        "\n",
+        "        for image in data.get('images', []):\n",
+        "            if image['nsfwLevel'] >= 4 and env == 'Kaggle':  # Filter NSFW images for Kaggle\n",
+        "                continue\n",
+        "            image_url = image['url']\n",
+        "            image_extension = image_url.split('.')[-1]\n",
+        "            image_name = f\"{model_name.split('.')[0]}.preview.{image_extension}\" if image_url else None\n",
+        "            return image_url, image_name\n",
         "\n",
-        "            image_name = f\"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}\" if image_url else None\n",
-        "        return image_url, image_name\n",
+        "        return None, None\n",
         "\n",
         "    image_url, image_name = get_image_info(data, model_type, model_name)\n",
         "\n",
-        "    return f\"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}\", clean_url, model_type, model_name, image_url, image_name, data\n",
+        "    return f\"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}\", download_url, model_type, model_name, image_url, image_name, data\n",
         "\n",
         "''' Main Download Code '''\n",
         "\n",
@@ -382,14 +389,10 @@
         "\n",
         "    \"\"\" Formatted info output \"\"\"\n",
         "    model_name_or_basename = file_name if file_name else basename\n",
-        "    format_output(clean_url or url, dst_dir, model_name_or_basename)\n",
-        "\n",
-        "    # print(url, dst_dir, model_name_or_basename)\n",
-        "    if 'civitai' in url:\n",
-        "        if not data:\n",
-        "            print(\"\\033[31m[Data Info]:\\033[0m Failed to retrieve data from the API.\\n\")\n",
-        "        if data and image_name:\n",
-        "            print(f\"\\033[32m[Preview DL]:\\033[0m {image_name} - {image_url}\\n\")\n",
+        "    try:\n",
+        "        format_output(clean_url or url, dst_dir, model_name_or_basename, image_url, image_name)\n",
+        "    except UnboundLocalError:\n",
+        "        format_output(clean_url or url, dst_dir, model_name_or_basename, None, None)\n",
         "\n",
         "    # =====================\n",
         "    def run_aria2c(url, dst_dir, file_name=None, args=\"\", header=\"\"):\n",

files_cells/notebooks/en/launch_en.ipynb

@@ -29,6 +29,7 @@
         "import time\n",
         "import json\n",
         "import requests\n",
+        "import subprocess\n",
         "import cloudpickle as pickle\n",
         "from datetime import timedelta\n",
         "from IPython.display import clear_output\n",
@@ -55,6 +56,17 @@
         "change_webui = settings.get('change_webui', \"\")\n",
         "\n",
         "\n",
+        "# ========================== OTHER ==========================\n",
+        "def is_package_installed(package_name):\n",
+        "    try:\n",
+        "        subprocess.run([\"npm\", \"ls\", \"-g\", package_name], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)\n",
+        "        return True\n",
+        "    except subprocess.CalledProcessError:\n",
+        "        return False\n",
+        "\n",
+        "lt_tunnel = is_package_installed('localtunnel')\n",
+        "\n",
+        "\n",
         "# ======================== TUNNEL V2 ========================\n",
         "print('Please Wait...')\n",
         "\n",
@@ -80,7 +92,9 @@
         "tunnel_port = 1834\n",
         "tunnel = tunnel_class(tunnel_port)\n",
         "tunnel.add_tunnel(command=\"cl tunnel --url localhost:{port}\", name=\"cl\", pattern=re.compile(r\"[\\w-]+\\.trycloudflare\\.com\"))\n",
-        "tunnel.add_tunnel(command=\"lt --port {port}\", name=\"lt\", pattern=re.compile(r\"[\\w-]+\\.loca\\.lt\"), note=\"Password : \" + \"\\033[32m\" + public_ipv4 + \"\\033[0m\" + \" rerun cell if 404 error.\")\n",
+        "\n",
+        "if lt_tunnel:\n",
+        "    tunnel.add_tunnel(command=\"lt --port {port}\", name=\"lt\", pattern=re.compile(r\"[\\w-]+\\.loca\\.lt\"), note=\"Password : \" + \"\\033[32m\" + public_ipv4 + \"\\033[0m\" + \" rerun cell if 404 error.\")\n",
         "\n",
         "if zrok_token:\n",
         "    !zrok enable {zrok_token} &> /dev/null\n",
@@ -89,7 +103,7 @@
         "clear_output()\n",
         "\n",
         "\n",
-        "# =============== Automatic Fixing Path V3 ===============\n",
+        "# ================= Automatic Fixing Path V3 ================\n",
         "paths_to_check = {\n",
         "    \"tagger_hf_cache_dir\": f\"{webui_path}/models/interrogators/\",\n",
         "    \"additional_networks_extra_lora_path\": f\"{webui_path}/models/Lora/\",\n",
@@ -110,8 +124,6 @@
         "            sed_command = f\"sed -i 's|\\\"{key}\\\": \\\".*\\\"|\\\"{key}\\\": \\\"{value}\\\"|' {config_path}\"\n",
         "            os.system(sed_command)\n",
         "\n",
-        "    if env == 'Kaggle':\n",
-        "        !sed -i 's/\"civitai_interface\\/NSFW content\\/visible\": true,/\"civitai_interface\\/NSFW content\\/visible\": false,/' {webui_path}/ui-config.json\n",
         "\n",
         "with tunnel:\n",
         "    %cd {webui_path}\n",

files_cells/notebooks/en/widgets_en.ipynb

@@ -116,6 +116,7 @@
         "'''\n",
         "display(HTML(JS))\n",
         "\n",
+        "\n",
         "# ==================== WIDGETS V2 ====================\n",
         "HR = widgets.HTML('<hr>')\n",
         "\n",
@@ -245,8 +246,6 @@
         "\n",
         "if free_plan and env == \"Google Colab\": # remove ngrok from colab\n",
         "    additional_widget_list.remove(ngrok_widget)\n",
-        "if os.path.exists(webui_path): # remove selection after selection ;3\n",
-        "    choose_changes_widget.children = [widget for widget in choose_changes_widget.children if widget != change_webui_widget]\n",
         "\n",
         "all_additional_box = factory.create_vbox(additional_widget_list, class_names=[\"container\", \"image_3\"])\n",
         "factory.display(all_additional_box)\n",
@@ -296,6 +295,24 @@
         "factory.display(save_button)\n",
         "\n",
         "\n",
+        "# ==================== OTHER FUNC ====================\n",
+        "# Setup UI\n",
+        "def setup_webui(change_webui_widget):\n",
+        "    if 'SDW_UI' in os.environ:\n",
+        "        UI = os.environ['SDW_UI']\n",
+        "    else:\n",
+        "        os.environ['SDW_UI'] = change_webui_widget\n",
+        "        os.environ['SDW_OLD_UI'] = change_webui_widget\n",
+        "\n",
+        "    UI = os.getenv('SDW_UI')\n",
+        "    OLD_UI = os.getenv('SDW_OLD_UI')\n",
+        "\n",
+        "    if UI != change_webui_widget:\n",
+        "        os.environ['SDW_UI'] = change_webui_widget\n",
+        "\n",
+        "    # print(f'Current UI: {os.getenv(\"SDW_UI\")}, Old UI: {os.getenv(\"SDW_OLD_UI\")}')\n",
+        "\n",
+        "\n",
         "# ============ Load / Save - Settings V2 ============\n",
         "SETTINGS_FILE = f'{root_path}/settings.json'\n",
         "\n",
@@ -329,6 +346,7 @@
         "    widgets.Widget.close_all()\n",
         "\n",
         "def save_data(button):\n",
+        "    setup_webui(change_webui_widget.value)\n",
         "    save_settings()\n",
         "    hide_widgets()\n",
         "\n",

files_cells/notebooks/ru/downloading_ru.ipynb

@@ -34,6 +34,9 @@
         "webui_path = os.getenv('WEBUI_PATH')\n",
         "free_plan = os.getenv('FREE_PLAN')\n",
         "\n",
+        "OLD_UI = os.getenv('SDW_OLD_UI')\n",
+        "UI = os.getenv('SDW_UI')\n",
+        "\n",
         "\n",
         "# ================ LIBRARIES V2 ================\n",
         "flag_file = f\"{root_path}/libraries_installed.txt\"\n",
@@ -115,6 +118,18 @@
         "\n",
         "\n",
         "# ================= MAIN CODE =================\n",
+        "# WebUI switch\n",
+        "if os.path.exists(webui_path):\n",
+        "    if UI != OLD_UI:\n",
+        "        print(f'Переключение веб-интерфейса с \\033[33m{OLD_UI}\\033[0m на \\033[33m{UI}\\033[0m...', end='')\n",
+        "        shutil.rmtree(webui_path)\n",
+        "        # Updating the environment variable after a change\n",
+        "        os.environ['SDW_OLD_UI'] = UI\n",
+        "        print('\\rГотово!')\n",
+        "        time.sleep(2)\n",
+        "        clear_output()\n",
+        "\n",
+        "# Unpacking the WebUI\n",
         "if not os.path.exists(webui_path):\n",
         "    start_install = int(time.time())\n",
         "    print(\"⌚ Распаковка Stable Diffusion...\" if change_webui != 'Forge' else \"⌚ Распаковка Stable Diffusion (Forge)...\", end='')\n",
@@ -221,7 +236,7 @@
         "    padding = (terminal_width - len(text)) // 2\n",
         "    return f\"\\033[1m\\033[36m{' ' * padding}{text}{' ' * padding}\\033[0m\\033[32m\"\n",
         "\n",
-        "def format_output(url, dst_dir, file_name):\n",
+        "def format_output(url, dst_dir, file_name, image_name=None, image_url=None):\n",
         "    info = center_text(f\"[{file_name.split('.')[0]}]\")\n",
         "    separation_line = '\\033[32m' + '---' * 20\n",
         "\n",
@@ -230,89 +245,81 @@
         "    print(f\"\\033[33mSAVE DIR: \\033[34m{dst_dir}\")\n",
         "    print(f\"\\033[33mFILE NAME: \\033[34m{file_name}\\033[0m\")\n",
         "\n",
+        "    if 'civitai' in url and image_url:\n",
+        "        print(f\"\\033[32m[Preview DL]:\\033[0m {image_name} - {image_url}\\n\")\n",
+        "\n",
         "''' GET CivitAi API - DATA '''\n",
         "\n",
         "def CivitAi_API(url, file_name=None):\n",
-        "    SUPPORT_TYPES = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')\n",
+        "    SUPPORT_TYPES = ('Checkpoint', 'TextualInversion', 'LORA')\n",
         "    CIVITAI_TOKEN = \"62c0c5956b2f9defbd844d754000180b\"\n",
         "\n",
         "    url = url.split('?token=')[0] if '?token=' in url else url\n",
         "    url = url.replace('?type=', f'?token={CIVITAI_TOKEN}&type=') if '?type=' in url else f\"{url}?token={CIVITAI_TOKEN}\"\n",
         "\n",
         "    def get_model_data(url):\n",
-        "        if \"civitai.com/models/\" in url:\n",
-        "            if '?modelVersionId=' in url:\n",
-        "                version_id = url.split('?modelVersionId=')[1]\n",
-        "                return requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\").json()\n",
+        "        base_url = \"https://civitai.com/api/v1\"\n",
+        "        try:\n",
+        "            if \"civitai.com/models/\" in url:\n",
+        "                if '?modelVersionId=' in url:\n",
+        "                    version_id = url.split('?modelVersionId=')[1]\n",
+        "                else:\n",
+        "                    model_id = url.split('/models/')[1].split('/')[0]\n",
+        "                    model_data = requests.get(f\"{base_url}/models/{model_id}\").json()\n",
+        "                    version_id = model_data['modelVersions'][0].get('id')\n",
         "            else:\n",
-        "                model_id = url.split('/models/')[1].split('/')[0]\n",
-        "                return requests.get(f\"https://civitai.com/api/v1/models/{model_id}\").json()\n",
-        "        else:\n",
-        "            version_id = url.split('/models/')[1].split('/')[0]\n",
-        "            return requests.get(f\"https://civitai.com/api/v1/model-versions/{version_id}\").json()\n",
+        "                version_id = url.split('/models/')[1].split('/')[0]\n",
+        "\n",
+        "            return requests.get(f\"{base_url}/model-versions/{version_id}\").json()\n",
+        "        except (KeyError, IndexError, requests.RequestException) as e:\n",
+        "            return None\n",
         "\n",
         "    data = get_model_data(url)\n",
         "\n",
         "    if not data:\n",
-        "        return None, None, None, None, None, None, None\n",
+        "        print(\"\\033[31m[Data Info]:\\033[0m Failed to retrieve data from the API.\\n\")\n",
+        "        return 'None', None, None, None, None, None, None\n",
         "\n",
         "    def extract_model_info(url, data):\n",
-        "        if \"civitai.com/models/\" in url:\n",
-        "            if '?modelVersionId=' in url:\n",
-        "                model_type = data['model']['type']\n",
-        "                model_name = data['files'][0]['name']\n",
-        "            else:\n",
-        "                model_type = data['type']\n",
-        "                model_name = data['modelVersions'][0]['files'][0]['name']\n",
-        "        elif 'type=' in url:\n",
-        "            model_type = parse_qs(urlparse(url).query).get('type', [''])[0]\n",
-        "            if 'model' in model_type.lower():\n",
-        "                model_name = data['files'][0]['name']\n",
-        "            else:\n",
+        "        model_type = data['model']['type']\n",
+        "        model_name = data['files'][0]['name']\n",
+        "\n",
+        "        if 'type=' in url:\n",
+        "            url_model_type = parse_qs(urlparse(url).query).get('type', [''])[0].lower()\n",
+        "            if 'vae' in url_model_type:\n",
+        "                model_type = data['files'][1]['type']\n",
         "                model_name = data['files'][1]['name']\n",
-        "        else:\n",
-        "            model_type = data['model']['type']\n",
-        "            model_name = data['files'][0]['name']\n",
+        "\n",
         "        return model_type, model_name\n",
         "\n",
         "    model_type, model_name = extract_model_info(url, data)\n",
         "    model_name = file_name or model_name\n",
         "\n",
-        "    def get_download_url(url, data, model_type):\n",
-        "        if \"civitai.com/models/\" in url:\n",
-        "            if '?modelVersionId=' in url:\n",
-        "                return data.get('downloadUrl')\n",
-        "            else:\n",
-        "                return data[\"modelVersions\"][0].get(\"downloadUrl\", \"\")\n",
-        "        elif 'type=' in url:\n",
-        "            if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):\n",
-        "                return data['files'][0]['downloadUrl']\n",
-        "            else:\n",
-        "                return data['files'][1]['downloadUrl']\n",
+        "    def get_download_url(data, model_type):\n",
+        "        if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):\n",
+        "            return data['files'][0]['downloadUrl']\n",
         "        else:\n",
-        "            return data.get('downloadUrl')\n",
+        "            return data['files'][1]['downloadUrl']\n",
         "\n",
-        "    download_url = get_download_url(url, data, model_type)\n",
-        "    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url)\n",
+        "    download_url = get_download_url(data, model_type)\n",
         "\n",
         "    def get_image_info(data, model_type, model_name):\n",
-        "        image_url, image_name = None, None\n",
-        "        if any(t in model_type for t in SUPPORT_TYPES):\n",
-        "            try:\n",
-        "                images = data.get('images') or data['modelVersions'][0].get('images', [])\n",
-        "                if env == 'Kaggle':\n",
-        "                    image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)\n",
-        "                else:\n",
-        "                    image_url = images[0]['url'] if images else None\n",
-        "            except KeyError:\n",
-        "                pass\n",
+        "        if not any(t in model_type for t in SUPPORT_TYPES):\n",
+        "            return None, None\n",
+        "\n",
+        "        for image in data.get('images', []):\n",
+        "            if image['nsfwLevel'] >= 4 and env == 'Kaggle':  # Filter NSFW images for Kaggle\n",
+        "                continue\n",
+        "            image_url = image['url']\n",
+        "            image_extension = image_url.split('.')[-1]\n",
+        "            image_name = f\"{model_name.split('.')[0]}.preview.{image_extension}\" if image_url else None\n",
+        "            return image_url, image_name\n",
         "\n",
-        "            image_name = f\"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}\" if image_url else None\n",
-        "        return image_url, image_name\n",
+        "        return None, None\n",
         "\n",
         "    image_url, image_name = get_image_info(data, model_type, model_name)\n",
         "\n",
-        "    return f\"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}\", clean_url, model_type, model_name, image_url, image_name, data\n",
+        "    return f\"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}\", download_url, model_type, model_name, image_url, image_name, data\n",
         "\n",
         "''' Main Download Code '''\n",
         "\n",
@@ -382,14 +389,10 @@
         "\n",
         "    \"\"\" Formatted info output \"\"\"\n",
         "    model_name_or_basename = file_name if file_name else basename\n",
-        "    format_output(clean_url or url, dst_dir, model_name_or_basename)\n",
-        "\n",
-        "    # print(url, dst_dir, model_name_or_basename)\n",
-        "    if 'civitai' in url:\n",
-        "        if not data:\n",
-        "            print(\"\\033[31m[Data Info]:\\033[0m Failed to retrieve data from the API.\\n\")\n",
-        "        if data and image_name:\n",
-        "            print(f\"\\033[32m[Preview DL]:\\033[0m {image_name} - {image_url}\\n\")\n",
+        "    try:\n",
+        "        format_output(clean_url or url, dst_dir, model_name_or_basename, image_url, image_name)\n",
+        "    except UnboundLocalError:\n",
+        "        format_output(clean_url or url, dst_dir, model_name_or_basename, None, None)\n",
         "\n",
         "    # =====================\n",
         "    def run_aria2c(url, dst_dir, file_name=None, args=\"\", header=\"\"):\n",

files_cells/notebooks/ru/launch_ru.ipynb

@@ -29,6 +29,7 @@
         "import time\n",
         "import json\n",
         "import requests\n",
+        "import subprocess\n",
         "import cloudpickle as pickle\n",
         "from datetime import timedelta\n",
         "from IPython.display import clear_output\n",
@@ -55,6 +56,17 @@
         "change_webui = settings.get('change_webui', \"\")\n",
         "\n",
         "\n",
+        "# ========================== OTHER ==========================\n",
+        "def is_package_installed(package_name):\n",
+        "    try:\n",
+        "        subprocess.run([\"npm\", \"ls\", \"-g\", package_name], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)\n",
+        "        return True\n",
+        "    except subprocess.CalledProcessError:\n",
+        "        return False\n",
+        "\n",
+        "lt_tunnel = is_package_installed('localtunnel')\n",
+        "\n",
+        "\n",
         "# ======================== TUNNEL V2 ========================\n",
         "print('Please Wait...')\n",
         "\n",
@@ -80,7 +92,9 @@
         "tunnel_port = 1834\n",
         "tunnel = tunnel_class(tunnel_port)\n",
         "tunnel.add_tunnel(command=\"cl tunnel --url localhost:{port}\", name=\"cl\", pattern=re.compile(r\"[\\w-]+\\.trycloudflare\\.com\"))\n",
-        "tunnel.add_tunnel(command=\"lt --port {port}\", name=\"lt\", pattern=re.compile(r\"[\\w-]+\\.loca\\.lt\"), note=\"Password : \" + \"\\033[32m\" + public_ipv4 + \"\\033[0m\" + \" rerun cell if 404 error.\")\n",
+        "\n",
+        "if lt_tunnel:\n",
+        "    tunnel.add_tunnel(command=\"lt --port {port}\", name=\"lt\", pattern=re.compile(r\"[\\w-]+\\.loca\\.lt\"), note=\"Password : \" + \"\\033[32m\" + public_ipv4 + \"\\033[0m\" + \" rerun cell if 404 error.\")\n",
         "\n",
         "if zrok_token:\n",
         "    !zrok enable {zrok_token} &> /dev/null\n",
@@ -89,7 +103,7 @@
         "clear_output()\n",
         "\n",
         "\n",
-        "# =============== Automatic Fixing Path V3 ===============\n",
+        "# ================= Automatic Fixing Path V3 ================\n",
         "paths_to_check = {\n",
         "    \"tagger_hf_cache_dir\": f\"{webui_path}/models/interrogators/\",\n",
         "    \"additional_networks_extra_lora_path\": f\"{webui_path}/models/Lora/\",\n",
@@ -110,8 +124,6 @@
         "            sed_command = f\"sed -i 's|\\\"{key}\\\": \\\".*\\\"|\\\"{key}\\\": \\\"{value}\\\"|' {config_path}\"\n",
         "            os.system(sed_command)\n",
         "\n",
-        "    if env == 'Kaggle':\n",
-        "        !sed -i 's/\"civitai_interface\\/NSFW content\\/visible\": true,/\"civitai_interface\\/NSFW content\\/visible\": false,/' {webui_path}/ui-config.json\n",
         "\n",
         "with tunnel:\n",
         "    %cd {webui_path}\n",

files_cells/notebooks/ru/widgets_ru.ipynb

@@ -116,6 +116,7 @@
         "'''\n",
         "display(HTML(JS))\n",
         "\n",
+        "\n",
         "# ==================== WIDGETS V2 ====================\n",
         "HR = widgets.HTML('<hr>')\n",
         "\n",
@@ -245,8 +246,6 @@
         "\n",
         "if free_plan and env == \"Google Colab\": # remove ngrok from colab\n",
         "    additional_widget_list.remove(ngrok_widget)\n",
-        "if os.path.exists(webui_path): # remove selection after selection ;3\n",
-        "    choose_changes_widget.children = [widget for widget in choose_changes_widget.children if widget != change_webui_widget]\n",
         "\n",
         "all_additional_box = factory.create_vbox(additional_widget_list, class_names=[\"container\", \"image_3\"])\n",
         "factory.display(all_additional_box)\n",
@@ -296,6 +295,24 @@
         "factory.display(save_button)\n",
         "\n",
         "\n",
+        "# ==================== OTHER FUNC ====================\n",
+        "# Setup UI\n",
+        "def setup_webui(change_webui_widget):\n",
+        "    if 'SDW_UI' in os.environ:\n",
+        "        UI = os.environ['SDW_UI']\n",
+        "    else:\n",
+        "        os.environ['SDW_UI'] = change_webui_widget\n",
+        "        os.environ['SDW_OLD_UI'] = change_webui_widget\n",
+        "\n",
+        "    UI = os.getenv('SDW_UI')\n",
+        "    OLD_UI = os.getenv('SDW_OLD_UI')\n",
+        "\n",
+        "    if UI != change_webui_widget:\n",
+        "        os.environ['SDW_UI'] = change_webui_widget\n",
+        "\n",
+        "    # print(f'Current UI: {os.getenv(\"SDW_UI\")}, Old UI: {os.getenv(\"SDW_OLD_UI\")}')\n",
+        "\n",
+        "\n",
         "# ============ Load / Save - Settings V2 ============\n",
         "SETTINGS_FILE = f'{root_path}/settings.json'\n",
         "\n",
@@ -329,6 +346,7 @@
         "    widgets.Widget.close_all()\n",
         "\n",
         "def save_data(button):\n",
+        "    setup_webui(change_webui_widget.value)\n",
         "    save_settings()\n",
         "    hide_widgets()\n",
         "\n",

files_cells/python/en/downloading_en.py

@@ -24,6 +24,9 @@ root_path = os.getenv('ROOT_PATH')
 webui_path = os.getenv('WEBUI_PATH')
 free_plan = os.getenv('FREE_PLAN')
 
+OLD_UI = os.getenv('SDW_OLD_UI')
+UI = os.getenv('SDW_UI')
+
 
 # ================ LIBRARIES V2 ================
 flag_file = f"{root_path}/libraries_installed.txt"
@@ -105,6 +108,18 @@ except:
 
 
 # ================= MAIN CODE =================
+# WebUI switch
+if os.path.exists(webui_path):
+    if UI != OLD_UI:
+        print(f'Switching the WebUI from \033[33m{OLD_UI}\033[0m to \033[33m{UI}\033[0m...', end='')
+        shutil.rmtree(webui_path)
+        # Updating the environment variable after a change
+        os.environ['SDW_OLD_UI'] = UI
+        print('\rDone!')
+        time.sleep(2)
+        clear_output()
+
+# Unpacking the WebUI
 if not os.path.exists(webui_path):
     start_install = int(time.time())
     print("⌚ Unpacking Stable Diffusion..." if change_webui != 'Forge' else "⌚ Unpacking Stable Diffusion (Forge)...", end='')
@@ -211,7 +226,7 @@ def center_text(text, terminal_width=45):
     padding = (terminal_width - len(text)) // 2
     return f"\033[1m\033[36m{' ' * padding}{text}{' ' * padding}\033[0m\033[32m"
 
-def format_output(url, dst_dir, file_name):
+def format_output(url, dst_dir, file_name, image_name=None, image_url=None):
     info = center_text(f"[{file_name.split('.')[0]}]")
     separation_line = '\033[32m' + '---' * 20
 
@@ -220,89 +235,81 @@ def format_output(url, dst_dir, file_name):
     print(f"\033[33mSAVE DIR: \033[34m{dst_dir}")
     print(f"\033[33mFILE NAME: \033[34m{file_name}\033[0m")
 
+    if 'civitai' in url and image_url:
+        print(f"\033[32m[Preview DL]:\033[0m {image_name} - {image_url}\n")
+
 ''' GET CivitAi API - DATA '''
 
 def CivitAi_API(url, file_name=None):
-    SUPPORT_TYPES = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')
+    SUPPORT_TYPES = ('Checkpoint', 'TextualInversion', 'LORA')
     CIVITAI_TOKEN = "62c0c5956b2f9defbd844d754000180b"
 
     url = url.split('?token=')[0] if '?token=' in url else url
     url = url.replace('?type=', f'?token={CIVITAI_TOKEN}&type=') if '?type=' in url else f"{url}?token={CIVITAI_TOKEN}"
 
     def get_model_data(url):
-        if "civitai.com/models/" in url:
-            if '?modelVersionId=' in url:
-                version_id = url.split('?modelVersionId=')[1]
-                return requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}").json()
+        base_url = "https://civitai.com/api/v1"
+        try:
+            if "civitai.com/models/" in url:
+                if '?modelVersionId=' in url:
+                    version_id = url.split('?modelVersionId=')[1]
+                else:
+                    model_id = url.split('/models/')[1].split('/')[0]
+                    model_data = requests.get(f"{base_url}/models/{model_id}").json()
+                    version_id = model_data['modelVersions'][0].get('id')
             else:
-                model_id = url.split('/models/')[1].split('/')[0]
-                return requests.get(f"https://civitai.com/api/v1/models/{model_id}").json()
-        else:
-            version_id = url.split('/models/')[1].split('/')[0]
-            return requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}").json()
+                version_id = url.split('/models/')[1].split('/')[0]
+
+            return requests.get(f"{base_url}/model-versions/{version_id}").json()
+        except (KeyError, IndexError, requests.RequestException) as e:
+            return None
 
     data = get_model_data(url)
 
     if not data:
-        return None, None, None, None, None, None, None
+        print("\033[31m[Data Info]:\033[0m Failed to retrieve data from the API.\n")
+        return 'None', None, None, None, None, None, None
 
     def extract_model_info(url, data):
-        if "civitai.com/models/" in url:
-            if '?modelVersionId=' in url:
-                model_type = data['model']['type']
-                model_name = data['files'][0]['name']
-            else:
-                model_type = data['type']
-                model_name = data['modelVersions'][0]['files'][0]['name']
-        elif 'type=' in url:
-            model_type = parse_qs(urlparse(url).query).get('type', [''])[0]
-            if 'model' in model_type.lower():
-                model_name = data['files'][0]['name']
-            else:
+        model_type = data['model']['type']
+        model_name = data['files'][0]['name']
+
+        if 'type=' in url:
+            url_model_type = parse_qs(urlparse(url).query).get('type', [''])[0].lower()
+            if 'vae' in url_model_type:
+                model_type = data['files'][1]['type']
                 model_name = data['files'][1]['name']
-        else:
-            model_type = data['model']['type']
-            model_name = data['files'][0]['name']
+
         return model_type, model_name
 
     model_type, model_name = extract_model_info(url, data)
     model_name = file_name or model_name
 
-    def get_download_url(url, data, model_type):
-        if "civitai.com/models/" in url:
-            if '?modelVersionId=' in url:
-                return data.get('downloadUrl')
-            else:
-                return data["modelVersions"][0].get("downloadUrl", "")
-        elif 'type=' in url:
-            if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):
-                return data['files'][0]['downloadUrl']
-            else:
-                return data['files'][1]['downloadUrl']
+    def get_download_url(data, model_type):
+        if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):
+            return data['files'][0]['downloadUrl']
         else:
-            return data.get('downloadUrl')
+            return data['files'][1]['downloadUrl']
 
-    download_url = get_download_url(url, data, model_type)
-    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url)
+    download_url = get_download_url(data, model_type)
 
     def get_image_info(data, model_type, model_name):
-        image_url, image_name = None, None
-        if any(t in model_type for t in SUPPORT_TYPES):
-            try:
-                images = data.get('images') or data['modelVersions'][0].get('images', [])
-                if env == 'Kaggle':
-                    image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)
-                else:
-                    image_url = images[0]['url'] if images else None
-            except KeyError:
-                pass
+        if not any(t in model_type for t in SUPPORT_TYPES):
+            return None, None
+
+        for image in data.get('images', []):
+            if image['nsfwLevel'] >= 4 and env == 'Kaggle':  # Filter NSFW images for Kaggle
+                continue
+            image_url = image['url']
+            image_extension = image_url.split('.')[-1]
+            image_name = f"{model_name.split('.')[0]}.preview.{image_extension}" if image_url else None
+            return image_url, image_name
 
-            image_name = f"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}" if image_url else None
-        return image_url, image_name
+        return None, None
 
     image_url, image_name = get_image_info(data, model_type, model_name)
 
-    return f"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}", clean_url, model_type, model_name, image_url, image_name, data
+    return f"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}", download_url, model_type, model_name, image_url, image_name, data
 
 ''' Main Download Code '''
 
@@ -372,14 +379,10 @@ def manual_download(url, dst_dir, file_name):
 
     """ Formatted info output """
     model_name_or_basename = file_name if file_name else basename
-    format_output(clean_url or url, dst_dir, model_name_or_basename)
-
-    # print(url, dst_dir, model_name_or_basename)
-    if 'civitai' in url:
-        if not data:
-            print("\033[31m[Data Info]:\033[0m Failed to retrieve data from the API.\n")
-        if data and image_name:
-            print(f"\033[32m[Preview DL]:\033[0m {image_name} - {image_url}\n")
+    try:
+        format_output(clean_url or url, dst_dir, model_name_or_basename, image_url, image_name)
+    except UnboundLocalError:
+        format_output(clean_url or url, dst_dir, model_name_or_basename, None, None)
 
     # =====================
     def run_aria2c(url, dst_dir, file_name=None, args="", header=""):

files_cells/python/en/launch_en.py

@@ -5,6 +5,7 @@ import re
 import time
 import json
 import requests
+import subprocess
 import cloudpickle as pickle
 from datetime import timedelta
 from IPython.display import clear_output
@@ -31,6 +32,17 @@ commandline_arguments = settings.get('commandline_arguments', "")
 change_webui = settings.get('change_webui', "")
 
 
+# ========================== OTHER ==========================
+def is_package_installed(package_name):
+    try:
+        subprocess.run(["npm", "ls", "-g", package_name], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        return True
+    except subprocess.CalledProcessError:
+        return False
+
+lt_tunnel = is_package_installed('localtunnel')
+
+
 # ======================== TUNNEL V2 ========================
 print('Please Wait...')
 
@@ -56,7 +68,9 @@ tunnel_class = pickle.load(open(f"{root_path}/new_tunnel", "rb"), encoding="utf-
 tunnel_port = 1834
 tunnel = tunnel_class(tunnel_port)
 tunnel.add_tunnel(command="cl tunnel --url localhost:{port}", name="cl", pattern=re.compile(r"[\w-]+\.trycloudflare\.com"))
-tunnel.add_tunnel(command="lt --port {port}", name="lt", pattern=re.compile(r"[\w-]+\.loca\.lt"), note="Password : " + "\033[32m" + public_ipv4 + "\033[0m" + " rerun cell if 404 error.")
+
+if lt_tunnel:
+    tunnel.add_tunnel(command="lt --port {port}", name="lt", pattern=re.compile(r"[\w-]+\.loca\.lt"), note="Password : " + "\033[32m" + public_ipv4 + "\033[0m" + " rerun cell if 404 error.")
 
 if zrok_token:
     get_ipython().system('zrok enable {zrok_token} &> /dev/null')
@@ -65,7 +79,7 @@ if zrok_token:
 clear_output()
 
 
-# =============== Automatic Fixing Path V3 ===============
+# ================= Automatic Fixing Path V3 ================
 paths_to_check = {
     "tagger_hf_cache_dir": f"{webui_path}/models/interrogators/",
     "additional_networks_extra_lora_path": f"{webui_path}/models/Lora/",
@@ -86,8 +100,6 @@ if os.path.exists(config_path):
             sed_command = f"sed -i 's|\"{key}\": \".*\"|\"{key}\": \"{value}\"|' {config_path}"
             os.system(sed_command)
 
-    if env == 'Kaggle':
-        get_ipython().system('sed -i \'s/"civitai_interface\\/NSFW content\\/visible": true,/"civitai_interface\\/NSFW content\\/visible": false,/\' {webui_path}/ui-config.json')
 
 with tunnel:
     get_ipython().run_line_magic('cd', '{webui_path}')

files_cells/python/en/widgets_en.py

@@ -92,6 +92,7 @@ function toggleContainer() {
 '''
 display(HTML(JS))
 
+
 # ==================== WIDGETS V2 ====================
 HR = widgets.HTML('<hr>')
 
@@ -221,8 +222,6 @@ additional_widget_list = [additional_header,
 
 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
-    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)
@@ -272,6 +271,24 @@ save_button = factory.create_button('Save', class_name="button_save")
 factory.display(save_button)
 
 
+# ==================== OTHER FUNC ====================
+# Setup UI
+def setup_webui(change_webui_widget):
+    if 'SDW_UI' in os.environ:
+        UI = os.environ['SDW_UI']
+    else:
+        os.environ['SDW_UI'] = change_webui_widget
+        os.environ['SDW_OLD_UI'] = change_webui_widget
+
+    UI = os.getenv('SDW_UI')
+    OLD_UI = os.getenv('SDW_OLD_UI')
+
+    if UI != change_webui_widget:
+        os.environ['SDW_UI'] = change_webui_widget
+
+    # print(f'Current UI: {os.getenv("SDW_UI")}, Old UI: {os.getenv("SDW_OLD_UI")}')
+
+
 # ============ Load / Save - Settings V2 ============
 SETTINGS_FILE = f'{root_path}/settings.json'
 
@@ -305,6 +322,7 @@ def hide_widgets():
     widgets.Widget.close_all()
 
 def save_data(button):
+    setup_webui(change_webui_widget.value)
     save_settings()
     hide_widgets()
 

files_cells/python/ru/downloading_ru.py

@@ -24,6 +24,9 @@ root_path = os.getenv('ROOT_PATH')
 webui_path = os.getenv('WEBUI_PATH')
 free_plan = os.getenv('FREE_PLAN')
 
+OLD_UI = os.getenv('SDW_OLD_UI')
+UI = os.getenv('SDW_UI')
+
 
 # ================ LIBRARIES V2 ================
 flag_file = f"{root_path}/libraries_installed.txt"
@@ -105,6 +108,18 @@ except:
 
 
 # ================= MAIN CODE =================
+# WebUI switch
+if os.path.exists(webui_path):
+    if UI != OLD_UI:
+        print(f'Переключение веб-интерфейса с \033[33m{OLD_UI}\033[0m на \033[33m{UI}\033[0m...', end='')
+        shutil.rmtree(webui_path)
+        # Updating the environment variable after a change
+        os.environ['SDW_OLD_UI'] = UI
+        print('\rГотово!')
+        time.sleep(2)
+        clear_output()
+
+# Unpacking the WebUI
 if not os.path.exists(webui_path):
     start_install = int(time.time())
     print("⌚ Распаковка Stable Diffusion..." if change_webui != 'Forge' else "⌚ Распаковка Stable Diffusion (Forge)...", end='')
@@ -211,7 +226,7 @@ def center_text(text, terminal_width=45):
     padding = (terminal_width - len(text)) // 2
     return f"\033[1m\033[36m{' ' * padding}{text}{' ' * padding}\033[0m\033[32m"
 
-def format_output(url, dst_dir, file_name):
+def format_output(url, dst_dir, file_name, image_name=None, image_url=None):
     info = center_text(f"[{file_name.split('.')[0]}]")
     separation_line = '\033[32m' + '---' * 20
 
@@ -220,89 +235,81 @@ def format_output(url, dst_dir, file_name):
     print(f"\033[33mSAVE DIR: \033[34m{dst_dir}")
     print(f"\033[33mFILE NAME: \033[34m{file_name}\033[0m")
 
+    if 'civitai' in url and image_url:
+        print(f"\033[32m[Preview DL]:\033[0m {image_name} - {image_url}\n")
+
 ''' GET CivitAi API - DATA '''
 
 def CivitAi_API(url, file_name=None):
-    SUPPORT_TYPES = ('Checkpoint', 'Model', 'TextualInversion', 'LORA')
+    SUPPORT_TYPES = ('Checkpoint', 'TextualInversion', 'LORA')
     CIVITAI_TOKEN = "62c0c5956b2f9defbd844d754000180b"
 
     url = url.split('?token=')[0] if '?token=' in url else url
     url = url.replace('?type=', f'?token={CIVITAI_TOKEN}&type=') if '?type=' in url else f"{url}?token={CIVITAI_TOKEN}"
 
     def get_model_data(url):
-        if "civitai.com/models/" in url:
-            if '?modelVersionId=' in url:
-                version_id = url.split('?modelVersionId=')[1]
-                return requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}").json()
+        base_url = "https://civitai.com/api/v1"
+        try:
+            if "civitai.com/models/" in url:
+                if '?modelVersionId=' in url:
+                    version_id = url.split('?modelVersionId=')[1]
+                else:
+                    model_id = url.split('/models/')[1].split('/')[0]
+                    model_data = requests.get(f"{base_url}/models/{model_id}").json()
+                    version_id = model_data['modelVersions'][0].get('id')
             else:
-                model_id = url.split('/models/')[1].split('/')[0]
-                return requests.get(f"https://civitai.com/api/v1/models/{model_id}").json()
-        else:
-            version_id = url.split('/models/')[1].split('/')[0]
-            return requests.get(f"https://civitai.com/api/v1/model-versions/{version_id}").json()
+                version_id = url.split('/models/')[1].split('/')[0]
+
+            return requests.get(f"{base_url}/model-versions/{version_id}").json()
+        except (KeyError, IndexError, requests.RequestException) as e:
+            return None
 
     data = get_model_data(url)
 
     if not data:
-        return None, None, None, None, None, None, None
+        print("\033[31m[Data Info]:\033[0m Failed to retrieve data from the API.\n")
+        return 'None', None, None, None, None, None, None
 
     def extract_model_info(url, data):
-        if "civitai.com/models/" in url:
-            if '?modelVersionId=' in url:
-                model_type = data['model']['type']
-                model_name = data['files'][0]['name']
-            else:
-                model_type = data['type']
-                model_name = data['modelVersions'][0]['files'][0]['name']
-        elif 'type=' in url:
-            model_type = parse_qs(urlparse(url).query).get('type', [''])[0]
-            if 'model' in model_type.lower():
-                model_name = data['files'][0]['name']
-            else:
+        model_type = data['model']['type']
+        model_name = data['files'][0]['name']
+
+        if 'type=' in url:
+            url_model_type = parse_qs(urlparse(url).query).get('type', [''])[0].lower()
+            if 'vae' in url_model_type:
+                model_type = data['files'][1]['type']
                 model_name = data['files'][1]['name']
-        else:
-            model_type = data['model']['type']
-            model_name = data['files'][0]['name']
+
         return model_type, model_name
 
     model_type, model_name = extract_model_info(url, data)
     model_name = file_name or model_name
 
-    def get_download_url(url, data, model_type):
-        if "civitai.com/models/" in url:
-            if '?modelVersionId=' in url:
-                return data.get('downloadUrl')
-            else:
-                return data["modelVersions"][0].get("downloadUrl", "")
-        elif 'type=' in url:
-            if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):
-                return data['files'][0]['downloadUrl']
-            else:
-                return data['files'][1]['downloadUrl']
+    def get_download_url(data, model_type):
+        if any(t.lower() in model_type.lower() for t in SUPPORT_TYPES):
+            return data['files'][0]['downloadUrl']
         else:
-            return data.get('downloadUrl')
+            return data['files'][1]['downloadUrl']
 
-    download_url = get_download_url(url, data, model_type)
-    clean_url = re.sub(r'[?&]token=[^&]*', '', download_url)
+    download_url = get_download_url(data, model_type)
 
     def get_image_info(data, model_type, model_name):
-        image_url, image_name = None, None
-        if any(t in model_type for t in SUPPORT_TYPES):
-            try:
-                images = data.get('images') or data['modelVersions'][0].get('images', [])
-                if env == 'Kaggle':
-                    image_url = next((image['url'] for image in images if image['nsfwLevel'] < 4), None)
-                else:
-                    image_url = images[0]['url'] if images else None
-            except KeyError:
-                pass
+        if not any(t in model_type for t in SUPPORT_TYPES):
+            return None, None
+
+        for image in data.get('images', []):
+            if image['nsfwLevel'] >= 4 and env == 'Kaggle':  # Filter NSFW images for Kaggle
+                continue
+            image_url = image['url']
+            image_extension = image_url.split('.')[-1]
+            image_name = f"{model_name.split('.')[0]}.preview.{image_extension}" if image_url else None
+            return image_url, image_name
 
-            image_name = f"{model_name.split('.')[0]}.preview.{image_url.split('.')[-1]}" if image_url else None
-        return image_url, image_name
+        return None, None
 
     image_url, image_name = get_image_info(data, model_type, model_name)
 
-    return f"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}", clean_url, model_type, model_name, image_url, image_name, data
+    return f"{download_url}{'&' if '?' in download_url else '?'}token={CIVITAI_TOKEN}", download_url, model_type, model_name, image_url, image_name, data
 
 ''' Main Download Code '''
 
@@ -372,14 +379,10 @@ def manual_download(url, dst_dir, file_name):
 
     """ Formatted info output """
     model_name_or_basename = file_name if file_name else basename
-    format_output(clean_url or url, dst_dir, model_name_or_basename)
-
-    # print(url, dst_dir, model_name_or_basename)
-    if 'civitai' in url:
-        if not data:
-            print("\033[31m[Data Info]:\033[0m Failed to retrieve data from the API.\n")
-        if data and image_name:
-            print(f"\033[32m[Preview DL]:\033[0m {image_name} - {image_url}\n")
+    try:
+        format_output(clean_url or url, dst_dir, model_name_or_basename, image_url, image_name)
+    except UnboundLocalError:
+        format_output(clean_url or url, dst_dir, model_name_or_basename, None, None)
 
     # =====================
     def run_aria2c(url, dst_dir, file_name=None, args="", header=""):

files_cells/python/ru/launch_ru.py

@@ -5,6 +5,7 @@ import re
 import time
 import json
 import requests
+import subprocess
 import cloudpickle as pickle
 from datetime import timedelta
 from IPython.display import clear_output
@@ -31,6 +32,17 @@ commandline_arguments = settings.get('commandline_arguments', "")
 change_webui = settings.get('change_webui', "")
 
 
+# ========================== OTHER ==========================
+def is_package_installed(package_name):
+    try:
+        subprocess.run(["npm", "ls", "-g", package_name], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        return True
+    except subprocess.CalledProcessError:
+        return False
+
+lt_tunnel = is_package_installed('localtunnel')
+
+
 # ======================== TUNNEL V2 ========================
 print('Please Wait...')
 
@@ -56,7 +68,9 @@ tunnel_class = pickle.load(open(f"{root_path}/new_tunnel", "rb"), encoding="utf-
 tunnel_port = 1834
 tunnel = tunnel_class(tunnel_port)
 tunnel.add_tunnel(command="cl tunnel --url localhost:{port}", name="cl", pattern=re.compile(r"[\w-]+\.trycloudflare\.com"))
-tunnel.add_tunnel(command="lt --port {port}", name="lt", pattern=re.compile(r"[\w-]+\.loca\.lt"), note="Password : " + "\033[32m" + public_ipv4 + "\033[0m" + " rerun cell if 404 error.")
+
+if lt_tunnel:
+    tunnel.add_tunnel(command="lt --port {port}", name="lt", pattern=re.compile(r"[\w-]+\.loca\.lt"), note="Password : " + "\033[32m" + public_ipv4 + "\033[0m" + " rerun cell if 404 error.")
 
 if zrok_token:
     get_ipython().system('zrok enable {zrok_token} &> /dev/null')
@@ -65,7 +79,7 @@ if zrok_token:
 clear_output()
 
 
-# =============== Automatic Fixing Path V3 ===============
+# ================= Automatic Fixing Path V3 ================
 paths_to_check = {
     "tagger_hf_cache_dir": f"{webui_path}/models/interrogators/",
     "additional_networks_extra_lora_path": f"{webui_path}/models/Lora/",
@@ -86,8 +100,6 @@ if os.path.exists(config_path):
             sed_command = f"sed -i 's|\"{key}\": \".*\"|\"{key}\": \"{value}\"|' {config_path}"
             os.system(sed_command)
 
-    if env == 'Kaggle':
-        get_ipython().system('sed -i \'s/"civitai_interface\\/NSFW content\\/visible": true,/"civitai_interface\\/NSFW content\\/visible": false,/\' {webui_path}/ui-config.json')
 
 with tunnel:
     get_ipython().run_line_magic('cd', '{webui_path}')

files_cells/python/ru/widgets_ru.py

@@ -92,6 +92,7 @@ function toggleContainer() {
 '''
 display(HTML(JS))
 
+
 # ==================== WIDGETS V2 ====================
 HR = widgets.HTML('<hr>')
 
@@ -221,8 +222,6 @@ additional_widget_list = [additional_header,
 
 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
-    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)
@@ -272,6 +271,24 @@ save_button = factory.create_button('Сохранить', class_name="button_sav
 factory.display(save_button)
 
 
+# ==================== OTHER FUNC ====================
+# Setup UI
+def setup_webui(change_webui_widget):
+    if 'SDW_UI' in os.environ:
+        UI = os.environ['SDW_UI']
+    else:
+        os.environ['SDW_UI'] = change_webui_widget
+        os.environ['SDW_OLD_UI'] = change_webui_widget
+
+    UI = os.getenv('SDW_UI')
+    OLD_UI = os.getenv('SDW_OLD_UI')
+
+    if UI != change_webui_widget:
+        os.environ['SDW_UI'] = change_webui_widget
+
+    # print(f'Current UI: {os.getenv("SDW_UI")}, Old UI: {os.getenv("SDW_OLD_UI")}')
+
+
 # ============ Load / Save - Settings V2 ============
 SETTINGS_FILE = f'{root_path}/settings.json'
 
@@ -305,6 +322,7 @@ def hide_widgets():
     widgets.Widget.close_all()
 
 def save_data(button):
+    setup_webui(change_webui_widget.value)
     save_settings()
     hide_widgets()