aboutsummaryrefslogtreecommitdiffstats
path: root/modules/ui_extra_networks.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-02-19 09:41:29 +0000
committerGitHub <noreply@github.com>2023-02-19 09:41:29 +0000
commitdfb3b8f398239876cdfc03657680e50c76a1fed8 (patch)
tree30b5b4e5a179e9d88e75ac0abdcf0090e1c74397 /modules/ui_extra_networks.py
parentedb10092de516dda5271130ed53628387780a859 (diff)
parent43137773227344dd79b56c227dbc9fb54ac7c337 (diff)
downloadstable-diffusion-webui-gfx803-dfb3b8f398239876cdfc03657680e50c76a1fed8.tar.gz
stable-diffusion-webui-gfx803-dfb3b8f398239876cdfc03657680e50c76a1fed8.tar.bz2
stable-diffusion-webui-gfx803-dfb3b8f398239876cdfc03657680e50c76a1fed8.zip
Merge branch 'master' into weighted-learning
Diffstat (limited to 'modules/ui_extra_networks.py')
-rw-r--r--modules/ui_extra_networks.py10
1 files changed, 8 insertions, 2 deletions
diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index 90abec0a..71f1d81f 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -76,6 +76,10 @@ class ExtraNetworksPage:
while subdir.startswith("/"):
subdir = subdir[1:]
+ is_empty = len(os.listdir(x)) == 0
+ if not is_empty and not subdir.endswith("/"):
+ subdir = subdir + "/"
+
subdirs[subdir] = 1
if subdirs:
@@ -94,11 +98,13 @@ class ExtraNetworksPage:
dirs = "".join([f"<li>{x}</li>" for x in self.allowed_directories_for_previews()])
items_html = shared.html("extra-networks-no-cards.html").format(dirs=dirs)
+ self_name_id = self.name.replace(" ", "_")
+
res = f"""
-<div id='{tabname}_{self.name}_subdirs' class='extra-network-subdirs extra-network-subdirs-{view}'>
+<div id='{tabname}_{self_name_id}_subdirs' class='extra-network-subdirs extra-network-subdirs-{view}'>
{subdirs_html}
</div>
-<div id='{tabname}_{self.name}_cards' class='extra-network-{view}'>
+<div id='{tabname}_{self_name_id}_cards' class='extra-network-{view}'>
{items_html}
</div>
"""