Add application file

webui.py

@@ -234,21 +234,22 @@ weight_pymss_root = os.getenv("weight_pymss_root")
 outside_index_root = os.getenv("outside_index_root")
 
 # PyMSS functionality is temporarily disabled.
-# def weight_names():
-#     return sorted(
-#         name for name in os.listdir(weight_root) if name.endswith(".pth")
-#     )
-#
-#
-#def refresh_weight_choices(previous_names=None, force=False):
-#    current_names = tuple(weight_names())
-#    if force or current_names != previous_names:
-#        return current_names, change_choices()
-#    return current_names, {"__type__": "update"}
+def weight_names():
+    return sorted(
+        name for name in os.listdir(weight_root) if name.endswith(".pth")
+    )
 
 
-names = weight_names()
-pymss_names = PYMSS_MODEL_CHOICES
+def refresh_weight_choices(previous_names=None, force=False):
+    current_names = tuple(weight_names())
+    if force or current_names != previous_names:
+        return current_names, change_choices()
+    return current_names, {"__type__": "update"}
+
+
+# Names and PyMSS model choices are not used outside training flows.
+# names = weight_names()
+# pymss_names = PYMSS_MODEL_CHOICES
 
 
 def change_choices():
@@ -367,109 +368,109 @@ def render_pymss_progress(percent=0, label="等待开始", state="idle"):
 #     save_root_ins,
 #     format0,
 # ):
-    progress_state = {
-        "percent": 0.0,
-        "label": "正在准备 PyMSS 分离任务",
-        "state": "running",
-    }
-    busy = False
-
-    def update_progress(event):
-        nonlocal busy
-        event_type = event.get("event")
-        file_count = max(1, int(event.get("file_count") or 1))
-        file_index = max(1, int(event.get("file_index") or 1))
-        message = str(event.get("message") or "")
-
-        if event_type == "progress":
-            done = max(0.0, float(event.get("done") or 0))
-            total = max(1.0, float(event.get("total") or 1))
-            file_fraction = min(1.0, done / total)
-            progress_state["percent"] = (
-                (file_index - 1 + file_fraction) / file_count * 100
-            )
-            progress_state["label"] = "文件 %s/%s · %s · %.0f/%.0f 秒" % (
-                file_index,
-                file_count,
-                message or "正在处理音频",
-                done,
-                total,
-            )
-            progress_state["state"] = "running"
-        elif event_type == "file_start":
-            progress_state["percent"] = (file_index - 1) / file_count * 100
-            progress_state["label"] = message
-            progress_state["state"] = "running"
-        elif event_type == "file":
-            progress_state["percent"] = file_index / file_count * 100
-            progress_state["label"] = (
-                message.splitlines()[0] if message else "文件处理结束"
-            )
-            progress_state["state"] = "running" if event.get("ok") else "failed"
-        elif event_type == "done":
-            successful = int(event.get("successful") or 0)
-            failed = int(event.get("failed") or 0)
-            progress_state["percent"] = 100.0
-            progress_state["label"] = "分离完成：成功 %s，失败 %s" % (
-                successful,
-                failed,
-            )
-            progress_state["state"] = "done" if failed == 0 else "failed"
-        elif event_type == "retry_fp32":
-            progress_state["percent"] = 0.0
-            progress_state["label"] = message
-            progress_state["state"] = "running"
-        elif event_type == "cancelled":
-            progress_state["label"] = message or "PyMSS 分离任务已停止"
-            progress_state["state"] = "stopped"
-        elif event_type in {"fatal", "busy"}:
-            progress_state["label"] = (
-                message.splitlines()[0] if message else "PyMSS 分离任务失败"
-            )
-            progress_state["state"] = "failed"
-            busy = event_type == "busy"
-        elif event_type in {"preparing", "precision_attempt", "status"}:
-            progress_state["label"] = message
-            progress_state["state"] = "running"
-
-    last_info = ""
-    try:
-        for info in _pymss_separate_core(
-            model_name,
-            inp_root,
-            save_root_vocal,
-            paths,
-            save_root_ins,
-            format0,
-            event_callback=update_progress,
-        ):
-            last_info = info
-            start_button = button_update()
-            stop_button = button_update()
-            if not busy:
-                start_button = button_update(visible=False)
-                stop_button = button_update(visible=True)
-            yield (
-                info,
-                render_pymss_progress(**progress_state),
-                start_button,
-                stop_button,
-            )
-    except Exception:
-        last_info = "失败\n%s" % traceback.format_exc()
-        progress_state.update(
-            {"label": "PyMSS 分离任务失败", "state": "failed"}
-        )
-        logger.exception("PyMSS WebUI task failed")
-
-    start_button = button_update()
-    stop_button = button_update()
-    if not busy:
-        start_button = button_update(visible=True)
-        stop_button = button_update(visible=False)
-    yield (last_info, render_pymss_progress(**progress_state), start_button, stop_button)
-
-
+#     progress_state = {
+#         "percent": 0.0,
+#         "label": "正在准备 PyMSS 分离任务",
+#         "state": "running",
+#     }
+#     busy = False
+#
+#     def update_progress(event):
+#         nonlocal busy
+#         event_type = event.get("event")
+#         file_count = max(1, int(event.get("file_count") or 1))
+#         file_index = max(1, int(event.get("file_index") or 1))
+#         message = str(event.get("message") or "")
+#
+#         if event_type == "progress":
+#             done = max(0.0, float(event.get("done") or 0))
+#             total = max(1.0, float(event.get("total") or 1))
+#             file_fraction = min(1.0, done / total)
+#             progress_state["percent"] = (
+#                 (file_index - 1 + file_fraction) / file_count * 100
+#             )
+#             progress_state["label"] = "文件 %s/%s · %s · %.0f/%.0f 秒" % (
+#                 file_index,
+#                 file_count,
+#                 message or "正在处理音频",
+#                 done,
+#                 total,
+#             )
+#             progress_state["state"] = "running"
+#         elif event_type == "file_start":
+#             progress_state["percent"] = (file_index - 1) / file_count * 100
+#             progress_state["label"] = message
+#             progress_state["state"] = "running"
+#         elif event_type == "file":
+#             progress_state["percent"] = file_index / file_count * 100
+#             progress_state["label"] = (
+#                 message.splitlines()[0] if message else "文件处理结束"
+#             )
+#             progress_state["state"] = "running" if event.get("ok") else "failed"
+#         elif event_type == "done":
+#             successful = int(event.get("successful") or 0)
+#             failed = int(event.get("failed") or 0)
+#             progress_state["percent"] = 100.0
+#             progress_state["label"] = "分离完成：成功 %s，失败 %s" % (
+#                 successful,
+#                 failed,
+#             )
+#             progress_state["state"] = "done" if failed == 0 else "failed"
+#         elif event_type == "retry_fp32":
+#             progress_state["percent"] = 0.0
+#             progress_state["label"] = message
+#             progress_state["state"] = "running"
+#         elif event_type == "cancelled":
+#             progress_state["label"] = message or "PyMSS 分离任务已停止"
+#             progress_state["state"] = "stopped"
+#         elif event_type in {"fatal", "busy"}:
+#             progress_state["label"] = (
+#                 message.splitlines()[0] if message else "PyMSS 分离任务失败"
+#             )
+#             progress_state["state"] = "failed"
+#             busy = event_type == "busy"
+#         elif event_type in {"preparing", "precision_attempt", "status"}:
+#             progress_state["label"] = message
+#             progress_state["state"] = "running"
+#
+#     last_info = ""
+#     try:
+#         for info in _pymss_separate_core(
+#             model_name,
+#             inp_root,
+#             save_root_vocal,
+#             paths,
+#             save_root_ins,
+#             format0,
+#             event_callback=update_progress,
+#         ):
+#             last_info = info
+#             start_button = button_update()
+#             stop_button = button_update()
+#             if not busy:
+#                 start_button = button_update(visible=False)
+#                 stop_button = button_update(visible=True)
+#             yield (
+#                 info,
+#                 render_pymss_progress(**progress_state),
+#                 start_button,
+#                 stop_button,
+#             )
+#     except Exception:
+#         last_info = "失败\n%s" % traceback.format_exc()
+#         progress_state.update(
+#             {"label": "PyMSS 分离任务失败", "state": "failed"}
+#         )
+#         logger.exception("PyMSS WebUI task failed")
+#
+#     start_button = button_update()
+#     stop_button = button_update()
+#     if not busy:
+#         start_button = button_update(visible=True)
+#         stop_button = button_update(visible=False)
+#     yield (last_info, render_pymss_progress(**progress_state), start_button, stop_button)
+#
+#
 # PyMSS functionality is temporarily disabled.
 # def stop_pymss_webui():
 #     return (