"""
Runware Bytedance Provider Settings Node
Provides Bytedance-specific settings for image/video generation
"""

from typing import Optional, Dict, Any

class RunwareBytedanceProviderSettings:
    """Runware Bytedance Provider Settings Node"""
    
    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {},
            "optional": {
                "useCameraFixed": ("BOOLEAN", {
                    "tooltip": "Enable to include cameraFixed parameter in provider settings",
                    "default": False,
                }),
                "cameraFixed": ("BOOLEAN", {
                    "tooltip": "Enable to fix camera position during video generation",
                    "default": False,
                }),
                "useMaxSequentialImages": ("BOOLEAN", {
                    "tooltip": "Enable to include maxSequentialImages parameter in provider settings",
                    "default": False,
                }),
                "maxSequentialImages": ("INT", {
                    "tooltip": "Maximum number of sequential images to generate (1-15)",
                    "default": 1,
                    "min": 1,
                    "max": 15,
                }),
                "useFastMode": ("BOOLEAN", {
                    "tooltip": "Enable to include fastMode parameter in provider settings",
                    "default": False,
                }),
                "fastMode": ("BOOLEAN", {
                    "tooltip": "When enabled, speeds up generation by sacrificing some effects. Default: false. RTF: 25-28 (fast) vs 35 (normal). Supported by OmniHuman 1.5.",
                    "default": False,
                }),
                "useAudio": ("BOOLEAN", {
                    "tooltip": "Enable to include audio parameter in provider settings.",
                    "default": False,
                }),
                "audio": ("BOOLEAN", {
                    "tooltip": "Enable audio generation support when available.",
                    "default": False,
                }),
                "useDraft": ("BOOLEAN", {
                    "tooltip": "Enable to include draft parameter in provider settings",
                    "default": False,
                }),
                "draft": ("BOOLEAN", {
                    "tooltip": "Enable draft mode for Bytedance provider",
                    "default": False,
                }),
                "useOptimizePromptMode": ("BOOLEAN", {
                    "tooltip": "Enable to include optimizePromptMode parameter in provider settings",
                    "default": False,
                }),
                "optimizePromptMode": (["standard", "fast"], {
                    "tooltip": "Prompt optimization mode. standard: higher quality, longer time. fast: faster generation, average quality.",
                    "default": "standard",
                }),
            }
        }
    
    RETURN_TYPES = ("RUNWAREPROVIDERSETTINGS",)
    RETURN_NAMES = ("providerSettings",)
    FUNCTION = "create_provider_settings"
    CATEGORY = "Runware/Provider Settings"
    
    def create_provider_settings(self, **kwargs) -> tuple[Dict[str, Any]]:
        """Create Bytedance provider settings"""
        
        # Get control parameters
        useCameraFixed = kwargs.get("useCameraFixed", False)
        useMaxSequentialImages = kwargs.get("useMaxSequentialImages", False)
        useFastMode = kwargs.get("useFastMode", False)
        useAudio = kwargs.get("useAudio", False)
        useDraft = kwargs.get("useDraft", False)
        useOptimizePromptMode = kwargs.get("useOptimizePromptMode", False)
        
        # Get value parameters
        cameraFixed = kwargs.get("cameraFixed", False)
        maxSequentialImages = kwargs.get("maxSequentialImages", 1)
        fastMode = kwargs.get("fastMode", False)
        audio = kwargs.get("audio", False)
        draft = kwargs.get("draft", False)
        optimizePromptMode = kwargs.get("optimizePromptMode", "standard")
        
        # Build settings dictionary - only include what is enabled
        settings = {}
        
        # Add cameraFixed only if useCameraFixed is enabled
        if useCameraFixed:
            settings["cameraFixed"] = cameraFixed
        
        # Add maxSequentialImages only if useMaxSequentialImages is enabled
        if useMaxSequentialImages:
            settings["maxSequentialImages"] = maxSequentialImages
        
        # Add fastMode only if useFastMode is enabled
        if useFastMode:
            settings["fastMode"] = fastMode

        # Add audio only if useAudio is enabled
        if useAudio:
            settings["audio"] = audio

        # Add draft only if useDraft is enabled
        if useDraft:
            settings["draft"] = draft

        # Add optimizePromptMode only if useOptimizePromptMode is enabled
        if useOptimizePromptMode:
            settings["optimizePromptMode"] = optimizePromptMode
        
        # Clean up None values
        settings = {k: v for k, v in settings.items() if v is not None}
        
        return (settings,)

# Node class mappings
NODE_CLASS_MAPPINGS = {
    "RunwareBytedanceProviderSettings": RunwareBytedanceProviderSettings,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "RunwareBytedanceProviderSettings": "Runware Bytedance Provider Settings",
}