# Import all the node classes from the nodes directory
from .nodes.prompt_machine import Prompt_Machine, Prompt_Style_Machine
from .nodes.resolution_machine import Resolution_Machine
from .nodes.save_machine import Save_Machine
from .nodes.upscale_machine import Upscale_Machine
from .nodes.prompt_autocomplete import PromptAutocomplete
from .nodes.latent_machine import Latent_Machine
from .nodes.preview_machine import Preview_Machine

# A dictionary that maps class names to class objects
NODE_CLASS_MAPPINGS = {
    "Prompt_Machine": Prompt_Machine,
    "Prompt_Style_Machine": Prompt_Style_Machine,
    "Resolution_Machine": Resolution_Machine,
    "Save_Machine": Save_Machine,
    "Upscale_Machine": Upscale_Machine,
    "PromptAutocomplete": PromptAutocomplete,
    "Latent_Machine": Latent_Machine,
    "Preview_Machine": Preview_Machine,
}

# A dictionary that maps display names to class names
NODE_DISPLAY_NAME_MAPPINGS = {
    "Prompt_Machine": "Prompt Machine",
    "Prompt_Style_Machine": "Prompt Style Machine",
    "Resolution_Machine": "Resolution Machine",
    "Save_Machine": "Save Machine",
    "Upscale_Machine": "Upscale Machine",
    "PromptAutocomplete": "Prompt Autocomplete",
    "Latent_Machine": "Latent Machine",
    "Preview_Machine": "Preview Machine",
}

# A dictionary that contains web directory information
WEB_DIRECTORY = "./js"

__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']