Update app.py

app.py

@@ -41,6 +41,35 @@ except ImportError:
     HF_HUB_AVAILABLE = False
     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):
+    _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)
+        return cfg
+    _WB._get_config = _patched_get_config
+    _WB._patched_for_custom_palette = True
+# ===== monkey-patch 끝 =====
+
+import json, pathlib
+
+palette_path = pathlib.Path(__file__).parent / "best_ai_palette.json"
+with open(palette_path, "r", encoding="utf-8") as f:
+    best_ai_palette = json.load(f)
+
 # -------------------------------------------------------------------
 # 🛠️  헬퍼 함수들
 # -------------------------------------------------------------------
@@ -1065,7 +1094,9 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
                     label="",
                     info="Drag nodes → Connect edges → Edit properties → Auto-save!",
                     value=workflow_value,
-                    elem_id="main_workflow"
+                    elem_id="main_workflow",
+                    custom_palette=best_ai_palette   # ← 새로 추가
+
                 )
 
                 # ---------- 저장 로직 ----------