♻️ What's changed?

files_cells/notebooks/en/launch_en.ipynb

@@ -33,7 +33,6 @@
         "from datetime import timedelta\n",
         "from IPython.display import clear_output\n",
         "\n",
-        "\n",
         "#  ================= DETECT ENV =================\n",
         "def detect_environment():\n",
         "    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)\n",
@@ -45,24 +44,23 @@
         "    for env_var, (environment, path) in environments.items():\n",
         "        if env_var in os.environ:\n",
         "            return environment, path, free_plan\n",
+        "    return 'Unknown', '/unknown/path', free_plan\n",
         "\n",
         "env, root_path, free_plan = detect_environment()\n",
         "webui_path = f\"{root_path}/sdw\"\n",
-        "#  ----------------------------------------------\n",
         "\n",
         "def load_settings():\n",
         "    SETTINGS_FILE = f'{root_path}/settings.json'\n",
         "    if os.path.exists(SETTINGS_FILE):\n",
         "        with open(SETTINGS_FILE, 'r') as f:\n",
-        "            settings = json.load(f)\n",
-        "        return settings\n",
+        "            return json.load(f)\n",
+        "    return {}\n",
         "\n",
         "settings = load_settings()\n",
-        "ngrok_token = settings['ngrok_token']\n",
-        "zrok_token = settings['zrok_token']\n",
-        "commandline_arguments = settings['commandline_arguments']\n",
-        "change_webui = settings['change_webui']\n",
-        "\n",
+        "ngrok_token = settings.get('ngrok_token', \"\")\n",
+        "zrok_token = settings.get('zrok_token', \"\")\n",
+        "commandline_arguments = settings.get('commandline_arguments', \"\")\n",
+        "change_webui = settings.get('change_webui', \"\")\n",
         "\n",
         "# ======================== TUNNEL V2 ========================\n",
         "print('Please Wait...')\n",
@@ -71,60 +69,55 @@
         "    try:\n",
         "        url = f'https://api64.ipify.org?format=json&{version}=true'\n",
         "        response = requests.get(url)\n",
-        "        data = response.json()\n",
-        "        public_ip = data['ip']\n",
-        "        return public_ip\n",
+        "        return response.json().get('ip', 'N/A')\n",
         "    except Exception as e:\n",
         "        print(f\"Error getting public {version} address:\", e)\n",
         "\n",
         "# Check if public IP is already saved, if not then get it\n",
-        "try:\n",
-        "    with open(f\"{root_path}/public_ip.txt\", \"r\") as file:\n",
+        "public_ip_file = f\"{root_path}/public_ip.txt\"\n",
+        "if os.path.exists(public_ip_file):\n",
+        "    with open(public_ip_file, 'r') as file:\n",
         "        public_ipv4 = file.read().strip()\n",
-        "except FileNotFoundError:\n",
+        "else:\n",
         "    public_ipv4 = get_public_ip(version='ipv4')\n",
-        "    with open(f\"{root_path}/public_ip.txt\", \"w\") as file:\n",
+        "    with open(public_ip_file, 'w') as file:\n",
         "        file.write(public_ipv4)\n",
         "\n",
         "tunnel_class = pickle.load(open(f\"{root_path}/new_tunnel\", \"rb\"), encoding=\"utf-8\")\n",
-        "tunnel_port = 1734\n",
+        "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",
-        "''' add zrok tunnel '''\n",
         "if zrok_token:\n",
-        "    get_ipython().system('zrok enable {zrok_token} &> /dev/null')\n",
+        "    !zrok enable {zrok_token} &> /dev/null\n",
         "    tunnel.add_tunnel(command=\"zrok share public http://localhost:{port}/ --headless\", name=\"zrok\", pattern=re.compile(r\"[\\w-]+\\.share\\.zrok\\.io\"))\n",
         "\n",
         "clear_output()\n",
         "\n",
-        "\n",
         "# =============== Automatic Fixing Path V3 ===============\n",
-        "paths_to_check = [\n",
-        "    (\"tagger_hf_cache_dir\", f\"{webui_path}/models/interrogators/\"),\n",
-        "    (\"additional_networks_extra_lora_path\", f\"{webui_path}/models/Lora/\"),\n",
-        "    (\"ad_extra_models_dir\", f\"{webui_path}/models/adetailer/\"),\n",
-        "    (\"sd_checkpoint_hash\", \"\"),\n",
-        "    (\"sd_model_checkpoint\", \"\"),\n",
-        "    (\"sd_vae\", \"None\")\n",
-        "]\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",
+        "    \"ad_extra_models_dir\": f\"{webui_path}/models/adetailer/\",\n",
+        "    \"sd_checkpoint_hash\": \"\",\n",
+        "    \"sd_model_checkpoint\": \"\",\n",
+        "    \"sd_vae\": \"None\"\n",
+        "}\n",
         "\n",
         "config_path = f'{webui_path}/ui-config.json'\n",
         "\n",
-        "with open(config_path, 'r') as file:\n",
-        "    config_data = json.load(file)\n",
+        "if os.path.exists(config_path):\n",
+        "    with open(config_path, 'r') as file:\n",
+        "        config_data = json.load(file)\n",
         "\n",
-        "for key, value in paths_to_check:\n",
-        "    if key in config_data and config_data[key] != value:\n",
-        "        sed_command = f\"sed -i 's|\\\"{key}\\\": \\\".*\\\"|\\\"{key}\\\": \\\"{value}\\\"|' {config_path}\"\n",
-        "        os.system(sed_command)\n",
-        "\n",
-        "# Additional check for Kaggle\n",
-        "if env == 'Kaggle':\n",
-        "   get_ipython().system('sed -i \\'s/\"civitai_interface\\\\/NSFW content\\\\/value\":.*/\"civitai_interface\\\\/NSFW content\\\\/value\": false/g\\' {webui_path}/ui-config.json')\n",
-        "#  -------------------------------------------------------\n",
+        "    for key, value in paths_to_check.items():\n",
+        "        if key in config_data and config_data[key] != value:\n",
+        "            sed_command = f\"sed -i 's|\\\"{key}\\\": \\\".*\\\"|\\\"{key}\\\": \\\"{value}\\\"|' {config_path}\"\n",
+        "            os.system(sed_command)\n",
         "\n",
+        "    if env == 'Kaggle':\n",
+        "        get_ipython().system('sed -i \\'s/\"civitai_interface\\\\/NSFW content\\\\/value\":.*/\"civitai_interface\\\\/NSFW content\\\\/value\": false/g\\' {webui_path}/ui-config.json')\n",
         "\n",
         "with tunnel:\n",
         "    %cd {webui_path}\n",
@@ -135,19 +128,15 @@
         "    if env != \"Google Colab\":\n",
         "        commandline_arguments += f' --encrypt-pass={tunnel_port} --api'\n",
         "\n",
-        "    # -- FORGE --\n",
         "    if change_webui == 'Forge':\n",
         "        commandline_arguments += ' --cuda-stream --pin-shared-memory'\n",
         "\n",
         "    !COMMANDLINE_ARGS=\"{commandline_arguments}\" python launch.py\n",
         "\n",
-        "\n",
-        "# after runnig\n",
         "start_colab = float(open(f'{webui_path}/static/colabTimer.txt', 'r').read())\n",
         "time_since_start = str(timedelta(seconds=time.time()-start_colab)).split('.')[0]\n",
         "print(f\"\\n⌚️ \\033[0mYou have been conducting this session for - \\033[33m{time_since_start}\\033[0m\\n\\n\")\n",
         "\n",
-        "''' del zrok tunnel '''\n",
         "if zrok_token:\n",
         "    !zrok disable &> /dev/null"
       ]

files_cells/notebooks/ru/launch_ru.ipynb

@@ -33,7 +33,6 @@
         "from datetime import timedelta\n",
         "from IPython.display import clear_output\n",
         "\n",
-        "\n",
         "#  ================= DETECT ENV =================\n",
         "def detect_environment():\n",
         "    free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)\n",
@@ -45,24 +44,23 @@
         "    for env_var, (environment, path) in environments.items():\n",
         "        if env_var in os.environ:\n",
         "            return environment, path, free_plan\n",
+        "    return 'Unknown', '/unknown/path', free_plan\n",
         "\n",
         "env, root_path, free_plan = detect_environment()\n",
         "webui_path = f\"{root_path}/sdw\"\n",
-        "#  ----------------------------------------------\n",
         "\n",
         "def load_settings():\n",
         "    SETTINGS_FILE = f'{root_path}/settings.json'\n",
         "    if os.path.exists(SETTINGS_FILE):\n",
         "        with open(SETTINGS_FILE, 'r') as f:\n",
-        "            settings = json.load(f)\n",
-        "        return settings\n",
+        "            return json.load(f)\n",
+        "    return {}\n",
         "\n",
         "settings = load_settings()\n",
-        "ngrok_token = settings['ngrok_token']\n",
-        "zrok_token = settings['zrok_token']\n",
-        "commandline_arguments = settings['commandline_arguments']\n",
-        "change_webui = settings['change_webui']\n",
-        "\n",
+        "ngrok_token = settings.get('ngrok_token', \"\")\n",
+        "zrok_token = settings.get('zrok_token', \"\")\n",
+        "commandline_arguments = settings.get('commandline_arguments', \"\")\n",
+        "change_webui = settings.get('change_webui', \"\")\n",
         "\n",
         "# ======================== TUNNEL V2 ========================\n",
         "print('Please Wait...')\n",
@@ -71,60 +69,55 @@
         "    try:\n",
         "        url = f'https://api64.ipify.org?format=json&{version}=true'\n",
         "        response = requests.get(url)\n",
-        "        data = response.json()\n",
-        "        public_ip = data['ip']\n",
-        "        return public_ip\n",
+        "        return response.json().get('ip', 'N/A')\n",
         "    except Exception as e:\n",
         "        print(f\"Error getting public {version} address:\", e)\n",
         "\n",
         "# Check if public IP is already saved, if not then get it\n",
-        "try:\n",
-        "    with open(f\"{root_path}/public_ip.txt\", \"r\") as file:\n",
+        "public_ip_file = f\"{root_path}/public_ip.txt\"\n",
+        "if os.path.exists(public_ip_file):\n",
+        "    with open(public_ip_file, 'r') as file:\n",
         "        public_ipv4 = file.read().strip()\n",
-        "except FileNotFoundError:\n",
+        "else:\n",
         "    public_ipv4 = get_public_ip(version='ipv4')\n",
-        "    with open(f\"{root_path}/public_ip.txt\", \"w\") as file:\n",
+        "    with open(public_ip_file, 'w') as file:\n",
         "        file.write(public_ipv4)\n",
         "\n",
         "tunnel_class = pickle.load(open(f\"{root_path}/new_tunnel\", \"rb\"), encoding=\"utf-8\")\n",
-        "tunnel_port = 1734\n",
+        "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",
-        "''' add zrok tunnel '''\n",
         "if zrok_token:\n",
-        "    get_ipython().system('zrok enable {zrok_token} &> /dev/null')\n",
+        "    !zrok enable {zrok_token} &> /dev/null\n",
         "    tunnel.add_tunnel(command=\"zrok share public http://localhost:{port}/ --headless\", name=\"zrok\", pattern=re.compile(r\"[\\w-]+\\.share\\.zrok\\.io\"))\n",
         "\n",
         "clear_output()\n",
         "\n",
-        "\n",
         "# =============== Automatic Fixing Path V3 ===============\n",
-        "paths_to_check = [\n",
-        "    (\"tagger_hf_cache_dir\", f\"{webui_path}/models/interrogators/\"),\n",
-        "    (\"additional_networks_extra_lora_path\", f\"{webui_path}/models/Lora/\"),\n",
-        "    (\"ad_extra_models_dir\", f\"{webui_path}/models/adetailer/\"),\n",
-        "    (\"sd_checkpoint_hash\", \"\"),\n",
-        "    (\"sd_model_checkpoint\", \"\"),\n",
-        "    (\"sd_vae\", \"None\")\n",
-        "]\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",
+        "    \"ad_extra_models_dir\": f\"{webui_path}/models/adetailer/\",\n",
+        "    \"sd_checkpoint_hash\": \"\",\n",
+        "    \"sd_model_checkpoint\": \"\",\n",
+        "    \"sd_vae\": \"None\"\n",
+        "}\n",
         "\n",
         "config_path = f'{webui_path}/ui-config.json'\n",
         "\n",
-        "with open(config_path, 'r') as file:\n",
-        "    config_data = json.load(file)\n",
+        "if os.path.exists(config_path):\n",
+        "    with open(config_path, 'r') as file:\n",
+        "        config_data = json.load(file)\n",
         "\n",
-        "for key, value in paths_to_check:\n",
-        "    if key in config_data and config_data[key] != value:\n",
-        "        sed_command = f\"sed -i 's|\\\"{key}\\\": \\\".*\\\"|\\\"{key}\\\": \\\"{value}\\\"|' {config_path}\"\n",
-        "        os.system(sed_command)\n",
-        "\n",
-        "# Additional check for Kaggle\n",
-        "if env == 'Kaggle':\n",
-        "   get_ipython().system('sed -i \\'s/\"civitai_interface\\\\/NSFW content\\\\/value\":.*/\"civitai_interface\\\\/NSFW content\\\\/value\": false/g\\' {webui_path}/ui-config.json')\n",
-        "#  -------------------------------------------------------\n",
+        "    for key, value in paths_to_check.items():\n",
+        "        if key in config_data and config_data[key] != value:\n",
+        "            sed_command = f\"sed -i 's|\\\"{key}\\\": \\\".*\\\"|\\\"{key}\\\": \\\"{value}\\\"|' {config_path}\"\n",
+        "            os.system(sed_command)\n",
         "\n",
+        "    if env == 'Kaggle':\n",
+        "        get_ipython().system('sed -i \\'s/\"civitai_interface\\\\/NSFW content\\\\/value\":.*/\"civitai_interface\\\\/NSFW content\\\\/value\": false/g\\' {webui_path}/ui-config.json')\n",
         "\n",
         "with tunnel:\n",
         "    %cd {webui_path}\n",
@@ -135,19 +128,15 @@
         "    if env != \"Google Colab\":\n",
         "        commandline_arguments += f' --encrypt-pass={tunnel_port} --api'\n",
         "\n",
-        "    # -- FORGE --\n",
         "    if change_webui == 'Forge':\n",
         "        commandline_arguments += ' --cuda-stream --pin-shared-memory'\n",
         "\n",
         "    !COMMANDLINE_ARGS=\"{commandline_arguments}\" python launch.py\n",
         "\n",
-        "\n",
-        "# after runnig\n",
         "start_colab = float(open(f'{webui_path}/static/colabTimer.txt', 'r').read())\n",
         "time_since_start = str(timedelta(seconds=time.time()-start_colab)).split('.')[0]\n",
         "print(f\"\\n⌚️ \\033[0mВы проводите эту сессию в течение - \\033[33m{time_since_start}\\033[0m\\n\\n\")\n",
         "\n",
-        "''' del zrok tunnel '''\n",
         "if zrok_token:\n",
         "    !zrok disable &> /dev/null"
       ]

files_cells/python/en/launch_en.py

@@ -9,7 +9,6 @@ import cloudpickle as pickle
 from datetime import timedelta
 from IPython.display import clear_output
 
-
 #  ================= DETECT ENV =================
 def detect_environment():
     free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
@@ -21,24 +20,23 @@ def detect_environment():
     for env_var, (environment, path) in environments.items():
         if env_var in os.environ:
             return environment, path, free_plan
+    return 'Unknown', '/unknown/path', free_plan
 
 env, root_path, free_plan = detect_environment()
 webui_path = f"{root_path}/sdw"
-#  ----------------------------------------------
 
 def load_settings():
     SETTINGS_FILE = f'{root_path}/settings.json'
     if os.path.exists(SETTINGS_FILE):
         with open(SETTINGS_FILE, 'r') as f:
-            settings = json.load(f)
-        return settings
+            return json.load(f)
+    return {}
 
 settings = load_settings()
-ngrok_token = settings['ngrok_token']
-zrok_token = settings['zrok_token']
-commandline_arguments = settings['commandline_arguments']
-change_webui = settings['change_webui']
-
+ngrok_token = settings.get('ngrok_token', "")
+zrok_token = settings.get('zrok_token', "")
+commandline_arguments = settings.get('commandline_arguments', "")
+change_webui = settings.get('change_webui', "")
 
 # ======================== TUNNEL V2 ========================
 print('Please Wait...')
@@ -47,60 +45,55 @@ def get_public_ip(version='ipv4'):
     try:
         url = f'https://api64.ipify.org?format=json&{version}=true'
         response = requests.get(url)
-        data = response.json()
-        public_ip = data['ip']
-        return public_ip
+        return response.json().get('ip', 'N/A')
     except Exception as e:
         print(f"Error getting public {version} address:", e)
 
 # Check if public IP is already saved, if not then get it
-try:
-    with open(f"{root_path}/public_ip.txt", "r") as file:
+public_ip_file = f"{root_path}/public_ip.txt"
+if os.path.exists(public_ip_file):
+    with open(public_ip_file, 'r') as file:
         public_ipv4 = file.read().strip()
-except FileNotFoundError:
+else:
     public_ipv4 = get_public_ip(version='ipv4')
-    with open(f"{root_path}/public_ip.txt", "w") as file:
+    with open(public_ip_file, 'w') as file:
         file.write(public_ipv4)
 
 tunnel_class = pickle.load(open(f"{root_path}/new_tunnel", "rb"), encoding="utf-8")
-tunnel_port = 1734
+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.")
 
-''' add zrok tunnel '''
 if zrok_token:
     get_ipython().system('zrok enable {zrok_token} &> /dev/null')
     tunnel.add_tunnel(command="zrok share public http://localhost:{port}/ --headless", name="zrok", pattern=re.compile(r"[\w-]+\.share\.zrok\.io"))
 
 clear_output()
 
-
 # =============== 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/"),
-    ("ad_extra_models_dir", f"{webui_path}/models/adetailer/"),
-    ("sd_checkpoint_hash", ""),
-    ("sd_model_checkpoint", ""),
-    ("sd_vae", "None")
-]
+paths_to_check = {
+    "tagger_hf_cache_dir": f"{webui_path}/models/interrogators/",
+    "additional_networks_extra_lora_path": f"{webui_path}/models/Lora/",
+    "ad_extra_models_dir": f"{webui_path}/models/adetailer/",
+    "sd_checkpoint_hash": "",
+    "sd_model_checkpoint": "",
+    "sd_vae": "None"
+}
 
 config_path = f'{webui_path}/ui-config.json'
 
-with open(config_path, 'r') as file:
-    config_data = json.load(file)
+if os.path.exists(config_path):
+    with open(config_path, 'r') as file:
+        config_data = json.load(file)
 
-for key, value in paths_to_check:
-    if key in config_data and config_data[key] != value:
-        sed_command = f"sed -i 's|\"{key}\": \".*\"|\"{key}\": \"{value}\"|' {config_path}"
-        os.system(sed_command)
-
-# Additional check for Kaggle
-if env == 'Kaggle':
-   get_ipython().system('sed -i \'s/"civitai_interface\\/NSFW content\\/value":.*/"civitai_interface\\/NSFW content\\/value": false/g\' {webui_path}/ui-config.json')
-#  -------------------------------------------------------
+    for key, value in paths_to_check.items():
+        if key in config_data and config_data[key] != value:
+            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\\/value":.*/"civitai_interface\\/NSFW content\\/value": false/g\' {webui_path}/ui-config.json')
 
 with tunnel:
     get_ipython().run_line_magic('cd', '{webui_path}')
@@ -111,19 +104,15 @@ with tunnel:
     if env != "Google Colab":
         commandline_arguments += f' --encrypt-pass={tunnel_port} --api'
 
-    # -- FORGE --
     if change_webui == 'Forge':
         commandline_arguments += ' --cuda-stream --pin-shared-memory'
 
     get_ipython().system('COMMANDLINE_ARGS="{commandline_arguments}" python launch.py')
 
-
-# after runnig
 start_colab = float(open(f'{webui_path}/static/colabTimer.txt', 'r').read())
 time_since_start = str(timedelta(seconds=time.time()-start_colab)).split('.')[0]
 print(f"\n⌚️ \033[0mYou have been conducting this session for - \033[33m{time_since_start}\033[0m\n\n")
 
-''' del zrok tunnel '''
 if zrok_token:
     get_ipython().system('zrok disable &> /dev/null')
 

files_cells/python/ru/launch_ru.py

@@ -9,7 +9,6 @@ import cloudpickle as pickle
 from datetime import timedelta
 from IPython.display import clear_output
 
-
 #  ================= DETECT ENV =================
 def detect_environment():
     free_plan = (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024 ** 3) <= 20)
@@ -21,24 +20,23 @@ def detect_environment():
     for env_var, (environment, path) in environments.items():
         if env_var in os.environ:
             return environment, path, free_plan
+    return 'Unknown', '/unknown/path', free_plan
 
 env, root_path, free_plan = detect_environment()
 webui_path = f"{root_path}/sdw"
-#  ----------------------------------------------
 
 def load_settings():
     SETTINGS_FILE = f'{root_path}/settings.json'
     if os.path.exists(SETTINGS_FILE):
         with open(SETTINGS_FILE, 'r') as f:
-            settings = json.load(f)
-        return settings
+            return json.load(f)
+    return {}
 
 settings = load_settings()
-ngrok_token = settings['ngrok_token']
-zrok_token = settings['zrok_token']
-commandline_arguments = settings['commandline_arguments']
-change_webui = settings['change_webui']
-
+ngrok_token = settings.get('ngrok_token', "")
+zrok_token = settings.get('zrok_token', "")
+commandline_arguments = settings.get('commandline_arguments', "")
+change_webui = settings.get('change_webui', "")
 
 # ======================== TUNNEL V2 ========================
 print('Please Wait...')
@@ -47,60 +45,55 @@ def get_public_ip(version='ipv4'):
     try:
         url = f'https://api64.ipify.org?format=json&{version}=true'
         response = requests.get(url)
-        data = response.json()
-        public_ip = data['ip']
-        return public_ip
+        return response.json().get('ip', 'N/A')
     except Exception as e:
         print(f"Error getting public {version} address:", e)
 
 # Check if public IP is already saved, if not then get it
-try:
-    with open(f"{root_path}/public_ip.txt", "r") as file:
+public_ip_file = f"{root_path}/public_ip.txt"
+if os.path.exists(public_ip_file):
+    with open(public_ip_file, 'r') as file:
         public_ipv4 = file.read().strip()
-except FileNotFoundError:
+else:
     public_ipv4 = get_public_ip(version='ipv4')
-    with open(f"{root_path}/public_ip.txt", "w") as file:
+    with open(public_ip_file, 'w') as file:
         file.write(public_ipv4)
 
 tunnel_class = pickle.load(open(f"{root_path}/new_tunnel", "rb"), encoding="utf-8")
-tunnel_port = 1734
+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.")
 
-''' add zrok tunnel '''
 if zrok_token:
     get_ipython().system('zrok enable {zrok_token} &> /dev/null')
     tunnel.add_tunnel(command="zrok share public http://localhost:{port}/ --headless", name="zrok", pattern=re.compile(r"[\w-]+\.share\.zrok\.io"))
 
 clear_output()
 
-
 # =============== 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/"),
-    ("ad_extra_models_dir", f"{webui_path}/models/adetailer/"),
-    ("sd_checkpoint_hash", ""),
-    ("sd_model_checkpoint", ""),
-    ("sd_vae", "None")
-]
+paths_to_check = {
+    "tagger_hf_cache_dir": f"{webui_path}/models/interrogators/",
+    "additional_networks_extra_lora_path": f"{webui_path}/models/Lora/",
+    "ad_extra_models_dir": f"{webui_path}/models/adetailer/",
+    "sd_checkpoint_hash": "",
+    "sd_model_checkpoint": "",
+    "sd_vae": "None"
+}
 
 config_path = f'{webui_path}/ui-config.json'
 
-with open(config_path, 'r') as file:
-    config_data = json.load(file)
+if os.path.exists(config_path):
+    with open(config_path, 'r') as file:
+        config_data = json.load(file)
 
-for key, value in paths_to_check:
-    if key in config_data and config_data[key] != value:
-        sed_command = f"sed -i 's|\"{key}\": \".*\"|\"{key}\": \"{value}\"|' {config_path}"
-        os.system(sed_command)
-
-# Additional check for Kaggle
-if env == 'Kaggle':
-   get_ipython().system('sed -i \'s/"civitai_interface\\/NSFW content\\/value":.*/"civitai_interface\\/NSFW content\\/value": false/g\' {webui_path}/ui-config.json')
-#  -------------------------------------------------------
+    for key, value in paths_to_check.items():
+        if key in config_data and config_data[key] != value:
+            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\\/value":.*/"civitai_interface\\/NSFW content\\/value": false/g\' {webui_path}/ui-config.json')
 
 with tunnel:
     get_ipython().run_line_magic('cd', '{webui_path}')
@@ -111,19 +104,15 @@ with tunnel:
     if env != "Google Colab":
         commandline_arguments += f' --encrypt-pass={tunnel_port} --api'
 
-    # -- FORGE --
     if change_webui == 'Forge':
         commandline_arguments += ' --cuda-stream --pin-shared-memory'
 
     get_ipython().system('COMMANDLINE_ARGS="{commandline_arguments}" python launch.py')
 
-
-# after runnig
 start_colab = float(open(f'{webui_path}/static/colabTimer.txt', 'r').read())
 time_since_start = str(timedelta(seconds=time.time()-start_colab)).split('.')[0]
 print(f"\n⌚️ \033[0mВы проводите эту сессию в течение - \033[33m{time_since_start}\033[0m\n\n")
 
-''' del zrok tunnel '''
 if zrok_token:
     get_ipython().system('zrok disable &> /dev/null')