Fix web directory.

Comfy-Org · Dec 28, 2024 · f199d0e · f199d0e
1 parent 4ff5458
commit f199d0e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/hooks/post_gen_project.py b/hooks/post_gen_project.py
@@ -28,7 +28,7 @@ def init_git():
         Pth("LICENSE").unlink()
 
     if "{{cookiecutter.__include_web}}" == "False":
-        shutil.rmtree("src/{{cookiecutter.project_slug}}/web")
+        shutil.rmtree("web")
 
     if init_git():
         print("✓ Git repository initialized and remote configured")

diff --git a/{{cookiecutter.project_name}}/__init__.py b/{{cookiecutter.project_name}}/__init__.py
@@ -12,8 +12,8 @@
 __email__ = "{{cookiecutter.email}}"
 __version__ = "{{cookiecutter.version}}"
 
-from .src.nodes import NODE_CLASS_MAPPINGS
-from .src.nodes import NODE_DISPLAY_NAME_MAPPINGS
+from .src.{{cookiecutter.project_slug}}.nodes import NODE_CLASS_MAPPINGS
+from .src.{{cookiecutter.project_slug}}.nodes import NODE_DISPLAY_NAME_MAPPINGS
 
 {% if cookiecutter.__include_web -%}
 WEB_DIRECTORY = "./web"

diff --git a/{{cookiecutter.project_name}}/src/{{cookiecutter.project_slug}}/nodes.py b/{{cookiecutter.project_name}}/src/{{cookiecutter.project_slug}}/nodes.py
@@ -103,8 +103,6 @@ def test(self, image, string_field, int_field, float_field, print_to_screen):
     #def IS_CHANGED(s, image, string_field, int_field, float_field, print_to_screen):
     #    return ""
 
-# Set the web directory, any .js file in that directory will be loaded by the frontend as a frontend extension
-# WEB_DIRECTORY = "./somejs"
 
 # A dictionary that contains all nodes you want to export with their names
 # NOTE: names should be globally unique