nodes/ComfyAssets-ComfyUI_PromptManager
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
104 Commits 4 Branches 1 Tag
180ae12f4dee35fd062b344dd2c46fa968536e85
T
Clone
Open with VS Code Open with VSCodium Open with Intellij IDEA
Download ZIP Download TAR.GZ Download BUNDLE

README.md

ComfyUI Prompt Manager

A comprehensive ComfyUI custom node that extends the standard text encoder with persistent prompt storage, advanced search capabilities, automatic image gallery system, and powerful ComfyUI workflow metadata analysis using SQLite.

Overview

Prompt Manager Interface

ComfyUI Prompt Manager functions as a drop-in replacement for ComfyUI's standard CLIPTextEncode node while adding comprehensive prompt management and automatic image tracking features:

  • 🔄 Drop-in Replacement: Works exactly like the standard text encoder
  • 💾 Persistent Storage: Automatically saves all prompts to a local SQLite database
  • 🔍 Advanced Search: Query past prompts with text search, category filtering, and metadata
  • 🖼️ Automatic Image Gallery: Automatically links generated images to their prompts
  • 🏷️ Rich Metadata: Add categories, tags, ratings, notes, and workflow names to prompts
  • 🚫 Duplicate Prevention: Uses SHA256 hashing to detect and prevent duplicate storage
  • 🌐 Advanced Web Interface: Comprehensive admin dashboard with metadata analysis
  • 📊 Analytics: Track prompt usage patterns and effectiveness over time
  • 🔬 Workflow Analysis: Extract and analyze ComfyUI workflow data from PNG images
  • 📋 Metadata Viewer: Standalone tool for analyzing ComfyUI-generated images
  • 🛠️ System Management: Built-in diagnostics, backup/restore, and maintenance tools

Image Gallery

Features

Core Functionality

  • Text Encoding: Standard CLIP text encoding for ComfyUI workflows
  • Auto-Save: Every prompt is automatically saved to the database
  • Metadata Support: Optional categories, tags, ratings (1-5), notes, and workflow names
  • Hash-based Deduplication: Prevents storing identical prompts multiple times

🖼️ Automatic Image Gallery

Prompt Editor

  • Smart Image Detection: Automatically monitors ComfyUI output directory for new images
  • Intelligent Linking: Links generated images to their corresponding prompts based on execution timing
  • Metadata Extraction: Extracts and stores ComfyUI workflow data and generation parameters
  • Gallery Interface: Beautiful web-based gallery with thumbnail grid view
  • Full-size Image Viewer: Modal overlay with navigation controls and keyboard shortcuts
  • Image Navigation: Browse through images with arrow keys or navigation buttons
  • Image Counter: Shows current position in gallery (e.g., "3 / 7")
  • Integrated Metadata Panel: View workflow data directly in the image viewer
  • File Path Display: Shows actual file paths in output directory for easy access

🔬 ComfyUI Workflow Analysis

  • PNG Metadata Parsing: Advanced PNG chunk analysis to extract embedded ComfyUI data
  • Workflow Data Extraction: Complete ComfyUI workflow JSON recovery from images
  • Prompt Recovery: Extract positive/negative prompts from any ComfyUI-generated image
  • Generation Parameters: Recover steps, CFG scale, sampler, seed, and model information
  • Multi-Node Support: Handles PromptManager, CLIPTextEncode, and other node types
  • Copy & Export: One-click copying of prompts and workflow JSON download
  • Standalone Viewer: Dedicated metadata analysis tool at /prompt_manager/metadata.html
  • Drag & Drop Interface: Easy analysis by dropping PNG files directly into the viewer
  • Error Handling: Robust parsing with NaN value cleaning and malformed JSON recovery

📋 Standalone Metadata Viewer

  • Independent Tool: Access at http://localhost:8188/prompt_manager/metadata.html
  • Real-time Analysis: Instant PNG metadata extraction and display
  • Comprehensive Display: Shows all discoverable ComfyUI workflow information
  • Export Capabilities: Download workflow JSON files for reuse
  • File Path Integration: Click to reveal file locations in your system
  • Visual Interface: Clean, organized presentation of complex workflow data

Search & Retrieval

  • Full-text search across all stored prompts
  • Category filtering for organized prompt collections
  • Tag-based search with support for multiple tags
  • Rating filters to find your best prompts
  • Date range filtering for temporal searches
  • Recent prompts quick access
  • Top-rated prompts for quality discovery
  • Image search by file properties and metadata

🌐 Advanced Web Interface

  • Admin Dashboard: Comprehensive browser-based management interface with modern dark theme
  • Responsive Design: Works on desktop, tablet, and mobile devices
  • Real-time Search: Instant search results as you type
  • Bulk Operations: Edit multiple prompts simultaneously with batch tools
  • Settings Panel: Configure behavior and display options
  • System Management: Complete toolset for database operations:
    • 🔍 Metadata Viewer: Launch standalone PNG analysis tool
    • 📸 Scan Images: Bulk scan output directories for ComfyUI images
    • 💾 Backup/Restore: Database backup and recovery operations
    • 🔧 Maintenance: Database optimization and cleanup tools including orphaned prompt removal
    • 📋 Logs: Comprehensive logging system with download capability
    • 🔍 Diagnostics: System health checks and troubleshooting
  • Export Tools: Download prompts and metadata in various formats
  • Progress Tracking: Real-time progress indicators for long operations
  • Error Handling: Graceful error display with recovery suggestions

Database Management

  • SQLite backend for reliable local storage
  • Automatic schema creation and management
  • Database optimization with proper indexing and maintenance tools
  • Export functionality to JSON or CSV formats
  • Backup and restore capabilities
  • Relationship tracking between prompts and generated images
  • Orphaned prompt cleanup to remove prompts without linked images

Installation

For ComfyUI Users

  1. Clone the repository into your ComfyUI custom_nodes directory:

    cd ComfyUI/custom_nodes/
git clone https://github.com/ComfyAssets/ComfyUI_PromptManager
cd ComfyUI_PromptManager

  2. Install dependencies:

    pip install -r requirements.txt

  3. Restart ComfyUI to load the new node

  4. Add the node to your workflow:

    • Look for "Prompt Manager" in the conditioning category
    • Use it exactly like the standard "CLIP Text Encode" node

  5. Access the web interface:

    • Admin Dashboard: http://localhost:8188/prompt_manager/admin - Full management interface
    • Metadata Viewer: http://localhost:8188/prompt_manager/metadata.html - Standalone PNG analysis tool
    • Simple Interface: http://localhost:8188/prompt_manager/ - Basic prompt browser

Prerequisites

  • ComfyUI installation
  • Python 3.8 or higher
  • SQLite3 (included with Python)
  • watchdog library for automatic image monitoring

Usage

Basic Usage

Replace any CLIPTextEncode node with PromptManager:

  1. Add the node to your workflow
  2. Connect CLIP model (same as standard text encoder)
  3. Enter your prompt in the text field
  4. Optionally add metadata:
    • Category: "portraits", "landscapes", "abstract", etc.
    • Tags: "detailed, anime, masterpiece" (comma-separated)
    • Rating: 1-5 stars for prompt quality
    • Notes: Any additional information
    • Workflow Name: Name of your workflow for organization

The node will encode your text, automatically save it to the database, and link any generated images to the prompt.

🖼️ Using the Image Gallery

The image gallery automatically captures and links generated images:

  1. Generate images using workflows with the Prompt Manager node
  2. Open the admin interface at http://localhost:8188/prompt_manager/admin
  3. Click the "🖼️ Gallery" button on any prompt to view its images
  4. Click any thumbnail to open the full-size image viewer with integrated metadata panel
  5. Navigate images using:
    • Arrow keys (←/→) for keyboard navigation
    • Navigation buttons for mouse control
    • ESC key to close the viewer
  6. View metadata in the right panel showing:
    • File path in output directory
    • Model/checkpoint information
    • Prompts (positive and negative)
    • Generation parameters (steps, CFG, sampler, seed)
    • Raw workflow data with download option

🔬 Analyzing ComfyUI Images

Use the standalone metadata viewer to analyze any ComfyUI-generated PNG:

  1. Open the metadata viewer at http://localhost:8188/prompt_manager/metadata.html
  2. Drag and drop any PNG file with ComfyUI workflow data
  3. View extracted information:
    • Prompts: Copy positive/negative prompts with one click
    • Parameters: See all generation settings used
    • Model Info: Identify checkpoint and settings
    • Workflow: Download complete workflow JSON
  4. Copy data using the built-in copy buttons
  5. Export workflow as JSON file for reuse in ComfyUI

📸 Bulk Image Scanning

Import existing ComfyUI images into your database:

  1. Open admin dashboard and click "📸 Scan Images"
  2. Start scan to analyze your entire output directory
  3. Monitor progress with real-time status updates
  4. Review results showing:
    • Images found and processed
    • Prompts extracted and linked
    • Any errors or issues encountered
  5. Access imported data through the normal gallery interface

🌐 Web Interface Features

The comprehensive web interface provides:

  • Search Bar: Real-time search across all prompts
  • Filter Options: Filter by category, tags, rating, and date
  • Bulk Operations: Select multiple prompts for batch editing
  • Export Tools: Download your prompt collection
  • Settings Panel: Configure auto-save and display options
  • Diagnostics: System health checks and troubleshooting

🔧 Database Maintenance

Keep your database optimized with built-in maintenance tools:

  1. Open the admin dashboard and click "🔧 Maintenance"
  2. Select maintenance operations:
    • Remove Duplicates: Find and merge duplicate prompts while preserving images
    • Vacuum Database: Optimize database file size and performance
    • Clean Orphaned Images: Remove image records for missing files
    • 🚮 Prune Orphaned Prompts: Remove prompts that have no linked images
    • Check Consistency: Verify database integrity and relationships
    • Database Statistics: Generate detailed database reports
  3. Run selected operations and monitor progress in real-time
  4. Review results showing detailed information about each operation

Database Location

By default, the database is saved as example_prompts.db in the node directory. This file contains all your prompts and linked images and can be backed up or shared.

Searching Prompts

Use the web interface for intuitive searching, or access the database directly:

from database.operations import PromptDatabase

db = PromptDatabase()

# Search for landscape prompts
results = db.search_prompts(text="landscape", limit=10)

# Find highly rated prompts
results = db.search_prompts(rating_min=4)

# Search by category and tags
results = db.search_prompts(
    category="portraits",
    tags=["anime", "detailed"]
)

# Get recent prompts
recent = db.get_recent_prompts(limit=20)

# Get images for a prompt
images = db.get_prompt_images(prompt_id="123")

Examples

Example 1: Basic Prompt Storage

Input: "A beautiful sunset over a mountain lake"
Category: "landscapes"
Tags: "nature, sunset, mountains, water"
Rating: 5
Notes: "Perfect for peaceful scenes"

Example 2: Character Prompt

Input: "Portrait of a cyberpunk hacker with neon implants"
Category: "characters"
Tags: "cyberpunk, portrait, sci-fi, neon"
Rating: 4
Workflow: "character_generator_v2"

Example 3: Abstract Art

Input: "Swirling colors in an abstract geometric pattern"
Category: "abstract"
Tags: "geometric, colorful, pattern, modern"
Rating: 3
Notes: "Good for experimental art"

Database Schema

-- Prompts table
CREATE TABLE prompts (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    text TEXT NOT NULL,
    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
    updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
    workflow_name TEXT,
    category TEXT,
    tags TEXT,  -- JSON array of tags
    rating INTEGER CHECK(rating >= 1 AND rating <= 5),
    notes TEXT,
    hash TEXT UNIQUE  -- SHA256 hash for deduplication
);

-- Generated images table
CREATE TABLE generated_images (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    prompt_id TEXT NOT NULL,
    image_path TEXT NOT NULL,
    filename TEXT NOT NULL,
    generation_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
    file_size INTEGER,
    width INTEGER,
    height INTEGER,
    format TEXT,
    workflow_data TEXT,  -- JSON workflow metadata
    prompt_metadata TEXT,  -- JSON prompt parameters
    parameters TEXT,  -- JSON generation parameters
    FOREIGN KEY (prompt_id) REFERENCES prompts(id)
);

Architecture

Core Components

  • prompt_manager.py - Main ComfyUI node implementation
  • database/models.py - Database schema and connection management
  • database/operations.py - CRUD operations and search functionality
  • py/api.py - Web API endpoints for the interface
  • py/config.py - Configuration management
  • utils/hashing.py - SHA256 hashing for deduplication
  • utils/validators.py - Input validation and sanitization
  • utils/image_monitor.py - Automatic image detection system
  • utils/prompt_tracker.py - Prompt execution tracking
  • utils/metadata_extractor.py - PNG metadata analysis engine
  • utils/logging_config.py - Comprehensive logging system
  • utils/diagnostics.py - System diagnostics and health checks
  • web/admin.html - Advanced admin dashboard with metadata panel
  • web/index.html - Simple web interface
  • web/prompt_manager.js - JavaScript functionality
  • web/metadata.html - Standalone PNG metadata viewer

File Structure

ComfyUI_PromptManager/
├── __init__.py                    # Node registration
├── prompt_manager.py             # Main node implementation
├── database/
│   ├── __init__.py
│   ├── models.py                 # Database schema
│   └── operations.py             # Database operations
├── py/
│   ├── __init__.py
│   ├── api.py                    # Web API endpoints
│   └── config.py                 # Configuration
├── utils/
│   ├── __init__.py
│   ├── hashing.py               # Hashing utilities
│   ├── validators.py            # Input validation
│   ├── image_monitor.py         # Automatic image detection
│   ├── prompt_tracker.py        # Prompt execution tracking
│   ├── metadata_extractor.py    # PNG metadata analysis
│   ├── logging_config.py        # Logging system
│   └── diagnostics.py           # System diagnostics
├── web/
│   ├── admin.html               # Advanced admin dashboard
│   ├── index.html               # Simple web interface
│   ├── metadata.html            # Standalone metadata viewer
│   └── prompt_manager.js        # JavaScript functionality
├── tests/
│   ├── __init__.py
│   └── test_basic.py            # Test suite
├── requirements.txt             # Dependencies
├── example_usage.py            # Standalone examples
├── example_prompts.db          # Example database
└── README.md                   # This file

Configuration

Database Settings

You can customize the database path by modifying the configuration:

# In py/config.py
DATABASE_PATH = "custom_path/prompts.db"

Gallery & Monitoring Settings

Configure the automatic image detection and gallery system:

# Gallery monitoring configuration (GalleryConfig class)
MONITORING_ENABLED = True
MONITORING_DIRECTORIES = []  # Auto-detect ComfyUI output if empty
SUPPORTED_EXTENSIONS = ['.png', '.jpg', '.jpeg', '.webp', '.gif']
PROCESSING_DELAY = 2.0  # Delay before processing new images
PROMPT_TIMEOUT = 120  # Seconds to keep prompt context active
CLEANUP_INTERVAL = 300  # Seconds between cleanup of expired prompts

# Performance settings
MAX_CONCURRENT_PROCESSING = 3
METADATA_EXTRACTION_TIMEOUT = 10  # Seconds for PNG analysis
IMAGES_PER_PAGE = 20
THUMBNAIL_SIZE = 256

Web Interface Settings

Configure the web interface behavior:

# Web UI settings (PromptManagerConfig class)
RESULT_TIMEOUT = 5  # Seconds to auto-hide results in ComfyUI node
WEBUI_DISPLAY_MODE = 'newtab'  # 'popup' or 'newtab' for Web UI button
SHOW_TEST_BUTTON = False  # Show API test button in node UI

Performance Tuning

  • The database automatically creates indexes for optimal search performance
  • Regular VACUUM operations keep the database optimized
  • Consider backing up the database periodically

Advanced Features

Export Your Prompts

from database.operations import PromptDatabase

db = PromptDatabase()

# Export to JSON
db.export_prompts("my_prompts.json", format="json")

# Export to CSV
db.export_prompts("my_prompts.csv", format="csv")

Database Statistics

info = db.model.get_database_info()
print(f"Total prompts: {info['total_prompts']}")
print(f"Average rating: {info['average_rating']}")

Backup and Restore

# Create backup
db.model.backup_database("backup_prompts.db")

# The database file can be copied directly for backup

Development

Running Tests

cd KikoTextEncode
python -m pytest tests/ -v

Code Style

The project follows PEP 8 guidelines with:

  • Black formatter (88 character line limit)
  • Type hints for all functions
  • Comprehensive docstrings
  • Proper error handling

Contributing

  1. Fork the repository
  2. Create a feature branch
  3. Add tests for new functionality
  4. Ensure all tests pass
  5. Submit a pull request

Troubleshooting

Common Issues

Database Permission Errors

  • Ensure the ComfyUI process has write permissions to the node directory
  • Check that the database file isn't locked by another process

Import Errors

  • Verify ComfyUI is properly installed
  • Check that all required dependencies are available

Performance Issues

  • Run VACUUM on the database occasionally
  • Consider archiving old prompts if the database becomes very large

Debug Mode

For debugging, you can enable verbose logging in the node:

# Add to kiko_text_encode.py
import logging
logging.basicConfig(level=logging.DEBUG)

License

MIT License - see LICENSE file for details.

Support

  • Issues: Report bugs and request features via GitHub Issues
  • Documentation: See the Wiki for detailed guides
  • Community: Join the discussion in ComfyUI Discord

Roadmap

Completed in v3.0.0

  • ✅ PNG Metadata Analysis: Complete ComfyUI workflow extraction from images
  • ✅ Standalone Metadata Viewer: Dedicated tool for analyzing any ComfyUI image
  • ✅ Advanced Admin Dashboard: Comprehensive management interface with modern UI
  • ✅ Integrated Metadata Panel: Real-time workflow analysis in image viewer
  • ✅ Bulk Image Scanning: Mass import of existing ComfyUI images
  • ✅ System Management Tools: Backup, restore, diagnostics, and maintenance
  • ✅ Enhanced Error Handling: Robust PNG parsing with NaN value cleaning

Planned Features

  • ☁️ Cloud Sync: Optional cloud backup and sync
  • 🤝 Collaboration: Share prompt collections with other users
  • 🧠 AI Suggestions: Recommend similar prompts based on metadata analysis
  • 📈 Advanced Analytics: Detailed usage statistics and trends with workflow insights
  • 🔌 Plugin System: Support for third-party extensions and custom analyzers
  • 🎨 Enhanced Batch Processing: Advanced bulk operations with metadata editing
  • 🔄 Workflow Templates: Save and reuse common workflow patterns
  • 📊 Visual Analytics: Charts and graphs for prompt effectiveness analysis

Integration Ideas

  • Auto-tagging: Use AI to automatically categorize prompts
  • Workflow linking: Connect prompts to specific workflow templates
  • Image analysis: Analyze generated images to improve suggestions
  • Version control: Track prompt iterations and effectiveness

Changelog

v3.0.0 (Metadata Analysis Release)

  • 🔬 PNG Metadata Analysis: Advanced ComfyUI workflow extraction from PNG images
  • 📋 Standalone Metadata Viewer: Dedicated tool at /web/metadata.html for analyzing any ComfyUI image
  • 🖼️ Integrated Metadata Panel: Real-time workflow analysis in the image viewer
  • 📸 Bulk Image Scanning: Mass import and analysis of existing ComfyUI output directories
  • 🛠️ System Management Suite: Complete backup, restore, maintenance, and diagnostic tools
  • 🚮 Orphaned Prompt Cleanup: Remove prompts that have no linked images for database optimization
  • 📋 Advanced Logging: Comprehensive logging system with web-based log viewer
  • 🎨 Enhanced Admin Dashboard: Modern dark theme with improved UX and management tools
  • 🔧 Robust Error Handling: Advanced PNG parsing with NaN value cleaning and error recovery
  • 📁 File Path Integration: Direct file path display and system integration
  • ⚡ Performance Optimizations: Improved metadata extraction and database operations

v2.0.0 (Gallery Release)

  • 🖼️ Automatic Image Gallery: Complete image tracking and gallery system
  • 🌐 Advanced Web Interface: Comprehensive admin dashboard with responsive design
  • 📱 Image Viewer: Full-screen modal with navigation and keyboard shortcuts
  • 🔍 Enhanced Search: Real-time search with advanced filtering options
  • ⚡ Performance Improvements: Optimized database operations
  • 🛠️ Basic Diagnostics: Initial system diagnostics and health monitoring
  • 📊 Bulk Operations: Multi-select prompt editing and management

v1.0.0 (Initial Release)

  • Core text encoding with database storage
  • Search and filtering functionality
  • Metadata support (categories, tags, ratings, notes)
  • Hash-based deduplication
  • SQLite backend with optimized schema
  • Basic web interface
  • Export functionality
Reference in New Issue View Git Blame Copy Permalink
S
Description
No description provided
Readme MIT
4.2 MiB
Languages
Python 49.6%
JavaScript 37%
HTML 12.5%
CSS 0.8%