diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-04-29 22:13:48 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-04-29 22:13:48 +0300 |
| commit | 80987c36f9bfa33092ac7c75624b25d839cb2a06 (patch) | |
| tree | 6f1acaae50a49544a08384b52739df3e2d803a10 /javascript | |
| parent | 2e78e65a22bfa6b116ae18d12fdcb85ec8acd727 (diff) | |
| parent | 15c4e78b44e14d70b82dcb5b563251fed7be0766 (diff) | |
Merge pull request #7595 from siutin/feature/restore-progress
restore the progress from session lost / tab reload
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 [] } + |