comfyui-pose-selector

Pose Selector for ComfyUI - Developer Guide

Этот модуль предназначен для выбора текстовых промптов (поз) из модульной JSON-библиотеки с визуальным превью и автоматической подстановкой в текст через тег {POSE}.

Архитектура модуля

1. Категории и Данные (js/categories/)

• Библиотека разделена на .json файлы, каждый из которых представляет отдельную категорию.
• Стандарты именования (Strict Rules):
  • Имя позы (name): Должно начинаться с имени JSON-файла + порядковый номер (например, в Classic.json позы называются Classic 01, Classic 02).
  • Изображение (image): Имя файла изображения в папке poses/ должно строго соответствовать имени позы (например, Classic 01.jpg).
  • Исключение из рандома: Если имя JSON-файла начинается с подчеркивания (например, _Hidden.json), категория будет доступна для ручного выбора, но не будет участвовать в глобальном рандоме (random_from_all).

2. Backend (pose_selector.py)

• Динамический импорт: При запуске сканирует JSON-файлы, сортирует их по алфавиту и формирует списки для валидации в ComfyUI.
• Входы:
  • text: Текст с тегом {POSE}.
  • category & pose_name: Выбор категории и конкретной позы.
  • random_from_category (BOOLEAN): Рандом внутри выбранной рубрики.
  • random_from_all (BOOLEAN): Рандом по всей библиотеке (за исключением файлов с префиксом _).
  • sequential_selection (BOOLEAN): Последовательный перебор поз. Использует формулу index = seed % count. Работает как для отдельной категории, так и для всего списка (если включен random_from_all).
  • seed (INT): Контроль случайности или индекс для последовательного выбора.
  • undressing_mode (BOOLEAN): Динамический режим подстановки одежды.
  • Поля подстановки (clothes, undrwr_color, undrwr1...undrwr4): Динамические текстовые поля.
• Логика работы:
  • Замена переменных (Undressing Mode): Если включен этот режим, скрипт ищет в сыром тексте позы теги {CLOTHES}, {UNDRWR1}...{UNDRWR4} и заменяет их на значения из соответствующих полей (с автоматическим приклеиванием общего цвета из undrwr_color).
  • Умная очистка: Если какая-то переменная (например, undrwr4) пуста, скрипт удаляет её тег и автоматически вычищает висячие запятые и двойные пробелы (например, bra, , panties. превращается в bra, panties.).
  • IS_CHANGED: Возвращает seed, если включен рандом или последовательный перебор, чтобы форсировать выполнение ноды.
  • Синхронизация: Через PromptServer отправляет событие pose_selector_update для мгновенного обновления UI при серверном рандоме.

3. Frontend (js/pose_selector.js)

• Динамическая видимость полей: Скрипт отслеживает состояние чекбокса undressing_mode. При включении он бережно восстанавливает скрытые текстовые поля (w.type, w.computeSize, w.draw и display: block для HTML-элементов). При выключении поля сжимаются (height=0) и их draw-функция блокируется (() => {}), предотвращая перехлест HTML-элементов над другими виджетами (DOM visibility isolation).
• Динамическая фильтрация: Обновляет список pose_name при смене category.
• Кэширование: Все запросы к JSON-файлам дополняются временной меткой (?t=...), что предотвращает использование устаревших закэшированных данных браузером после правок JSON.
• Интерфейс:
  • Превью: Отрисовывается в кастомном виджете pose_preview.
  • Кнопки Random: Позволяют вручную перемешать позы прямо в интерфейсе без запуска генерации.

Установка и Окружение

1. Копировать папку comfyui-pose-selector в ComfyUI/custom_nodes/.
2. JSON-файлы — в js/categories/, изображения — в poses/.