slahiriandClaude Opus 4.5
9ce2d13bb4
Major version bump for Z-Image node and multi-provider support release. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
"""
|
|
ComfyUI-AI-Photography-Toolkit
|
|
A collection of AI-powered photography and image generation tools for ComfyUI.
|
|
All nodes are prefixed with SID_ for easy identification.
|
|
|
|
Author: Siddhartha Lahiri
|
|
Version: 4.0.0
|
|
"""
|
|
|
|
__version__ = "4.0.0"
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import importlib.util
|
|
|
|
# Auto-install dependencies
|
|
def install_dependencies():
|
|
"""
|
|
Automatically install required dependencies if not already installed.
|
|
"""
|
|
dependencies = {
|
|
"anthropic": "anthropic>=0.39.0",
|
|
"yaml": "pyyaml>=6.0",
|
|
}
|
|
|
|
python_exe = sys.executable
|
|
|
|
for package_name, package_spec in dependencies.items():
|
|
# Check if package is already installed
|
|
if importlib.util.find_spec(package_name) is None:
|
|
print(f"\n{'='*60}")
|
|
print(f"Installing {package_name} for SID Photography Toolkit...")
|
|
print(f"{'='*60}")
|
|
try:
|
|
subprocess.check_call(
|
|
[python_exe, "-m", "pip", "install", package_spec],
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE
|
|
)
|
|
print(f"{package_name} installed successfully!")
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"ERROR: Failed to install {package_name}")
|
|
print(f"Please manually install: pip install {package_spec}")
|
|
print(f"Error: {e}")
|
|
print(f"{'='*60}\n")
|
|
|
|
# Install dependencies on module load
|
|
install_dependencies()
|
|
|
|
from typing_extensions import override
|
|
from comfy_api.latest import ComfyExtension, io
|
|
|
|
# Import all node classes
|
|
from .sid_zimage_prompt_generator import SID_ZImagePromptGenerator
|
|
|
|
|
|
class SIDPhotographyToolkitExtension(ComfyExtension):
|
|
"""
|
|
Extension class for SID Photography Toolkit.
|
|
Registers all SID_ prefixed nodes with ComfyUI.
|
|
"""
|
|
|
|
@override
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
"""
|
|
Return list of all nodes in this extension.
|
|
Add new nodes here as they are created.
|
|
"""
|
|
return [
|
|
SID_ZImagePromptGenerator,
|
|
# Future nodes will be added here
|
|
# SID_ZImagePromptEnhancer,
|
|
]
|
|
|
|
|
|
async def comfy_entrypoint() -> SIDPhotographyToolkitExtension:
|
|
"""
|
|
ComfyUI calls this function to load the extension and its nodes.
|
|
This is the entry point for the custom node package.
|
|
"""
|
|
print("\n" + "="*60)
|
|
print(f"Loading SID Photography Toolkit v{__version__} for ComfyUI")
|
|
print("="*60)
|
|
print("Nodes loaded:")
|
|
print(" - SID_ZImagePromptGenerator: Z-Image compatible narrative prompts")
|
|
print("="*60 + "\n")
|
|
|
|
return SIDPhotographyToolkitExtension()
|