From 72754c6cb0a13d20140d2d5c20f23f745f49bfc0 Mon Sep 17 00:00:00 2001
From: shrimbly <falloon.willie@gmail.com>
Date: Tue, 31 Mar 2026 11:23:35 +1300
Subject: [PATCH] fix: add audio-to-video badge and generateVideo to audio drop
 targets

Add missing "audio-to-video" case to getCapabilityBadges() so the
capability badge renders in the model explorer. Add generateVideo to
AUDIO_TARGET_OPTIONS in ConnectionDropMenu so dropping an audio
connection offers Generate Video as a target node.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 src/components/ConnectionDropMenu.tsx       | 9 +++++++++
 src/components/modals/ModelSearchDialog.tsx | 4 ++++
 2 files changed, 13 insertions(+)

diff --git a/src/components/ConnectionDropMenu.tsx b/src/components/ConnectionDropMenu.tsx
index 8b3a60f4..ca8ff240 100644
--- a/src/components/ConnectionDropMenu.tsx
+++ b/src/components/ConnectionDropMenu.tsx
@@ -515,6 +515,15 @@ const AUDIO_TARGET_OPTIONS: MenuOption[] = [
       </svg>
     ),
   },
+  {
+    type: "generateVideo",
+    label: "Generate Video",
+    icon: (
+      <svg className="w-4 h-4" fill="none" viewBox="0 0 24 24" stroke="currentColor" strokeWidth={1.5}>
+        <path strokeLinecap="round" strokeLinejoin="round" d="m15.75 10.5 4.72-4.72a.75.75 0 0 1 1.28.53v11.38a.75.75 0 0 1-1.28.53l-4.72-4.72M4.5 18.75h9a2.25 2.25 0 0 0 2.25-2.25v-9a2.25 2.25 0 0 0-2.25-2.25h-9A2.25 2.25 0 0 0 2.25 7.5v9a2.25 2.25 0 0 0 2.25 2.25Z" />
+      </svg>
+    ),
+  },
   {
     type: "output",
     label: "Output",
diff --git a/src/components/modals/ModelSearchDialog.tsx b/src/components/modals/ModelSearchDialog.tsx
index 7ba4f3f5..febf97f1 100644
--- a/src/components/modals/ModelSearchDialog.tsx
+++ b/src/components/modals/ModelSearchDialog.tsx
@@ -539,6 +539,10 @@ export function ModelSearchDialog({
           color = "bg-fuchsia-500/20 text-fuchsia-300";
           label = "txt→audio";
           break;
+        case "audio-to-video":
+          color = "bg-violet-500/20 text-violet-300";
+          label = "audio→vid";
+          break;
       }
 
       if (label) {