diff --git a/__init__.py b/__init__.py index d4eae97..c6ab693 100644 --- a/__init__.py +++ b/__init__.py @@ -10,8 +10,10 @@ comfy_path = os.path.dirname(folder_paths.__file__) tk_nodes_path = os.path.join(os.path.dirname(__file__)) js_dest_path = os.path.join(comfy_path, "web", "extensions", "SDPromptReader") -if not os.path.exists(js_dest_path): - os.makedirs(js_dest_path) +os.makedirs(js_dest_path, exist_ok=True) -js_src_path = os.path.join(tk_nodes_path, "js", "promptDisplay.js") -shutil.copy(js_src_path, js_dest_path) +files_to_copy = ["promptDisplay.js", "parameterDisplay.js", "seedGen.js"] + +for file in files_to_copy: + js_src_path = os.path.join(tk_nodes_path, "js", file) + shutil.copy(js_src_path, js_dest_path)