Merge pull request #8366 from akx/extra-net-descs (c239b3d7) · Commits · github_fork / Stable Diffusion Webui

extensions-builtin/Lora/ui_extra_networks_lora.py

+2 −9

Original line number	Diff line number	Diff line
		@@ -15,18 +15,11 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
		def list_items(self):
		for name, lora_on_disk in lora.available_loras.items():
		path, ext = os.path.splitext(lora_on_disk.filename)
		previews = [path + ".png", path + ".preview.png"]

		preview = None
		for file in previews:
		if os.path.isfile(file):
		preview = self.link_preview(file)
		break

		yield {
		"name": name,
		"filename": path,
		"preview": preview,
		"preview": self._find_preview(path),
		"description": self._find_description(path),
		"search_term": self.search_terms_from_path(lora_on_disk.filename),
		"prompt": json.dumps(f"<lora:{name}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),
		"local_preview": path + ".png",

+1 −0

Original line number	Diff line number	Diff line
		@@ -7,6 +7,7 @@
		<span style="display:none" class='search_term'>{search_term}</span>
		</div>
		<span class='name'>{name}</span>
		<span class='description'>{description}</span>
		</div>
		</div>

+25 −0

Original line number	Diff line number	Diff line
		import glob
		import os.path
		import urllib.parse
		from functools import lru_cache
		from pathlib import Path
		from typing import Optional

		from modules import shared
		import gradio as gr
		@@ -130,6 +132,7 @@ class ExtraNetworksPage:
		"tabname": json.dumps(tabname),
		"local_preview": json.dumps(item["local_preview"]),
		"name": item["name"],
		"description": (item.get("description") or ""),
		"card_clicked": onclick,
		"save_card_preview": '"' + html.escape(f"""return saveCardPreview(event, {json.dumps(tabname)}, {json.dumps(item["local_preview"])})""") + '"',
		"search_term": item.get("search_term", ""),
		@@ -137,6 +140,28 @@ class ExtraNetworksPage:

		return self.card_page.format(**args)

		def _find_preview(self, path: str) -> Optional[str]:
		"""
		Find a preview PNG for a given path (without extension) and call link_preview on it.
		"""
		for file in [path + ".png", path + ".preview.png"]:
		if os.path.isfile(file):
		return self.link_preview(file)
		return None

		@lru_cache(maxsize=512)
		def _find_description(self, path: str) -> Optional[str]:
		"""
		Find and read a description file for a given path (without extension).
		"""
		for file in [f"{path}.txt", f"{path}.description.txt"]:
		try:
		with open(file, "r", encoding="utf-8", errors="replace") as f:
		return f.read()
		except OSError:
		pass
		return None


		def intialize():
		extra_pages.clear()

+2 −10

Original line number	Diff line number	Diff line
		import html
		import json
		import os
		import urllib.parse

		from modules import shared, ui_extra_networks, sd_models

		@@ -17,18 +16,11 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
		checkpoint: sd_models.CheckpointInfo
		for name, checkpoint in sd_models.checkpoints_list.items():
		path, ext = os.path.splitext(checkpoint.filename)
		previews = [path + ".png", path + ".preview.png"]

		preview = None
		for file in previews:
		if os.path.isfile(file):
		preview = self.link_preview(file)
		break

		yield {
		"name": checkpoint.name_for_extra,
		"filename": path,
		"preview": preview,
		"preview": self._find_preview(path),
		"description": self._find_description(path),
		"search_term": self.search_terms_from_path(checkpoint.filename) + " " + (checkpoint.sha256 or ""),
		"onclick": '"' + html.escape(f"""return selectCheckpoint({json.dumps(name)})""") + '"',
		"local_preview": path + ".png",

+2 −8

Original line number	Diff line number	Diff line
		@@ -14,18 +14,12 @@ class ExtraNetworksPageHypernetworks(ui_extra_networks.ExtraNetworksPage):
		def list_items(self):
		for name, path in shared.hypernetworks.items():
		path, ext = os.path.splitext(path)
		previews = [path + ".png", path + ".preview.png"]

		preview = None
		for file in previews:
		if os.path.isfile(file):
		preview = self.link_preview(file)
		break

		yield {
		"name": name,
		"filename": path,
		"preview": preview,
		"preview": self._find_preview(path),
		"description": self._find_description(path),
		"search_term": self.search_terms_from_path(path),
		"prompt": json.dumps(f"<hypernet:{name}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),
		"local_preview": path + ".png",