extraNetworks.js 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712
  1. function toggleCss(key, css, enable) {
  2. var style = document.getElementById(key);
  3. if (enable && !style) {
  4. style = document.createElement('style');
  5. style.id = key;
  6. style.type = 'text/css';
  7. document.head.appendChild(style);
  8. }
  9. if (style && !enable) {
  10. document.head.removeChild(style);
  11. }
  12. if (style) {
  13. style.innerHTML == '';
  14. style.appendChild(document.createTextNode(css));
  15. }
  16. }
  17. function setupExtraNetworksForTab(tabname) {
  18. function registerPrompt(tabname, id) {
  19. var textarea = gradioApp().querySelector("#" + id + " > label > textarea");
  20. if (!activePromptTextarea[tabname]) {
  21. activePromptTextarea[tabname] = textarea;
  22. }
  23. textarea.addEventListener("focus", function() {
  24. activePromptTextarea[tabname] = textarea;
  25. });
  26. }
  27. var tabnav = gradioApp().querySelector('#' + tabname + '_extra_tabs > div.tab-nav');
  28. var controlsDiv = document.createElement('DIV');
  29. controlsDiv.classList.add('extra-networks-controls-div');
  30. tabnav.appendChild(controlsDiv);
  31. tabnav.insertBefore(controlsDiv, null);
  32. var this_tab = gradioApp().querySelector('#' + tabname + '_extra_tabs');
  33. this_tab.querySelectorAll(":scope > [id^='" + tabname + "_']").forEach(function(elem) {
  34. // tabname_full = {tabname}_{extra_networks_tabname}
  35. var tabname_full = elem.id;
  36. var search = gradioApp().querySelector("#" + tabname_full + "_extra_search");
  37. var sort_dir = gradioApp().querySelector("#" + tabname_full + "_extra_sort_dir");
  38. var refresh = gradioApp().querySelector("#" + tabname_full + "_extra_refresh");
  39. var currentSort = '';
  40. // If any of the buttons above don't exist, we want to skip this iteration of the loop.
  41. if (!search || !sort_dir || !refresh) {
  42. return; // `return` is equivalent of `continue` but for forEach loops.
  43. }
  44. var applyFilter = function(force) {
  45. var searchTerm = search.value.toLowerCase();
  46. gradioApp().querySelectorAll('#' + tabname + '_extra_tabs div.card').forEach(function(elem) {
  47. var searchOnly = elem.querySelector('.search_only');
  48. var text = Array.prototype.map.call(elem.querySelectorAll('.search_terms, .description'), function(t) {
  49. return t.textContent.toLowerCase();
  50. }).join(" ");
  51. var visible = text.indexOf(searchTerm) != -1;
  52. if (searchOnly && searchTerm.length < 4) {
  53. visible = false;
  54. }
  55. if (visible) {
  56. elem.classList.remove("hidden");
  57. } else {
  58. elem.classList.add("hidden");
  59. }
  60. });
  61. applySort(force);
  62. };
  63. var applySort = function(force) {
  64. var cards = gradioApp().querySelectorAll('#' + tabname_full + ' div.card');
  65. var parent = gradioApp().querySelector('#' + tabname_full + "_cards");
  66. var reverse = sort_dir.dataset.sortdir == "Descending";
  67. var activeSearchElem = gradioApp().querySelector('#' + tabname_full + "_controls .extra-network-control--sort.extra-network-control--enabled");
  68. var sortKey = activeSearchElem ? activeSearchElem.dataset.sortkey : "default";
  69. var sortKeyDataField = "sort" + sortKey.charAt(0).toUpperCase() + sortKey.slice(1);
  70. var sortKeyStore = sortKey + "-" + sort_dir.dataset.sortdir + "-" + cards.length;
  71. if (sortKeyStore == currentSort && !force) {
  72. return;
  73. }
  74. currentSort = sortKeyStore;
  75. var sortedCards = Array.from(cards);
  76. sortedCards.sort(function(cardA, cardB) {
  77. var a = cardA.dataset[sortKeyDataField];
  78. var b = cardB.dataset[sortKeyDataField];
  79. if (!isNaN(a) && !isNaN(b)) {
  80. return parseInt(a) - parseInt(b);
  81. }
  82. return (a < b ? -1 : (a > b ? 1 : 0));
  83. });
  84. if (reverse) {
  85. sortedCards.reverse();
  86. }
  87. parent.innerHTML = '';
  88. var frag = document.createDocumentFragment();
  89. sortedCards.forEach(function(card) {
  90. frag.appendChild(card);
  91. });
  92. parent.appendChild(frag);
  93. };
  94. search.addEventListener("input", function() {
  95. applyFilter();
  96. });
  97. applySort();
  98. applyFilter();
  99. extraNetworksApplySort[tabname_full] = applySort;
  100. extraNetworksApplyFilter[tabname_full] = applyFilter;
  101. var controls = gradioApp().querySelector("#" + tabname_full + "_controls");
  102. controlsDiv.insertBefore(controls, null);
  103. if (elem.style.display != "none") {
  104. extraNetworksShowControlsForPage(tabname, tabname_full);
  105. }
  106. });
  107. registerPrompt(tabname, tabname + "_prompt");
  108. registerPrompt(tabname, tabname + "_neg_prompt");
  109. }
  110. function extraNetworksMovePromptToTab(tabname, id, showPrompt, showNegativePrompt) {
  111. if (!gradioApp().querySelector('.toprow-compact-tools')) return; // only applicable for compact prompt layout
  112. var promptContainer = gradioApp().getElementById(tabname + '_prompt_container');
  113. var prompt = gradioApp().getElementById(tabname + '_prompt_row');
  114. var negPrompt = gradioApp().getElementById(tabname + '_neg_prompt_row');
  115. var elem = id ? gradioApp().getElementById(id) : null;
  116. if (showNegativePrompt && elem) {
  117. elem.insertBefore(negPrompt, elem.firstChild);
  118. } else {
  119. promptContainer.insertBefore(negPrompt, promptContainer.firstChild);
  120. }
  121. if (showPrompt && elem) {
  122. elem.insertBefore(prompt, elem.firstChild);
  123. } else {
  124. promptContainer.insertBefore(prompt, promptContainer.firstChild);
  125. }
  126. if (elem) {
  127. elem.classList.toggle('extra-page-prompts-active', showNegativePrompt || showPrompt);
  128. }
  129. }
  130. function extraNetworksShowControlsForPage(tabname, tabname_full) {
  131. gradioApp().querySelectorAll('#' + tabname + '_extra_tabs .extra-networks-controls-div > div').forEach(function(elem) {
  132. var targetId = tabname_full + "_controls";
  133. elem.style.display = elem.id == targetId ? "" : "none";
  134. });
  135. }
  136. function extraNetworksUnrelatedTabSelected(tabname) { // called from python when user selects an unrelated tab (generate)
  137. extraNetworksMovePromptToTab(tabname, '', false, false);
  138. extraNetworksShowControlsForPage(tabname, null);
  139. }
  140. function extraNetworksTabSelected(tabname, id, showPrompt, showNegativePrompt, tabname_full) { // called from python when user selects an extra networks tab
  141. extraNetworksMovePromptToTab(tabname, id, showPrompt, showNegativePrompt);
  142. extraNetworksShowControlsForPage(tabname, tabname_full);
  143. }
  144. function applyExtraNetworkFilter(tabname_full) {
  145. var doFilter = function() {
  146. var applyFunction = extraNetworksApplyFilter[tabname_full];
  147. if (applyFunction) {
  148. applyFunction(true);
  149. }
  150. };
  151. setTimeout(doFilter, 1);
  152. }
  153. function applyExtraNetworkSort(tabname_full) {
  154. var doSort = function() {
  155. extraNetworksApplySort[tabname_full](true);
  156. };
  157. setTimeout(doSort, 1);
  158. }
  159. var extraNetworksApplyFilter = {};
  160. var extraNetworksApplySort = {};
  161. var activePromptTextarea = {};
  162. function setupExtraNetworks() {
  163. setupExtraNetworksForTab('txt2img');
  164. setupExtraNetworksForTab('img2img');
  165. }
  166. var re_extranet = /<([^:^>]+:[^:]+):[\d.]+>(.*)/s;
  167. var re_extranet_g = /<([^:^>]+:[^:]+):[\d.]+>/g;
  168. var re_extranet_neg = /\(([^:^>]+:[\d.]+)\)/;
  169. var re_extranet_g_neg = /\(([^:^>]+:[\d.]+)\)/g;
  170. function tryToRemoveExtraNetworkFromPrompt(textarea, text, isNeg) {
  171. var m = text.match(isNeg ? re_extranet_neg : re_extranet);
  172. var replaced = false;
  173. var newTextareaText;
  174. var extraTextBeforeNet = opts.extra_networks_add_text_separator;
  175. if (m) {
  176. var extraTextAfterNet = m[2];
  177. var partToSearch = m[1];
  178. var foundAtPosition = -1;
  179. newTextareaText = textarea.value.replaceAll(isNeg ? re_extranet_g_neg : re_extranet_g, function(found, net, pos) {
  180. m = found.match(isNeg ? re_extranet_neg : re_extranet);
  181. if (m[1] == partToSearch) {
  182. replaced = true;
  183. foundAtPosition = pos;
  184. return "";
  185. }
  186. return found;
  187. });
  188. if (foundAtPosition >= 0) {
  189. if (extraTextAfterNet && newTextareaText.substr(foundAtPosition, extraTextAfterNet.length) == extraTextAfterNet) {
  190. newTextareaText = newTextareaText.substr(0, foundAtPosition) + newTextareaText.substr(foundAtPosition + extraTextAfterNet.length);
  191. }
  192. if (newTextareaText.substr(foundAtPosition - extraTextBeforeNet.length, extraTextBeforeNet.length) == extraTextBeforeNet) {
  193. newTextareaText = newTextareaText.substr(0, foundAtPosition - extraTextBeforeNet.length) + newTextareaText.substr(foundAtPosition);
  194. }
  195. }
  196. } else {
  197. newTextareaText = textarea.value.replaceAll(new RegExp(`((?:${extraTextBeforeNet})?${text})`, "g"), "");
  198. replaced = (newTextareaText != textarea.value);
  199. }
  200. if (replaced) {
  201. textarea.value = newTextareaText;
  202. return true;
  203. }
  204. return false;
  205. }
  206. function updatePromptArea(text, textArea, isNeg) {
  207. if (!tryToRemoveExtraNetworkFromPrompt(textArea, text, isNeg)) {
  208. textArea.value = textArea.value + opts.extra_networks_add_text_separator + text;
  209. }
  210. updateInput(textArea);
  211. }
  212. function cardClicked(tabname, textToAdd, textToAddNegative, allowNegativePrompt) {
  213. if (textToAddNegative.length > 0) {
  214. updatePromptArea(textToAdd, gradioApp().querySelector("#" + tabname + "_prompt > label > textarea"));
  215. updatePromptArea(textToAddNegative, gradioApp().querySelector("#" + tabname + "_neg_prompt > label > textarea"), true);
  216. } else {
  217. var textarea = allowNegativePrompt ? activePromptTextarea[tabname] : gradioApp().querySelector("#" + tabname + "_prompt > label > textarea");
  218. updatePromptArea(textToAdd, textarea);
  219. }
  220. }
  221. function saveCardPreview(event, tabname, filename) {
  222. var textarea = gradioApp().querySelector("#" + tabname + '_preview_filename > label > textarea');
  223. var button = gradioApp().getElementById(tabname + '_save_preview');
  224. textarea.value = filename;
  225. updateInput(textarea);
  226. button.click();
  227. event.stopPropagation();
  228. event.preventDefault();
  229. }
  230. function extraNetworksSearchButton(tabname, extra_networks_tabname, event) {
  231. var searchTextarea = gradioApp().querySelector("#" + tabname + "_" + extra_networks_tabname + "_extra_search");
  232. var button = event.target;
  233. var text = button.classList.contains("search-all") ? "" : button.textContent.trim();
  234. searchTextarea.value = text;
  235. updateInput(searchTextarea);
  236. }
  237. function extraNetworksTreeProcessFileClick(event, btn, tabname, extra_networks_tabname) {
  238. /**
  239. * Processes `onclick` events when user clicks on files in tree.
  240. *
  241. * @param event The generated event.
  242. * @param btn The clicked `tree-list-item` button.
  243. * @param tabname The name of the active tab in the sd webui. Ex: txt2img, img2img, etc.
  244. * @param extra_networks_tabname The id of the active extraNetworks tab. Ex: lora, checkpoints, etc.
  245. */
  246. // NOTE: Currently unused.
  247. return;
  248. }
  249. function extraNetworksTreeProcessDirectoryClick(event, btn, tabname, extra_networks_tabname) {
  250. /**
  251. * Processes `onclick` events when user clicks on directories in tree.
  252. *
  253. * Here is how the tree reacts to clicks for various states:
  254. * unselected unopened directory: Directory is selected and expanded.
  255. * unselected opened directory: Directory is selected.
  256. * selected opened directory: Directory is collapsed and deselected.
  257. * chevron is clicked: Directory is expanded or collapsed. Selected state unchanged.
  258. *
  259. * @param event The generated event.
  260. * @param btn The clicked `tree-list-item` button.
  261. * @param tabname The name of the active tab in the sd webui. Ex: txt2img, img2img, etc.
  262. * @param extra_networks_tabname The id of the active extraNetworks tab. Ex: lora, checkpoints, etc.
  263. */
  264. var ul = btn.nextElementSibling;
  265. // This is the actual target that the user clicked on within the target button.
  266. // We use this to detect if the chevron was clicked.
  267. var true_targ = event.target;
  268. function _expand_or_collapse(_ul, _btn) {
  269. // Expands <ul> if it is collapsed, collapses otherwise. Updates button attributes.
  270. if (_ul.hasAttribute("hidden")) {
  271. _ul.removeAttribute("hidden");
  272. _btn.dataset.expanded = "";
  273. } else {
  274. _ul.setAttribute("hidden", "");
  275. delete _btn.dataset.expanded;
  276. }
  277. }
  278. function _remove_selected_from_all() {
  279. // Removes the `selected` attribute from all buttons.
  280. var sels = document.querySelectorAll("div.tree-list-content");
  281. [...sels].forEach(el => {
  282. delete el.dataset.selected;
  283. });
  284. }
  285. function _select_button(_btn) {
  286. // Removes `data-selected` attribute from all buttons then adds to passed button.
  287. _remove_selected_from_all();
  288. _btn.dataset.selected = "";
  289. }
  290. function _update_search(_tabname, _extra_networks_tabname, _search_text) {
  291. // Update search input with select button's path.
  292. var search_input_elem = gradioApp().querySelector("#" + tabname + "_" + extra_networks_tabname + "_extra_search");
  293. search_input_elem.value = _search_text;
  294. updateInput(search_input_elem);
  295. }
  296. // If user clicks on the chevron, then we do not select the folder.
  297. if (true_targ.matches(".tree-list-item-action--leading, .tree-list-item-action-chevron")) {
  298. _expand_or_collapse(ul, btn);
  299. } else {
  300. // User clicked anywhere else on the button.
  301. if ("selected" in btn.dataset && !(ul.hasAttribute("hidden"))) {
  302. // If folder is select and open, collapse and deselect button.
  303. _expand_or_collapse(ul, btn);
  304. delete btn.dataset.selected;
  305. _update_search(tabname, extra_networks_tabname, "");
  306. } else if (!(!("selected" in btn.dataset) && !(ul.hasAttribute("hidden")))) {
  307. // If folder is open and not selected, then we don't collapse; just select.
  308. // NOTE: Double inversion sucks but it is the clearest way to show the branching here.
  309. _expand_or_collapse(ul, btn);
  310. _select_button(btn, tabname, extra_networks_tabname);
  311. _update_search(tabname, extra_networks_tabname, btn.dataset.path);
  312. } else {
  313. // All other cases, just select the button.
  314. _select_button(btn, tabname, extra_networks_tabname);
  315. _update_search(tabname, extra_networks_tabname, btn.dataset.path);
  316. }
  317. }
  318. }
  319. function extraNetworksTreeOnClick(event, tabname, extra_networks_tabname) {
  320. /**
  321. * Handles `onclick` events for buttons within an `extra-network-tree .tree-list--tree`.
  322. *
  323. * Determines whether the clicked button in the tree is for a file entry or a directory
  324. * then calls the appropriate function.
  325. *
  326. * @param event The generated event.
  327. * @param tabname The name of the active tab in the sd webui. Ex: txt2img, img2img, etc.
  328. * @param extra_networks_tabname The id of the active extraNetworks tab. Ex: lora, checkpoints, etc.
  329. */
  330. var btn = event.currentTarget;
  331. var par = btn.parentElement;
  332. if (par.dataset.treeEntryType === "file") {
  333. extraNetworksTreeProcessFileClick(event, btn, tabname, extra_networks_tabname);
  334. } else {
  335. extraNetworksTreeProcessDirectoryClick(event, btn, tabname, extra_networks_tabname);
  336. }
  337. }
  338. function extraNetworksControlSortOnClick(event, tabname, extra_networks_tabname) {
  339. /** Handles `onclick` events for Sort Mode buttons. */
  340. var self = event.currentTarget;
  341. var parent = event.currentTarget.parentElement;
  342. parent.querySelectorAll('.extra-network-control--sort').forEach(function(x) {
  343. x.classList.remove('extra-network-control--enabled');
  344. });
  345. self.classList.add('extra-network-control--enabled');
  346. applyExtraNetworkSort(tabname + "_" + extra_networks_tabname);
  347. }
  348. function extraNetworksControlSortDirOnClick(event, tabname, extra_networks_tabname) {
  349. /**
  350. * Handles `onclick` events for the Sort Direction button.
  351. *
  352. * Modifies the data attributes of the Sort Direction button to cycle between
  353. * ascending and descending sort directions.
  354. *
  355. * @param event The generated event.
  356. * @param tabname The name of the active tab in the sd webui. Ex: txt2img, img2img, etc.
  357. * @param extra_networks_tabname The id of the active extraNetworks tab. Ex: lora, checkpoints, etc.
  358. */
  359. if (event.currentTarget.dataset.sortdir == "Ascending") {
  360. event.currentTarget.dataset.sortdir = "Descending";
  361. event.currentTarget.setAttribute("title", "Sort descending");
  362. } else {
  363. event.currentTarget.dataset.sortdir = "Ascending";
  364. event.currentTarget.setAttribute("title", "Sort ascending");
  365. }
  366. applyExtraNetworkSort(tabname + "_" + extra_networks_tabname);
  367. }
  368. function extraNetworksControlTreeViewOnClick(event, tabname, extra_networks_tabname) {
  369. /**
  370. * Handles `onclick` events for the Tree View button.
  371. *
  372. * Toggles the tree view in the extra networks pane.
  373. *
  374. * @param event The generated event.
  375. * @param tabname The name of the active tab in the sd webui. Ex: txt2img, img2img, etc.
  376. * @param extra_networks_tabname The id of the active extraNetworks tab. Ex: lora, checkpoints, etc.
  377. */
  378. var button = event.currentTarget;
  379. button.classList.toggle("extra-network-control--enabled");
  380. var show = !button.classList.contains("extra-network-control--enabled");
  381. var pane = gradioApp().getElementById(tabname + "_" + extra_networks_tabname + "_pane");
  382. pane.classList.toggle("extra-network-dirs-hidden", show);
  383. }
  384. function extraNetworksControlRefreshOnClick(event, tabname, extra_networks_tabname) {
  385. /**
  386. * Handles `onclick` events for the Refresh Page button.
  387. *
  388. * In order to actually call the python functions in `ui_extra_networks.py`
  389. * to refresh the page, we created an empty gradio button in that file with an
  390. * event handler that refreshes the page. So what this function here does
  391. * is it manually raises a `click` event on that button.
  392. *
  393. * @param event The generated event.
  394. * @param tabname The name of the active tab in the sd webui. Ex: txt2img, img2img, etc.
  395. * @param extra_networks_tabname The id of the active extraNetworks tab. Ex: lora, checkpoints, etc.
  396. */
  397. var btn_refresh_internal = gradioApp().getElementById(tabname + "_" + extra_networks_tabname + "_extra_refresh_internal");
  398. btn_refresh_internal.dispatchEvent(new Event("click"));
  399. }
  400. var globalPopup = null;
  401. var globalPopupInner = null;
  402. function closePopup() {
  403. if (!globalPopup) return;
  404. globalPopup.style.display = "none";
  405. }
  406. function popup(contents) {
  407. if (!globalPopup) {
  408. globalPopup = document.createElement('div');
  409. globalPopup.classList.add('global-popup');
  410. var close = document.createElement('div');
  411. close.classList.add('global-popup-close');
  412. close.addEventListener("click", closePopup);
  413. close.title = "Close";
  414. globalPopup.appendChild(close);
  415. globalPopupInner = document.createElement('div');
  416. globalPopupInner.classList.add('global-popup-inner');
  417. globalPopup.appendChild(globalPopupInner);
  418. gradioApp().querySelector('.main').appendChild(globalPopup);
  419. }
  420. globalPopupInner.innerHTML = '';
  421. globalPopupInner.appendChild(contents);
  422. globalPopup.style.display = "flex";
  423. }
  424. var storedPopupIds = {};
  425. function popupId(id) {
  426. if (!storedPopupIds[id]) {
  427. storedPopupIds[id] = gradioApp().getElementById(id);
  428. }
  429. popup(storedPopupIds[id]);
  430. }
  431. function extraNetworksFlattenMetadata(obj) {
  432. const result = {};
  433. // Convert any stringified JSON objects to actual objects
  434. for (const key of Object.keys(obj)) {
  435. if (typeof obj[key] === 'string') {
  436. try {
  437. const parsed = JSON.parse(obj[key]);
  438. if (parsed && typeof parsed === 'object') {
  439. obj[key] = parsed;
  440. }
  441. } catch (error) {
  442. continue;
  443. }
  444. }
  445. }
  446. // Flatten the object
  447. for (const key of Object.keys(obj)) {
  448. if (typeof obj[key] === 'object' && obj[key] !== null) {
  449. const nested = extraNetworksFlattenMetadata(obj[key]);
  450. for (const nestedKey of Object.keys(nested)) {
  451. result[`${key}/${nestedKey}`] = nested[nestedKey];
  452. }
  453. } else {
  454. result[key] = obj[key];
  455. }
  456. }
  457. // Special case for handling modelspec keys
  458. for (const key of Object.keys(result)) {
  459. if (key.startsWith("modelspec.")) {
  460. result[key.replaceAll(".", "/")] = result[key];
  461. delete result[key];
  462. }
  463. }
  464. // Add empty keys to designate hierarchy
  465. for (const key of Object.keys(result)) {
  466. const parts = key.split("/");
  467. for (let i = 1; i < parts.length; i++) {
  468. const parent = parts.slice(0, i).join("/");
  469. if (!result[parent]) {
  470. result[parent] = "";
  471. }
  472. }
  473. }
  474. return result;
  475. }
  476. function extraNetworksShowMetadata(text) {
  477. try {
  478. let parsed = JSON.parse(text);
  479. if (parsed && typeof parsed === 'object') {
  480. parsed = extraNetworksFlattenMetadata(parsed);
  481. const table = createVisualizationTable(parsed, 0);
  482. popup(table);
  483. return;
  484. }
  485. } catch (error) {
  486. console.error(error);
  487. }
  488. var elem = document.createElement('pre');
  489. elem.classList.add('popup-metadata');
  490. elem.textContent = text;
  491. popup(elem);
  492. return;
  493. }
  494. function requestGet(url, data, handler, errorHandler) {
  495. var xhr = new XMLHttpRequest();
  496. var args = Object.keys(data).map(function(k) {
  497. return encodeURIComponent(k) + '=' + encodeURIComponent(data[k]);
  498. }).join('&');
  499. xhr.open("GET", url + "?" + args, true);
  500. xhr.onreadystatechange = function() {
  501. if (xhr.readyState === 4) {
  502. if (xhr.status === 200) {
  503. try {
  504. var js = JSON.parse(xhr.responseText);
  505. handler(js);
  506. } catch (error) {
  507. console.error(error);
  508. errorHandler();
  509. }
  510. } else {
  511. errorHandler();
  512. }
  513. }
  514. };
  515. var js = JSON.stringify(data);
  516. xhr.send(js);
  517. }
  518. function extraNetworksCopyCardPath(event) {
  519. navigator.clipboard.writeText(event.target.getAttribute("data-clipboard-text"));
  520. event.stopPropagation();
  521. }
  522. function extraNetworksRequestMetadata(event, extraPage) {
  523. var showError = function() {
  524. extraNetworksShowMetadata("there was an error getting metadata");
  525. };
  526. var cardName = event.target.parentElement.parentElement.getAttribute("data-name");
  527. requestGet("./sd_extra_networks/metadata", {page: extraPage, item: cardName}, function(data) {
  528. if (data && data.metadata) {
  529. extraNetworksShowMetadata(data.metadata);
  530. } else {
  531. showError();
  532. }
  533. }, showError);
  534. event.stopPropagation();
  535. }
  536. var extraPageUserMetadataEditors = {};
  537. function extraNetworksEditUserMetadata(event, tabname, extraPage) {
  538. var id = tabname + '_' + extraPage + '_edit_user_metadata';
  539. var editor = extraPageUserMetadataEditors[id];
  540. if (!editor) {
  541. editor = {};
  542. editor.page = gradioApp().getElementById(id);
  543. editor.nameTextarea = gradioApp().querySelector("#" + id + "_name" + ' textarea');
  544. editor.button = gradioApp().querySelector("#" + id + "_button");
  545. extraPageUserMetadataEditors[id] = editor;
  546. }
  547. var cardName = event.target.parentElement.parentElement.getAttribute("data-name");
  548. editor.nameTextarea.value = cardName;
  549. updateInput(editor.nameTextarea);
  550. editor.button.click();
  551. popup(editor.page);
  552. event.stopPropagation();
  553. }
  554. function extraNetworksRefreshSingleCard(page, tabname, name) {
  555. requestGet("./sd_extra_networks/get-single-card", {page: page, tabname: tabname, name: name}, function(data) {
  556. if (data && data.html) {
  557. var card = gradioApp().querySelector(`#${tabname}_${page.replace(" ", "_")}_cards > .card[data-name="${name}"]`);
  558. var newDiv = document.createElement('DIV');
  559. newDiv.innerHTML = data.html;
  560. var newCard = newDiv.firstElementChild;
  561. newCard.style.display = '';
  562. card.parentElement.insertBefore(newCard, card);
  563. card.parentElement.removeChild(card);
  564. }
  565. });
  566. }
  567. window.addEventListener("keydown", function(event) {
  568. if (event.key == "Escape") {
  569. closePopup();
  570. }
  571. });
  572. /**
  573. * Setup custom loading for this script.
  574. * We need to wait for all of our HTML to be generated in the extra networks tabs
  575. * before we can actually run the `setupExtraNetworks` function.
  576. * The `onUiLoaded` function actually runs before all of our extra network tabs are
  577. * finished generating. Thus we needed this new method.
  578. *
  579. */
  580. var uiAfterScriptsCallbacks = [];
  581. var uiAfterScriptsTimeout = null;
  582. var executedAfterScripts = false;
  583. function scheduleAfterScriptsCallbacks() {
  584. clearTimeout(uiAfterScriptsTimeout);
  585. uiAfterScriptsTimeout = setTimeout(function() {
  586. executeCallbacks(uiAfterScriptsCallbacks);
  587. }, 200);
  588. }
  589. onUiLoaded(function() {
  590. var mutationObserver = new MutationObserver(function(m) {
  591. let existingSearchfields = gradioApp().querySelectorAll("[id$='_extra_search']").length;
  592. let neededSearchfields = gradioApp().querySelectorAll("[id$='_extra_tabs'] > .tab-nav > button").length - 2;
  593. if (!executedAfterScripts && existingSearchfields >= neededSearchfields) {
  594. mutationObserver.disconnect();
  595. executedAfterScripts = true;
  596. scheduleAfterScriptsCallbacks();
  597. }
  598. });
  599. mutationObserver.observe(gradioApp(), {childList: true, subtree: true});
  600. });
  601. uiAfterScriptsCallbacks.push(setupExtraNetworks);