v2

Dockerfile

@@ -9,15 +9,56 @@ 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
+# Clone custom node repositories with error handling and debug output
+RUN echo "Cloning ComfyUI-YoloWorld-EfficientSAM..." && \
+    git clone https://github.com/Fannovel16/ComfyUI-YoloWorld-EfficientSAM.git custom_nodes/ComfyUI-YoloWorld-EfficientSAM || \
+    echo "Failed to clone ComfyUI-YoloWorld-EfficientSAM"
+
+RUN echo "Cloning comfyui-inpaint-nodes..." && \
+    git clone https://github.com/Acly/comfyui-inpaint-nodes.git custom_nodes/comfyui-inpaint-nodes || \
+    echo "Failed to clone comfyui-inpaint-nodes"
+
+RUN echo "Cloning rgthree-comfy..." && \
+    git clone https://github.com/rgthree/rgthree-comfy.git custom_nodes/rgthree-comfy || \
+    echo "Failed to clone rgthree-comfy"
+
+# List contents of custom_nodes/ for debugging
+RUN echo "Contents of custom_nodes/:" && \
+    ls -la custom_nodes/ && \
+    echo "Contents of ComfyUI-YoloWorld-EfficientSAM:" && \
+    ls -la custom_nodes/ComfyUI-YoloWorld-EfficientSAM || echo "ComfyUI-YoloWorld-EfficientSAM not found" && \
+    echo "Contents of comfyui-inpaint-nodes:" && \
+    ls -la custom_nodes/comfyui-inpaint-nodes || echo "comfyui-inpaint-nodes not found" && \
+    echo "Contents of rgthree-comfy:" && \
+    ls -la custom_nodes/rgthree-comfy || echo "rgthree-comfy not found"
 
 # 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
+RUN if [ -f "custom_nodes/ComfyUI-YoloWorld-EfficientSAM/requirements.txt" ]; then \
+        echo "Installing dependencies for ComfyUI-YoloWorld-EfficientSAM..." && \
+        pip install -r custom_nodes/ComfyUI-YoloWorld-EfficientSAM/requirements.txt || \
+        echo "Failed to install dependencies for ComfyUI-YoloWorld-EfficientSAM"; \
+    else \
+        echo "No requirements.txt found for ComfyUI-YoloWorld-EfficientSAM"; \
+    fi
+
+RUN if [ -f "custom_nodes/comfyui-inpaint-nodes/requirements.txt" ]; then \
+        echo "Installing dependencies for comfyui-inpaint-nodes..." && \
+        pip install -r custom_nodes/comfyui-inpaint-nodes/requirements.txt || \
+        echo "Failed to install dependencies for comfyui-inpaint-nodes"; \
+    else \
+        echo "No requirements.txt found for comfyui-inpaint-nodes"; \
+    fi
+
+RUN if [ -f "custom_nodes/rgthree-comfy/requirements.txt" ]; then \
+        echo "Installing dependencies for rgthree-comfy..." && \
+        pip install -r custom_nodes/rgthree-comfy/requirements.txt || \
+        echo "Failed to install dependencies for rgthree-comfy"; \
+    else \
+        echo "No requirements.txt found for rgthree-comfy"; \
+    fi
+
+# Install additional dependencies for ComfyUI-YoloWorld-EfficientSAM
+RUN pip install efficient-sam
 
 # Install dependencies
 RUN pip install --no-cache-dir --upgrade pip

app.py

@@ -180,6 +180,8 @@ def import_custom_nodes() -> None:
     import execution
     from nodes import init_extra_nodes
     import server
+    import importlib.util
+    import traceback
 
     # Debug: List contents of custom_nodes/
     print("Custom nodes directory contents:", os.listdir('custom_nodes'))
@@ -187,6 +189,24 @@ def import_custom_nodes() -> None:
         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))
+            # Check if __init__.py exists
+            init_path = os.path.join(node_path, '__init__.py')
+            print(f"__init__.py exists in {node_dir}: {os.path.exists(init_path)}")
+            if os.path.exists(init_path):
+                # Attempt to import the module
+                try:
+                    module_name = f"custom_nodes.{node_dir}"
+                    spec = importlib.util.spec_from_file_location(module_name, init_path)
+                    if spec is None:
+                        print(f"Failed to create spec for {module_name}")
+                        continue
+                    module = importlib.util.module_from_spec(spec)
+                    sys.modules[module_name] = module
+                    spec.loader.exec_module(module)
+                    print(f"Successfully imported {module_name}")
+                except Exception as e:
+                    print(f"Failed to import {node_dir}: {str(e)}")
+                    traceback.print_exc()
 
     # Creating a new event loop and setting it as the default loop
     loop = asyncio.new_event_loop()
@@ -200,7 +220,11 @@ def import_custom_nodes() -> None:
     execution.PromptQueue(server_instance)
 
     # Initializing custom nodes
-    init_extra_nodes()
+    try:
+        init_extra_nodes()
+    except Exception as e:
+        print(f"Error in init_extra_nodes(): {str(e)}")
+        traceback.print_exc()
 
     print("Custom Nodes Loaded:", NODE_CLASS_MAPPINGS.keys())