ComfyUI-Daydream-Scope/__init__.py at main · daydreamlive/ComfyUI-Daydream-Scope · 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
"""
ComfyUI Scope - Video generation pipelines for ComfyUI.

This node pack is a wrapper around the Scope API, providing an alternate
UI for Scope within ComfyUI.

Nodes:
- ScopeModelLoader: Load pipeline model (handles VACE, LoRA, VAE automatically)
- ScopeLoraLoader: Configure LoRA (optional, chainable)
- ScopeVACE: Configure VACE conditioning (optional)
- ScopeSampler: Main generation node (batch, blocking)
- ScribblePreprocessor: Extract scribble/contour line art from images

Example workflow:
    LoadVideo -> ScopeModelLoader -> ScopeSampler -> output
    LoadImage -> ScribblePreprocessor -> ScopeVACE -> ScopeSampler -> output

Note: InteractiveScopeSampler is disabled in this POC release.
"""

from __future__ import annotations

from comfy_api.latest import ComfyExtension, io

from .nodes import (
    ScopeModelLoader,
    ScopeLoraLoader,
    ScopeVACENode,
    ScopeSampler,
    ScribblePreprocessor,
    ScopePromptBlendNode,
    ScopeKeyframeNode,
    ScopeKeyframePeakCyclerNode,
)


class ScopeExtension(ComfyExtension):
    """ComfyUI extension providing Scope video generation nodes."""

    async def get_node_list(self) -> list[type[io.ComfyNode]]:
        return [
            ScopeModelLoader,
            ScopeLoraLoader,
            ScopeVACENode,
            ScopeSampler,
            ScribblePreprocessor,
            ScopePromptBlendNode,
            ScopeKeyframeNode,
            ScopeKeyframePeakCyclerNode,
        ]


async def comfy_entrypoint() -> ScopeExtension:
    """V3 API entrypoint for ComfyUI."""
    return ScopeExtension()


# V1 API compatibility
NODE_CLASS_MAPPINGS = {
    "ScopeModelLoader": ScopeModelLoader,
    "ScopeLoraLoader": ScopeLoraLoader,
    "ScopeVACE": ScopeVACENode,
    "ScopeSampler": ScopeSampler,
    "ScopeScribblePreprocessor": ScribblePreprocessor,
    "ScopePromptBlend": ScopePromptBlendNode,
    "ScopeKeyframe": ScopeKeyframeNode,
    "ScopeKeyframePeakCycler": ScopeKeyframePeakCyclerNode,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "ScopeModelLoader": "Scope Model Loader",
    "ScopeLoraLoader": "Scope LoRA",
    "ScopeVACE": "Scope VACE",
    "ScopeSampler": "Scope Sampler",
    "ScopeScribblePreprocessor": "Scribble Preprocessor (Scope)",
    "ScopePromptBlend": "Scope Prompt Blend",
    "ScopeKeyframe": "Scope Keyframe",
    "ScopeKeyframePeakCycler": "Scope Keyframe Peak Cycler",
}

__all__ = [
    "comfy_entrypoint",
    "ScopeExtension",
    "NODE_CLASS_MAPPINGS",
    "NODE_DISPLAY_NAME_MAPPINGS",
]