From 067b92c7cfd092bf0d5c8f8d73a295735b921208 Mon Sep 17 00:00:00 2001 From: AIrjen Date: Mon, 7 Aug 2023 21:32:58 +0200 Subject: [PATCH] Try and fix it --- OneButtonPromptNodes.py | 6 ++---- __init__.py | 8 ++++++++ 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/OneButtonPromptNodes.py b/OneButtonPromptNodes.py index c0fe8a5..c1763bd 100644 --- a/OneButtonPromptNodes.py +++ b/OneButtonPromptNodes.py @@ -4,10 +4,8 @@ import folder_paths from datetime import datetime import uuid -# adding Folder_2/subfolder to the system path -# sys.path.insert(0, '/home/amninder/Desktop/project/Folder_2/subfolder') -custom_nodes_path = os.path.dirname(__file__) -onebuttonprompt_path = os.path.join(custom_nodes_path, "..", "onebuttonprompt") +custom_nodes_path = os.path.join(folder_paths.base_path, "custom_nodes") +onebuttonprompt_path = os.path.join(custom_nodes_path, "onebuttonprompt") sys.path.insert(0, onebuttonprompt_path) from build_dynamic_prompt import * diff --git a/__init__.py b/__init__.py index 206c4c1..ab17cd6 100644 --- a/__init__.py +++ b/__init__.py @@ -1,3 +1,11 @@ +import os +import sys +import folder_paths + +custom_nodes_path = os.path.join(folder_paths.base_path, "custom_nodes") +onebuttonprompt_path = os.path.join(custom_nodes_path, "onebuttonprompt") +sys.path.insert(0, onebuttonprompt_path) + from .OneButtonPromptNodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] \ No newline at end of file