diff --git a/index.html b/index.html index c6506d3..f1c7e36 100644 --- a/index.html +++ b/index.html @@ -354,7 +354,7 @@ src="js/ui/tool/generic.js?v=2bcd36d" type="text/javascript"> - + diff --git a/js/ui/tool/dream.js b/js/ui/tool/dream.js index b01cfb5..8d04257 100644 --- a/js/ui/tool/dream.js +++ b/js/ui/tool/dream.js @@ -835,16 +835,21 @@ const dream_generate_callback = async (bb, resolution, state) => { bbCanvas.height = bb.h; const bbCtx = bbCanvas.getContext("2d"); - const reqCanvas = document.createElement("canvas"); - reqCanvas.width = request.width; - reqCanvas.height = request.height; - const reqCtx = reqCanvas.getContext("2d"); + const maskCanvas = document.createElement("canvas"); + maskCanvas.width = request.width; + maskCanvas.height = request.height; + const maskCtx = maskCanvas.getContext("2d"); + + const initCanvas = document.createElement("canvas"); + initCanvas.width = request.width; + initCanvas.height = request.height; + const initCtx = initCanvas.getContext("2d"); bbCtx.fillStyle = "#000F"; // Get init image - reqCtx.fillRect(0, 0, request.width, request.height); - reqCtx.drawImage( + initCtx.fillRect(0, 0, request.width, request.height); + initCtx.drawImage( visibleCanvas, 0, 0, @@ -855,7 +860,7 @@ const dream_generate_callback = async (bb, resolution, state) => { request.width, request.height ); - request.init_images = [reqCanvas.toDataURL()]; + request.init_images = [initCanvas.toDataURL()]; // Get mask image bbCtx.fillStyle = "#000F"; @@ -909,8 +914,8 @@ const dream_generate_callback = async (bb, resolution, state) => { bbCtx.fillStyle = "#FFFF"; bbCtx.fillRect(0, 0, bb.w, bb.h); - reqCtx.clearRect(0, 0, reqCanvas.width, reqCanvas.height); - reqCtx.drawImage( + maskCtx.clearRect(0, 0, maskCanvas.width, maskCanvas.height); + maskCtx.drawImage( bbCanvas, 0, 0, @@ -921,7 +926,7 @@ const dream_generate_callback = async (bb, resolution, state) => { request.width, request.height ); - request.mask = reqCanvas.toDataURL(); + request.mask = maskCanvas.toDataURL(); // Dream _generate("img2img", request, bb, {