extraNetworks.js 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179
  1. function setupExtraNetworksForTab(tabname){
  2. gradioApp().querySelector('#'+tabname+'_extra_tabs').classList.add('extra-networks')
  3. var tabs = gradioApp().querySelector('#'+tabname+'_extra_tabs > div')
  4. var search = gradioApp().querySelector('#'+tabname+'_extra_search textarea')
  5. var refresh = gradioApp().getElementById(tabname+'_extra_refresh')
  6. search.classList.add('search')
  7. tabs.appendChild(search)
  8. tabs.appendChild(refresh)
  9. search.addEventListener("input", function(evt){
  10. searchTerm = search.value.toLowerCase()
  11. gradioApp().querySelectorAll('#'+tabname+'_extra_tabs div.card').forEach(function(elem){
  12. text = elem.querySelector('.name').textContent.toLowerCase() + " " + elem.querySelector('.search_term').textContent.toLowerCase()
  13. elem.style.display = text.indexOf(searchTerm) == -1 ? "none" : ""
  14. })
  15. });
  16. }
  17. var activePromptTextarea = {};
  18. function setupExtraNetworks(){
  19. setupExtraNetworksForTab('txt2img')
  20. setupExtraNetworksForTab('img2img')
  21. function registerPrompt(tabname, id){
  22. var textarea = gradioApp().querySelector("#" + id + " > label > textarea");
  23. if (! activePromptTextarea[tabname]){
  24. activePromptTextarea[tabname] = textarea
  25. }
  26. textarea.addEventListener("focus", function(){
  27. activePromptTextarea[tabname] = textarea;
  28. });
  29. }
  30. registerPrompt('txt2img', 'txt2img_prompt')
  31. registerPrompt('txt2img', 'txt2img_neg_prompt')
  32. registerPrompt('img2img', 'img2img_prompt')
  33. registerPrompt('img2img', 'img2img_neg_prompt')
  34. }
  35. onUiLoaded(setupExtraNetworks)
  36. var re_extranet = /<([^:]+:[^:]+):[\d\.]+>/;
  37. var re_extranet_g = /\s+<([^:]+:[^:]+):[\d\.]+>/g;
  38. function tryToRemoveExtraNetworkFromPrompt(textarea, text){
  39. var m = text.match(re_extranet)
  40. if(! m) return false
  41. var partToSearch = m[1]
  42. var replaced = false
  43. var newTextareaText = textarea.value.replaceAll(re_extranet_g, function(found, index){
  44. m = found.match(re_extranet);
  45. if(m[1] == partToSearch){
  46. replaced = true;
  47. return ""
  48. }
  49. return found;
  50. })
  51. if(replaced){
  52. textarea.value = newTextareaText
  53. return true;
  54. }
  55. return false
  56. }
  57. function cardClicked(tabname, textToAdd, allowNegativePrompt){
  58. var textarea = allowNegativePrompt ? activePromptTextarea[tabname] : gradioApp().querySelector("#" + tabname + "_prompt > label > textarea")
  59. if(! tryToRemoveExtraNetworkFromPrompt(textarea, textToAdd)){
  60. textarea.value = textarea.value + opts.extra_networks_add_text_separator + textToAdd
  61. }
  62. updateInput(textarea)
  63. }
  64. function saveCardPreview(event, tabname, filename){
  65. var textarea = gradioApp().querySelector("#" + tabname + '_preview_filename > label > textarea')
  66. var button = gradioApp().getElementById(tabname + '_save_preview')
  67. textarea.value = filename
  68. updateInput(textarea)
  69. button.click()
  70. event.stopPropagation()
  71. event.preventDefault()
  72. }
  73. function extraNetworksSearchButton(tabs_id, event){
  74. searchTextarea = gradioApp().querySelector("#" + tabs_id + ' > div > textarea')
  75. button = event.target
  76. text = button.classList.contains("search-all") ? "" : button.textContent.trim()
  77. searchTextarea.value = text
  78. updateInput(searchTextarea)
  79. }
  80. var globalPopup = null;
  81. var globalPopupInner = null;
  82. function popup(contents){
  83. if(! globalPopup){
  84. globalPopup = document.createElement('div')
  85. globalPopup.onclick = function(){ globalPopup.style.display = "none"; };
  86. globalPopup.classList.add('global-popup');
  87. var close = document.createElement('div')
  88. close.classList.add('global-popup-close');
  89. close.onclick = function(){ globalPopup.style.display = "none"; };
  90. close.title = "Close";
  91. globalPopup.appendChild(close)
  92. globalPopupInner = document.createElement('div')
  93. globalPopupInner.onclick = function(event){ event.stopPropagation(); return false; };
  94. globalPopupInner.classList.add('global-popup-inner');
  95. globalPopup.appendChild(globalPopupInner)
  96. gradioApp().appendChild(globalPopup);
  97. }
  98. globalPopupInner.innerHTML = '';
  99. globalPopupInner.appendChild(contents);
  100. globalPopup.style.display = "flex";
  101. }
  102. function extraNetworksShowMetadata(text){
  103. elem = document.createElement('pre')
  104. elem.classList.add('popup-metadata');
  105. elem.textContent = text;
  106. popup(elem);
  107. }
  108. function requestGet(url, data, handler, errorHandler){
  109. var xhr = new XMLHttpRequest();
  110. var args = Object.keys(data).map(function(k){ return encodeURIComponent(k) + '=' + encodeURIComponent(data[k]) }).join('&')
  111. xhr.open("GET", url + "?" + args, true);
  112. xhr.onreadystatechange = function () {
  113. if (xhr.readyState === 4) {
  114. if (xhr.status === 200) {
  115. try {
  116. var js = JSON.parse(xhr.responseText);
  117. handler(js)
  118. } catch (error) {
  119. console.error(error);
  120. errorHandler()
  121. }
  122. } else{
  123. errorHandler()
  124. }
  125. }
  126. };
  127. var js = JSON.stringify(data);
  128. xhr.send(js);
  129. }
  130. function extraNetworksRequestMetadata(event, extraPage, cardName){
  131. showError = function(){ extraNetworksShowMetadata("there was an error getting metadata"); }
  132. requestGet("./sd_extra_networks/metadata", {"page": extraPage, "item": cardName}, function(data){
  133. if(data && data.metadata){
  134. extraNetworksShowMetadata(data.metadata)
  135. } else{
  136. showError()
  137. }
  138. }, showError)
  139. event.stopPropagation()
  140. }