Extensions/comfyui-naia-bridge

ComfyUI Extension

comfyui-naia-bridge

A ComfyUI extension with 6 custom nodes.

By DNT-LAB·Created 4 months ago·Updated 3 months ago· 0

DNT-LAB/comfyui-naia-bridge

Nodes6

On cloudLocal install

CategoryNAIA Bridge/API, NAIA Bridge/Prompt

Stars0

Updated3 months ago

Nodes (6)

NAIA Check Health

NAIA Bridge/API

NAIA Prompt Fetch (WebSocket)

NAIA Bridge/Prompt

NAIA Read Prompt Engineering

NAIA Bridge/Engineering

NAIA Request Random Prompt

NAIA Bridge/API

NAIA Request Random Prompt (with Override)

NAIA Bridge/API

NAIA Request Random Prompt (with Seed)

NAIA Bridge/API

Readme

ComfyUI NAIA 2.0 Bridge

ComfyUI 워크플로우에서 NAIA 2.0의 Remote Web API를 통해 프롬프트를 동기적으로 받아오거나 Prompt Engineering 설정을 override하는 커스텀 노드 패키지.

Bridge ComfyUI workflows to NAIA 2.0's prompt generation engine via its Remote Web API.

✨ 주요 기능 / Features

✅ NAIA의 랜덤 프롬프트 엔진을 ComfyUI에서 동기적으로 호출 (request-response 매칭)
✅ ComfyUI 워크플로우 실행마다 NAIA가 새 프롬프트 생성 → CLIP Text Encode로 직결
✅ Prompt Engineering 설정 per-request override (NAIA 데스크톱 UI 불변)
✅ NAIA 자체 이미지 생성과 병렬 작동 가능 (자동 생성 체크 시)
✅ WebSocket 캐시를 통한 실시간 편집 상태 미러
✅ 연결 진단 및 에러 메시지 한국어 지원

📋 사전 요구사항 / Prerequisites

1. NAIA 2.0 (패치 적용 버전)

본 노드는 NAIA의 core/remote_api_server.py 에 ComfyUI 전용 sync API 패치가 적용되어 있어야 동작합니다.

POST /api/comfyui/random
GET /api/comfyui/health
peng_override per-request 지원

NAIA 2.0 v2.x (해당 패치 포함 버전) 필요. NAIA Settings > Web Session 체크 또는 --web-session 플래그로 실행.

2. ComfyUI

v0.3+ 권장 (frontend 1.42.14+ 권장, tooltip/placeholder 정상 렌더)

3. Python 의존성

websocket-client>=1.5
requests>=2.28

ComfyUI-Manager 사용 시 자동 설치됨.

🚀 설치 / Installation

방법 1: ComfyUI-Manager — 검색 설치 (권장)

가장 쉬운 방법. ComfyUI Registry 에 등록되어 있어 Manager 내 검색으로 바로 설치 가능합니다.

ComfyUI 상단 Manager 버튼 클릭
Custom Nodes Manager 열기
검색창에 comfyui-naia-bridge 또는 naia 입력
목록에서 ComfyUI NAIA 2.0 Bridge 찾아 Install 클릭
의존성 자동 설치 완료 후 ComfyUI 완전 재시작 (백엔드 + 브라우저 탭)

방법 2: comfy-cli 로 설치

pip install comfy-cli  # 미설치 시
comfy node install comfyui-naia-bridge

방법 3: ComfyUI-Manager "Install via Git URL"

Manager의 security_level 이 normal 이하인 환경에서만 가능합니다 (strong 이상은 Registry 경로를 써야 함).

Manager → Install via Git URL
URL: https://github.com/DNT-LAB/comfyui-naia-bridge
Install → ComfyUI 재시작

방법 4: 수동 설치 (git 직접 사용)

cd ComfyUI/custom_nodes
git clone https://github.com/DNT-LAB/comfyui-naia-bridge.git
cd comfyui-naia-bridge
pip install -r requirements.txt

ComfyUI 재시작.

업데이트 / Update

ComfyUI-Manager 사용 시: Manager의 Try Update 또는 Update All
comfy-cli 사용 시: comfy node install comfyui-naia-bridge@latest

수동 설치 시:

cd ComfyUI/custom_nodes/comfyui-naia-bridge
git pull
pip install -r requirements.txt

📖 사용법 / How to Use

Step 1: NAIA 사전 설정

NAIA 사전 설정 — 검색 + Web Session 활성화

두 가지를 먼저 준비합니다.

1-A. 검색 실행 (좌측 패널)

NAIA의 랜덤 프롬프트는 검색 결과 집합에서 무작위로 1개 row를 뽑아 생성됩니다. 검색이 선행되지 않으면 요청 실패합니다.

검색 키워드 (예: 1girl), 필요 시 제외 키워드 입력
Rating 필터 체크 (Explicit / NSFW / Sensitive / General)
데이터셋 버전 선택 (25.09 권장)
검색 버튼 클릭 → 하단에 검색: N / 남음: N 카운터가 0 이상이어야 함

1-B. Web Session 활성화 (우측 Settings 탭)

Web Session 활성화 체크 → 브라우저에서 http://localhost:7243 접속 가능 확인
자동 시작 체크 시 다음 NAIA 실행부터 자동 기동
CLI 사용자: python NAIA_cold_v4.py --web-session 으로도 가능

Step 2: ComfyUI 워크플로우 구성

워크플로우 예시 — NAIA Request Random Prompt → CLIP Text Encode

우클릭 → Add Node → NAIA Bridge → NAIA Request Random Prompt
기본 설정: use_naia_settings = true (NAIA 데스크톱 UI 그대로 사용)
CLIP Text Encode 추가 (Positive용 1개면 충분)
핵심: CLIPTextEncode 의 text 위젯 우클릭 → "위젯을 입력으로 변환" (영어: Convert Widget to Input) — 이게 없으면 STRING 출력을 꽂을 수 없음
NAIA Request Random Prompt 의 prompt 슬롯을 Positive CLIPTextEncode 의 text 에 연결

나머지는 표준 ComfyUI 파이프라인:

Load Checkpoint ─ MODEL/CLIP/VAE
                   │
NAIA Request Random Prompt ─ prompt ─→ CLIP Text Encode (Positive)
                                           │
CLIP Text Encode (Negative, 직접 입력) ────┤
                                           ↓
                                        KSampler → VAE Decode → Save Image

Queue Prompt 실행 → 매번 NAIA가 새 랜덤 프롬프트 생성 → ComfyUI가 이미지 생성

⚠️ Negative Prompt 연결은 선택사항 (기본: 연결 X)

negative_prompt 출력은 기본적으로 연결하지 않는 것을 권장합니다. 이유:

NAIA의 negative_prompt 필드는 NAIA 메인 UI에서 사용자가 직접 입력한 고정 값
랜덤 요청이어도 이 값은 바뀌지 않음 → 매번 같은 negative를 받게 됨 (의미 X)
ComfyUI 쪽의 Negative CLIP Text Encode에 직접 원하는 negative prompt를 입력하세요

예외 — NAIA의 Conditional Prompt 모듈을 적극 활용하는 경우: 이 모듈은 프롬프트 context에 따라 negative를 동적으로 변경합니다. 이 기능을 사용 중이라면 negative_prompt 출력을 Negative CLIP Text Encode 에 연결해 NAIA가 관리하는 negative를 받아 오세요.

프리뷰 확인 (선택)

Show Text (ComfyUI-Custom-Scripts) 또는 rgthree 의 Display String 노드를 prompt 출력에 연결하면 NAIA가 보낸 원본 프롬프트를 확인할 수 있습니다.

📦 제공 노드 / Nodes

노드 메뉴: Add Node → NAIA Bridge → ...

1. `NAIA Request Random Prompt` (메인 노드)

카테고리: NAIA Bridge/API

NAIA에 새 랜덤 프롬프트를 동기 요청. 20+ 입력 위젯으로 모든 기능 통합.

| 입력 | 기본값 | 설명 | |---|---|---| | use_naia_settings | true | true: NAIA 데스크톱 UI 그대로. false: 아래 위젯 값으로 override | | pre_prompt | "" | override 시 선행 프롬프트 | | post_prompt | "" | override 시 후행 프롬프트 | | auto_hide | "" | override 시 자동 숨김 태그 | | remove_author ... 15종 | skip | 전처리 옵션 (skip/on/off) | | host | 127.0.0.1 | NAIA 호스트 | | port | 7243 | NAIA 포트 |

출력: (prompt, negative_prompt, width, height) — 문자열 2개 + 정수 2개.

prompt, negative_prompt: CLIP Text Encode에 연결 (아래 참고).
width, height: NAIA 추천 해상도. EmptyLatentImage(또는 EmptySD3LatentImage 등)의 width/height 슬롯에 직접 연결 가능.
- NAIA 표준 fallback chain 미러: ① source_row 원본 해상도 (auto_fit) → ② NAIA resolution_combo random pick (random_resolution). NAIA 데스크톱 UI 토글 상태 무관 항상 적용.
- 1MP 정규화: 응답 width × height가 NAI 표준 1MP (1024² = 1,048,576) 초과 시 비율 유지하며 8배수 가장 가까운 값으로 자동 정규화 (대부분의 SDXL/Flux/SD3/NAI 모델이 1MP 학습 기준). 1MP 이하는 변경 없음.
- 노드 출력 INT를 EmptyLatentImage 위젯에 연결하려면 우클릭 → "Convert Widget to Input" 필요.

2. `NAIA Request Random Prompt (with Override)` (변형 노드)

카테고리: NAIA Bridge/API

NAIA Request Random Prompt 노드의 확장형 변형 노드. 출력으로 제공하는 4가지 값(prompt, negative_prompt, width, height)을 직접 입력(Widget 또는 Input)으로도 연동할 수 있으며, 개별 오버라이드 및 캐싱 유지 기능이 추가되어 더욱 유연한 워크플로우 제어가 가능합니다.

브릿지 바이패스 (use_naia_bridge):
- false (off)로 설정 시 NAIA 서버 요청을 아예 시도하지 않고, 입력된 4가지 값을 그대로 즉시 출력으로 통과시킵니다.
- 이 상태에서는 입력값이 바뀌지 않았다면 ComfyUI의 실행 캐싱(Caching)이 완벽히 유지되어 노드가 불필요하게 재실행되지 않습니다.
개별 오버라이드 제어 (override_*):
- override_prompt, override_negative, override_width, override_height 토글을 제공합니다.
- true (on): NAIA 서버에서 실시간으로 가져온 랜덤 프롬프트 및 추천 해상도로 덮어씁니다.
- false (off): 위젯으로 직접 입력받았거나 외부 노드에서 온 입력값을 그대로 보존하여 출력합니다.

3. `NAIA Request Random Prompt (with Seed)` (시드 연동형 간소화 노드)

카테고리: NAIA Bridge/API

use_naia_settings = true 고정 상태로 복잡한 15종 전처리 및 프리/포스트 프롬프트 위젯을 모두 제외하여 외관을 대폭 간소화한 노드입니다. 시드(Seed) 입력을 직접 지원하여 ComfyUI의 표준 캐싱 메커니즘과 완벽히 연동됩니다.

시드 연동형 자동 재사용:
- seed 입력 값이 동일하게 유지될 경우, NAIA 서버를 중복 요청하지 않고 이전 실행 시점에 생성되었던 프롬프트, 네거티브, 가로/세로 해상도를 완벽히 재사용(ComfyUI 네이티브 캐싱)합니다.
- seed 값이 변경(또는 Randomize)되면, ComfyUI가 자동으로 감지하여 NAIA 서버로부터 새로운 랜덤 프롬프트를 동기 요청하여 받아옵니다.

4. `NAIA Prompt Fetch (WebSocket)`

카테고리: NAIA Bridge/Prompt

NAIA 메인 UI 또는 Web Remote에서 현재 편집/표시 중인 프롬프트를 WS 캐시에서 가져옴. 랜덤 요청이 아니라 "지금 화면에 있는 값"을 미러.

5. `NAIA Read Prompt Engineering`

카테고리: NAIA Bridge/Engineering

NAIA의 Prompt Engineering 모듈 현재 상태 조회 (디버그/검토용). pre/post/auto_hide, 15종 전처리 JSON, 프리셋 목록 반환.

6. `NAIA Check Health`

카테고리: NAIA Bridge/API

NAIA 서버 연결 진단. 실패 시 raise 안 하고 (ok=False, error_json) 반환 → 조건부 게이팅 노드로 활용 가능.

⚠️ 알려진 동작 / Notes

use_naia_settings = false 의 all-or-nothing 규약: override 시 pre/post/auto_hide/preprocessing 전부 노드 위젯 값으로 대체됨. 일부만 지정해도 나머지는 빈 값/OFF로 처리. 데스크톱 UI 값을 부분 사용하려면 NAIA Read Prompt Engineering 노드로 먼저 조회 후 위젯에 수동 복사.
# 주석 자동 제거: NAIA 출력의 #랜덤프롬프트 같은 줄 단위 주석은 브리지가 제거 후 반환. NovelAI 이스케이프 syntax #(...) 는 보호됨.
추천 해상도는 항상 활성: ComfyUI 요청은 NAIA의 random_resolution / auto_fit_resolution UI 토글과 무관하게 항상 추천값을 계산해 응답합니다 (source_row 우선, 없으면 random fallback). NAIA 메인 UI 설정은 변경되지 않습니다.
첫 실행 시 ~3초 지연: WS 초기 연결 + prompt_sync 수신 대기. 이후 요청은 <0.3초.
멀티 NAIA 인스턴스: (host, port) 싱글턴 설계로 이론상 가능하나 충분한 테스트 미실시.

🧭 트러블슈팅 / Troubleshooting

"NAIA 서버 연결 실패"

NAIA 실행 중인지 확인
Settings > Web Session 체크 또는 CLI --web-session 플래그 확인
포트 7243 (기본) 방화벽 차단 여부

"prompt_sync 메시지를 7초 이내 받지 못했습니다"

NAIA 메인 UI에서 프롬프트 편집 한 번 또는 랜덤 생성 한 번 실행
Web Session 활성화 직후 최초 접속 시 발생 가능

ComfyUI-Manager 설치 후 노드가 안 보임

ComfyUI 완전 재시작 (백엔드 + 프론트엔드 브라우저 탭)
ComfyUI/custom_nodes/comfyui-naia-bridge/ 폴더 확인
콘솔 로그에서 import 에러 확인

"Convert Widget to Input" 메뉴가 안 보임

ComfyUI frontend 1.42.14+ 필요
python -m pip install -r ComfyUI/requirements.txt 로 업데이트

📄 라이선스 / License

이 노드는 NAIA 2.0 (GPL-3.0) 서버와 HTTP/WebSocket으로 통신합니다. GPL 전파 대상 아님 (loose coupling aggregation).

🙏 크레딧 / Credits

NAIA 2.0: https://github.com/DNT-LAB/NAIA2.0
ComfyUI: https://github.com/comfyanonymous/ComfyUI

comfyui-naia-bridge

Nodes (6)

ComfyUI NAIA 2.0 Bridge

✨ 주요 기능 / Features

📋 사전 요구사항 / Prerequisites

1. NAIA 2.0 (패치 적용 버전)

2. ComfyUI

3. Python 의존성

🚀 설치 / Installation

방법 1: ComfyUI-Manager — 검색 설치 (권장)

방법 2: comfy-cli 로 설치

방법 3: ComfyUI-Manager "Install via Git URL"

방법 4: 수동 설치 (git 직접 사용)

업데이트 / Update

📖 사용법 / How to Use

Step 1: NAIA 사전 설정

Step 2: ComfyUI 워크플로우 구성

⚠️ Negative Prompt 연결은 선택사항 (기본: 연결 X)

프리뷰 확인 (선택)

📦 제공 노드 / Nodes

1. NAIA Request Random Prompt (메인 노드)

2. NAIA Request Random Prompt (with Override) (변형 노드)

3. NAIA Request Random Prompt (with Seed) (시드 연동형 간소화 노드)

4. NAIA Prompt Fetch (WebSocket)

5. NAIA Read Prompt Engineering

6. NAIA Check Health

⚠️ 알려진 동작 / Notes

🧭 트러블슈팅 / Troubleshooting

"NAIA 서버 연결 실패"

"prompt_sync 메시지를 7초 이내 받지 못했습니다"

ComfyUI-Manager 설치 후 노드가 안 보임

"Convert Widget to Input" 메뉴가 안 보임

📄 라이선스 / License

🙏 크레딧 / Credits

1. `NAIA Request Random Prompt` (메인 노드)

2. `NAIA Request Random Prompt (with Override)` (변형 노드)

3. `NAIA Request Random Prompt (with Seed)` (시드 연동형 간소화 노드)

4. `NAIA Prompt Fetch (WebSocket)`

5. `NAIA Read Prompt Engineering`

6. `NAIA Check Health`