ComfyUI_LTXRelight/__init__.py at main · gregtee2/ComfyUI_LTXRelight · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
"""LTX Relight ComfyUI nodes.

Current v7 workflow:
LTXRelightLoadHDRI -> LTXRelightSphereCard -> LTXRelightBuildSphereCardRef
-> LTXRelightSampler.

Also includes delta recovery, lens helpers, and exact-pixel LTX crop tools.
"""
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
from .align_nodes import (
	ALIGN_NODE_CLASS_MAPPINGS,
	ALIGN_NODE_DISPLAY_NAME_MAPPINGS,
)
from .delta_recover_node import (
	NODE_CLASS_MAPPINGS as DELTA_NODE_CLASS_MAPPINGS,
	NODE_DISPLAY_NAME_MAPPINGS as DELTA_NODE_DISPLAY_NAME_MAPPINGS,
)
from .reformat import (
	NODE_CLASS_MAPPINGS as REFORMAT_NODE_CLASS_MAPPINGS,
	NODE_DISPLAY_NAME_MAPPINGS as REFORMAT_NODE_DISPLAY_NAME_MAPPINGS,
)

NODE_CLASS_MAPPINGS = {
	**NODE_CLASS_MAPPINGS,
	**ALIGN_NODE_CLASS_MAPPINGS,
	**DELTA_NODE_CLASS_MAPPINGS,
	**REFORMAT_NODE_CLASS_MAPPINGS,
}
NODE_DISPLAY_NAME_MAPPINGS = {
	**NODE_DISPLAY_NAME_MAPPINGS,
	**ALIGN_NODE_DISPLAY_NAME_MAPPINGS,
	**DELTA_NODE_DISPLAY_NAME_MAPPINGS,
	**REFORMAT_NODE_DISPLAY_NAME_MAPPINGS,
}

# Register the HDRI panorama viewer HTTP endpoints (/ltx_relight/preview, /ltx_relight/info).
# Side-effect import: the module wires itself into PromptServer at import time.
from . import ltx_hdri_viewer_endpoints  # noqa: F401
from . import ltx_crop_editor_endpoints  # noqa: F401

# Tells ComfyUI to serve files from the ./web folder as static JS extensions.
WEB_DIRECTORY = "./web"

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