Add files via upload

This commit is contained in:
Oleh
2024-05-07 14:02:58 +03:00
committed by GitHub
parent cbb3fb74d1
commit c129afc4aa
3 changed files with 89 additions and 0 deletions
+45
View File
@@ -0,0 +1,45 @@
import folder_paths
import re
class EmbeddingsNameLoader:
def __init__(self):
pass
@classmethod
def INPUT_TYPES(s):
return {"required": { "text": ("STRING", {"multiline": True}),
}}
RETURN_TYPES = ("STRING", )
RETURN_NAMES = ("STRING", )
FUNCTION = "run"
CATEGORY = "Embeddings Tools"
def run(self, text):
emb_files = folder_paths.get_filename_list("embeddings")
output_text = text
for file_name in emb_files:
base_name = file_name.split('\\')[-1].split('.')[0]
#output_text = output_text.replace(base_name, f"embedding:{base_name}")
pattern = re.compile(re.escape(base_name), re.IGNORECASE)
output_text = pattern.sub(f"embedding:{base_name}", output_text)
return (output_text, )
NODE_CLASS_MAPPINGS = {
"EmbeddingsNameLoader": EmbeddingsNameLoader,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"EmbeddingsNameLoader": "Load Embeddings by Name",
}
+32
View File
@@ -0,0 +1,32 @@
import folder_paths
class EmbendingList:
def __init__(self):
pass
@classmethod
def INPUT_TYPES(s):
return {"required": { }}
RETURN_TYPES = ("STRING",)
RETURN_NAMES = ("Embeddings List (STRING)",)
FUNCTION = "run"
CATEGORY = "Embeddings Tools"
def run(self):
emb_files = folder_paths.get_filename_list("embeddings")
emb_files_str = "\n".join(emb_files)
return (emb_files_str, )
NODE_CLASS_MAPPINGS = {
"EmbendingList": EmbendingList,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"EmbendingList": "Embeddings List",
}
+12
View File
@@ -0,0 +1,12 @@
from .EmbeddingsNameLoader import EmbeddingsNameLoader
from .EmbendingList import EmbendingList
NODE_CLASS_MAPPINGS = {
"EmbeddingsNameLoader": EmbeddingsNameLoader,
"EmbendingList": EmbendingList,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"EmbeddingsNameLoader": "Load Embeddings by Name",
"EmbendingList": "Embeddings List",
}