Add H3 beat prompt builder node

dumas_h3_beat_prompt.py

@@ -0,0 +1,85 @@
+import json
+
+
+_DEFAULT_BEAT = "Describe this beat."
+_DEFAULT_STATE = {"beats": [{"text": _DEFAULT_BEAT}]}
+
+
+def _clone_default_state():
+    return {"beats": [{"text": _DEFAULT_BEAT}]}
+
+
+def _parse_beat_prompt_state(value):
+    if isinstance(value, dict):
+        raw = value
+    else:
+        text = str(value or "").strip()
+        if not text:
+            return _clone_default_state()
+        try:
+            raw = json.loads(text)
+        except Exception:
+            return _clone_default_state()
+
+    beats = []
+    for item in list(raw.get("beats") or []):
+        if isinstance(item, dict):
+            text = str(item.get("text") or "")
+        else:
+            text = str(item or "")
+        beats.append({"text": text})
+
+    if not beats:
+        return _clone_default_state()
+    return {"beats": beats}
+
+
+def _assemble_beat_prompt(state):
+    parsed = _parse_beat_prompt_state(state)
+    chunks = []
+    for beat in parsed["beats"]:
+        text = str(beat.get("text") or "").strip()
+        if text:
+            chunks.append(text)
+    return "\n\n".join(chunks)
+
+
+class DumasH3BeatPromptNode:
+    DESCRIPTION = (
+        "Build a MiniMax H3 prompt from one textbox per beat, with a front-end beat "
+        "editor that can append directive examples such as wardrobe, seconds, exit, and music."
+    )
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("prompt",)
+    FUNCTION = "build_prompt"
+    CATEGORY = "Dumas/MiniMax"
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {},
+            "hidden": {
+                "H3BeatPromptState": ("STRING", {"default": json.dumps(_DEFAULT_STATE)}),
+            },
+        }
+
+    def build_prompt(self, H3BeatPromptState=""):
+        return (_assemble_beat_prompt(H3BeatPromptState),)
+
+
+NODE_CLASS_MAPPINGS = {
+    "DumasH3BeatPrompt": DumasH3BeatPromptNode,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "DumasH3BeatPrompt": "Dumas H3 Beat Prompt",
+}
+
+
+__all__ = [
+    "DumasH3BeatPromptNode",
+    "NODE_CLASS_MAPPINGS",
+    "NODE_DISPLAY_NAME_MAPPINGS",
+    "_assemble_beat_prompt",
+    "_parse_beat_prompt_state",
+]