diff --git a/js/index.js b/js/index.js index fb874af..94b9493 100644 --- a/js/index.js +++ b/js/index.js @@ -24,11 +24,16 @@ app.registerExtension({ Object.defineProperty(loraNameWidget, 'value', { get() { - const ret = originalDescriptor.get.call(loraNameWidget); + const ret = originalDescriptor.get?.call(loraNameWidget) || originalDescriptor.value || ''; return ret }, set(value) { - originalDescriptor.set.call(loraNameWidget, value); + if (originalDescriptor.set) { + originalDescriptor.set.call(loraNameWidget, value); + } else { + originalDescriptor.value = value; + } + const body = new FormData(); body.append('lora_name',value); api diff --git a/lora_info.py b/lora_info.py index 3d38f4b..95b61b8 100644 --- a/lora_info.py +++ b/lora_info.py @@ -5,6 +5,9 @@ import json import server from server import PromptServer from aiohttp import web +import os + +db_path = os.path.join(os.path.curdir, 'custom_nodes', 'lora-info', 'db.json') def load_json_from_file(file_path): try: @@ -43,7 +46,7 @@ def calculate_sha256(file_path): return sha256_hash.hexdigest() def get_lora_info(lora_name): - db = load_json_from_file('./custom_nodes/lora-info/db.json') + db = load_json_from_file(db_path) output = None examplePrompt = None trainedWords = None @@ -96,7 +99,7 @@ def get_lora_info(lora_name): "examplePrompt": examplePrompt, "baseModel": baseModel } - save_dict_to_json(db, './custom_nodes/lora-info/db.json') + save_dict_to_json(db, db_path) return (output, trainedWords, examplePrompt, baseModel)