forked from extern/easydiffusion
ef90832aea
* New engine.js first draft. * Small fixes... * Bump version for cache... * Improved cancellation code. * Cleaning * Wrong argument used in Task.waitUntil * session_id needs to always match SD.sessionId * Removed passing explicit Session ID from UI. Use SD.sessionID to replace. * Cleaning... Removed a disabled line and a hardcoded value. * Fix return if tasks are still waiting. * Added checkbox to reverse processing order. * Fixed progress not displaying properly. * Renamed reverse label. * Only hide progress bar inside onCompleted. * Thanks to rbertus2000 for helping testing and debugging! * Resolve async promises when used optionally. * when removed var should have used let, not const. * Renamed getTaskErrorHandler to onTaskErrorHandler to better reflect actual implementation. * Switched to the unsafer and less git friendly end of lines comma as requested in review. * Raised SERVER_STATE_VALIDITY_DURATION to 90 seconds to match the changes to Beta. * Added logging. * Added one more hook before those inside the SD engine. * Added selftest.plugin.js as part of core. * Removed a tests that wasn't yet implemented... * Groupped task stopping and abort in single function. * Added optional test for plugins. * Allow prompt text to be selected. * Added comment. * Improved isServerAvailable for better mobile usage and added comments for easier debugging. * Comments... * Normalized EVENT_STATUS_CHANGED to follow the same pattern as the other events. * Disable plugins if editorModifierTagsList is not defined. * Adds a new ServiceContainer to register IOC handlers. * Added expect test for a missing dependency in a ServiceContainer * Moved all event code in it's own sub class for easier reuse. * Removed forgotten unused var... * Allow getPrompts to be reused be plugins. * Renamed EventSource to GenericEventSource to avoid redefining an existing class name. * Added missing time argument to debounce * Added output_quality to engine.js * output_quality need to be an int. * Fixed typo. * Replaced the default euler_a by dpm2 to work with both SD1.# and SD2 * Remove generic completed tasks from plugins on generator complete. * dpm2 starts at step 2, replaced with plms to start at step 1. * Merge error * Merge error * changelog Co-authored-by: Marc-Andre Ferland <madrang@gmail.com>
73 lines
2.5 KiB
JavaScript
73 lines
2.5 KiB
JavaScript
const PLUGIN_API_VERSION = "1.0"
|
|
|
|
const PLUGINS = {
|
|
/**
|
|
* Register new buttons to show on each output image.
|
|
*
|
|
* Example:
|
|
* PLUGINS['IMAGE_INFO_BUTTONS'].push({
|
|
* text: 'Make a Similar Image',
|
|
* on_click: function(origRequest, image) {
|
|
* let newTaskRequest = getCurrentUserRequest()
|
|
* newTaskRequest.reqBody = Object.assign({}, origRequest, {
|
|
* init_image: image.src,
|
|
* prompt_strength: 0.7,
|
|
* seed: Math.floor(Math.random() * 10000000)
|
|
* })
|
|
* newTaskRequest.seed = newTaskRequest.reqBody.seed
|
|
* createTask(newTaskRequest)
|
|
* },
|
|
* filter: function(origRequest, image) {
|
|
* // this is an optional function. return true/false to show/hide the button
|
|
* // if this function isn't set, the button will always be visible
|
|
* return true
|
|
* }
|
|
* })
|
|
*/
|
|
IMAGE_INFO_BUTTONS: [],
|
|
MODIFIERS_LOAD: [],
|
|
TASK_CREATE: [],
|
|
OUTPUTS_FORMATS: new ServiceContainer(
|
|
function png() { return (reqBody) => new SD.RenderTask(reqBody) }
|
|
, function jpeg() { return (reqBody) => new SD.RenderTask(reqBody) }
|
|
),
|
|
}
|
|
PLUGINS.OUTPUTS_FORMATS.register = function(...args) {
|
|
const service = ServiceContainer.prototype.register.apply(this, args)
|
|
if (typeof outputFormatField !== 'undefined') {
|
|
const newOption = document.createElement("option")
|
|
newOption.setAttribute("value", service.name)
|
|
newOption.innerText = service.name
|
|
outputFormatField.appendChild(newOption)
|
|
}
|
|
return service
|
|
}
|
|
|
|
function loadScript(url) {
|
|
const script = document.createElement('script')
|
|
const promiseSrc = new PromiseSource()
|
|
script.addEventListener('error', () => promiseSrc.reject(new Error(`Script "${url}" couldn't be loaded.`)))
|
|
script.addEventListener('load', () => promiseSrc.resolve(url))
|
|
script.src = url + '?t=' + Date.now()
|
|
|
|
console.log('loading script', url)
|
|
document.head.appendChild(script)
|
|
|
|
return promiseSrc.promise
|
|
}
|
|
|
|
async function loadUIPlugins() {
|
|
try {
|
|
const res = await fetch('/get/ui_plugins')
|
|
if (!res.ok) {
|
|
console.error(`Error HTTP${res.status} while loading plugins list. - ${res.statusText}`)
|
|
return
|
|
}
|
|
const plugins = await res.json()
|
|
const loadingPromises = plugins.map(loadScript)
|
|
return await Promise.allSettled(loadingPromises)
|
|
} catch (e) {
|
|
console.log('error fetching plugin paths', e)
|
|
}
|
|
}
|