Merge branch 'dev/ptrottier/remix_integration' into 'main'

Implement RTX Remix Integration bits

See merge request lightspeedrtx/lss-ai-tools/comfyui-rtx_remix!35

Author: po-trottier

.prettierrc

@@ -0,0 +1,10 @@
+{
+  "printWidth": 120,
+  "tabWidth": 2,
+  "useTabs": false,
+  "semi": true,
+  "singleQuote": false,
+  "trailingComma": "es5",
+  "bracketSpacing": true,
+  "arrowParens": "always"
+}

CHANGELOG.md

@@ -15,6 +15,22 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Removed
 
+## [2.0.0] - 2025-11-21
+
+### Added
+
+- **RTX Remix Save Texture Node**: Saves textures to the output directory with a subfolder based on the job ID or prompt timestamp if not job ID is provided
+- **RTX Remix Integration Workflow**: A new workflow for integrating ComfyUI into the RTX Remix Toolkit without using REST API nodes
+- **RTX Remix Front-End**: Front-end ComfyUI implementation to implement a better workflow for users
+    - Ability to tag input slots and output nodes with metadata for RTX Remix Toolkit to use
+    - Ability to export the workflow (and associated API Workflow) to the user directory for use in the RTX Remix Toolkit integration
+    - Update the UI to indicate tagged nodes and slots
+
+### Changed
+
+- Moved all nodes previously existing nodes to the REST API sub-menu and sub-directory
+- Updated display names to include the 🌐 prefix to indicate REST API nodes
+
 ## [1.1.1] - 2025-07-25
 
 ### Fixed

README.md

@@ -1 +1 @@
-2024.0.0
+2.0.0

VERSION.md

@@ -14,6 +14,31 @@
 * See the License for the specific language governing permissions and
 * limitations under the License.
 """
-from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
 
-__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
+from __future__ import annotations
+
+__all__ = ["WEB_DIRECTORY", "comfy_entrypoint"]
+
+from comfy_api.latest import io, ComfyExtension
+
+# Import nodes
+from .nodes import RTX_REMIX_NODES
+
+# Define the API endpoints
+from .api import *
+
+# Define the UI components
+WEB_DIRECTORY = "./web"
+
+
+class RTXRemixExtension(ComfyExtension):
+    """RTX Remix Extension for ComfyUI"""
+
+    async def get_node_list(self) -> list[type[io.ComfyNode]]:
+        """Return all nodes for this extension"""
+        return RTX_REMIX_NODES
+
+
+async def comfy_entrypoint() -> RTXRemixExtension:
+    """V3 entry point for ComfyUI to discover this extension's nodes"""
+    return RTXRemixExtension()

__init__.py

@@ -0,0 +1,27 @@
+"""
+* SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+* SPDX-License-Identifier: Apache-2.0
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+* https://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+"""
+
+from .ui import get_all_ui, get_node_ui
+from .workflows import get_workflow, list_workflows, save_workflow
+
+__all__ = [
+    "get_all_ui",
+    "get_node_ui",
+    "get_workflow",
+    "list_workflows",
+    "save_workflow",
+]

api/__init__.py

@@ -0,0 +1,18 @@
+"""
+* SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+* SPDX-License-Identifier: Apache-2.0
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+* https://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+"""
+
+__all__: list[str] = []

api/configs/__init__.py

@@ -0,0 +1,115 @@
+"""
+* SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+* SPDX-License-Identifier: Apache-2.0
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+* https://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+"""
+
+from __future__ import annotations
+
+__all__ = ["NODE_UI_CONFIGS", "NodeUIConfig"]
+
+from enum import Enum
+
+from ..models import (
+    DynamicHelp,
+    HelpContent,
+    NodeUI,
+    UrlHandler,
+    UrlParser,
+    WidgetVisibility,
+)
+from ...nodes.constant import ModelSource
+
+
+class NodeUIConfig(str, Enum):
+    RTX_REMIX_DOWNLOAD_MODEL = "RTXRemixDownloadModel"
+
+
+NODE_UI_CONFIGS: dict[NodeUIConfig, NodeUI] = {
+    NodeUIConfig.RTX_REMIX_DOWNLOAD_MODEL: NodeUI(
+        url_handler=UrlHandler(
+            source_field="model_source",
+            host_patterns={
+                ModelSource.HUGGINGFACE: "huggingface.co",
+                ModelSource.CIVITAI: "civitai.com",
+            },
+            parsers={
+                ModelSource.HUGGINGFACE: UrlParser(
+                    populate={"repo_id": "hf_repo_id", "filename": "hf_filename"},
+                    reset=["civitai_model_id", "custom_filename"],
+                ),
+                ModelSource.CIVITAI: UrlParser(
+                    populate={"version_id": "civitai_model_id"},
+                    reset=["hf_repo_id", "hf_filename", "custom_filename"],
+                ),
+                ModelSource.CUSTOM: UrlParser(
+                    populate={"filename": "custom_filename"},
+                    reset=["hf_repo_id", "hf_filename", "civitai_model_id"],
+                ),
+            },
+        ),
+        visibility_rules=[
+            # Show model_source only after a URL is entered
+            WidgetVisibility(
+                source_field="url",
+                show_when_filled=["model_source"],
+            ),
+            # Show source-specific fields based on detected source
+            WidgetVisibility(
+                source_field="model_source",
+                mapping={
+                    ModelSource.HUGGINGFACE: [
+                        "hf_repo_id",
+                        "hf_filename",
+                        "hf_token",
+                    ],
+                    ModelSource.CIVITAI: ["civitai_model_id", "civitai_api_key"],
+                    ModelSource.CUSTOM: ["custom_filename", "custom_hash"],
+                },
+            ),
+            # Show archive fields when extract_archive is enabled
+            WidgetVisibility(
+                source_field="extract_archive",
+                mapping={
+                    True: ["archive_model_filename", "extracted_model_hash"],
+                },
+            ),
+        ],
+        info_button=DynamicHelp(
+            source_field="model_source",
+            configs={
+                "": HelpContent(
+                    label="🛈 How to use this node",
+                    title="Download Model",
+                    template="rtx-remix-download-generic-info-template",
+                ),
+                ModelSource.HUGGINGFACE: HelpContent(
+                    label="🛈 HuggingFace Help",
+                    title="Download from HuggingFace",
+                    template="rtx-remix-download-huggingface-token-info-template",
+                ),
+                ModelSource.CIVITAI: HelpContent(
+                    label="🛈 CivitAI Help",
+                    title="Download from CivitAI",
+                    template="rtx-remix-download-civitai-apikey-info-template",
+                ),
+                ModelSource.CUSTOM: HelpContent(
+                    label="🛈 Custom URL Help",
+                    title="Custom URL Download",
+                    template="rtx-remix-download-custom-howto-info-template",
+                ),
+            },
+        ),
+    )
+}

api/configs/dynamic_ui.py

@@ -0,0 +1,27 @@
+"""
+* SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+* SPDX-License-Identifier: Apache-2.0
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+* https://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+"""
+
+__all__ = ["get_base_url", "WORKFLOWS_PREFIX", "API_WORKFLOWS_DIR", "WORKFLOWS_DIR", "REMIX_USER_ID"]
+
+WORKFLOWS_PREFIX = "workflows"
+API_WORKFLOWS_DIR = "api_workflows"
+WORKFLOWS_DIR = "workflows"
+REMIX_USER_ID = "rtx-remix"
+
+
+def get_base_url(version: int = 1) -> str:
+    return f"/rtx-remix/v{version}"

api/constants.py

@@ -0,0 +1,34 @@
+"""
+* SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+* SPDX-License-Identifier: Apache-2.0
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+* https://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+"""
+
+__all__ = ["PathType", "SourceType"]
+
+from enum import Enum
+
+
+class PathType(Enum):
+    """Enumeration for workflow path types."""
+
+    API = "api"
+    FULL = "full"
+
+
+class SourceType(Enum):
+    """Enumeration for workflow source types."""
+
+    USER = "user"
+    RTX_REMIX = "rtx-remix"