change condition for scheduleAfterScriptsCallbacks() to properly reflect the needed amount of search fields

1 files changed, 6 insertions, 3 deletions

diff --git a/javascript/extraNetworks.js b/javascript/extraNetworks.js
index 7ec6a04d..195525b0 100644
--- a/javascript/extraNetworks.js
+++ b/javascript/extraNetworks.js
@@ -622,10 +622,13 @@ function scheduleAfterScriptsCallbacks() {
     }, 200);
 }
 
-document.addEventListener("DOMContentLoaded", function() {
+onUiLoaded(function() {
     var mutationObserver = new MutationObserver(function(m) {
-        if (!executedAfterScripts &&
-            gradioApp().querySelectorAll("[id$='_extra_search']").length >= 6) {
+        let existingSearchfields = gradioApp().querySelectorAll("[id$='_extra_search']").length;
+        let neededSearchfields = gradioApp().querySelectorAll("[id$='_extra_tabs'] > .tab-nav > button").length - 2;
+
+        if (!executedAfterScripts && existingSearchfields >= neededSearchfields) {
+            mutationObserver.disconnect();
             executedAfterScripts = true;
             scheduleAfterScriptsCallbacks();
         }