Update app.py

app.py

@@ -7,6 +7,8 @@ import torch
 import gradio as gr
 from huggingface_hub import hf_hub_download
 from comfy import model_management
+import comfy_extras.nodes_model_advanced
+import comfy_extras.nodes_custom_sampler
 from PIL import Image
 
 # --- Helper Functions from original script ---
@@ -31,20 +33,24 @@ def find_path(name: str, path: str = None) -> str:
 
 def add_comfyui_directory_to_sys_path() -> None:
     comfyui_path = os.getcwd()
-    if comfyui_path is not None and os.path.isdir(comfyui_path):
-        sys.path.append(comfyui_path)
-        print(f"'{comfyui_path}' added to sys.path")
+    if "main.py" in os.listdir(comfyui_path):
+        if comfyui_path not in sys.path:
+            sys.path.append(comfyui_path)
+            print(f"'{comfyui_path}' added to sys.path")
 
 def add_extra_model_paths() -> None:
     try:
         from main import load_extra_path_config
-    except ImportError:
-        from utils.extra_config import load_extra_path_config
+    except (ImportError, ModuleNotFoundError):
+        print("Could not import from main.py, trying utils...")
+        try:
+            from utils.extra_config import load_extra_path_config
+        except (ImportError, ModuleNotFoundError):
+            print("Could not find load_extra_path_config function.")
+            return
     extra_model_paths = find_path("extra_model_paths.yaml")
-    if extra_model_paths is not None:
+    if extra_model_paths:
         load_extra_path_config(extra_model_paths)
-    else:
-        print("Could not find the extra_model_paths config file.")
 
 def import_custom_nodes() -> None:
     import asyncio
@@ -62,6 +68,7 @@ def import_custom_nodes() -> None:
 add_comfyui_directory_to_sys_path()
 add_extra_model_paths()
 import_custom_nodes()
+            
 from nodes import NODE_CLASS_MAPPINGS
 
 print("Downlading models from Hugging Face Hub...")