holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix img2img task restoration

Browse Source 
Fix source image, mask, and color profile restoration for use settings, copy/paste, and d&d.
This commit is contained in:
patriceac 2022-12-28 16:43:35 -08:00
parent 451ab7e84c
commit 19c16af5fa
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
ui/media/js/dnd.js
View File

@ -137,7 +137,14 @@ const TASK_MAPPING = {
readUI: () => (maskSetting.checked ? imageInpainter.getImg() : undefined), readUI: () => (maskSetting.checked ? imageInpainter.getImg() : undefined),
parse: (val) => val parse: (val) => val
}, },
preserve_init_image_color_profile: { name: 'Preserve Color Profile',
setUI: (preserve_init_image_color_profile) => {
applyColorCorrectionField.checked = parseBoolean(preserve_init_image_color_profile)
},
readUI: () => applyColorCorrectionField.checked,
parse: (val) => parseBoolean(val)
},
use_face_correction: { name: 'Use Face Correction', use_face_correction: { name: 'Use Face Correction',
setUI: (use_face_correction) => { setUI: (use_face_correction) => {
useFaceCorrectionField.checked = parseBoolean(use_face_correction) useFaceCorrectionField.checked = parseBoolean(use_face_correction)
@ -297,7 +304,11 @@ function restoreTaskToUI(task, fieldsToSkip) {
// restore the original tag // restore the original tag
promptField.value = task.reqBody.original_prompt || task.reqBody.prompt promptField.value = task.reqBody.original_prompt || task.reqBody.prompt
// properly reset checkboxes // properly reset missing fields
if (!('use_hypernetwork_model' in task.reqBody)) {
hypernetworkModelField.value = ""
hypernetworkModelField.dispatchEvent(new Event("change"))
}
if (!('use_face_correction' in task.reqBody)) { if (!('use_face_correction' in task.reqBody)) {
useFaceCorrectionField.checked = false useFaceCorrectionField.checked = false
} }
@ -306,6 +317,7 @@ function restoreTaskToUI(task, fieldsToSkip) {
} }
if (!('mask' in task.reqBody)) { if (!('mask' in task.reqBody)) {
maskSetting.checked = false maskSetting.checked = false
maskSetting.dispatchEvent(new Event("click"))
} }
upscaleModelField.disabled = !useUpscalingField.checked upscaleModelField.disabled = !useUpscalingField.checked
upscaleAmountField.disabled = !useUpscalingField.checked upscaleAmountField.disabled = !useUpscalingField.checked
@ -313,12 +325,17 @@ function restoreTaskToUI(task, fieldsToSkip) {
// Show the source picture if present // Show the source picture if present
initImagePreview.src = (task.reqBody.init_image == undefined ? '' : task.reqBody.init_image) initImagePreview.src = (task.reqBody.init_image == undefined ? '' : task.reqBody.init_image)
if (IMAGE_REGEX.test(initImagePreview.src)) { if (IMAGE_REGEX.test(initImagePreview.src)) {
initImagePreview.dispatchEvent(new Event("load"))
if (Boolean(task.reqBody.mask)) { if (Boolean(task.reqBody.mask)) {
setTimeout(() => { // add a delay to insure this happens AFTER the main image loads (which reloads the inpainter) setTimeout(() => { // add a delay to insure this happens AFTER the main image loads (which reloads the inpainter)
imageInpainter.setImg(task.reqBody.mask) imageInpainter.setImg(task.reqBody.mask)
}, 250) }, 250)
} }
} }
else
{
img2imgUnload()
}
} }
function readUI() { function readUI() {
const reqBody = {} const reqBody = {}