PLUS: защита от наборов, которые размазывают модель

plus/serve_muscriptor.py

@@ -382,6 +382,32 @@ def expand_instruments(text: str) -> list[str]:
         return WIDE_INSTRUMENTS.split(",")
     return [name.strip() for name in cleaned.split(",") if name.strip()]
 
+MAX_GROUPS = 10                      # шире модель размазывается: замерено
+BAND_LIST = "drums,electric_bass,acoustic_bass,chromatic_percussion,acoustic_piano,electric_piano,clean_electric_guitar,synth_pad".split(",")
+
+
+def guard_instruments(names: list[str], label: str = "") -> tuple[list[str], str]:
+    """Свести опасный набор к проверенному и объяснить почему."""
+    if not names:
+        return names, ""
+    band_alias = [n for n in names if n.lower() in ("band", "набор")]
+    if band_alias:
+        return list(BAND_LIST), "%s: это слово-псевдоним, беру проверенный набор" % label
+    known = set(ALL_INSTRUMENTS.split(","))
+    unknown = [n for n in names if n not in known]
+    if unknown:
+        names = [n for n in names if n in known]
+        if not names:
+            return list(BAND_LIST), ("%s: неизвестные группы %s, беру проверенный набор"
+                                     % (label, ", ".join(unknown[:4])))
+        return names, "%s: выкинул неизвестные группы %s" % (label, ", ".join(unknown[:4]))
+    if len(names) > MAX_GROUPS:
+        return list(BAND_LIST), (
+            "%s: %d групп из %d — модель размазывается (замер: 35 групп дали 0 нот,"
+            " 12 групп хуже на 9 попаданий). Беру проверенный набор из %d."
+            % (label or "набор", len(names), len(names), len(BAND_LIST)))
+    return names, ""
+
 def start_proxy(server_port: int, listen_port: int,
                 instruments: str = BAND_INSTRUMENTS,
                 multi_pass: str = "",
@@ -457,6 +483,17 @@ def main(argv: list[str] | None = None) -> int:
         CUSTOM_SSH = args.ssh
 
     ensure_package()
+    checked, warning = guard_instruments(expand_instruments(args.instruments),
+                                         "--instruments")
+    if warning:
+        say("      " + warning)
+        args.instruments = ",".join(checked)
+    if args.multi_pass:
+        checked_pass, warning = guard_instruments(expand_instruments(args.multi_pass),
+                                                  "--multi-pass")
+        if warning:
+            say("      " + warning)
+            args.multi_pass = ",".join(checked_pass)
     if not args.skip_audio:
         ensure_audio_tools()
     ensure_ssh()