feature multi document for SD

This commit is contained in:
zombieyang
2024-07-24 19:37:27 +08:00
parent bbb2541bca
commit 85a44c8dc7
8 changed files with 131 additions and 43 deletions
+8 -3
View File
@@ -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
)
-3
View File
@@ -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
+29
View File
@@ -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