ComfyUI-Lora-Manager/__init__.py at main · broken/ComfyUI-Lora-Manager · 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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
try:  # pragma: no cover - import fallback for pytest collection
    from .py.lora_manager import LoraManager
    from .py.nodes.lora_loader import LoraLoaderLM, LoraTextLoaderLM
    from .py.nodes.checkpoint_loader import CheckpointLoaderLM
    from .py.nodes.unet_loader import UNETLoaderLM
    from .py.nodes.trigger_word_toggle import TriggerWordToggleLM
    from .py.nodes.prompt import PromptLM
    from .py.nodes.text import TextLM
    from .py.nodes.lora_stacker import LoraStackerLM
    from .py.nodes.lora_stack_combiner import LoraStackCombinerLM
    from .py.nodes.save_image import SaveImageLM
    from .py.nodes.debug_metadata import DebugMetadataLM
    from .py.nodes.wanvideo_lora_select import WanVideoLoraSelectLM
    from .py.nodes.wanvideo_lora_select_from_text import WanVideoLoraTextSelectLM
    from .py.nodes.lora_pool import LoraPoolLM
    from .py.nodes.lora_randomizer import LoraRandomizerLM
    from .py.nodes.lora_cycler import LoraCyclerLM
    from .py.metadata_collector import init as init_metadata_collector
except (
    ImportError
):  # pragma: no cover - allows running under pytest without package install
    import importlib
    import pathlib
    import sys

    package_root = pathlib.Path(__file__).resolve().parent
    if str(package_root) not in sys.path:
        sys.path.append(str(package_root))

    PromptLM = importlib.import_module("py.nodes.prompt").PromptLM
    TextLM = importlib.import_module("py.nodes.text").TextLM
    LoraManager = importlib.import_module("py.lora_manager").LoraManager
    LoraLoaderLM = importlib.import_module("py.nodes.lora_loader").LoraLoaderLM
    LoraTextLoaderLM = importlib.import_module("py.nodes.lora_loader").LoraTextLoaderLM
    CheckpointLoaderLM = importlib.import_module(
        "py.nodes.checkpoint_loader"
    ).CheckpointLoaderLM
    UNETLoaderLM = importlib.import_module("py.nodes.unet_loader").UNETLoaderLM
    TriggerWordToggleLM = importlib.import_module(
        "py.nodes.trigger_word_toggle"
    ).TriggerWordToggleLM
    LoraStackerLM = importlib.import_module("py.nodes.lora_stacker").LoraStackerLM
    LoraStackCombinerLM = importlib.import_module(
        "py.nodes.lora_stack_combiner"
    ).LoraStackCombinerLM
    SaveImageLM = importlib.import_module("py.nodes.save_image").SaveImageLM
    DebugMetadataLM = importlib.import_module("py.nodes.debug_metadata").DebugMetadataLM
    WanVideoLoraSelectLM = importlib.import_module(
        "py.nodes.wanvideo_lora_select"
    ).WanVideoLoraSelectLM
    WanVideoLoraTextSelectLM = importlib.import_module(
        "py.nodes.wanvideo_lora_select_from_text"
    ).WanVideoLoraTextSelectLM
    LoraPoolLM = importlib.import_module("py.nodes.lora_pool").LoraPoolLM
    LoraRandomizerLM = importlib.import_module(
        "py.nodes.lora_randomizer"
    ).LoraRandomizerLM
    LoraCyclerLM = importlib.import_module("py.nodes.lora_cycler").LoraCyclerLM
    init_metadata_collector = importlib.import_module("py.metadata_collector").init

NODE_CLASS_MAPPINGS = {
    PromptLM.NAME: PromptLM,
    TextLM.NAME: TextLM,
    LoraLoaderLM.NAME: LoraLoaderLM,
    LoraTextLoaderLM.NAME: LoraTextLoaderLM,
    CheckpointLoaderLM.NAME: CheckpointLoaderLM,
    UNETLoaderLM.NAME: UNETLoaderLM,
    TriggerWordToggleLM.NAME: TriggerWordToggleLM,
    LoraStackerLM.NAME: LoraStackerLM,
    LoraStackCombinerLM.NAME: LoraStackCombinerLM,
    SaveImageLM.NAME: SaveImageLM,
    DebugMetadataLM.NAME: DebugMetadataLM,
    WanVideoLoraSelectLM.NAME: WanVideoLoraSelectLM,
    WanVideoLoraTextSelectLM.NAME: WanVideoLoraTextSelectLM,
    LoraPoolLM.NAME: LoraPoolLM,
    LoraRandomizerLM.NAME: LoraRandomizerLM,
    LoraCyclerLM.NAME: LoraCyclerLM,
}

WEB_DIRECTORY = "./web/comfyui"

# Check and build Vue widgets if needed (development mode)
try:
    from .py.vue_widget_builder import check_and_build_vue_widgets

    # Auto-build in development, warn only if fails
    check_and_build_vue_widgets(auto_build=True, warn_only=True)
except ImportError:
    # Fallback for pytest
    import importlib

    check_and_build_vue_widgets = importlib.import_module(
        "py.vue_widget_builder"
    ).check_and_build_vue_widgets
    check_and_build_vue_widgets(auto_build=True, warn_only=True)
except Exception as e:
    import logging

    logging.warning(f"[LoRA Manager] Vue widget build check skipped: {e}")

# Initialize metadata collector
init_metadata_collector()

# Register routes on import
LoraManager.add_routes()
__all__ = ["NODE_CLASS_MAPPINGS", "WEB_DIRECTORY"]