Upload 3 files

javascript/colab_sd_models.js

@@ -0,0 +1,244 @@
+document.addEventListener('DOMContentLoaded', () => {
+  const observer = new MutationObserver((mutations) => {
+    mutations.forEach((mutation) => {
+      if (mutation.addedNodes) {
+        mutation.addedNodes.forEach((node) => {
+          if (node.id === 'tab_models_list') {
+            // запуск кода после загрузки элементов
+            Array.from(document.querySelectorAll('#tabs > div.tab-nav > button')).find(button => button.textContent.includes('модели')).click();
+            // переопределение параметров колонок на основе числа моделей в категории
+            const models_checkbox_grids = document.querySelectorAll("#tab_models_list fieldset > div[data-testid='checkbox-group']");
+            models_checkbox_grids.forEach(models_checkbox_grid => {
+              const model_label = models_checkbox_grid.querySelectorAll('label');
+              const numColumns = model_label.length > 9 ? 'repeat(auto-fit, minmax(250px, 1fr))' : '1fr';
+              models_checkbox_grid.style.gridTemplateColumns = numColumns;
+            });
+            // подвсетка категории мужицких моделей (костаыль, потому что градио своим скриптом долбит каждые Nms переопределение классов)
+            const intervalId = setInterval(() => {
+              const modelsNavButtons = document.querySelectorAll("div#tab_models_list > div.gap > div.tabs > div.tab-nav > button");
+              const MaleCat = Array.from(modelsNavButtons).find(button => button.textContent.includes('мужские'));
+              if (MaleCat) {
+                MaleCat.setAttribute("id", "male_only");
+              }
+            }, 100);
+            // получение инфы по кд о свободном пространтсве из скрытого уродского текстбокса в красивый элементик в шапочке
+            const freespacetextarea = document.querySelector("#free_space_area > label > textarea");
+            const frespace_out = document.querySelector("#frespace_out");
+            let prevValue = freespacetextarea.value;
+            setInterval(function () {
+              const currentValue = freespacetextarea.value;
+              if (currentValue !== prevValue) {
+                frespace_out.innerHTML = `${currentValue}`;
+                prevValue = currentValue;
+              }
+            }, 100);
+            // скликивание реальной но скрытой уродской кнопки с обработчиком проверки свободного места при нажатии на фейковую но красивую кнопочку
+            const freespace_getButton = document.querySelector("#freespace_get");
+            const free_spaceOrigButton = document.querySelector("#free_space_button");
+            freespace_getButton.addEventListener("click", function () {
+              free_spaceOrigButton.click();
+            });
+            // копирование реальных элементов из вкладок с моделями в результатах поиска
+            const SearchBlock = document.querySelector('#clear_search_models_results').closest('label').closest('div').closest('div').closest('div');
+            const ModelDLHeaderBlock = document.querySelector('.models_dl_header').closest('div').parentNode.closest('div').closest('div');
+            const ModelDLHeaderContainer = document.querySelector('.models_dl_header').closest('div').parentNode;
+            ModelDLHeaderBlock.appendChild(SearchBlock);
+            // небольшой css-фикс
+            ModelDLHeaderContainer.style.cssText = `display: flex; flex-direction: row; align-items: center; justify-content: flex-start; flex-wrap: wrap;`;
+            document.querySelector('.models_dl_header').parentNode.style.marginRight = "50px";
+            // фильтрация моделей при вводе
+            const searchInput = document.querySelector('input[type="text"]');
+            const findedModels = document.querySelector('#finded_models');
+            const tabModelsList = document.querySelector('#tab_models_list');
+            const labels = tabModelsList.querySelectorAll('label');
+            const clearSearchResultsButton = document.querySelector("#clear_search_models_results");
+            searchInput.addEventListener('input', (event) => {
+              const searchTerm = event.target.value.toLowerCase();
+              findedModels.innerHTML = '';
+              if (searchTerm !== '') {
+                labels.forEach((label) => {
+                  const labelText = label.textContent.toLowerCase();
+                  if (labelText.includes(searchTerm)) {
+                    const clone = label.cloneNode(true);
+                    findedModels.appendChild(clone);
+                    const originalCheckbox = label.querySelector('input[type="checkbox"]');
+                    const clonedCheckbox = clone.querySelector('input[type="checkbox"]');
+                    clonedCheckbox.addEventListener('change', () => {
+                      originalCheckbox.click();
+                    });
+                  }
+                });
+              }
+            });
+            // обработчик на кнопочку очистки результатов
+            clearSearchResultsButton.addEventListener('click', () => {
+              findedModels.innerHTML = '';
+              searchInput.value = '';
+            });
+            // автоматическое скликивание скрытых кнопок для подгрузки установленных моделей и свободного места после загрузки дополнения через 1 сек
+            setTimeout(() => document.querySelector("#files_button").click(), 1000);
+            setTimeout(() => document.querySelector("#free_space_button").click(), 1000);
+            // файловый менеджер на тексбоксах
+            setTimeout(() => {
+              const filesArea = document.querySelector("#files_area > label > textarea");
+              const filesCheckbox = document.querySelector("#files_checkbox");
+              const deleteArea = document.querySelector("#delete_area > label > textarea");
+              // обновление списка чекбоксов с файлами при изменении списка путей в текстбоксе
+              function addCheckboxEventListeners() {
+                const delete_checkboxes = document.querySelectorAll("#files_checkbox > label > input[type=checkbox]");
+                delete_checkboxes.forEach(checkbox => {
+                  checkbox.addEventListener("change", event => {
+                    // отмеченные на удаление делаем красными и зачеркнутыми
+                    const delete_span = event.target.parentElement.querySelector("span");
+                    if (event.target.checked) {
+                      delete_span.style.textDecoration = "line-through";
+                      delete_span.style.color = "#ed5252";
+                    } else {
+                      delete_span.style.textDecoration = "none";
+                      delete_span.style.color = "";
+                    }
+                  });
+                });
+              }
+              // функция для обновления чекбоксов с файлами почти в реальном времени
+              function updateCheckboxes() {
+                while (filesCheckbox.firstChild) {
+                  filesCheckbox.removeChild(filesCheckbox.firstChild);
+                }
+                const fileNames = filesArea.value.split("\n").map(path => path.split("/").pop());
+                if (fileNames.length === 0 || (fileNames.length === 1 && fileNames[0] === "")) {
+                  const message = document.createElement("p");
+                  message.textContent = "ничего не найдено";
+                  filesCheckbox.appendChild(message);
+                } else {
+                  fileNames.forEach(fileName => {
+                    const checkbox = document.createElement("input");
+                    checkbox.type = "checkbox";
+                    checkbox.id = fileName;
+                    checkbox.addEventListener("change", event => {
+                      if (event.target.checked) {
+                        deleteArea.value += (deleteArea.value ? "\n" : "") + fileName;
+                      } else {
+                        const lines = deleteArea.value.split("\n");
+                        const index = lines.indexOf(fileName);
+                        if (index !== -1) {
+                          lines.splice(index, 1);
+                          deleteArea.value = lines.join("\n");
+                        }
+                      }
+                      deleteArea.dispatchEvent(new Event('input')); // имитация ввода, это самая важная часть кода
+                    });
+                    const label = document.createElement("label");
+                    label.htmlFor = fileName;
+                    label.className = "filecheckbox";
+                    const span = document.createElement("span");
+                    span.textContent = fileName;
+                    label.appendChild(checkbox);
+                    label.appendChild(span);
+                    filesCheckbox.appendChild(label);
+                  });
+                }
+                deleteArea.value = deleteArea.value.trim();
+                deleteArea.dispatchEvent(new Event('input')); // имитация ввода, без этого не работает
+                addCheckboxEventListeners()
+              }
+              // наблюдаем за скрытым тексбоксом с путями до моделей и обнов��ям чекбоксы
+              const observer = new MutationObserver(updateCheckboxes);
+              observer.observe(filesArea, { characterData: true, subtree: true });
+              updateCheckboxes();
+              // скликивание реальной но скрытой уродской кнопки с обработчиком обновления списка файлов при нажатии на фейковую но красивую кнопочку
+              const RefreshFilesButton = document.querySelector("#refresh_files_button");
+              RefreshFilesButton.addEventListener("click", () => {
+                document.querySelector("#files_button").click();
+                // задержки по 3 секунды необходимы, чтобы колаб одуплился
+                setTimeout(function () { updateCheckboxes(); }, 3000);
+              });
+              // при клике на фейковую кнопочку удаления - произойдет и обновление списка файлов с задержкой 3 сек
+              document.querySelector("#delete_button").addEventListener("click", function () {
+                setTimeout(function () {
+                  document.querySelector("#files_button").click();
+                  // задержки по 3 секунды необходимы, чтобы колаб одуплился
+                  setTimeout(function () { updateCheckboxes(); }, 3000);
+                }, 3000);
+              });
+              // скликивание реальной но скрытой уродской кнопки с обработчиком удаления моделей при нажатии на фейковую но красивую кнопочку
+              const OrigDelButton = document.querySelector("#delete_button");
+              const CustomDelButton = document.querySelector("#delete_files_button");
+              CustomDelButton.addEventListener("click", () => {
+                OrigDelButton.click();
+              });
+              // проверка выхлопа из функции загрузки в скрытом текстбоксе
+              setInterval(function () {
+                var DLresultText = document.querySelector("#downloads_result_text > span.finish_dl_func");
+                DLresultText.textContent = document.querySelector("#dlresultbox > label > textarea").value;
+                // функция скрытия прогрессбара
+                function checkDLresult(element, text) {
+                  if (element.textContent.includes(text)) {
+                    document.querySelector("div.downloads_result_container > div.models_porgress_loader").style.removeProperty("display");
+                    document.querySelector("#downloads_start_text").style.removeProperty("display");
+                    document.querySelector("#downloads_result_text > span.dl_progress_info").textContent = "чтобы новые файлы появились в выпадающем списке моделей, нужно обновить их список по соответсвующей кнопке";
+                  }
+                }
+                // просто скрываем прогрессбар если в выхлопе есть фраза о завершении или предупреждение
+                checkDLresult(DLresultText, "заверш");
+                checkDLresult(DLresultText, "слишком");
+                checkDLresult(DLresultText, "ОШИБКА");
+                // раскрашиваем текст сообщения о результате выполнения, если что-то пошло не так
+                if (DLresultText) {
+                  if (DLresultText.textContent.includes("слишком")) {
+                    DLresultText.style.setProperty("color", "#ff4f8b", "important");
+                  } else if (DLresultText.textContent.includes("ОШИБКА")) {
+                    DLresultText.style.setProperty("color", "#de2f2f", "important");
+                  } else if (DLresultText.textContent.includes("заверш")) {
+                    DLresultText.style.setProperty("color", "#99fb99", "important");
+                  }
+                }
+              }, 200);
+              // действия по клику на фейковую но видимую кнопку для скачивания
+              document.querySelector("#general_download_button").addEventListener("click", function () {
+                // очистка текстбокса от выхлопа предыдущего выполнения
+                var resultTextareaDL = document.querySelector("#dlresultbox > label > textarea");
+                resultTextareaDL.value = "";
+                var resultClearOut = new Event("input", { bubbles: true }); // без этого не будет работать обноволение .value
+                resultTextareaDL.dispatchEvent(resultClearOut);
+                // делаем прогрессбар и место для результирующего текста видимыми
+                const DLprogressBar = document.querySelector("div.downloads_result_container > div.models_porgress_loader");
+                DLprogressBar.style.setProperty("display", "block", "important");
+                const DLresultText = document.querySelector("#downloads_result_text");
+                DLresultText.style.setProperty("display", "block", "important");
+                document.querySelector("#downloads_start_text").style.setProperty("display", "block", "important");
+                // скликивание реальных но скрытых кнопок с обработчиками загрузки файлов по чекбоксам и кастомных ссылок при нажатии на фейковую но кнопочку
+                // формирование списка из кастомных ссылок
+                document.querySelector("#ownlinks_download_button").click();
+                setTimeout(function () {
+                  // через 3 сек. добавим его к списку ссылок из чекбоксов и отправим на загрузку вместе
+                  document.querySelector("#checkboxes_download_button").click();
+                }, 3000); // задержка, чтобы колаб успел одуплиться
+              });
+              // если кнопка загрузки уже нажата, запрещаем кликать еще раз пока функция загрузки не выплюнет ответ
+              var GendownloadButton = document.querySelector("#general_download_button");
+              var DLprogressBar = document.querySelector("div.downloads_result_container > div.models_porgress_loader");
+              if (GendownloadButton && DLprogressBar) {
+                var DLobserver = new MutationObserver(function (mutations) {
+                  mutations.forEach(function (mutation) {
+                    if (mutation.type === "attributes" && mutation.attributeName === "style") { // отслеживание видимости прогрессбара
+                      if (DLprogressBar.style.display === "block") {
+                        GendownloadButton.setAttribute("disabled", "disabled");
+                      } else {
+                        setTimeout(function () {
+                          GendownloadButton.removeAttribute("disabled");
+                        }, 3000); // не сразу даем кликнуть снова, а после ожидания 3 секунды, чтобы не закликали
+                      }
+                    }
+                  });
+                });
+                DLobserver.observe(DLprogressBar, { attributes: true });
+              }
+            }, 9000); // запуск скриптов через 9 секунд после загрузки вебуи, чтобы успели отработать скрипты других дополнений и градио
+          }
+        });
+      }
+    });
+  });
+  observer.observe(document.body, { childList: true, subtree: true });
+});

scripts/colab_sd_models.py

@@ -0,0 +1,242 @@
+import os
+import re
+import requests
+import shutil
+import subprocess
+from concurrent.futures import ThreadPoolExecutor, as_completed
+
+import gradio as gr
+from bs4 import BeautifulSoup
+from modules import script_callbacks, scripts, shared, sd_models
+from modules.paths_internal import data_path
+
+
+urlprefix = "https://huggingface.co/2ch/models/resolve/main/"
+models_json_data = requests.get(urlprefix + "colab_models.json").json()
+wget = "wget -nv -t 10 --show-progress --progress=bar:force -q --content-disposition "
+sdroot = data_path
+models_folder_path = sd_models.model_path
+loras_folder_path = shared.cmd_opts.lora_dir
+embeddings_folder_path = shared.cmd_opts.embeddings_dir
+ct = "token=542c1d6077168822e1b49e30e3437a5d"
+
+
+def on_ui_tabs():
+    with gr.Blocks() as models_list:
+        # шапка вкладки с поиском и свободным местом на диске
+        gr.HTML(
+            """<div class="models_top_container"><div class="models_top_header_text"><h1 class="models_dl_header">выбор и скачивание моделей</h1><p>учитывай весьма ограниченное пространство на диске в колабе!</p></div><div class="freespaceinfo"><div id="frespace_output"><span>свободно в колабе: <span id="frespace_out">нажми на кнопочку</span></div><div id="freespace_get"></div></div></div>""")
+        search_and_results = gr.HTML(
+            """<label id="models_search_field"><input type="text" id="models_search_input" class="svelte-1pie7s6" placeholder="начни вводить для поиска"/><span id="clear_search_models_results"></span></label><div id="finded_models"></div>""")
+        # вкладки с категориями
+        buttons = []
+        checkbox_groups = []
+        checkboxes = []
+        for category in models_json_data['categories']:
+            categories = ["models_A", "models_B", "models_C", "models_D", "models_E", "models_F", "models_G", "models_H", "models_I", "models_J", "models_K", "models_L", "models_M", "models_N", "models_O", "models_P", "models_Q"]
+            tab_names = ["аниме", "лайнарт", "женские", "игры и кино", "техника и космос", "крипота", "макро", "мемные", "мужские", "мультфильмы", "пиксельарт", "трехмерная графика", "универсальные", "фотореализм", "фурри", "футанари",
+                         "художественные"]
+            category_map = dict(zip(categories, tab_names))
+            tab_name = category_map.get(category)
+            with gr.Tab(tab_name):
+                checkbox_group = gr.CheckboxGroup(sorted(models_json_data['categories'][category], key=lambda x: x.lower()), label="")
+                checkboxes.append(checkbox_group)
+                checkbox_groups.append(checkbox_group)
+        # ссылки для реквеста добавления моделей
+        gr.HTML(
+            """<div class="request_new_models">не нашел нужную модель? запрос на добавление можно отправить в <a href="https://t.me/colabSDbot" target="_blank">бота</a> или <a href="https://t.me/stabdiff" target="_blank">группу</a>.</div>""")
+
+        # функция для формирования списка своих ссылок
+        def get_own_links(ownmodels, ownloras, ownembeddings):
+            urls = []
+            for text, dlpath in zip([ownmodels, ownloras, ownembeddings], [models_folder_path, loras_folder_path, embeddings_folder_path]):
+                lines = text.split("\n")
+                for line in lines:
+                    if line.strip():
+                        # url = wget+line+" -P "+dlpath
+                        link = line.strip() + (f'?{ct}' if '?' not in line else f'&{ct}') if 'civitai' in line else line.strip()
+                        url = f'{wget}"{link}" -P {dlpath}'
+                        urls.append(url)
+            own_urls = os.path.join(sdroot, "urls.txt")
+            with open(own_urls, "w") as f:
+                for url in urls:
+                    f.write(url + "\n")
+            print("список загрузки сформирован...")
+
+        # текстбоксы для указания своих ссылок
+        gr.HTML("""<div class="ownfiles_header"><h2>здесь можно указать прямые ссылки на загрузку моделей, лор и внедрений</h2></div>""")
+        with gr.Row():
+            plhd = """вставляй каждую ссылку с новой строки!\nпримеры ссылок:\nhttps://models.tensorplay.ai/104249\nhttps://civitai.com/api/download/models/110660\nhttps://huggingface.co/2ch/gay/resolve/main/lora/BettercocksFlaccid.safetensors"""
+            ownmodels = gr.Textbox(label="модели", placeholder=plhd, info="прямые ссылки на Checkpoints", lines=5, elem_id="ownmodels")
+            ownloras = gr.Textbox(label="лоры", placeholder=plhd, info="прямые ссылки на LoRas", lines=5, elem_id="ownloras")
+            ownembeddings = gr.Textbox(label="внедрения", placeholder=plhd, info="прямые ссылки на Textual Inversions", lines=5, elem_id="ownembeddings")
+        # кнопка при нажатии на которую происходит сначала клик на кнопку формирования списка своих ссылок, а потом на кнопку основной функции загрузки
+        download_button = gr.Button("скачать выбранные модели", elem_id="general_download_button")
+        # кнопка для формирования списка своих ссылок
+        button = gr.Button("скачать по ссылкам", elem_id="ownlinks_download_button")
+        button.click(get_own_links, inputs=[ownmodels, ownloras, ownembeddings])
+        # кнопка для отправки на загрузку
+        download_button = gr.Button("скачать отмеченные модели", elem_id="checkboxes_download_button")
+
+        # функция определения точки монтирования и свободного места на диске
+        def find_mount_point():
+            # __file__ = "."
+            path = os.path.realpath(__file__)
+            path = os.path.abspath(path)
+            while not os.path.ismount(path):
+                path = os.path.dirname(path)
+            return path
+
+        def free_space():
+            total, used, free = shutil.disk_usage(find_mount_point())
+            power = 2 ** 10
+            n = 0
+            power_labels = {0: '', 1: 'Кило', 2: 'Мега', 3: 'Гига', 4: 'Тера'}
+            while free > power:
+                free /= power
+                n += 1
+            return f"{free:.2f} {power_labels[n]}байт"
+
+        # функция вычисления общего размера загружаемых файлов в байтах
+        def get_file_size(url):
+            url = re.search(r'https?://\S+', url).group()
+
+            def contleght(url):
+                return int(requests.get(url, stream=True).headers.get('Content-Length', 0))
+
+            if "huggingface" in url:
+                try:
+                    try:
+                        file_size = int(next((pre.text.split('size ')[1].split('\n')[0] for pre in BeautifulSoup(requests.get(url.replace('resolve', 'blame')).text, 'html.parser').find_all('pre') if 'size' in pre.text)))
+                    except:
+                        file_size_text = BeautifulSoup(requests.get(url.replace('resolve', 'blob')).text, 'html.parser').find('strong', string='Size of remote file:').next_sibling.strip()
+                        file_size = int(float(file_size_text.split()[0]) * {'KB': 1024, 'MB': 1024 ** 2, 'GB': 1024 ** 3, 'TB': 1024 ** 4}[file_size_text.split()[1]])
+                except:
+                    file_size = contleght(url)
+                if file_size < 1048576: file_size = contleght(url)
+            elif "civitai" in url:
+                try:
+                    file_size = int(requests.get("https://civitai.com/api/v1/model-versions/" + url.split('/')[-1] + civitai_token).json()["files"][0]["sizeKB"] * 1024)
+                except:
+                    file_size = contleght(url)
+                if file_size < 1048576: file_size = contleght(url)
+            else:
+                file_size = contleght(url)
+            return file_size
+
+        # основная функция для загрузки отмеченных чекбоксов
+        def start_download(*checkbox_groups):
+            try:
+                urls = []
+                for checkbox_group in checkbox_groups:
+                    selected_choices = checkbox_group
+                    for choice in selected_choices:
+                        if choice in models_json_data["models"]:
+                            file_names = models_json_data["models"][choice].split(',')
+                            for file_name in file_names:
+                                url = wget + urlprefix + file_name.strip() + " -P " + models_folder_path
+                                urls.append(url)
+                own_urls = os.path.join(sdroot, "urls.txt")
+                with open(own_urls, "r") as f:
+                    own_links = f.read().splitlines()
+                urls.extend(own_links)
+                os.remove(own_urls)
+            except Exception as e:
+                print(f"ОШИБКА: {e}")
+                return f"ОШИБКА: {e}"
+            try:
+                def bytes_convert(size_bytes):
+                    if size_bytes >= 1073741824:
+                        return f"{round(size_bytes / 1073741824, 2)} Гб"
+                    else:
+                        return f"{round(size_bytes / 1048576, 2)} Мб"
+
+                def downloader(url):
+                    process = subprocess.Popen(url, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+                    while True:
+                        output = process.stdout.readline().decode('utf-8')
+                        if output == '' and process.poll() is not None:
+                            break
+                        if output:
+                            yield output.strip()
+                    return process.poll()
+
+                total_file_size = 0
+                with ThreadPoolExecutor(max_workers=len(urls)) as executor:
+                    futures = [executor.submit(get_file_size, url) for url in urls]
+                    for future in as_completed(futures):
+                        total_file_size += future.result()
+                total, used, free = shutil.disk_usage(find_mount_point())
+                if total_file_size <= (free - 1073741824):
+                    print(f"загрузка {bytes_convert(total_file_size)} уже началась, жди!")
+                    with ThreadPoolExecutor(max_workers=len(urls)) as executor:
+                        futures = [executor.submit(downloader, url) for url in urls]
+                        for future in as_completed(futures):
+                            for line in future.result():
+                                print(line)
+                    if os.path.exists(os.path.join(models_folder_path, "nullModel.ckpt")):
+                        try:
+                            os.remove(os.path.join(models_folder_path, "nullModel.ckpt"))
+                        except:
+                            pass
+                    return "функция загрузки завершила работу!"
+                else:
+                    print(f"слишком много файлов! ты пытаешься скачать {bytes_convert(total_file_size)}, имея свободных только {bytes_convert(free)} (и как минимум 1 Гб должен оставаться не занятым на диске!).")
+                    return f"слишком много файлов! ты пытаешься скачать {bytes_convert(total_file_size)}, имея свободных только {bytes_convert(free)} (и как минимум 1 Гб должен оставаться не занятым на диске!)."
+            except Exception as e:
+                print(f"ОШИБКА: {e}")
+                return f"ОШИБКА: {e}"
+
+        # скрытый текстбокс для вывода информации о результате
+        dlresultbox = gr.Textbox(label="", elem_id="dlresultbox")
+        download_button.click(start_download, inputs=checkbox_groups, outputs=dlresultbox)
+        # анимация процесса выполнения функции загрузки
+        gr.HTML(
+            """<div class="downloads_result_container"><div class="models_porgress_loader"></div><div id="downloads_start_text">задача по загрузке запущена, подробности в выводе ячейки в колабе...</div><div id="downloads_result_text"><span class="finish_dl_func"></span><span class="dl_progress_info"></span></div></div>""")
+        # скрытые элементы для взаимодействия с функцией определения места на диске
+        spacetextbox = gr.Textbox(label="", elem_id="free_space_area")
+        space_button = gr.Button("проверить свободное место", elem_id="free_space_button")
+        space_button.click(fn=free_space, outputs=spacetextbox)
+        # HTML-разметка для формирования чекбоксов для удаления файлов моделей
+        gr.HTML(
+            """<hr class="divider"/><div id="filemanager"><h2 class="current_models_files">файлы моделей которые можно удалить для освобождения места:</h2><div id="files_checkbox"></div><div class="filebuttons"><div id="delete_files_button"></div><div id="refresh_files_button"></div></div></div>""")
+
+        # функция получения путей до установленных файлов моделей
+        def get_models_paths():
+            file_paths = []
+            for root, dirs, files in os.walk(models_folder_path):
+                for file in files:
+                    file_path = os.path.join(root, file)
+                    file_paths.append(file_path)
+            return '\n'.join(file_paths)
+
+        # скрытые текстбокс и кнопка для получения списка файлов из папки моделей
+        filestextbox = gr.Textbox(label="", elem_id="files_area")
+        files_button = gr.Button("установленные модели", elem_id="files_button")
+        files_button.click(fn=get_models_paths, outputs=filestextbox)
+
+        # функция удаления списка отмеченных файлов
+        def del_models(inputs):
+            files_to_delete = inputs.split("\n")
+            for file in files_to_delete:
+                if file and file != "None":
+                    try:
+                        os.remove(os.path.join(models_folder_path, file))
+                        print(f"успешно удалена модель: {file}")
+                    except OSError as e:
+                        print(f"ОШИБКА: {e.filename} - {e.strerror}.")
+                else:
+                    print("удалять нечего, или ничего не выбрано для удаления")
+
+        # скрытые текстбокс и кнопка для формирования списка файлов на удаление
+        deletetextbox = gr.Textbox(label="", elem_id="delete_area")
+        delete_button = gr.Button("удалить", elem_id="delete_button")
+        delete_button.click(fn=del_models, inputs=deletetextbox, outputs=deletetextbox)
+        with gr.Accordion("описания моделей (не обновляется)", open=False):
+            gr.HTML(requests.get("https://raw.githubusercontent.com/PR0LAPSE/PR0LAPSE.github.io/main/models.html").text)
+    # возврат всех элементов
+    return (models_list, "модели", "models_list"),
+
+
+# вывод в вебуи
+script_callbacks.on_ui_tabs(on_ui_tabs)

style.css

@@ -0,0 +1,302 @@
+#free_space_area,
+#free_space_button,
+#files_area,
+#delete_area,
+#files_button,
+#delete_button,
+#ownlinks_download_button,
+#checkboxes_download_button,
+#dlresultbox,
+#downloads_result_text,
+#downloads_start_text,
+.models_porgress_loader {
+	display: none !important;
+}
+#models_search_input {
+	width: 500px !important;
+	float: none !important;
+	display: inline !important;
+	position: relative;
+	top: -25px;
+}
+#clear_search_models_results {
+	display: inline-block;
+	cursor: pointer !important;
+	--wh: 40px;
+	width: var(--wh);
+	height: var(--wh);
+	padding-top: 0px !important;
+	margin: 10px 0 0 10px!important;
+	position: relative;
+	top: -10px;
+	border-radius: 5px;
+	border: var(--button-border-width) solid var(--button-secondary-border-color);
+	text-align: center;
+	background-image: url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' width='40' height='40' viewBox='0 0 40 40'%3E%3Cpath d='m9.02,30.98L30.98,9.02m0,21.96L9.02,9.02' fill='none' opacity='.5' stroke='%23fff' stroke-linecap='round' stroke-linejoin='round' stroke-width='1.17'/%3E%3C/svg%3E");
+	background-color: var(--button-secondary-background-fill);
+}
+#finded_models {
+	position: absolute;
+	z-index: 2;
+	width: 500px;
+	top: 50px;
+	background: var(--input-background-fill);
+	color: var(--body-text-color);
+}
+#finded_models label {
+	background: none !important;
+	border: none;
+}
+#tab_models_list .tabs label {
+	background: none !important;
+	border: none;
+}
+#tab_models_list .tabs .tabitem div fieldset div {
+	display: grid;
+	grid-gap: 10px;
+}
+#tab_models_list div[data-testid='checkbox-group'] label span {
+	width: 275px;
+	font-size: 0.92em !important;
+	white-space: nowrap;
+	overflow: hidden;
+	text-overflow: ellipsis;
+}
+#tab_models_list div.tab-nav > button {
+	font-size: 1em !important;
+}
+#tab_models_list fieldset label:hover:not(.selected) {
+	color: var(--checkbox-background-color-selected)!important;
+}
+#tab_models_list fieldset .selected {
+	color: #60d43d !important;
+}
+#general_download_button {
+	border: var(--button-border-width) solid var(--button-primary-border-color);
+	background: var(--button-primary-background-fill);
+	color: var(--button-primary-text-color);
+	margin-top: 28px !important;
+}
+#tab_models_list button.secondary {
+	width: 70%;
+	margin: 0 auto;
+}
+#male_only {
+	color: #b8dabb !important;
+}
+.freespaceinfo {
+	width: fit-content;
+	display: flex;
+	flex-direction: row;
+	flex-wrap: nowrap;
+	justify-content: flex-start;
+	align-items: center;
+}
+#frespace_output {
+	width: fit-content;
+	height: 40px;
+	border: 1px solid var(--button-secondary-border-color);
+	margin: -30px 0px 0px 0px;
+	display: flex;
+	flex-direction: row;
+	flex-wrap: nowrap;
+	align-items: center;
+	padding: 0px 15px;
+	font-size: 1.2em;
+}
+#freespace_get {
+	width: 40px;
+	height: 40px;
+	border: 1px solid var(--button-secondary-border-color);
+	margin: -30px 50px 0px 10px;
+	cursor: pointer;
+	background-image: url("data:image/svg+xml,%3Csvg id='refresh' xmlns='http://www.w3.org/2000/svg' width='30' height='30' viewBox='0 0 36 36'%3E%3Cpath fill='%23fff' opacity='.5' d='m3.33,33v-2.81h6.09l-.7-.56c-2-1.59-3.45-3.33-4.36-5.2-.91-1.88-1.36-3.97-1.36-6.28,0-3.31.98-6.29,2.93-8.93,1.95-2.64,4.52-4.46,7.71-5.46v2.91c-2.34.91-4.23,2.41-5.67,4.52-1.44,2.11-2.16,4.43-2.16,6.96,0,1.97.37,3.68,1.1,5.13.73,1.45,1.73,2.71,2.98,3.77l1.41.98v-5.81h2.81v10.78H3.33Zm19.08-.7v-2.95c2.37-.91,4.27-2.41,5.67-4.52,1.41-2.11,2.11-4.43,2.11-6.96,0-1.5-.37-3.02-1.1-4.57-.73-1.55-1.7-2.91-2.88-4.1l-1.36-1.22v5.81h-2.81V3h10.78v2.81h-6.14l.7.66c1.88,1.75,3.28,3.63,4.22,5.63.94,2,1.41,3.92,1.41,5.77,0,3.31-.97,6.3-2.91,8.95-1.94,2.66-4.5,4.48-7.69,5.48Z'%3E%3C/path%3E%3C/svg%3E");
+	background-position: 50%;
+	background-repeat: no-repeat
+}
+#freespace_get:hover {
+	background-color: #00000078;
+	background-image: url("data:image/svg+xml,%3Csvg id='refresh' xmlns='http://www.w3.org/2000/svg' width='30' height='30' viewBox='0 0 36 36'%3E%3Cpath fill='%23fff' opacity='.7' d='m3.33,33v-2.81h6.09l-.7-.56c-2-1.59-3.45-3.33-4.36-5.2-.91-1.88-1.36-3.97-1.36-6.28,0-3.31.98-6.29,2.93-8.93,1.95-2.64,4.52-4.46,7.71-5.46v2.91c-2.34.91-4.23,2.41-5.67,4.52-1.44,2.11-2.16,4.43-2.16,6.96,0,1.97.37,3.68,1.1,5.13.73,1.45,1.73,2.71,2.98,3.77l1.41.98v-5.81h2.81v10.78H3.33Zm19.08-.7v-2.95c2.37-.91,4.27-2.41,5.67-4.52,1.41-2.11,2.11-4.43,2.11-6.96,0-1.5-.37-3.02-1.1-4.57-.73-1.55-1.7-2.91-2.88-4.1l-1.36-1.22v5.81h-2.81V3h10.78v2.81h-6.14l.7.66c1.88,1.75,3.28,3.63,4.22,5.63.94,2,1.41,3.92,1.41,5.77,0,3.31-.97,6.3-2.91,8.95-1.94,2.66-4.5,4.48-7.69,5.48Z'%3E%3C/path%3E%3C/svg%3E");
+}
+#freespace_get:active {
+	background-color: #00000078;
+	background-image: url("data:image/svg+xml,%3Csvg id='refresh' xmlns='http://www.w3.org/2000/svg' width='30' height='30' viewBox='0 0 36 36'%3E%3Cpath fill='%23fff' opacity='.85' d='m3.33,33v-2.81h6.09l-.7-.56c-2-1.59-3.45-3.33-4.36-5.2-.91-1.88-1.36-3.97-1.36-6.28,0-3.31.98-6.29,2.93-8.93,1.95-2.64,4.52-4.46,7.71-5.46v2.91c-2.34.91-4.23,2.41-5.67,4.52-1.44,2.11-2.16,4.43-2.16,6.96,0,1.97.37,3.68,1.1,5.13.73,1.45,1.73,2.71,2.98,3.77l1.41.98v-5.81h2.81v10.78H3.33Zm19.08-.7v-2.95c2.37-.91,4.27-2.41,5.67-4.52,1.41-2.11,2.11-4.43,2.11-6.96,0-1.5-.37-3.02-1.1-4.57-.73-1.55-1.7-2.91-2.88-4.1l-1.36-1.22v5.81h-2.81V3h10.78v2.81h-6.14l.7.66c1.88,1.75,3.28,3.63,4.22,5.63.94,2,1.41,3.92,1.41,5.77,0,3.31-.97,6.3-2.91,8.95-1.94,2.66-4.5,4.48-7.69,5.48Z'%3E%3C/path%3E%3C/svg%3E");
+	transform: rotate(90deg)
+}
+span#frespace_out {
+	color: #7ced7c
+}
+.ownfiles_header {
+	display: flex;
+}
+.ownfiles_header > span {
+	margin-right: 50px;
+}
+.ownfiles_header h2 {
+	font-size: 1.2em !important;
+	color: #deeedb !important;
+	font-weight: normal !important
+}
+#ownmodels > label > span {
+	color: #37e46f;
+}
+#ownloras > label > span {
+	color: #45b9ec;
+}
+#ownembeddings > label > span {
+	color: #f46ea2;
+}
+#ownmodels > label > textarea,
+#ownloras > label > textarea,
+#ownembeddings > label > textarea {
+	font-family: monospace;
+	font-size: 0.95em
+}
+#ownmodels > label > textarea {
+	color: #a7d0b4;
+}
+#ownloras > label > textarea {
+	color: #9bc5d9;
+}
+#ownembeddings > label > textarea {
+	color: #dcbfcb;
+}
+#filemanager {
+	height: max-content;
+}
+.current_models_files {
+	margin-bottom: 30px!important;
+	color: #eedbdb !important;
+}
+#files_checkbox {
+	width: max-content;
+	display: grid;
+	grid-template-columns: repeat(3, 1fr);
+	grid-gap: 35px;
+	grid-row-gap: 18px;
+}
+.filecheckbox input {
+	margin-right: 6px;
+	border: var(--checkbox-border-width) solid var(--checkbox-border-color) !important;
+	background-color: var(--checkbox-background-color) !important;
+	line-height: var(--line-sm) !important;
+}
+.filecheckbox input:checked {
+	border-color: #f47373 !important;
+	background-image: #de2f2f !important;
+	background-color: #de2f2f !important;
+}
+.filecheckbox span:hover {
+	color: #ff9494
+}
+.filebuttons {
+	display: flex;
+	flex-direction: row;
+	flex-wrap: wrap;
+	align-content: center;
+	align-items: center;
+	justify-content: center;
+	margin-top: 15px;
+	padding-top: 15px;
+	border-top: 2px solid #afafaf17;
+}
+#delete_files_button,
+#refresh_files_button {
+	width: 40px;
+	height: 40px;
+	border: 1px solid var(--button-secondary-border-color);
+	margin: 0 50px;
+	cursor: pointer;
+	background-color: transparent;
+	background-position: 50%;
+	background-repeat: no-repeat
+}
+#refresh_files_button {
+	background-image: url("data:image/svg+xml,%3Csvg id='refresh' xmlns='http://www.w3.org/2000/svg' width='30' height='30' viewBox='0 0 36 36'%3E%3Cpath fill='%23fff' opacity='.5' d='m3.33,33v-2.81h6.09l-.7-.56c-2-1.59-3.45-3.33-4.36-5.2-.91-1.88-1.36-3.97-1.36-6.28,0-3.31.98-6.29,2.93-8.93,1.95-2.64,4.52-4.46,7.71-5.46v2.91c-2.34.91-4.23,2.41-5.67,4.52-1.44,2.11-2.16,4.43-2.16,6.96,0,1.97.37,3.68,1.1,5.13.73,1.45,1.73,2.71,2.98,3.77l1.41.98v-5.81h2.81v10.78H3.33Zm19.08-.7v-2.95c2.37-.91,4.27-2.41,5.67-4.52,1.41-2.11,2.11-4.43,2.11-6.96,0-1.5-.37-3.02-1.1-4.57-.73-1.55-1.7-2.91-2.88-4.1l-1.36-1.22v5.81h-2.81V3h10.78v2.81h-6.14l.7.66c1.88,1.75,3.28,3.63,4.22,5.63.94,2,1.41,3.92,1.41,5.77,0,3.31-.97,6.3-2.91,8.95-1.94,2.66-4.5,4.48-7.69,5.48Z'%3E%3C/path%3E%3C/svg%3E");
+}
+#refresh_files_button:hover {
+	background-color: #00000078;
+	background-image: url("data:image/svg+xml,%3Csvg id='refresh' xmlns='http://www.w3.org/2000/svg' width='30' height='30' viewBox='0 0 36 36'%3E%3Cpath fill='%23fff' opacity='.7' d='m3.33,33v-2.81h6.09l-.7-.56c-2-1.59-3.45-3.33-4.36-5.2-.91-1.88-1.36-3.97-1.36-6.28,0-3.31.98-6.29,2.93-8.93,1.95-2.64,4.52-4.46,7.71-5.46v2.91c-2.34.91-4.23,2.41-5.67,4.52-1.44,2.11-2.16,4.43-2.16,6.96,0,1.97.37,3.68,1.1,5.13.73,1.45,1.73,2.71,2.98,3.77l1.41.98v-5.81h2.81v10.78H3.33Zm19.08-.7v-2.95c2.37-.91,4.27-2.41,5.67-4.52,1.41-2.11,2.11-4.43,2.11-6.96,0-1.5-.37-3.02-1.1-4.57-.73-1.55-1.7-2.91-2.88-4.1l-1.36-1.22v5.81h-2.81V3h10.78v2.81h-6.14l.7.66c1.88,1.75,3.28,3.63,4.22,5.63.94,2,1.41,3.92,1.41,5.77,0,3.31-.97,6.3-2.91,8.95-1.94,2.66-4.5,4.48-7.69,5.48Z'%3E%3C/path%3E%3C/svg%3E");
+}
+#refresh_files_button:active {
+	background-color: #00000078;
+	background-image: url("data:image/svg+xml,%3Csvg id='refresh' xmlns='http://www.w3.org/2000/svg' width='30' height='30' viewBox='0 0 36 36'%3E%3Cpath fill='%23fff' opacity='.85' d='m3.33,33v-2.81h6.09l-.7-.56c-2-1.59-3.45-3.33-4.36-5.2-.91-1.88-1.36-3.97-1.36-6.28,0-3.31.98-6.29,2.93-8.93,1.95-2.64,4.52-4.46,7.71-5.46v2.91c-2.34.91-4.23,2.41-5.67,4.52-1.44,2.11-2.16,4.43-2.16,6.96,0,1.97.37,3.68,1.1,5.13.73,1.45,1.73,2.71,2.98,3.77l1.41.98v-5.81h2.81v10.78H3.33Zm19.08-.7v-2.95c2.37-.91,4.27-2.41,5.67-4.52,1.41-2.11,2.11-4.43,2.11-6.96,0-1.5-.37-3.02-1.1-4.57-.73-1.55-1.7-2.91-2.88-4.1l-1.36-1.22v5.81h-2.81V3h10.78v2.81h-6.14l.7.66c1.88,1.75,3.28,3.63,4.22,5.63.94,2,1.41,3.92,1.41,5.77,0,3.31-.97,6.3-2.91,8.95-1.94,2.66-4.5,4.48-7.69,5.48Z'%3E%3C/path%3E%3C/svg%3E");
+	transform: rotate(90deg)
+}
+#delete_files_button {
+	background-image: url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' width='32' height='36' viewBox='0 0 32 36'%3E%3Cpath d='m10.05,27.05l5.95-6.05,6,6.05,2.35-2.4-5.95-6.05,5.95-6.05-2.35-2.4-6,6.05-5.95-6.05-2.4,2.4,6,6.05-6,6.05,2.4,2.4Zm-5,8.95c-.8,0-1.5-.3-2.1-.9s-.9-1.3-.9-2.1V4.5H0V1.5h9.4V0h13.2v1.5h9.4v3h-2.05v28.5c0,.8-.3,1.5-.9,2.1s-1.3.9-2.1.9H5.05ZM26.95,4.5H5.05v28.5h21.9V4.5Zm-21.9,0v28.5V4.5Z' opacity='.5' fill='%23fff'/%3E%3C/svg%3E");
+	background-size: 62%;
+}
+#delete_files_button:hover {
+	background-image: url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' width='32' height='36' viewBox='0 0 32 36'%3E%3Cpath d='m10.05,27.05l5.95-6.05,6,6.05,2.35-2.4-5.95-6.05,5.95-6.05-2.35-2.4-6,6.05-5.95-6.05-2.4,2.4,6,6.05-6,6.05,2.4,2.4Zm-5,8.95c-.8,0-1.5-.3-2.1-.9s-.9-1.3-.9-2.1V4.5H0V1.5h9.4V0h13.2v1.5h9.4v3h-2.05v28.5c0,.8-.3,1.5-.9,2.1s-1.3.9-2.1.9H5.05ZM26.95,4.5H5.05v28.5h21.9V4.5Zm-21.9,0v28.5V4.5Z' opacity='.7' fill='%23ff2727'/%3E%3C/svg%3E");
+	background-size: 62%;
+}
+#delete_files_button:active {
+	background-color: #00000078;
+	background-image: url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' width='32' height='36' viewBox='0 0 32 36'%3E%3Cpath d='m10.05,27.05l5.95-6.05,6,6.05,2.35-2.4-5.95-6.05,5.95-6.05-2.35-2.4-6,6.05-5.95-6.05-2.4,2.4,6,6.05-6,6.05,2.4,2.4Zm-5,8.95c-.8,0-1.5-.3-2.1-.9s-.9-1.3-.9-2.1V4.5H0V1.5h9.4V0h13.2v1.5h9.4v3h-2.05v28.5c0,.8-.3,1.5-.9,2.1s-1.3.9-2.1.9H5.05ZM26.95,4.5H5.05v28.5h21.9V4.5Zm-21.9,0v28.5V4.5Z' opacity='1' fill='%23ff2727'/%3E%3C/svg%3E");
+	background-size: 62%;
+}
+.models_porgress_loader {
+	width: 70%;
+	height: 8px;
+	display: block;
+	background: var(--block-label-border-color);
+	position: relative;
+	top: 0px;
+	overflow: hidden;
+	margin: 0 auto;
+}
+.models_porgress_loader::after {
+	content: '';
+	width: 0%;
+	height: 8px;
+	background-color: #ffffff2e;
+	background-image: linear-gradient(45deg, rgba(0, 0, 0, .25) 25%, transparent 25%, transparent 50%, rgba(0, 0, 0, 0.25) 50%, rgba(0, 0, 0, 0.25) 75%, transparent 75%, transparent);
+	background-size: 15px 15px;
+	position: absolute;
+	top: 0;
+	left: 0;
+	box-sizing: border-box;
+	animation: models_progress_bar_anim 10s cubic-bezier(0.45, 0.05, 0.55, 0.95) infinite;
+}
+@keyframes models_progress_bar_anim {
+	0% {
+		width: 0;
+	}
+	100% {
+		width: 100%;
+	}
+}
+#downloads_result_text,
+#downloads_start_text {
+	margin: 32px auto;
+	width: fit-content;
+	top: 100px;
+	font-size: 1.4em;
+}
+#downloads_start_text {
+	color: #81d0fb !important
+}
+#downloads_result_text > span {
+	display: block;
+	text-align: center;
+}
+#downloads_result_text > span.finish_dl_func {
+	color: #99fb99 !important
+}
+#downloads_result_text > span.dl_progress_info {
+	font-size: 0.7em!important;
+	margin-top: 10px;
+}
+#tab_models_list #colab_model_list.colab_model_list {
+	background: transparent !important;
+	background-image: none !important;
+}
+.request_new_models a {
+	color: var(--link-text-color) !important;
+	text-decoration: underline dashed !important
+}
+.request_new_models a:hover {
+	color: var(--link-text-color-hover) !important;
+	text-decoration: underline solid !important
+}