v1.9

Dockerfile

@@ -4,13 +4,21 @@ FROM python:3.10
 # Install git
 RUN apt-get update && apt-get install -y git
 
-
 WORKDIR /home/user/app
 
-
 # Copy all files to the container
 COPY . .
 
+# Clone custom node repositories
+RUN git clone https://github.com/Fannovel16/ComfyUI-YoloWorld-EfficientSAM.git custom_nodes/ComfyUI-YoloWorld-EfficientSAM
+RUN git clone https://github.com/Acly/comfyui-inpaint-nodes.git custom_nodes/comfyui-inpaint-nodes
+RUN git clone https://github.com/rgthree/rgthree-comfy.git custom_nodes/rgthree-comfy
+
+# Install dependencies for custom nodes (if they have requirements.txt)
+RUN if [ -f "custom_nodes/ComfyUI-YoloWorld-EfficientSAM/requirements.txt" ]; then pip install -r custom_nodes/ComfyUI-YoloWorld-EfficientSAM/requirements.txt; fi
+RUN if [ -f "custom_nodes/comfyui-inpaint-nodes/requirements.txt" ]; then pip install -r custom_nodes/comfyui-inpaint-nodes/requirements.txt; fi
+RUN if [ -f "custom_nodes/rgthree-comfy/requirements.txt" ]; then pip install -r custom_nodes/rgthree-comfy/requirements.txt; fi
+
 # Install dependencies
 RUN pip install --no-cache-dir --upgrade pip
 RUN pip install "inference[transformers,sam,clip,gaze,grounding-dino,yolo-world]" xformers torchvision torchaudio supervision onnxruntime onnxruntime-gpu --extra-index-url https://download.pytorch.org/whl/cu124
@@ -21,4 +29,4 @@ RUN pip install aiohttp==3.10.11
 EXPOSE 7860
 
 # Command to run the app
-CMD ["python", "app.py"]
+CMD ["python", "app.py"]

app.py

@@ -181,6 +181,13 @@ def import_custom_nodes() -> None:
     from nodes import init_extra_nodes
     import server
 
+    # Debug: List contents of custom_nodes/
+    print("Custom nodes directory contents:", os.listdir('custom_nodes'))
+    for node_dir in os.listdir('custom_nodes'):
+        node_path = os.path.join('custom_nodes', node_dir)
+        if os.path.isdir(node_path):
+            print(f"Contents of {node_dir}:", os.listdir(node_path))
+
     # Creating a new event loop and setting it as the default loop
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
@@ -292,14 +299,32 @@ def object_remover(object_name, image) :
     return "Error: Image generation failed."
 
 
-# yoloworld_modelloader_zho = NODE_CLASS_MAPPINGS["Yoloworld_ModelLoader_Zho"]()
-if "Yoloworld_ModelLoader_Zho" in NODE_CLASS_MAPPINGS:
-    yoloworld_modelloader_zho = NODE_CLASS_MAPPINGS["Yoloworld_ModelLoader_Zho"]()
-    yoloworld_modelloader_zho_3 = yoloworld_modelloader_zho.load_yolo_world_model(
-        yolo_world_model="yolo_world/m"
-    )
-else:
-    raise KeyError("❌ Error: 'Yoloworld_ModelLoader_Zho' not found. Check if custom nodes are loaded.")
+# Check for all required custom nodes
+required_nodes = [
+    "Yoloworld_ModelLoader_Zho",
+    "ESAM_ModelLoader_Zho",
+    "Yoloworld_ESAM_Zho",
+    "INPAINT_InpaintWithModel",
+    "INPAINT_LoadInpaintModel",
+    "Image Comparer (rgthree)",
+    "GrowMask",
+    "MaskToImage",
+    "VAEEncode",
+    "VAEDecode",
+    "VAELoader",
+    "LoadImage",
+    "SaveImage"
+]
+
+missing_nodes = [node for node in required_nodes if node not in NODE_CLASS_MAPPINGS]
+if missing_nodes:
+    raise KeyError(f"❌ Error: The following required nodes were not found: {missing_nodes}. Check if custom nodes are loaded correctly.")
+
+# Load the nodes
+yoloworld_modelloader_zho = NODE_CLASS_MAPPINGS["Yoloworld_ModelLoader_Zho"]()
+yoloworld_modelloader_zho_3 = yoloworld_modelloader_zho.load_yolo_world_model(
+    yolo_world_model="yolo_world/m"
+)
 
 esam_modelloader_zho = NODE_CLASS_MAPPINGS["ESAM_ModelLoader_Zho"]()
 esam_modelloader_zho_4 = esam_modelloader_zho.load_esam_model(device="CUDA")
@@ -314,12 +339,10 @@ vaeloader_20 = vaeloader.load_vae(
     vae_name="vae-ft-mse-840000-ema-pruned.safetensors"
 )
 
-
 loadimage = NODE_CLASS_MAPPINGS["LoadImage"]()
 yoloworld_esam_zho = NODE_CLASS_MAPPINGS["Yoloworld_ESAM_Zho"]()
 growmask = NODE_CLASS_MAPPINGS["GrowMask"]()
 inpaint_inpaintwithmodel = NODE_CLASS_MAPPINGS["INPAINT_InpaintWithModel"]()
-vaeloader = NODE_CLASS_MAPPINGS["VAELoader"]()
 vaeencode = NODE_CLASS_MAPPINGS["VAEEncode"]()
 masktoimage = NODE_CLASS_MAPPINGS["MaskToImage"]()
 vaedecode = NODE_CLASS_MAPPINGS["VAEDecode"]()
@@ -327,7 +350,6 @@ image_comparer_rgthree = NODE_CLASS_MAPPINGS["Image Comparer (rgthree)"]()
 saveimage = NODE_CLASS_MAPPINGS["SaveImage"]()
 
 
-
 #Add all the models that load a safetensors file
 model_loaders = [yoloworld_modelloader_zho_3,esam_modelloader_zho_4,inpaint_loadinpaintmodel_13,vaeloader_20]