diff options
Diffstat (limited to 'javascript')
-rw-r--r-- | javascript/hints.js | 2 | ||||
-rw-r--r-- | javascript/ui.js | 27 |
2 files changed, 28 insertions, 1 deletions
diff --git a/javascript/hints.js b/javascript/hints.js index 23d85710..1a3130f8 100644 --- a/javascript/hints.js +++ b/javascript/hints.js @@ -22,7 +22,7 @@ titles = { "\u{1f4cb}": "Apply selected styles to current prompt", "\u{1f4d2}": "Paste available values into the field", "\u{1f3b4}": "Show/hide extra networks", - + "\u{1F300}": "Restore progress", "Inpaint a part of image": "Draw a mask over an image, and the script will regenerate the masked area with content according to prompt", "SD upscale": "Upscale image normally, split result into tiles, improve each tile using img2img, merge whole image back", diff --git a/javascript/ui.js b/javascript/ui.js index 0ba92ef8..e50b44ee 100644 --- a/javascript/ui.js +++ b/javascript/ui.js @@ -362,6 +362,32 @@ function selectCheckpoint(name){ gradioApp().getElementById('change_checkpoint').click() } +function restoreProgress (task_tag) { + + if (task_tag) { + let successHandler = ({ current_task }) => { + if (current_task) { + showSubmitButtons(task_tag, false) + requestProgress(current_task, gradioApp().getElementById(`${task_tag}_gallery_container`), gradioApp().getElementById(`${task_tag}_gallery`), function(){ + showSubmitButtons(task_tag, true) + }) + } + } + + let errorHandler = e => window.alert(`invalid internal api respsonse. message: ${e}`) + + fetch("./internal/current_task") + .then(res => res.json()) + .then(successHandler) + .catch(errorHandler) + } + + var res = create_submit_args(arguments) + res[0] = 0 + return res + +} + function currentImg2imgSourceResolution(_, _, scaleBy){ var img = gradioApp().querySelector('#mode_img2img > div[style="display: block;"] img') return img ? [img.naturalWidth, img.naturalHeight, scaleBy] : [0, 0, scaleBy] @@ -377,3 +403,4 @@ function updateImg2imgResizeToTextAfterChangingImage(){ return [] } + |