Files

23 lines
1.1 KiB
Python

from .gemini_node import Gemini3ProImageGenNode, GeminiPromptGenerator, GeminiAspectRatioSelector, GeminiResolutionSelector, GeminiImageModelSelector, GeminiGridFormatSelector
NODE_CLASS_MAPPINGS = {
"Gemini3ProImageGenNode": Gemini3ProImageGenNode,
"GeminiPromptGenerator": GeminiPromptGenerator,
"GeminiAspectRatioSelector": GeminiAspectRatioSelector,
"GeminiResolutionSelector": GeminiResolutionSelector,
"GeminiImageModelSelector": GeminiImageModelSelector,
"GeminiGridFormatSelector": GeminiGridFormatSelector
}
NODE_DISPLAY_NAME_MAPPINGS = {
"Gemini3ProImageGenNode": "🤖 Gemini 3 Pro Image (API Fallback)",
"GeminiPromptGenerator": "🤖 Gemini 3 Pro (API Fallback)",
"GeminiAspectRatioSelector": "📐 Aspect Ratio Selector (API Fallback)",
"GeminiResolutionSelector": "📺 Resolution Selector (API Fallback)",
"GeminiImageModelSelector": "🧠 Image Model Selector (API Fallback)",
"GeminiGridFormatSelector": "🏁 Grid Format Selector (API Fallback)"
}
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]