From 36699396bf766c3adae00bdd931a59e56285747c Mon Sep 17 00:00:00 2001
From: spacepxl <spacetug93@gmail.com>
Date: Thu, 1 Feb 2024 23:47:49 -0500
Subject: [PATCH] catch opencv import error to fail gracefully without
 opencv-contrib

---
 import_error_install.bat | 4 ++--
 nodes.py                 | 6 +++++-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/import_error_install.bat b/import_error_install.bat
index 983a147..8790465 100644
--- a/import_error_install.bat
+++ b/import_error_install.bat
@@ -7,13 +7,13 @@ echo installing requirements...
 
 if exist "%python_exec%" (
     echo Installing with ComfyUI Portable
-	%python_exec% -s -m pip uninstall opencv-python opencv-contrib-python opencv-python-headless opencv-contrib-python-headless
+	%python_exec% -s -m pip uninstall -y opencv-python opencv-contrib-python opencv-python-headless opencv-contrib-python-headless
     for /f "delims=" %%i in (%requirements_txt%) do (
         %python_exec% -s -m pip install "%%i"
     )
 ) else (
     echo Installing with system Python
-	pip uninstall opencv-python opencv-contrib-python opencv-python-headless opencv-contrib-python-headless
+	pip uninstall -y opencv-python opencv-contrib-python opencv-python-headless opencv-contrib-python-headless
     for /f "delims=" %%i in (%requirements_txt%) do (
         pip install "%%i"
     )
diff --git a/nodes.py b/nodes.py
index 653e0e8..6fccc6d 100644
--- a/nodes.py
+++ b/nodes.py
@@ -5,9 +5,13 @@ import torch
 import torchvision.transforms
 import numpy as np
 import cv2
-from cv2.ximgproc import guidedFilter
 from pymatting import *
 
+try:
+    from cv2.ximgproc import guidedFilter
+except ImportError:
+    print("\033[33mUnable to import guidedFilter, make sure you have only opencv-contrib-python or run the import_error_install.bat script\033[m")
+
 
 class AlphaClean:
     def __init__(self):