небольшой рефакторинг

Bebra777228 · Jan 20, 2025 · 2bcde05 · 2bcde05
1 parent 51c35b0
commit 2bcde05
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 17 deletions.
diff --git a/app.py b/app.py
@@ -10,7 +10,7 @@
 
 from tabs.conversion import conversion_tab
 from tabs.edge_tts import edge_tts_tab
-from tabs.install import files_upload, url_download, zip_upload, install_hubert_tab, output_message
+from tabs.install import url_zip_download, zip_upload, files_upload, install_hubert_tab, output_message
 from tabs.welcome import welcome_tab
 
 DEFAULT_PORT = 4000
@@ -41,7 +41,7 @@
 
     with gr.Tab("Загрузка моделей"):
         with gr.Tab("Загрузка RVC моделей"):
-            url_download(output_message_component)
+            url_zip_download(output_message_component)
             zip_upload(output_message_component)
             files_upload(output_message_component)
             output_message_component.render()

diff --git a/tabs/install.py b/tabs/install.py
@@ -69,8 +69,8 @@ def download_and_replace_model(model_name, custom_url, progress=gr.Progress()):
         return f'Ошибка при установке модели "{model_name}": {str(e)}'
 
 
-def url_download(output_message):
-    with gr.Accordion("Загрузить по ссылке", open=False):
+def url_zip_download(output_message):
+    with gr.Accordion("Загрузить ZIP-файл по ссылке", open=False):
         gr.HTML(
             "<h3>"
             "Поддерживаемые сайты: "
@@ -83,7 +83,7 @@ def url_download(output_message):
         )
         with gr.Column():
             with gr.Group():
-                link = gr.Text(label="Ссылка на загрузку модели")
+                zip_link = gr.Text(label="Ссылка на загрузку ZIP-файла")
                 model_name = gr.Text(
                     label="Имя модели",
                     info="Дайте вашей загружаемой модели уникальное имя, отличное от других голосовых моделей.",
@@ -92,13 +92,13 @@ def url_download(output_message):
 
     download_btn.click(
         download_from_url,
-        inputs=[link, model_name],
+        inputs=[zip_link, model_name],
         outputs=output_message,
     )
 
 
 def zip_upload(output_message):
-    with gr.Accordion("Загрузить ZIP архивом", open=False):
+    with gr.Accordion("Загрузить ZIP-файл", open=False):
         with gr.Column():
             with gr.Group():
                 zip_file = gr.File(
@@ -118,7 +118,7 @@ def zip_upload(output_message):
 
 
 def files_upload(output_message):
-    with gr.Accordion("Загрузить файлами", open=False):
+    with gr.Accordion("Загрузить файлы .pth и .index", open=False):
         with gr.Column():
             with gr.Group():
                 with gr.Row(equal_height=False):
@@ -142,15 +142,14 @@ def files_upload(output_message):
 
 
 def install_hubert_tab():
-    with gr.Tab("Установка HuBERT моделей"):
-        gr.HTML("<center><h3>Не рекомендуется вносить изменения в этот раздел, если вы не проводили обучение RVC модели с использованием пользователькой HuBERT-модели.</h3></center>")
-        with gr.Row(variant="panel", equal_height=True):
-            with gr.Column(variant="panel"):
-                custom_url_checkbox = gr.Checkbox(label="Использовать другой HuBERT", value=False)
-                custom_url_textbox = gr.Textbox(label="URL модели", visible=False)
-                hubert_model_dropdown = gr.Dropdown(models, label="Список доступных HuBERT моделей:", visible=True)
-            hubert_download_btn = gr.Button("Установить!", variant="primary")
-        hubert_output_message = gr.Text(label="Сообщение вывода", interactive=False)
+    gr.HTML("<center><h3>Не рекомендуется вносить изменения в этот раздел, если вы не проводили обучение RVC модели с использованием пользователькой HuBERT-модели.</h3></center>")
+    with gr.Row(variant="panel", equal_height=True):
+        with gr.Column(variant="panel"):
+            custom_url_checkbox = gr.Checkbox(label="Использовать другой HuBERT", value=False)
+            custom_url_textbox = gr.Textbox(label="URL модели", visible=False)
+            hubert_model_dropdown = gr.Dropdown(models, label="Список доступных HuBERT моделей:", visible=True)
+        hubert_download_btn = gr.Button("Установить!", variant="primary")
+    hubert_output_message = gr.Text(label="Сообщение вывода", interactive=False)
 
     custom_url_checkbox.change(
         toggle_custom_url,