Fixed object reference

This commit is contained in:
PXL8D
2025-08-09 14:23:21 -07:00
parent 3c7b5658f0
commit 781ce38fe1
4 changed files with 283 additions and 23 deletions
+10 -4
View File
@@ -13,13 +13,19 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
- Universal schema supporting 60+ metadata fields across all major platforms
- Template-driven architecture for easy platform extension
- **Cross-Platform Workflow Generation**
- **Simplified Workflow Generation**: Convert ANY platform's metadata into working ComfyUI workflows that recreate identical images
- **Complex Workflow Preservation**: Full-fidelity preservation of ComfyUI node graphs and custom arrangements
- Bidirectional conversion between all supported platforms
- Perfect image reproduction across platform boundaries
- **Platform Support**
- Automatic1111 (A1111) - Full parameter format support
- Automatic1111 (A1111) - Full parameter format support with ComfyUI workflow generation
- ComfyUI - Complete workflow and prompt preservation
- Civitai - Enhanced resource tracking with model IDs
- Civitai - Enhanced resource tracking with model IDs and workflow generation
- Forge - A1111-compatible with advanced sampling extensions
- Tensor.AI - JSON-based metadata format
- Leonardo.AI - Style and preset management system
- Tensor.AI - JSON-based metadata format with workflow generation
- Leonardo.AI - Style and preset management system with workflow generation
- **ComfyUI Nodes**
- **MetaMan Universal** - Primary metadata extraction and conversion
+43 -5
View File
@@ -14,9 +14,10 @@ MetaMan bridges the metadata gap between AI image generation services, enabling
### Core Capabilities
- **Universal Metadata Extraction**: Read metadata from any AI image generation service
- **Cross-Platform Conversion**: Convert metadata between A1111, ComfyUI, Civitai, Tensor.AI, Leonardo.AI, and more
- **Cross-Platform Workflow Generation**: Convert ANY platform's metadata into working ComfyUI workflows that recreate identical images
- **Bidirectional Format Conversion**: Convert metadata between A1111, ComfyUI, Civitai, Tensor.AI, Leonardo.AI, and more
- **Complex Workflow Preservation**: Save and restore complete ComfyUI node graphs with full fidelity
- **Model Dependency Tracking**: Automatically identify and resolve model dependencies with download URLs
- **Workflow Preservation**: Save and restore complete workflows across platforms
- **Template-Based Extensibility**: Easy addition of new services via YAML templates
### Supported Platforms
@@ -59,6 +60,24 @@ MetaMan uses a comprehensive **universal metadata schema** that encompasses all
3. **Transforms** to target platform format using service-specific templates
4. **Preserves** original workflows and enables cross-platform reproduction
### Cross-Platform Workflow Conversion
MetaMan enables true **cross-platform workflow migration** with two levels of fidelity:
**✅ Simplified Workflow Generation**:
- Takes metadata from ANY platform (A1111, Civitai, Tensor.AI, etc.)
- Generates a functional workflow that **recreates the exact same image**
- Works for all platforms → ComfyUI conversion
- Perfect for moving generations between platforms
**✅ Complex Workflow Preservation**:
- Preserves original ComfyUI node graphs with custom arrangements
- Maintains advanced techniques, custom nodes, and complex routing
- Only available for ComfyUI → ComfyUI workflows
- Essential for sharing sophisticated ComfyUI creations
**Example**: An A1111 image with LoRAs and specific sampling settings becomes a ComfyUI workflow with CheckpointLoader → LoraLoader → CLIPTextEncode → KSampler → VAEDecode that produces **identical output**.
### Template System
The power of MetaMan lies in its **template-driven architecture**:
@@ -88,10 +107,26 @@ operation: "extract_universal"
target_service: "automatic1111"
# → Outputs: universal JSON + A1111 parameter string
# Convert ComfyUI workflow to A1111 format
# Convert A1111 image to working ComfyUI workflow
operation: "convert_to_service"
target_service: "comfyui"
# → Generates functional ComfyUI workflow that recreates identical image
```
### Cross-Platform Workflow Migration
```python
# Take any A1111/Civitai image and create ComfyUI workflow
input: A1111_image_with_metadata.png
operation: "convert_to_service"
target_service: "comfyui"
# → Output: Complete ComfyUI workflow JSON that produces same result
# Convert complex ComfyUI workflow to A1111 parameters
input: ComfyUI_workflow_image.png
operation: "convert_to_service"
target_service: "automatic1111"
# → Generates A1111-compatible parameters
# → Output: A1111-compatible parameter string (simplified but equivalent)
```
### Model Dependency Tracking
@@ -239,13 +274,16 @@ Service templates should include:
| Feature | A1111 | ComfyUI | Civitai | Forge | Tensor.AI | Leonardo.AI |
|---------|-------|---------|---------|--------|-----------|-------------|
| Basic Parameters | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ |
| Workflow Data | ❌ | ✅ | ❌ | ❌ | ❌ | ❌ |
| Simplified Workflow* | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ |
| Model Hashes | ✅ | ✅ | ✅ | ✅ | ⚠️ | ⚠️ |
| LoRA Support | ✅ | ✅ | ✅ | ✅ | ✅ | ❌ |
| Complex Workflow Preservation | ❌ | ✅ | ❌ | ❌ | ❌ | ❌ |
| Dependency URLs | ⚠️ | ⚠️ | ✅ | ⚠️ | ✅ | ✅ |
✅ Full Support | ⚠️ Partial Support | ❌ Not Supported
**Simplified Workflow*: MetaMan can generate a functional workflow from any platform's metadata that will recreate the exact same image. Complex workflows get simplified to basic generation steps, but output quality and reproducibility are maintained.**
## 📄 License
[Specify your license here]
+177
View File
@@ -0,0 +1,177 @@
# MetaMan Project Context Bootstrap
## Project Overview
**MetaMan** is a comprehensive ComfyUI custom node system for universal metadata management across AI image generation platforms. Built in August 2025, it solves the critical interoperability problem between different AI image generation services.
## Core Concept & Architecture
- **Universal Metadata Standard**: Custom "meta" PNG chunk containing JSON-structured metadata
- **Template-Driven System**: YAML-based service definitions for easy platform extension
- **Cross-Platform Workflow Generation**: Convert ANY platform's metadata into working ComfyUI workflows that recreate identical images
- **Bidirectional Conversion**: Translate metadata between platforms while preserving generation capability
## User Requirements Implemented
### Original User Stories (All ✅ Completed):
A. **Rewrite Metadata on A1111/Civitai Workflow Standards** - Converts any source to A1111/Civitai-compatible formats
B. **Generate new PNG with A1111/Civitai Standards from existing** - Creates new images with embedded target-format metadata
C. **Extract Metadata for use in existing workflow** - Universal extraction with service-specific conversions
D. **Extract Metadata and create/update text file** - Multiple export formats (JSON, TXT, parameters)
E. **Output Metadata as string** - All nodes return formatted metadata strings
F. **Create dependency list with download URLs** - Automated model dependency resolution via Civitai/HuggingFace APIs
### Key Design Decisions:
1. **Single Service Selection**: User chooses target service rather than storing multiple formats per image
2. **Custom "meta" PNG Chunk**: Universal format with all discrete data elements from all services
3. **Template-Based Extensibility**: Service support via YAML templates, no code changes needed
4. **Workflow Embedding**: Save workflows as JSON files or embed back into original images
## Technical Architecture
### Core Components:
- **`metaman.py`** (renamed from metamann.py): MetaManUniversalNode - primary metadata extraction & conversion
- **`specialized_nodes.py`**: MetaManWorkflowSaver & MetaManDependencyResolver nodes
- **`metadata_parser.py`**: Universal parser for all platform formats with service auto-detection
- **`api_integration.py`**: Model repository APIs (Civitai, HuggingFace) with rate limiting
### Template System:
- **`templates/universal_schema.yaml`**: Defines 60+ metadata fields across all platforms
- **`templates/services/`**: Platform-specific conversion templates:
- `automatic1111.yaml` - A1111 parameter format
- `comfyui.yaml` - ComfyUI workflow + prompt format
- `civitai.yaml` - Enhanced A1111 with resource tracking
- `forge.yaml` - Forge extensions and advanced sampling
- `tensor.ai.yaml` - JSON-based metadata
- `leonardo.ai.yaml` - Leonardo.AI preset system
### Supported Platforms:
✅ **Automatic1111 (A1111)** - Full parameter format support
✅ **ComfyUI** - Complete workflow preservation
✅ **Civitai** - Enhanced resource tracking with model IDs
✅ **Forge** - A1111-compatible with extensions
✅ **Tensor.AI** - JSON-based metadata format
✅ **Leonardo.AI** - Style and preset management
🔄 **SeaArt.AI** - Planned
🔄 **Midjourney** - Planned
## Key Terminology & Concepts
### Workflow Types:
- **Simplified Workflow**: Functional workflows generated from any platform metadata that recreate identical images
- **Complex Workflow Preservation**: Full-fidelity preservation of ComfyUI node graphs and custom arrangements
### Metadata Formats:
- **Universal Format**: MetaMan's comprehensive JSON schema covering all platforms
- **Service-Specific Format**: Platform-native metadata (A1111 parameters, ComfyUI workflow JSON, etc.)
- **PNG Chunks**: Text metadata embedded in PNG files (parameters, workflow, meta)
### Dependencies:
- **Model Dependencies**: Checkpoints, LoRAs, embeddings, VAE, ControlNet models
- **Dependency Resolution**: Multi-platform search with confidence scoring
- **Download URLs**: Direct links to model files with source platform info
## Project Structure
```
MetaMan/
├── Core Nodes
│ ├── __init__.py # Package entry point
│ ├── metaman.py # Universal metadata node
│ └── specialized_nodes.py # Workflow saver & dependency resolver
├── Utilities
│ ├── metadata_parser.py # Universal format parser
│ ├── api_integration.py # Model repository APIs
│ └── validation.py # Testing framework
├── Templates
│ ├── universal_schema.yaml # Universal metadata definition
│ └── services/ # Platform-specific templates (6 services)
├── Documentation
│ ├── README.md # Main documentation with badges
│ ├── SETUP.md # Installation & configuration
│ ├── CONTRIBUTING.md # Community contribution guidelines
│ └── CHANGELOG.md # Version history
├── Testing & Project Management
│ ├── test_installation.py # Installation verification
│ ├── validation.py # Template and functionality testing
│ ├── requirements.txt # Python dependencies
│ ├── LICENSE # MIT License
│ └── .gitignore # Git ignore rules
```
## Current Status (August 2025)
- ✅ **Complete codebase** ready for deployment
- ✅ **All user stories implemented** and tested
- ✅ **Documentation complete** with setup guides
- ✅ **Git repository prepared** for public release
- 🚀 **Ready for GitHub deployment** - just needs repo creation and push
## Technical Specifications
### Dependencies:
- Pillow >= 9.0.0 (image processing)
- PyYAML >= 6.0.0 (template processing)
- piexif >= 1.1.3 (EXIF metadata)
- requests >= 2.28.0 (API integration)
- jsonschema >= 4.17.0 (validation)
- torch (provided by ComfyUI)
### Node Interface:
**MetaMan Universal Node Operations:**
- `extract_universal` - Extract to universal format
- `convert_to_service` - Convert to specific service format
- `save_workflow` - Save workflow data
- `generate_dependencies` - Find model dependencies
- `export_metadata` - Export in various formats
**Input/Output Types:**
- Input: IMAGE, operation selection, service selection, optional parameters
- Output: IMAGE, universal_metadata (STRING), service_metadata (STRING), dependencies (STRING)
## Key Research & Terminology
### Platform-Specific Details:
- **A1111 Format**: Text-based parameters in PNG tEXt chunks, Latin-1 encoding
- **ComfyUI Format**: JSON workflow graphs with node connections and execution data
- **Civitai Extensions**: A1111 format + resource tracking with SHA256 hashes and model IDs
- **PNG Chunks**: tEXt (Latin-1), iTXt (UTF-8), custom "meta" chunk proposal
### Model Repository APIs:
- **Civitai API**: REST endpoints for model search, hash matching, download URLs
- **HuggingFace API**: Git-based model repositories with commit hashes
- **Search Priority**: Civitai → HuggingFace → Archives → Other sources
- **Rate Limiting**: Respect platform limits (1-2 seconds between requests)
### Sampler/Scheduler Mappings:
Complex mappings between platform naming conventions (e.g., A1111's "DPM++ 2M Karras" = ComfyUI's "dpmpp_2m" + "karras" scheduler)
## Important Clarifications Made
### Workflow Conversion Capabilities:
- **✅ ANY platform → ComfyUI workflow generation** (creates identical images)
- **✅ ComfyUI → ANY platform parameter conversion** (simplified but equivalent)
- **⚠️ Complex ComfyUI workflows lose node structure** when converted to linear platforms
- **🎯 Image reproduction is identical** across all conversion types
### Compatibility Matrix Meanings:
- **Simplified Workflow** ✅ = Can generate functional workflow that recreates exact image
- **Complex Workflow Preservation** ✅ = Can preserve original node arrangements and custom routing
- **Basic Parameters** ✅ = Standard generation parameters (prompt, steps, seed, etc.)
## File Locations
- **Project**: `/Users/pxl8d/Projects/MetaMan/`
- **ComfyUI Target**: `/Users/pxl8d/Art/ComfyUI/custom_nodes/`
- **Git Status**: Initialized repository ready for remote push
## Next Steps
1. Create GitHub repository (suggested name: `ComfyUI-MetaMan`)
2. Push codebase to public repository
3. Share with ComfyUI community
4. Consider ComfyUI-Manager integration
5. Community feedback and platform additions
## Research Terms & Context Covered
- A1111/Civitai metadata standards and PNG chunk formats
- ComfyUI workflow architecture and node-based systems
- Cross-platform compatibility challenges in AI image generation
- Model dependency tracking and resolution systems
- PNG metadata embedding techniques and encoding standards
- API rate limiting and repository search strategies
- Template-driven software architecture for extensibility
+53 -14
View File
@@ -21,15 +21,26 @@ def test_imports():
from PIL import Image, PngImagePlugin
print(" ✅ Basic dependencies OK")
# Test MetaMan imports
from . import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
print(" ✅ MetaMan node mappings OK")
# Test MetaMan imports (using absolute imports for testing)
import sys
import os
from .metaman import MetaManUniversalNode
from .specialized_nodes import MetaManWorkflowSaver, MetaManDependencyResolver
from .metadata_parser import metadata_parser
from .api_integration import model_resolver
print(" ✅ MetaMan modules OK")
# Add current directory to path for testing
current_dir = os.path.dirname(os.path.abspath(__file__))
if current_dir not in sys.path:
sys.path.insert(0, current_dir)
# Test importing MetaMan modules
import metaman
import specialized_nodes
import metadata_parser
import api_integration
print(" ✅ MetaMan modules imported OK")
# Test node class availability
from metaman import MetaManUniversalNode
from specialized_nodes import MetaManWorkflowSaver, MetaManDependencyResolver
print(" ✅ MetaMan node classes OK")
return True
@@ -98,8 +109,15 @@ def test_node_creation():
print("Testing node creation...")
try:
from .metaman import MetaManUniversalNode
from .specialized_nodes import MetaManWorkflowSaver, MetaManDependencyResolver
# Import with absolute imports
import sys
import os
current_dir = os.path.dirname(os.path.abspath(__file__))
if current_dir not in sys.path:
sys.path.insert(0, current_dir)
from metaman import MetaManUniversalNode
from specialized_nodes import MetaManWorkflowSaver, MetaManDependencyResolver
# Test node instantiation
universal_node = MetaManUniversalNode()
@@ -127,7 +145,14 @@ def test_metadata_parsing():
print("Testing metadata parsing...")
try:
from .metadata_parser import metadata_parser
# Import with absolute imports
import sys
import os
current_dir = os.path.dirname(os.path.abspath(__file__))
if current_dir not in sys.path:
sys.path.insert(0, current_dir)
from metadata_parser import metadata_parser
# Test A1111 format parsing
test_metadata = {
@@ -162,7 +187,14 @@ def test_api_integration():
print("Testing API integration...")
try:
from .api_integration import model_resolver, hash_verifier
# Import with absolute imports
import sys
import os
current_dir = os.path.dirname(os.path.abspath(__file__))
if current_dir not in sys.path:
sys.path.insert(0, current_dir)
from api_integration import model_resolver, hash_verifier
# Test resolver initialization
if hasattr(model_resolver, 'find_model_sources'):
@@ -190,12 +222,19 @@ def run_validation():
print("Running validation...")
try:
from .validation import run_validation
# Import with absolute imports
import sys
import os
current_dir = os.path.dirname(os.path.abspath(__file__))
if current_dir not in sys.path:
sys.path.insert(0, current_dir)
from validation import run_validation as validation_func
base_dir = Path(__file__).parent
templates_dir = str(base_dir / "templates")
report = run_validation(templates_dir)
report = validation_func(templates_dir)
print(" ✅ Validation completed")
print("\nValidation Report:")
print("-" * 50)