7.2 KiB
ComfyUI-Autocomplete-Plus
Warning
このカスタムノードは現在ベータ版です。予告なく機能が変更されたり不具合が発生する場合があります。
概要
ComfyUI-Autocomplete-Plus は、ComfyUI の任意のテキストエリアに複数の入力支援機能を提供するカスタムノードです。現在は主にDanbooruスタイルのタグ入力に対応しています。
特徴
- :zap:セットアップ不要: Danbooruタグに最適化された CSV データを自動でダウンロード
- :mag:オートコンプリート: テキスト入力中に、入力内容に基づいてタグ候補をリアルタイムで表示
- :file_cabinet:関連タグ表示機能: 選択したタグと関連性の高いタグを一覧表示
- :earth_asia:多言語対応: 日本語、中国語、韓国語での入力補完をサポート
- :computer_mouse:直感的な操作:
- マウスとキーボードどちらの操作にも対応
- カーソル位置や既存のテキストを考慮した自然なタグ挿入
- :art:デザイン: ComfyUIのライトテーマとダークテーマの両方に対応
- :pencil:ユーザーCSV: ユーザーが用意した CSV をオートコンプリート候補に追加可能
- 依存ライブラリゼロ: 外部ライブラリを使用せず、すべての入力支援処理がブラウザで動作
インストール
ベータ版では以下の手順でインストールできます。
- このリポジトリを ComfyUI の
custom_nodesフォルダにクローンまたはコピーします
git clone https://github.com/newtextdoc1111/ComfyUI-Autocomplete-Plus.git - ComfyUI を起動します。初回起動時のみ、必要な CSV データが HuggingFace から自動的にダウンロードされます
オートコンプリート
テキスト入力エリアで文字を入力すると、テキストに部分一致するタグを投稿数の多い順で表示します。上下キーで選択、EnterかTabキーで選択したタグを挿入できます。
- タグのエイリアスも検索対象に含まれます。日本語のひらがな、カタカナは区別せず検索されます
- タグのカテゴリ毎に色分けされます。色分けのルールは Danbooru と同じです
- 入力済みのタグはグレーアウトで表示されます。グレーアウトしたタグを確定した場合、代わりに入力済みのタグを選択状態にします
関連タグ
テキスト入力エリアの任意のタグを選択すると、関連性の高いタグを一覧表示します。UIは編集中のテキストエリアを基準に位置とサイズが自動で調整されます。
- 表示位置は、テキストエリアの下部を基本とし、空きスペースに応じて上下に自動調整されます
- ヘッダーの「↕️|↔️」ボタンで上下と左右の表示位置に切り替えられます
- ヘッダーの「📌|🎯」ボタンで表示する関連タグの固定状態を切り替えられます。固定状態で閉じたい場合はEscキーを押します
Ctrl+Shift+Spaceキーでカーソル位置の関連タグを表示できます
CSV データ
動作には基本となる CSV データが2つ必要です。これらは HuggingFace で管理されており、 ComfyUI にインストール後の初回起動時に自動でダウンロードされるのでセットアップは不要です。 基本 CSV ファイルはHuggingFaceで公開されているDanbooruデータセットを元にしているので、Danbooruサイトの投稿数や関連タグの情報と異なる場合があります。
Important
基本 CSV にはSFW, NSFW両方のタグが含まれています。
danbooru_tags.csv
タグ名、カテゴリ、投稿数、エイリアス(日本語、中国語、韓国語を含む)の情報を持つオートコンプリート用のタグ情報 CSV ファイルです。このカラム構成は DominikDoom/a1111-sd-webui-tagcomplete で使用されているものと同じです。
タグ情報は以下の条件でフィルタリングされています。
- 投稿数100件以上
- 投稿画像のスコアが5以上
- カテゴリが
general, character, copyrightのいずれか - タグ名に
(cosplay)が含まれていない
danbooru_tags_cooccurrence.
タグペアとその共起回数を記録した、関連タグ計算用の CSV ファイルです。
タグペアはタグ情報 CSV から、さらに以下の条件でフィルタリングされています。
- 共起回数が100件以上
ユーザーCSV
ユーザーが自身で用意したCSVを使用することも可能です。例として、よく使うメタタグを danbooru_tags_meta.csv の名前で data フォルダーに配置することでオートコンプリート候補に追加できます。
ヘッダー行はなくても構いません。反映にはブラウザのリロードが必要です。
メタタグの例
tag,category,count,alias
masterpiece,5,9999999,
best_quality,5,9999999,
high_quality,5,9999999,
normal_quality,5,9999999,
low_quality,5,9999999,
worst_quality,5,9999999,
Note
ユーザー CSV が複数ある場合アルファベット順に読み込まれます。同じタグが複数のファイルに存在する場合は先に読み込まれた方が保持されます。基本 CSV は最後にロードされます。
設定
オートコンプリート
- Enable Autocomplete: オートコンプリート機能の有効化/無効化
- Max suggestions: オートコンプリート候補の最大表示件数
関連タグ
- Enable Related Tags: 関連タグ機能の有効化/無効化
- Max related tags: 関連タグの最大表示件数
- Default Display Position: ComfyUI起動時のデフォルト表示位置
- Related Tags Trigger Mode : 関連タグの表示トリガー(クリック、Ctrl+クリック)
既知の問題
パフォーマンス
- CSV ファイルの容量が大きいため、ブラウザの起動時間が長くなる場合があります
- ブラウザ上で高速に動作させるためにメモリを消費します。ComfyUIが動作するスペックのマシンであれば問題にはならないと思います
オートコンプリート
関連タグ
- UIをピン留めした際のキーボード入力時、一部のキー以外はテキストエリアに入力されてしまう場合があります
クレジット
- ComfyUI-Custom-Node
- オートコンプリート機能の実装にあたり参考にしました
- DominikDoom/a1111-sd-webui-tagcomplete
- オートコンプリートの主要な機能や CSV の仕様で参考にしました