Added extra networks support

kekewind · Apr 11, 2023 · 74f1c74 · 74f1c74
1 parent b51580f
commit 74f1c74
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 28 deletions.
diff --git a/javascript/state.core.js b/javascript/state.core.js
@@ -38,6 +38,10 @@ state.core = (function () {
         'styles': 'styles'
     };
 
+    const TOGGLE_BUTTONS = {
+        'extra_networks': 'extra_networks',
+    };
+
     let store = null;
 
     function hasSetting(id, tab) {
@@ -59,44 +63,42 @@ state.core = (function () {
             .catch(error => console.error('[state]: Error getting JSON file:', error));
     }
 
+    function forEachElement(list, config, action) {
+        for (const [settingId, element] of Object.entries(list)) {
+            TABS.forEach(tab => {
+                if (config.hasSetting(settingId, tab)) {
+                    action(element, tab);
+                }
+            });
+        }
+    }
+
     function load(config) {
 
         store = new state.Store();
 
         loadUI();
         restoreTabs(config);
 
-        for (const [settingId, element] of Object.entries(ELEMENTS)) {
-            TABS.forEach(tab => {
-                if (config.hasSetting(settingId, tab)) {
-                    handleSavedInput(`${tab}_${element}`);
-                }
-            });
-        }
+        forEachElement(ELEMENTS, config, (element, tab) => {
+            handleSavedInput(`${tab}_${element}`);
+        });
 
-        for (const [settingId, element] of Object.entries(ELEMENTS_WITHOUT_PREFIX)) {
-            TABS.forEach(tab => {
-                if (config.hasSetting(settingId, tab)) {
-                    handleSavedInput(`${element}`);
-                }
-            });
-        }
+        forEachElement(ELEMENTS_WITHOUT_PREFIX, config, (element, tab) => {
+            handleSavedInput(`${element}`);
+        });
 
-        for (const [settingId, element] of Object.entries(SELECTS)) {
-            TABS.forEach(tab => {
-                if (config.hasSetting(settingId, tab)) {
-                    handleSavedSelects(`${tab}_${element}`);
-                }
-            });
-        }
+        forEachElement(SELECTS, config, (element, tab) => {
+            handleSavedSelects(`${tab}_${element}`);
+        });
 
-        for (const [settingId, element] of Object.entries(MULTI_SELECTS)) {
-            TABS.forEach(tab => {
-                if (config.hasSetting(settingId, tab)) {
-                    handleSavedMultiSelects(`${tab}_${element}`);
-                }
-            });
-        }
+        forEachElement(MULTI_SELECTS, config, (element, tab) => {
+            handleSavedMultiSelects(`${tab}_${element}`);
+        });
+
+        forEachElement(TOGGLE_BUTTONS, config, (element, tab) => {
+            handleToggleButton(`${tab}_${element}`);
+        });
 
         handleExtensions(config);
         handleSettingsPage();
@@ -208,6 +210,17 @@ state.core = (function () {
         state.utils.handleMultipleSelect(select, id, store);
     }
 
+    function handleToggleButton(id) {
+        const btn = gradioApp().querySelector(`button#${id}`);
+        if (! btn) { return; }
+        if (store.get(id) === 'true') {
+            state.utils.triggerMouseEvent(btn);
+        }
+        btn.addEventListener('click', function () {
+           store.set(id, Array.from(this.classList).indexOf('secondary-down') === -1);
+        });
+    }
+
     function handleExtensions(config) {
         if (config['state_extensions']) {
             config['state_extensions'].forEach(function (ext) {

diff --git a/scripts/state_settings.py b/scripts/state_settings.py
@@ -17,6 +17,7 @@ def on_ui_settings():
         "choices": [
             "prompt",
             "negative_prompt",
+            "extra_networks",
             "styles",
             "sampling",
             "sampling_steps",
@@ -42,6 +43,7 @@ def on_ui_settings():
         "choices": [
             "prompt",
             "negative_prompt",
+            "extra_networks",
             "styles",
             "sampling",
             "resize_mode",