diff --git a/src/components/nodes/GenerateImageNode.tsx b/src/components/nodes/GenerateImageNode.tsx index c0b06364..efc6ec6e 100644 --- a/src/components/nodes/GenerateImageNode.tsx +++ b/src/components/nodes/GenerateImageNode.tsx @@ -51,6 +51,19 @@ export function GenerateImageNode({ id, data, selected }: NodeProps { + if (nodeData.model && !nodeData.selectedModel) { + const displayName = nodeData.model === "nano-banana" ? "Nano Banana" : "Nano Banana Pro"; + const newSelectedModel: SelectedModel = { + provider: "gemini", + modelId: nodeData.model, + displayName, + }; + updateNodeData(id, { selectedModel: newSelectedModel }); + } + }, [id, nodeData.model, nodeData.selectedModel, updateNodeData]); + // Fetch models from external providers when provider changes useEffect(() => { const fetchModels = async () => {