feature multi document for SD
This commit is contained in:
@@ -64,7 +64,15 @@ def registerSDHTTPEndpoints(sdppp, app):
|
||||
name = addImageCache(Image.open(BytesIO(image.file.read())))
|
||||
return {'name': name}
|
||||
|
||||
from .sd_data import get_sd_document_data, set_sd_document_data, set_sd_special_get_bound_layer_options, set_sd_special_get_layer_options, set_sd_special_send_layer_options
|
||||
def registerSocketEvents(sdppp, sio):
|
||||
@sio.event
|
||||
async def c_set_sd_options(sid, data={}):
|
||||
set_sd_document_data(data['document_data'])
|
||||
set_sd_special_get_layer_options(data['special_get_layer_options'])
|
||||
set_sd_special_get_bound_layer_options(data['special_get_bound_layer_options'])
|
||||
set_sd_special_send_layer_options(data['special_send_layer_options'])
|
||||
|
||||
# only emit by sd webui instance
|
||||
@sio.event
|
||||
async def c_get_image(sid, data={}):
|
||||
@@ -76,9 +84,6 @@ def registerSocketEvents(sdppp, sio):
|
||||
use_layer_bounds = data['use_layer_bounds']
|
||||
sd_elem_id = data['sd']['elem_id']
|
||||
|
||||
# id = photoshopInstance.layer_name_to_id(layer)
|
||||
# bounds_id = photoshopInstance.layer_name_to_id(use_layer_bounds, id)
|
||||
|
||||
upload_name, opacity = await photoshopInstance.get_image(
|
||||
document_identify=document, layer_identify=layer, bound_layer_identify=use_layer_bounds
|
||||
)
|
||||
|
||||
@@ -48,14 +48,11 @@ def define_comfyui_nodes(sdppp):
|
||||
return np.random.rand()
|
||||
else:
|
||||
photoshopInstance = sdppp.get_ps_instance()
|
||||
print(1)
|
||||
is_changed, history_state_id = call_async_func_in_server_thread(
|
||||
photoshopInstance.check_layer_bounds_combo_changed(layer, use_layer_bounds)
|
||||
)
|
||||
print(2)
|
||||
if is_changed and history_state_id is None:
|
||||
return np.random.rand()
|
||||
print(3)
|
||||
photoshopInstance.update_comfyui_last_value(layer, use_layer_bounds, history_state_id)
|
||||
return history_state_id
|
||||
|
||||
|
||||
@@ -0,0 +1,29 @@
|
||||
|
||||
# need refactor
|
||||
sd_document_data = dict()
|
||||
def set_sd_document_data(data):
|
||||
global sd_document_data
|
||||
sd_document_data = data
|
||||
def get_sd_document_data():
|
||||
return sd_document_data
|
||||
|
||||
sd_special_get_layer_options = []
|
||||
def set_sd_special_get_layer_options(data):
|
||||
global sd_special_get_layer_options
|
||||
sd_special_get_layer_options = data
|
||||
def get_sd_special_get_layer_options():
|
||||
return sd_special_get_layer_options
|
||||
|
||||
sd_special_get_bound_layer_options = []
|
||||
def set_sd_special_get_bound_layer_options(data):
|
||||
global sd_special_get_bound_layer_options
|
||||
sd_special_get_bound_layer_options = data
|
||||
def get_sd_special_get_bound_layer_options():
|
||||
return sd_special_get_bound_layer_options
|
||||
|
||||
sd_special_send_layer_options = []
|
||||
def set_sd_special_send_layer_options(data):
|
||||
global sd_special_send_layer_options
|
||||
sd_special_send_layer_options = data
|
||||
def get_sd_special_send_layer_options():
|
||||
return sd_special_send_layer_options
|
||||
Reference in New Issue
Block a user