gero
e95a71a900
When GrabCut nodes fail to import (e.g., missing ultralytics dependency), the entire custom node package was failing to load. This meant users couldn't access the main TransparencyBackgroundRemover nodes at all. Changes: - Wrap grabcut_nodes import in try-except block - Provide clear error message when GrabCut nodes are disabled - Allow main TransparencyBackgroundRemover nodes to load independently - Add GRABCUT_AVAILABLE flag for runtime checks Behavior: - Main nodes always available (TransparencyBackgroundRemover, TransparencyBackgroundRemoverBatch) - GrabCut nodes optional (only load if dependencies available) - Clear messaging about what's enabled/disabled and how to fix
45 lines
2.0 KiB
Python
45 lines
2.0 KiB
Python
# Check for optional dependencies and provide helpful warnings
|
||
try:
|
||
import sklearn
|
||
SKLEARN_AVAILABLE = True
|
||
except ImportError:
|
||
SKLEARN_AVAILABLE = False
|
||
print("\n" + "="*70)
|
||
print("⚠️ TransparencyBackgroundRemover: scikit-learn not installed!")
|
||
print("="*70)
|
||
print("The node will work with reduced accuracy (color clustering disabled).")
|
||
print("\nTo install dependencies for best results, run ONE of:")
|
||
print(" • Windows (ComfyUI App): install.bat")
|
||
print(" • Linux/Mac: ./install.sh")
|
||
print(" • Manual: pip install scikit-learn")
|
||
print("\nInstallation files are in: custom_nodes/ComfyUI-TransparencyBackgroundRemover/")
|
||
print("="*70 + "\n")
|
||
|
||
# Import main transparency background remover nodes
|
||
from .nodes import NODE_CLASS_MAPPINGS as NODES_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as NODES_DISPLAY_MAPPINGS
|
||
|
||
# Try to import GrabCut nodes (optional - requires ultralytics)
|
||
try:
|
||
from .grabcut_nodes import NODE_CLASS_MAPPINGS as GRABCUT_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as GRABCUT_DISPLAY_MAPPINGS
|
||
GRABCUT_AVAILABLE = True
|
||
except ImportError as e:
|
||
# GrabCut nodes require additional dependencies (ultralytics, etc.)
|
||
GRABCUT_MAPPINGS = {}
|
||
GRABCUT_DISPLAY_MAPPINGS = {}
|
||
GRABCUT_AVAILABLE = False
|
||
print("\n" + "="*70)
|
||
print("ℹ️ TransparencyBackgroundRemover: GrabCut nodes disabled")
|
||
print("="*70)
|
||
print(f"Reason: {str(e)}")
|
||
print("\nGrabCut nodes require additional dependencies. To enable them:")
|
||
print(" • Install: pip install ultralytics")
|
||
print(" • Or run: pip install -r requirements.txt")
|
||
print("\nMain background remover nodes are still available!")
|
||
print("="*70 + "\n")
|
||
|
||
# Combine node mappings from both modules
|
||
NODE_CLASS_MAPPINGS = {**NODES_MAPPINGS, **GRABCUT_MAPPINGS}
|
||
NODE_DISPLAY_NAME_MAPPINGS = {**NODES_DISPLAY_MAPPINGS, **GRABCUT_DISPLAY_MAPPINGS}
|
||
|
||
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|