diff --git a/src/components/modals/ModelSearchDialog.tsx b/src/components/modals/ModelSearchDialog.tsx index 55029b51..0b097867 100644 --- a/src/components/modals/ModelSearchDialog.tsx +++ b/src/components/modals/ModelSearchDialog.tsx @@ -227,6 +227,21 @@ export function ModelSearchDialog({ } }; + // Get display name with suffix for fal.ai models to differentiate variants + const getDisplayName = (model: ProviderModel): string => { + if (model.provider === "fal") { + // Extract the last segment of the ID (e.g., "effects" from "kling-video/v1.6/pro/effects") + const segments = model.id.split("/"); + const lastSegment = segments[segments.length - 1]; + + // Only add suffix if it's not already in the name (case-insensitive) + if (lastSegment && !model.name.toLowerCase().includes(lastSegment.toLowerCase())) { + return `${model.name} - ${lastSegment}`; + } + } + return model.name; + }; + // Get model page URL for the provider's website const getModelUrl = (model: ProviderModel): string => { if (model.provider === "replicate") { @@ -483,9 +498,9 @@ export function ModelSearchDialog({ {/* Model Info */}