From 5843edac8acec803f31f367050d463fd1f621658 Mon Sep 17 00:00:00 2001
From: shrimbly <falloon.willie@gmail.com>
Date: Wed, 18 Feb 2026 22:07:49 +1300
Subject: [PATCH] refactor: simplify mapFalCategory with lookup map

Replace 3 if-statements mapping fal audio categories to a single
record lookup, making it easier to add new audio categories.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 src/app/api/models/route.ts | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/app/api/models/route.ts b/src/app/api/models/route.ts
index 9563d188..cd43b6b9 100644
--- a/src/app/api/models/route.ts
+++ b/src/app/api/models/route.ts
@@ -802,15 +802,15 @@ async function fetchWaveSpeedModels(apiKey: string): Promise<ProviderModel[]> {
 
 // ============ Fal.ai Helpers ============
 
+const FAL_AUDIO_CATEGORIES: Record<string, ModelCapability> = {
+  "text-to-speech": "text-to-audio",
+  "text-to-music": "text-to-audio",
+  "text-to-sound-effects": "text-to-audio",
+};
+
 function mapFalCategory(category: string): ModelCapability | null {
-  if (category === "text-to-speech") {
-    return "text-to-audio";
-  }
-  if (category === "text-to-music") {
-    return "text-to-audio";
-  }
-  if (category === "text-to-sound-effects") {
-    return "text-to-audio";
+  if (category in FAL_AUDIO_CATEGORIES) {
+    return FAL_AUDIO_CATEGORIES[category];
   }
   if (RELEVANT_CATEGORIES.includes(category)) {
     return category as ModelCapability;