61 lines
1.2 KiB
Python
61 lines
1.2 KiB
Python
from . import loader, loras
|
|
from copy import deepcopy
|
|
import logging
|
|
import json
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class DarkStringToJSON(object):
|
|
"""Convert string input data to JSON."""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"string_in": (
|
|
"STRING",
|
|
{"ForceInput": True},
|
|
),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("JSON",)
|
|
RETURN_NAMES = ("json_out",)
|
|
FUNCTION = "action"
|
|
|
|
CATEGORY = "DarkPrompt"
|
|
|
|
def action(self, string_in):
|
|
return (json.loads(string_in),)
|
|
|
|
|
|
class DarkTextToJSON(object):
|
|
"""Convert text input to JSON."""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"text_in": (
|
|
"TEXT",
|
|
{"ForceInput": True},
|
|
),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("JSON",)
|
|
RETURN_NAMES = ("json_out",)
|
|
FUNCTION = "action"
|
|
|
|
CATEGORY = "DarkPrompt"
|
|
|
|
def action(self, text_in):
|
|
return (json.loads(text_in),)
|