modules/webui/js/index.js

// based on https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/v1.6.0/script.js
function gradioApp() {
  const elems = document.getElementsByTagName("gradio-app");
  const elem = elems.length == 0 ? document : elems[0];

  if (elem !== document) {
    elem.getElementById = function (id) {
      return document.getElementById(id);
    };
  }
  return elem.shadowRoot ? elem.shadowRoot : elem;
}

/**
 * Get the currently selected top-level UI tab button (e.g. the button that says "Extras").
 */
function get_uiCurrentTab() {
  return gradioApp().querySelector("#tabs > .tab-nav > button.selected");
}

/**
 * Get the first currently visible top-level UI tab content (e.g. the div hosting the "txt2img" UI).
 */
function get_uiCurrentTabContent() {
  return gradioApp().querySelector(
    '#tabs > .tabitem[id^=tab_]:not([style*="display: none"])'
  );
}

var uiUpdateCallbacks = [];
var uiAfterUpdateCallbacks = [];
var uiLoadedCallbacks = [];
var uiTabChangeCallbacks = [];
var optionsChangedCallbacks = [];
var uiAfterUpdateTimeout = null;
var uiCurrentTab = null;

/**
 * Register callback to be called at each UI update.
 * The callback receives an array of MutationRecords as an argument.
 */
function onUiUpdate(callback) {
  uiUpdateCallbacks.push(callback);
}

/**
 * Register callback to be called soon after UI updates.
 * The callback receives no arguments.
 *
 * This is preferred over `onUiUpdate` if you don't need
 * access to the MutationRecords, as your function will
 * not be called quite as often.
 */
function onAfterUiUpdate(callback) {
  uiAfterUpdateCallbacks.push(callback);
}

/**
 * Register callback to be called when the UI is loaded.
 * The callback receives no arguments.
 */
function onUiLoaded(callback) {
  uiLoadedCallbacks.push(callback);
}

/**
 * Register callback to be called when the UI tab is changed.
 * The callback receives no arguments.
 */
function onUiTabChange(callback) {
  uiTabChangeCallbacks.push(callback);
}

/**
 * Register callback to be called when the options are changed.
 * The callback receives no arguments.
 * @param callback
 */
function onOptionsChanged(callback) {
  optionsChangedCallbacks.push(callback);
}

function executeCallbacks(queue, arg) {
  for (const callback of queue) {
    try {
      callback(arg);
    } catch (e) {
      console.error("error running callback", callback, ":", e);
    }
  }
}

/**
 * Schedule the execution of the callbacks registered with onAfterUiUpdate.
 * The callbacks are executed after a short while, unless another call to this function
 * is made before that time. IOW, the callbacks are executed only once, even
 * when there are multiple mutations observed.
 */
function scheduleAfterUiUpdateCallbacks() {
  clearTimeout(uiAfterUpdateTimeout);
  uiAfterUpdateTimeout = setTimeout(function () {
    executeCallbacks(uiAfterUpdateCallbacks);
  }, 200);
}

var executedOnLoaded = false;

document.addEventListener("DOMContentLoaded", function () {
  var mutationObserver = new MutationObserver(function (m) {
    if (!executedOnLoaded && gradioApp().querySelector("#generate_button")) {
      executedOnLoaded = true;
      executeCallbacks(uiLoadedCallbacks);
    }

    executeCallbacks(uiUpdateCallbacks, m);
    scheduleAfterUiUpdateCallbacks();
    const newTab = get_uiCurrentTab();
    if (newTab && newTab !== uiCurrentTab) {
      uiCurrentTab = newTab;
      executeCallbacks(uiTabChangeCallbacks);
    }
  });
  mutationObserver.observe(gradioApp(), { childList: true, subtree: true });
});

var onAppend = function (elem, f) {
  var observer = new MutationObserver(function (mutations) {
    mutations.forEach(function (m) {
      if (m.addedNodes.length) {
        f(m.addedNodes);
      }
    });
  });
  observer.observe(elem, { childList: true });
};

function addObserverIfDesiredNodeAvailable(querySelector, callback) {
  var elem = document.querySelector(querySelector);
  if (!elem) {
    window.setTimeout(
      () => addObserverIfDesiredNodeAvailable(querySelector, callback),
      1000
    );
    return;
  }

  onAppend(elem, callback);
}

/**
 * Show reset button on toast "Connection errored out."
 */
addObserverIfDesiredNodeAvailable(".toast-wrap", function (added) {
  added.forEach(function (element) {
    if (element.innerText.includes("Connection errored out.")) {
      window.setTimeout(function () {
        // FIXME: 这几个button好像是没有...a1111里面的gradio版本和我们的不一样
        document.getElementById("reset_button")?.classList.remove("hidden");
        document.getElementById("generate_button")?.classList.add("hidden");
        document.getElementById("skip_button")?.classList.add("hidden");
        document.getElementById("stop_button")?.classList.add("hidden");
      });
    }
  });
});

/**
 * Add a ctrl+enter as a shortcut to start a generation
 */
document.addEventListener("keydown", function (e) {
  const isModifierKey = e.metaKey || e.ctrlKey || e.altKey;
  const isEnterKey = e.key == "Enter" || e.keyCode == 13;

  if (isModifierKey && isEnterKey) {
    const generateButton = gradioApp().querySelector(
      "button:not(.hidden)[id=generate_button]"
    );
    if (generateButton) {
      generateButton.click();
      e.preventDefault();
      return;
    }

    const stopButton = gradioApp().querySelector(
      "button:not(.hidden)[id=stop_button]"
    );
    if (stopButton) {
      stopButton.click();
      e.preventDefault();
      return;
    }
  }
});

/**
 * checks that a UI element is not in another hidden element or tab content
 */
function uiElementIsVisible(el) {
  if (el === document) {
    return true;
  }

  const computedStyle = getComputedStyle(el);
  const isVisible = computedStyle.display !== "none";

  if (!isVisible) return false;
  return uiElementIsVisible(el.parentNode);
}

function uiElementInSight(el) {
  const clRect = el.getBoundingClientRect();
  const windowHeight = window.innerHeight;
  const isOnScreen = clRect.bottom > 0 && clRect.top < windowHeight;

  return isOnScreen;
}

function playNotification() {
  gradioApp().querySelector("#audio_notification audio")?.play();
}

function set_theme(theme) {
  var gradioURL = window.location.href;
  if (!gradioURL.includes("?__theme=")) {
    window.location.replace(gradioURL + "?__theme=" + theme);
  }
}

function htmlDecode(input) {
  var doc = new DOMParser().parseFromString(input, "text/html");
  return doc.documentElement.textContent;
}