From 23b7ef009234378a46d794de0daa014e26fc7ca5 Mon Sep 17 00:00:00 2001
From: shrimbly <falloon.willie@gmail.com>
Date: Mon, 20 Apr 2026 07:07:54 +1200
Subject: [PATCH] fix: preserve audio handle type in GenerateAudioNode dynamic
 inputs

The input schema mapping was forcing all non-image inputs to "text",
misclassifying audio inputs. Now correctly maps audio type so handles
get the right data-handletype, CSS dot color, and label color.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 src/components/nodes/GenerateAudioNode.tsx | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/components/nodes/GenerateAudioNode.tsx b/src/components/nodes/GenerateAudioNode.tsx
index 2ba9ac0f..8829284a 100644
--- a/src/components/nodes/GenerateAudioNode.tsx
+++ b/src/components/nodes/GenerateAudioNode.tsx
@@ -229,8 +229,9 @@ export function GenerateAudioNode({ id, data, selected }: NodeProps<GenerateAudi
     if (!nodeData.inputSchema || nodeData.inputSchema.length === 0) return null;
 
     return nodeData.inputSchema.map((input, index) => {
-      const handleType = input.type === "image" ? "image" : "text";
+      const handleType = input.type === "image" ? "image" : input.type === "audio" ? "audio" : "text";
       const topPx = 50 + (index - nodeData.inputSchema!.length / 2 + 0.5) * 20;
+      const handleColor = handleType === "image" ? "var(--handle-color-image)" : handleType === "audio" ? "var(--handle-color-audio)" : "var(--handle-color-text)";
       return (
         <React.Fragment key={input.name}>
           <Handle
@@ -238,13 +239,10 @@ export function GenerateAudioNode({ id, data, selected }: NodeProps<GenerateAudi
             position={Position.Left}
             id={input.name}
             data-handletype={handleType}
-            style={{
-              background: handleType === "image" ? "rgb(34, 197, 94)" : "rgb(251, 191, 36)",
-              top: `${topPx}px`,
-            }}
+            style={{ top: `${topPx}px` }}
             title={input.label}
           />
-          <HandleLabel label={input.label} side="target" color={handleType === "image" ? "var(--handle-color-image)" : "var(--handle-color-text)"} top={`${topPx - 18}px`} visible={showLabels} />
+          <HandleLabel label={input.label} side="target" color={handleColor} top={`${topPx - 18}px`} visible={showLabels} />
         </React.Fragment>
       );
     });