from .modules.imageInference import txt2img
from .modules.outpaintSettings import outpaintSettings
from .modules.bgremoval import bgremoval
from .modules.photoMaker import photoMaker
from .modules.upscaler import upscaler
from .modules.imageUpscalerSettings import RunwareImageUpscalerSettings
from .modules.modelSearch import modelSearch
from .modules.bridges import mainRoute
from .modules.controlNet import controlNet
from .modules.multiInference import multiInference
from .modules.runwareBFL import runwareKontext
from .modules.runwareImagen import runwareImagen
from .modules.loraSearch import loraSearch
from .modules.loraCombine import loraCombine
from .modules.refiner import refiner
from .modules.imageMasking import imageMasking
from .modules.controlNetPreprocessor import controlNetPreprocessor
from .modules.apiManager import apiManager
from .modules.imageCaptioning import imageCaptioning
from .modules.videoTranscription import videoTranscription
from .modules.controlNetCombine import controlNetCombine
from .modules.embeddingSearch import embeddingSearch
from .modules.embeddingsCombine import embeddingsCombine
from .modules.ipAdapter import ipAdapter
from .modules.ipAdapterCombine import ipAdapterCombine
from .modules.vaeSearch import vaeSearch
from .modules.referenceImages import referenceImages
from .modules.imageInferenceInputs import imageInferenceInputs
from .modules.imageInferenceAdvancedFeatures import RunwareImageInferenceAdvancedFeatures
from .modules.imageWatermarkAdvancedFeature import RunwareWatermarkAdvancedFeature
from .modules.imageRegionalPromptingAdvancedFeature import RunwareRegionalPromptingAdvancedFeature
from .modules.imageRegionalPromptingAdvancedFeatureRegions import RunwareRegionalPromptingAdvancedFeatureRegions
from .modules.videoInference import txt2vid
from .modules.videoInferenceSpeechInput import RunwareVideoInferenceSpeechInput
from .modules.videoModelSearch import videoModelSearch
from .modules.videoSettings import RunwareVideoSettings
from .modules.videoInferenceSettingsTts import RunwareVideoInferenceSettingsTts
from .modules.videoInferenceSettingsActiveSpeakerDetection import RunwareVideoInferenceSettingsActiveSpeakerDetection
from .modules.videoInferenceSettingsActiveSpeakerBoundingBoxes import RunwareVideoInferenceSettingsActiveSpeakerBoundingBoxes
from .modules.videoInferenceSettingsSegments import RunwareVideoInferenceSettingsSegments
from .modules.frameImages import RunwareFrameImages
from .modules.providerSettings.pixverseProviderSettings import RunwarePixverseProviderSettings
from .modules.providerSettings.openaiProviderSettings import RunwareOpenAIProviderSettings
from .modules.providerSettings.bytedanceProviderSettings import RunwareBytedanceProviderSettings
from .modules.providerSettings.briaProviderSettings import RunwareBriaProviderSettings
from .modules.providerSettings.briaProviderMask import RunwareBriaProviderMask
from .modules.audioInference import RunwareAudioInference
from .modules.audioInferenceSpeech import RunwareAudioInferenceSpeech
from .modules.audioInferenceSpeechVoices import RunwareAudioInferenceSpeechVoices
from .modules.audioModelSearch import RunwareAudioModelSearch
from .modules.textModelSearch import RunwareTextModelSearch
from .modules.textInferenceMessages import RunwareTextInferenceMessages
from .modules.textInferenceMessage import RunwareTextInferenceMessage
from .modules.textInference import RunwareTextInference
from .modules.textInferenceSettings import RunwareTextInferenceSettings
from .modules.textInferenceToolChoice import RunwareTextInferenceToolChoice
from .modules.textInferenceSettingsCache import RunwareTextInferenceSettingsCache
from .modules.textInferenceInputs import RunwareTextInferenceInputs
from .modules.textInferenceInputsImages import RunwareTextInferenceInputsImages
from .modules.textInferenceInputsVideos import RunwareTextInferenceInputsVideos
from .modules.audioSections import RunwareAudioSections
from .modules.audioInferenceInputs import audioInferenceInputs
from .modules.audioSettings import RunwareAudioSettings
from .modules.audioSettingsVoiceModify import RunwareAudioSettingsVoiceModify
from .modules.providerSettings.elevenlabsProviderSettings import RunwareElevenLabsProviderSettings
from .modules.providerSettings.elevenlabsProviderSettingsSections import RunwareElevenLabsProviderSettingsSections
from .modules.acceleratorOptions import acceleratorOptions
from .modules.providerSettings.viduProviderSettings import RunwareViduProviderSettings
from .modules.mediaUpload import runwareMediaUpload
from .modules.inputAudios import inputAudios
from .modules.audioInput import RunwareAudioInput
from .modules.speechInput import RunwareSpeechInput
from .modules.speechInputCombine import RunwareSpeechInputCombine
from .modules.referenceVideos import referenceVideos
from .modules.referenceVoices import referenceVoices
from .modules.referenceAudios import referenceAudios
from .modules.videoInferenceInputs import videoInferenceInputs
from .modules.videoInferenceElements import (
    RunwareVideoInferenceElements,
    RunwareVideoInferenceElementsCombine,
)
from .modules.providerSettings.lightricksProviderSettings import RunwareLightricksProviderSettings
from .modules.providerSettings.minimaxProviderSettings import RunwareMiniMaxProviderSettings
from .modules.providerSettings.klingProviderSettings import RunwareKlingProviderSettings
from .modules.providerSettings.klingMultiPromptSegment import RunwareKlingMultiPromptSegment
from .modules.providerSettings.klingProviderSettingsMultiPrompt import RunwareKlingProviderSettingsMultiPrompt
from .modules.providerSettings.runwayProviderSettings import RunwareRunwayProviderSettings
from .modules.providerSettings.lumaProviderSettings import RunwareLumaProviderSettings
from .modules.providerSettings.midjourneyProviderSettings import RunwareMidjourneyProviderSettings
from .modules.providerSettings.mireloProviderSettings import RunwareMireloProviderSettings
from .modules.providerSettings.alibabaProviderSettings import RunwareAlibabaProviderSettings
from .modules.providerSettings.googleProviderSettings import RunwareGoogleProviderSettings
from .modules.providerSettings.blackForestProviderSettings import RunwareBlackForestProviderSettings
from .modules.providerSettings.xaiProviderSettings import RunwareXAIProviderSettings
from .modules.providerSettings.recraftProviderSettings import RunwareRecraftProviderSettings
from .modules.providerSettings.recraftProviderSettingsColor import RunwareRecraftColor
from .modules.providerSettings.sourcefulProviderSettings import RunwareSourcefulProviderSettings
from .modules.providerSettings.sourcefulProviderSettingsFonts import RunwareSourcefulProviderSettingsFonts
from .modules.ultralyticsInputs import RunwareUltralyticsInputs
from .modules.providerSettings.syncSegment import RunwareSyncSegment
from .modules.providerSettings.syncProviderSettings import RunwareSyncProviderSettings
from .modules.vectorize import vectorize
from .modules.videoBgRemoval import videoBgRemoval
from .modules.videoUpscaler import videoUpscaler
from .modules.videoInputsReferences import videoInputsReferences
from .modules.videoInputsReferenceImagesMultipleImagesConnector import RunwareVideoInputsReferenceImagesMultipleImagesConnector
from .modules.videoInputsFrame import RunwareVideoInputsFrameImages
from .modules.safetyInputs import safetyInputs
from .modules.settings import RunwareSettings
from .modules.imageInferenceSettingsColorPalette import RunwareImageInferenceSettingsColorPalette
from .modules.imageInferenceSettingsMoodboards import RunwareImageInferenceSettingsMoodboards
from .modules.videoAdvancedFeatureInputs import videoAdvancedFeatureInputs
from .modules.advancedFeatureSettings.wanAnimateAdvancedFeatureSettings import RunwareWanAnimateAdvancedFeatureSettings
from .modules.saveImage import RunwareSaveImage
from .modules.videoOutputs import RunwareVideoInferenceOutputs
from .modules.threeDInference import threeDInference
from .modules.threeDInferenceInputs import threeDInferenceInputs
from .modules.threeDInferenceSettings import Runware3DInferenceSettings
from .modules.threeDInferenceSettingsSparseStructure import Runware3DInferenceSettingsSparseStructure
from .modules.threeDInferenceSettingsShapeSlat import Runware3DInferenceSettingsShapeSlat
from .modules.threeDInferenceSettingsTexSlat import Runware3DInferenceSettingsTexSlat
from .modules.save3D import RunwareSave3D
from .modules.saveText import RunwareSaveText
from .modules.loadMesh import RunwareLoadMesh

RUNWARE_COMFYUI_VERSION = "1.4.0 Beta"

RESET_COLOR = "\033[0m"
BLUE_COLOR = "\033[94m"
GREEN_COLOR = "\033[92m"
print(BLUE_COLOR + "##############################################################" + RESET_COLOR)
print(GREEN_COLOR + "  Runware ComfyUI Inference Services Are Loaded Successfully" + RESET_COLOR)
print(GREEN_COLOR + "  Version: " + RUNWARE_COMFYUI_VERSION + " | Maintained by: Runware Inc" + RESET_COLOR)
print(GREEN_COLOR + "  Official Website: https://my.runware.ai" + RESET_COLOR)
print(BLUE_COLOR + "##############################################################" + RESET_COLOR)

NODE_CLASS_MAPPINGS = {
    "Runware Image Inference": txt2img,
    "Runware Outpaint": outpaintSettings,
    "Runware Background Removal": bgremoval,
    "Runware PhotoMaker V2": photoMaker,
    "Runware Image Upscaler": upscaler,
    "Runware Image Upscaler Settings": RunwareImageUpscalerSettings,
    "Runware Model Search": modelSearch,
    "Runware Kontext Inference": runwareKontext,
    "Runware Imagen Inference": runwareImagen,
    "Runware Multi Inference": multiInference,
    "Runware Lora Search": loraSearch,
    "Runware Embedding Search": embeddingSearch,
    "Runware VAE Search": vaeSearch,
    "Runware Embeddings Combine": embeddingsCombine,
    "Runware ControlNet": controlNet,
    "Runware Lora Combine": loraCombine,
    "Runware Refiner": refiner,
    "Runware Image Masking": imageMasking,
    "Runware ControlNet PreProcessor": controlNetPreprocessor,
    "Runware API Manager": apiManager,
    "Runware Image Caption": imageCaptioning,
    "Runware Video Transcription": videoTranscription,
    "Runware ControlNet Combine": controlNetCombine,
    "Runware IPAdapter": ipAdapter,
    "Runware IPAdapters Combine": ipAdapterCombine,
    "Runware Reference Images": referenceImages,
    "Runware Image Inference Inputs": imageInferenceInputs,
    "Runware Image  Advanced Feature Input": RunwareImageInferenceAdvancedFeatures,
    "Runware Watermark Advanced Feature": RunwareWatermarkAdvancedFeature,
    "Runware Regional Prompting Advanced Feature": RunwareRegionalPromptingAdvancedFeature,
    "Runware Regional Prompting Advanced Feature Regions": RunwareRegionalPromptingAdvancedFeatureRegions,
    "Runware Video Inference": txt2vid,
    "Runware Video Model Search": videoModelSearch,
    "Runware Frame Images": RunwareFrameImages,
    "Runware Pixverse Provider Settings": RunwarePixverseProviderSettings,
    "Runware OpenAI Provider Settings": RunwareOpenAIProviderSettings,
    "Runware Bytedance Provider Settings": RunwareBytedanceProviderSettings,
    "Runware Bria Provider Settings": RunwareBriaProviderSettings,
    "Runware Bria Provider Mask": RunwareBriaProviderMask,
    "Runware Audio Inference": RunwareAudioInference,
    "Runware Audio Inference Speech": RunwareAudioInferenceSpeech,
    "Runware Audio Inference Speech Voices": RunwareAudioInferenceSpeechVoices,
    "Runware Audio Model Search": RunwareAudioModelSearch,
    "Runware Text Model Search": RunwareTextModelSearch,
    "Runware Text Inference Messages": RunwareTextInferenceMessages,
    "Runware Text Inference Message": RunwareTextInferenceMessage,
    "Runware Text Inference": RunwareTextInference,
    "Runware Text Inference Settings": RunwareTextInferenceSettings,
    "Runware Text Inference Tool Choice": RunwareTextInferenceToolChoice,
    "Runware Text Inference Settings Cache": RunwareTextInferenceSettingsCache,
    "Runware Text Inference Inputs": RunwareTextInferenceInputs,
    "Runware Text Inference Inputs Images": RunwareTextInferenceInputsImages,
    "Runware Text Inference Inputs Videos": RunwareTextInferenceInputsVideos,
    "Runware Audio Sections": RunwareAudioSections,
    "Runware Audio Inference Inputs": audioInferenceInputs,
    "Runware Audio Inference Settings": RunwareAudioSettings,
    "Runware Audio Inference Settings Voice Modify": RunwareAudioSettingsVoiceModify,
    "Runware ElevenLabs Provider Settings": RunwareElevenLabsProviderSettings,
    "Runware ElevenLabs Provider Settings Sections": RunwareElevenLabsProviderSettingsSections,
    "Runware Accelerator Options": acceleratorOptions,
    "Runware Vidu Provider Settings": RunwareViduProviderSettings,
    "Runware Media Upload": runwareMediaUpload,
    "Runware Input Audios": inputAudios,
    "Runware Video Audio Input": RunwareAudioInput,
    "Runware Video Speech Input": RunwareSpeechInput,
    "Runware Speech Inputs Combine": RunwareSpeechInputCombine,
    # Backward-compatible aliases for existing workflows
    "Runware Reference Videos": referenceVideos,
    "Runware Reference Voices": referenceVoices,
    "Runware Video Inputs References": videoInputsReferences,
    "Runware Video Inference Inputs Reference Videos": referenceVideos,
    "Runware Video Inference Inputs Reference Voices": referenceVoices,
    "Runware Video Inference Inputs Reference Audios": referenceAudios,
    "Runware Video Inference Inputs": videoInferenceInputs,
    "Runware Video Inference Elements": RunwareVideoInferenceElements,
    "Runware Video Inference Elements Combine": RunwareVideoInferenceElementsCombine,
    "Runware Video Inference Speech Input": RunwareVideoInferenceSpeechInput,
    "Runware Video Inference Settings": RunwareVideoSettings,
    "Runware Video Inference Settings TTS": RunwareVideoInferenceSettingsTts,
    "Runware Video Inference Settings Active Speaker Detection": RunwareVideoInferenceSettingsActiveSpeakerDetection,
    "Runware Video Inference Settings Active Speaker Bounding Boxes": RunwareVideoInferenceSettingsActiveSpeakerBoundingBoxes,
    "Runware Video Inference Settings Segments": RunwareVideoInferenceSettingsSegments,
    "Runware Lightricks Provider Settings": RunwareLightricksProviderSettings,
    "Runware MiniMax Provider Settings": RunwareMiniMaxProviderSettings,
    "Runware Luma Provider Settings": RunwareLumaProviderSettings,
    "Runware KlingAI Provider Settings": RunwareKlingProviderSettings,
    "Runware Kling MultiPrompt Segment": RunwareKlingMultiPromptSegment,
    "Runware Kling Provider Settings MultiPrompt": RunwareKlingProviderSettingsMultiPrompt,
    "Runware Runway Provider Settings": RunwareRunwayProviderSettings,
    "Runware Midjourney Provider Settings": RunwareMidjourneyProviderSettings,
    "Runware Mirelo Provider Settings": RunwareMireloProviderSettings,
    "Runware Alibaba Provider Settings": RunwareAlibabaProviderSettings,
    "Runware Google Provider Settings": RunwareGoogleProviderSettings,
    "Runware BlackForest Labs Provider Settings": RunwareBlackForestProviderSettings,
    "Runware xAI Provider Settings": RunwareXAIProviderSettings,
    "Runware Recraft Provider Settings": RunwareRecraftProviderSettings,
    "Runware Provider Settings Recraft Color": RunwareRecraftColor,
    "Runware Sourceful Provider Settings": RunwareSourcefulProviderSettings,
    "Runware Sourceful Provider Settings Fonts": RunwareSourcefulProviderSettingsFonts,
    "Runware Image Inference Ultralytics": RunwareUltralyticsInputs,
    "Runware Sync Segment": RunwareSyncSegment,
    "Runware Sync Provider Settings": RunwareSyncProviderSettings,
    "Runware Vectorize": vectorize,
    "Runware Video Background Removal": videoBgRemoval,
    "Runware Video Upscaler": videoUpscaler,
    "Runware Video Inference Inputs Reference Images": videoInputsReferences,
    "Runware Video Inference Inputs Reference Images Multiple Images Connector": RunwareVideoInputsReferenceImagesMultipleImagesConnector,
    "Runware Video Inputs Frame Images": RunwareVideoInputsFrameImages,
    "Runware Safety Inputs": safetyInputs,
    "Runware Settings": RunwareSettings,
    "Runware Image Inference Settings Color Palette": RunwareImageInferenceSettingsColorPalette,
    "Runware Image Inference Settings Moodboards": RunwareImageInferenceSettingsMoodboards,
    "Runware Video Advanced Feature Inputs": videoAdvancedFeatureInputs,
    "Runware Wan Animate Advanced Feature Settings": RunwareWanAnimateAdvancedFeatureSettings,
    "Runware Save Image": RunwareSaveImage,
    "Runware Video Inference Outputs": RunwareVideoInferenceOutputs,
    "Runware 3D Inference": threeDInference,
    "Runware 3D Inference Inputs": threeDInferenceInputs,
    "Runware 3D Inference Settings": Runware3DInferenceSettings,
    "Runware 3D Inference Settings Sparse Structure": Runware3DInferenceSettingsSparseStructure,
    "Runware 3D Inference Settings Shape Slat": Runware3DInferenceSettingsShapeSlat,
    "Runware 3D Inference Settings Tex Slat": Runware3DInferenceSettingsTexSlat,
    "Runware Save 3D": RunwareSave3D,
    "Runware Save Text": RunwareSaveText,
    "Runware Load Mesh": RunwareLoadMesh,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "Runware Settings": "Runware Image Inference Settings",
    "Runware Model Search": "Runware Model",
    "Runware Lora Search": "Runware Lora",
    "Runware Embedding Search": "Runware Embedding",
    "Runware VAE Search": "Runware VAE",
    "Runware Multi Inference": "Runware Multi Inference [BETA]",
    "Runware Video Model Search": "Runware Video Model",
    "Runware Audio Model Search": "Runware Audio Model",
    "Runware Text Model Search": "Runware Text Model",
    "Runware Image Inference Settings Color Palette": "Runware Image Inference Color Palette",
    "Runware Video Transcription": "Runware Video Caption",
    "Runware Video Inputs References": "Runware Video Inference Inputs Reference Images",
    "Runware Video Inference Inputs Reference Images": "Runware Video Inference Inputs Reference Images",
    "Runware Reference Voices": "Runware Video Inference Inputs Reference Voices",
    "Runware Video Inference Inputs Reference Voices": "Runware Video Inference Inputs Reference Voices",
    "Runware Reference Videos": "Runware Video Inference Inputs Reference Videos",
    "Runware Video Inference Inputs Reference Videos": "Runware Video Inference Inputs Reference Videos",
    "Runware Reference Audios": "Runware Video Inference Inputs Reference Audios",
    "Runware Video Inference Inputs Reference Audios": "Runware Video Inference Inputs Reference Audios",
    "Runware Video Inference Inputs Reference Images Multiple Images Connector": "Runware Video Inference Inputs Reference Images Multiple Images Connector",
}

WEB_DIRECTORY = "./clientlibs"
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]