Add option in settings to enable/disable tree view by default.

3 files changed, 10 insertions, 2 deletions

diff --git a/html/extra-networks-pane.html b/html/extra-networks-pane.html
index 73dad2ab..9f5b3ece 100644
--- a/html/extra-networks-pane.html
+++ b/html/extra-networks-pane.html
@@ -29,7 +29,7 @@
         </div>
         <div
             id="{tabname}_{extra_networks_tabname}_extra_tree_view"
-            class="extra-network-control--tree-view extra-network-control--enabled"
+            class="extra-network-control--tree-view {tree_view_btn_extra_class}"
             title="Enable Tree View"
             onclick="extraNetworksControlTreeViewOnClick(event, '{tabname}', '{extra_networks_tabname}');"
         >
@@ -45,7 +45,7 @@
         </div>
     </div>
     <div class="extra-network-pane-content">
-        <div id='{tabname}_{extra_networks_tabname}_tree' class='extra-network-tree'>
+        <div id='{tabname}_{extra_networks_tabname}_tree' class='extra-network-tree {tree_view_div_extra_class}'>
             {tree_html}
         </div>
         <div id='{tabname}_{extra_networks_tabname}_cards' class='extra-network-cards'>
diff --git a/modules/shared_options.py b/modules/shared_options.py
index 63488f4e..e0a6d977 100644
--- a/modules/shared_options.py
+++ b/modules/shared_options.py
@@ -251,6 +251,7 @@ options_templates.update(options_section(('extra_networks', "Extra Networks", "s
     "extra_networks_card_show_desc": OptionInfo(True, "Show description on card"),

     "extra_networks_card_order_field": OptionInfo("Path", "Default order field for Extra Networks cards", gr.Dropdown, {"choices": ['Path', 'Name', 'Date Created', 'Date Modified']}).needs_reload_ui(),

     "extra_networks_card_order": OptionInfo("Ascending", "Default order for Extra Networks cards", gr.Dropdown, {"choices": ['Ascending', 'Descending']}).needs_reload_ui(),

+    "extra_networks_tree_view_default_enabled": OptionInfo(False, "Enables the Extra Networks directory tree view by default").needs_reload_ui(),

     "extra_networks_add_text_separator": OptionInfo(" ", "Extra networks separator").info("extra text to add before <...> when adding extra network to prompt"),

     "ui_extra_networks_tab_reorder": OptionInfo("", "Extra networks tab order").needs_reload_ui(),

     "textual_inversion_print_at_load": OptionInfo(False, "Print a list of Textual Inversion embeddings when loading model"),

diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index 4c8a4074..80160b84 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -529,6 +529,11 @@ class ExtraNetworksPage:
         data_sortdir = shared.opts.extra_networks_card_order

         data_sortmode = shared.opts.extra_networks_card_order_field.lower().replace("sort", "").replace(" ", "_").rstrip("_").strip()

         data_sortkey = f"{data_sortmode}-{data_sortdir}-{len(self.items)}"

+        tree_view_btn_extra_class = ""

+        tree_view_div_extra_class = "hidden"

+        if shared.opts.extra_networks_tree_view_default_enabled:

+            tree_view_btn_extra_class = "extra-network-control--enabled"

+            tree_view_div_extra_class = ""

 

         return self.pane_tpl.format(

             **{

@@ -537,6 +542,8 @@ class ExtraNetworksPage:
                 "data_sortmode": data_sortmode,

                 "data_sortkey": data_sortkey,

                 "data_sortdir": data_sortdir,

+                "tree_view_btn_extra_class": tree_view_btn_extra_class,

+                "tree_view_div_extra_class": tree_view_div_extra_class,

                 "tree_html": self.create_tree_view_html(tabname),

                 "items_html": self.create_card_view_html(tabname),

             }