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"]