extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-12-26 08:58:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

Allow downloading the controlnet preview image

Browse Source
This commit is contained in:
cmdr2 2023-08-18 16:24:17 +05:30
parent 846bb2134e
commit e80db71d1c
2 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ui/media/css/main.css
View File

@ -1439,7 +1439,7 @@ div.top-right {
right: 8px;
}
button.useForControlnetBtn {
.task-fs-initimage .top-right button {
margin-top: 6px;
}

15
ui/media/js/main.js
View File

@ -1362,7 +1362,7 @@ async function onTaskStart(task) {
}
/* Hover effect for the init image in the task list */
function createInitImageHover(taskEntry) {
function createInitImageHover(taskEntry, task) {
taskEntry.querySelectorAll(".task-initimg").forEach((thumb) => {
let thumbimg = thumb.querySelector("img")
let img = createElement("img", { src: thumbimg.src })
@ -1371,7 +1371,9 @@ function createInitImageHover(taskEntry) {
div.innerHTML = `
<button class="useAsInputBtn">Use as Input</button>
<br>
<button class="useForControlnetBtn">Use for Controlnet</button>`
<button class="useForControlnetBtn">Use for Controlnet</button>
<br>
<button class="downloadPreviewImg">Download</button>`
div.querySelector(".useAsInputBtn").addEventListener("click", (e) => {
e.preventDefault()
onUseAsInputClick(null, img)
@ -1380,6 +1382,13 @@ function createInitImageHover(taskEntry) {
e.preventDefault()
controlImagePreview.src = img.src
})
div.querySelector(".downloadPreviewImg").addEventListener("click", (e) => {
e.preventDefault()
const name = "image." + task.reqBody["output_format"]
const blob = dataURItoBlob(img.src)
saveAs(blob, name)
})
thumb.querySelector(".task-fs-initimage").appendChild(div)
})
return
@ -1474,7 +1483,7 @@ function createTask(task) {
</div>`
if (task.reqBody.init_image !== undefined || task.reqBody.control_image !== undefined) {
createInitImageHover(taskEntry)
createInitImageHover(taskEntry, task)
}
if (task.reqBody.control_image !== undefined && task.reqBody.control_filter_to_apply !== undefined) {