ComfyUI Extension: ComfyUI-Danbooru-Gallery

Authored by Aaalice233

Created 2 days ago

Updated about 11 hours ago

8 stars

A danbooru gallery for ComfyUI

Custom Nodes (0)

README

ComfyUI Danbooru Gallery

中文 | English

中文版本

简介

一个基于 Danbooru API 的 ComfyUI 图像搜索和画廊插件，支持通过标签搜索图像、查看详细信息、内容过滤等功能。

主要特性

🔍 标签搜索: 支持通过 Danbooru 标签进行精确搜索
📄 分页加载: 高效的分页机制，提升浏览体验
💡 智能补全: 输入时自动补全热门标签
🎨 图像预览: 高质量图像预览和下载
🔧 设置管理: 多语言、黑名单、提示词过滤设置
🎯 ComfyUI 集成: 完美集成到 ComfyUI 工作流
🌊 瀑布流布局: 响应式瀑布流图像排列
📊 分级过滤: 支持按图像分级过滤内容
🏷️ 类别选择: 可选择输出的标签类别
📝 格式化选项: 支持括号转义和下划线替换
🎖️ 排行榜模式: 支持按热度排序显示
🚫 黑名单过滤: 自定义过滤不需要的内容
📱 响应式设计: 自适应不同屏幕尺寸
🔐 用户认证: 支持 Danbooru 用户名和 API 密钥认证
⭐ 收藏功能: 添加和移除图像收藏，支持云端同步
🌐 网络检测: 自动检测与 Danbooru 的网络连接状态
🈳 中英对照: 支持标签中英文互译和中文搜索匹配
⚙️ 高级设置: 调试模式、页面大小设置、缓存配置

快速安装

方法一：自动安装（推荐）

# 1. 将插件放到 ComfyUI/custom_nodes/ 目录
git clone https://github.com/comfyui-extensions/comfyui-danbooru-gallery.git

# 2. 运行安装脚本
cd comfyui-danbooru-gallery
python install.py

# 3. 重启 ComfyUI

方法二：手动安装

# 安装依赖
pip install -r requirements.txt

系统要求

Python: 3.8+
ComfyUI: 最新版本

核心依赖

requests>=2.28.0 - HTTP请求库
aiohttp>=3.8.0 - 异步HTTP客户端
Pillow>=9.0.0 - 图像处理库
torch>=1.12.0 - PyTorch框架
numpy>=1.21.0 - 数值计算库

基本使用

在 ComfyUI 中添加 "Danbooru Images Gallery" 节点
双击节点打开画廊界面
输入搜索标签（如：1girl blue_eyes smile）
选择评分过滤器并搜索
点击图像选择并导入到节点

详细功能说明

🔐 用户认证系统

支持 Danbooru 用户名和 API 密钥认证
认证后可使用收藏功能和高级功能
自动验证认证状态和网络连接

⭐ 收藏功能

添加/移除图像收藏
支持云端同步收藏列表
本地缓存收藏状态

🌐 网络连接检测

自动检测与 Danbooru 服务器的连接状态
显示网络错误信息和超时处理

⚙️ 高级设置

多语言支持: 中英文界面切换
黑名单管理: 自定义过滤不需要的标签
提示词过滤: 自动过滤水印、用户名等标签
调试模式: 启用详细日志输出
页面大小: 自定义每页显示的图像数量

🈳 中英对照系统

中英互译: 自动翻译英文标签为中文描述
中文搜索: 支持输入中文直接搜索对应英文标签
模糊匹配: 支持中文拼音和部分字符匹配
批量翻译: 高效的批量标签翻译处理
实时提示: 自动补全时显示中文翻译
多数据源: 支持JSON和CSV格式的翻译数据

翻译功能详解

支持的翻译数据格式

JSON格式 (zh_cn/all_tags_cn.json): 英文标签到中文的键值对映射
CSV格式 (zh_cn/danbooru.csv): 英文标签,中文翻译的CSV文件
角色CSV (zh_cn/wai_characters.csv): 中文角色名,英文标签的CSV文件

翻译功能特性

智能搜索: 支持精确匹配、前缀匹配、包含匹配和模糊匹配
缓存优化: 翻译结果缓存，提高响应速度
下划线处理: 自动处理有无下划线的标签变体
中文索引: 构建中文字符索引，支持快速搜索
权重排序: 根据匹配度为搜索结果排序

使用方法

中文搜索: 在搜索框中直接输入中文（如"女孩"），系统会自动匹配对应的英文标签
翻译显示: 在中文界面下，标签悬浮提示和自动补全会显示中文翻译
智能补全: 输入英文标签时，自动显示对应的中文翻译

标签语法

普通标签：tag_name
排除标签：-tag_name  
多标签组合：tag1 tag2 tag3
示例：1girl blue_eyes smile -blurry

项目结构

ComfyUI-Danbooru-Gallery/
├── __init__.py                 # 插件入口
├── danbooru_gallery.py         # 主要后端逻辑
├── install.py                  # 智能安装脚本
├── requirements.txt            # 依赖清单
├── pyproject.toml             # 项目配置
├── js/
│   └── danbooru_gallery.js     # 前端界面
├── zh_cn/                      # 中文翻译数据
│   ├── all_tags_cn.json        # JSON格式翻译数据
│   ├── danbooru.csv            # CSV格式翻译数据
│   └── wai_characters.csv      # 角色翻译数据
├── cache/                      # 缓存目录
└── README.md                   # 说明文档

故障排除

连接问题: 检查网络连接和 API 密钥
图像加载失败: 确认磁盘空间和图像 URL
插件不显示: 检查目录位置和依赖安装
性能问题: 检查控制台日志获取详细信息

English Version

Overview

A ComfyUI plugin for browsing and importing images from Danbooru using its API, with features including tag-based search, image preview, content filtering, and more.

Key Features

🔍 Tag Search: Precise search using Danbooru tags
📄 Pagination: Efficient pagination for better browsing
💡 Intelligent Autocomplete: Autocomplete popular tags while typing
🎨 Image Preview: High-quality image preview and download
🔧 Settings: Multi-language, blacklist, and prompt filtering
🎯 ComfyUI Integration: Seamless integration with ComfyUI workflow
🌊 Waterfall Layout: Responsive waterfall image arrangement
📊 Rating Filter: Filter content by image rating
🏷️ Category Selection: Choose which tag categories to output
📝 Formatting Options: Support bracket escaping and underscore replacement
🎖️ Ranking Mode: Display images sorted by popularity
🚫 Blacklist Filter: Custom filtering of unwanted content
📱 Responsive Design: Adaptive to different screen sizes
🔐 User Authentication: Support for Danbooru username and API key authentication
⭐ Favorites: Add and remove image favorites with cloud synchronization
🌐 Network Detection: Automatic detection of network connection to Danbooru
🈳 Chinese-English Bilingual: Support for tag translation and Chinese search matching
⚙️ Advanced Settings: Debug mode, page size settings, cache configuration

Quick Installation

Method 1: Automatic Installation (Recommended)

# 1. Clone to ComfyUI/custom_nodes/ directory
git clone https://github.com/comfyui-extensions/comfyui-danbooru-gallery.git

# 2. Run installation script
cd comfyui-danbooru-gallery
python install.py

# 3. Restart ComfyUI

Method 2: Manual Installation

# Install dependencies
pip install -r requirements.txt

System Requirements

Python: 3.8+
ComfyUI: Latest version

Core Dependencies

requests>=2.28.0 - HTTP request library
aiohttp>=3.8.0 - Async HTTP client
Pillow>=9.0.0 - Image processing library
torch>=1.12.0 - PyTorch framework
numpy>=1.21.0 - Numerical computing library

Basic Usage

Add "Danbooru Images Gallery" node in ComfyUI
Double-click the node to open gallery interface
Enter search tags (e.g., 1girl blue_eyes smile)
Select rating filter and search
Click images to select and import to node

Detailed Features

🔐 User Authentication System

Support for Danbooru username and API key authentication
Access to favorites and advanced features after authentication
Automatic authentication status and network connection verification

⭐ Favorites Feature

Add/remove image favorites
Cloud synchronization of favorites list
Local caching of favorite status

🌐 Network Connection Detection

Automatic detection of connection status to Danbooru server
Display network error messages and timeout handling

⚙️ Advanced Settings

Multi-language Support: Chinese/English interface switching
Blacklist Management: Custom filtering of unwanted tags
Prompt Filtering: Automatic filtering of watermarks, usernames, etc.
Debug Mode: Enable detailed logging output
Page Size: Customize number of images displayed per page

🈳 Chinese-English Bilingual System

Bidirectional Translation: Automatic translation of English tags to Chinese descriptions
Chinese Search: Support for searching with Chinese input to find corresponding English tags
Fuzzy Matching: Support for Chinese pinyin and partial character matching
Batch Translation: Efficient batch tag translation processing
Real-time Hints: Display Chinese translations during autocomplete
Multiple Data Sources: Support for JSON and CSV format translation data

Translation Features

Supported Translation Data Formats

JSON Format (zh_cn/all_tags_cn.json): English tag to Chinese key-value mapping
CSV Format (zh_cn/danbooru.csv): English tag, Chinese translation CSV file
Character CSV (zh_cn/wai_characters.csv): Chinese character name, English tag CSV file

Translation System Features

Intelligent Search: Support for exact match, prefix match, contains match, and fuzzy match
Cache Optimization: Translation result caching for improved response speed
Underscore Handling: Automatic handling of tag variants with/without underscores
Chinese Indexing: Build Chinese character index for fast searching
Weight Sorting: Sort search results by matching degree

Usage

Chinese Search: Enter Chinese directly in search box (e.g., "女孩"), system will automatically match corresponding English tags
Translation Display: In Chinese interface, tag tooltips and autocomplete show Chinese translations
Smart Completion: When typing English tags, automatically display corresponding Chinese translations

Tag Syntax

Normal tags: tag_name
Exclude tags: -tag_name
Multiple tags: tag1 tag2 tag3
Example: 1girl blue_eyes smile -blurry

Project Structure

ComfyUI-Danbooru-Gallery/
├── __init__.py                 # Plugin entry point
├── danbooru_gallery.py         # Main backend logic
├── install.py                  # Smart installation script
├── requirements.txt            # Dependency list
├── pyproject.toml             # Project configuration
├── js/
│   └── danbooru_gallery.js     # Frontend interface
├── zh_cn/                      # Chinese translation data
│   ├── all_tags_cn.json        # JSON format translation data
│   ├── danbooru.csv            # CSV format translation data
│   └── wai_characters.csv      # Character translation data
├── cache/                      # Cache directory
└── README.md                   # Documentation

Troubleshooting

Connection Issues: Check network and API key
Image Loading Fails: Verify disk space and image URLs
Plugin Not Showing: Check directory location and dependencies
Performance Issues: Check console logs for detailed information

开发 | Development

技术栈 | Tech Stack

Backend: Python + aiohttp + requests
Frontend: JavaScript + ComfyUI UI
Cache: File system cache
API: Danbooru REST API

贡献 | Contributing

欢迎提交 Issue 和 Pull Request！
Issues and Pull Requests are welcome!

许可证 | License

MIT License

致谢 | Acknowledgments

感谢 Danbooru 提供优秀的 API | Thanks to Danbooru for the excellent API
感谢 ComfyUI 社区 | Thanks to the ComfyUI community
参考了 ComfyUI_Civitai_Gallery 项目 | Inspired by ComfyUI_Civitai_Gallery project

翻译文件来源 | Translation Data Sources

danbooru-diffusion-prompt-builder - Danbooru 标签翻译数据
zh_CN-Tags - 中文标签数据
ComfyUI_Mira - 角色翻译数据