Update app.py

app.py

@@ -42,27 +42,31 @@ except ImportError:
     print("Huggingface Hub not available. Install with: pip install huggingface-hub")
 
 
-# app.py 맨 위
+# ───────────────────────────────────────────────────────────
 from gradio_workflowbuilder import WorkflowBuilder as _WB
 
-# ===== monkey-patch 시작 =====
 if not getattr(_WB, "_patched_for_custom_palette", False):
+    # 1) __init__ 확장 ─ custom_palette 보관
     _orig_init = _WB.__init__
     def _patched_init(self, *args, custom_palette=None, **kwargs):
         self._custom_palette = custom_palette or []
         _orig_init(self, *args, **kwargs)
     _WB.__init__ = _patched_init
 
-    _orig_get_config = _WB._get_config
-    def _patched_get_config(self):
-        cfg = _orig_get_config(self)
-        if self._custom_palette:
-            cfg["palette"] = (cfg.get("palette", []) +
-                              self._custom_palette)
+    # 2) config 생성 시 팔레트 합치기
+    cfg_method = "_get_config" if hasattr(_WB, "_get_config") else "get_config"
+    _orig_get_cfg = getattr(_WB, cfg_method)
+
+    def _patched_get_cfg(self, *args, **kwargs):
+        cfg = _orig_get_cfg(self, *args, **kwargs)
+        if getattr(self, "_custom_palette", None):
+            cfg["palette"] = cfg.get("palette", []) + self._custom_palette
         return cfg
-    _WB._get_config = _patched_get_config
+
+    setattr(_WB, cfg_method, _patched_get_cfg)
     _WB._patched_for_custom_palette = True
-# ===== monkey-patch 끝 =====
+# ───────────────────────────────────────────────────────────
+
 
 import json, pathlib