`Merge`: Merge all masks and inpaint
`Merge and Invert`: Merge all masks and Invert, then inpaint | | + +Applied in this order: x, y offset → erosion/dilation → merge/invert. + +#### Inpainting + +Each option corresponds to a corresponding option on the inpaint tab. Therefore, please refer to the inpaint tab for usage details on how to use each option. + +## ControlNet Inpainting + +You can use the ControlNet extension if you have ControlNet installed and ControlNet models. + +Support `inpaint, scribble, lineart, openpose, tile` controlnet models. Once you choose a model, the preprocessor is set automatically. It works separately from the model set by the Controlnet extension. + +## Advanced Options + +API request example: [wiki/API](https://github.com/Bing-su/adetailer/wiki/API) + +`ui-config.json` entries: [wiki/ui-config.json](https://github.com/Bing-su/adetailer/wiki/ui-config.json) + +`[SEP], [SKIP]` tokens: [wiki/Advanced](https://github.com/Bing-su/adetailer/wiki/Advanced) + +## Media + +- 🎥 [どこよりも詳しいAfter Detailer (adetailer)の使い方① 【Stable Diffusion】](https://youtu.be/sF3POwPUWCE) +- 🎥 [どこよりも詳しいAfter Detailer (adetailer)の使い方② 【Stable Diffusion】](https://youtu.be/urNISRdbIEg) + +## Model + +| Model | Target | mAP 50 | mAP 50-95 | +| --------------------- | --------------------- | ----------------------------- | ----------------------------- | +| face_yolov8n.pt | 2D / realistic face | 0.660 | 0.366 | +| face_yolov8s.pt | 2D / realistic face | 0.713 | 0.404 | +| hand_yolov8n.pt | 2D / realistic hand | 0.767 | 0.505 | +| person_yolov8n-seg.pt | 2D / realistic person | 0.782 (bbox)
0.761 (mask) | 0.555 (bbox)
0.460 (mask) | +| person_yolov8s-seg.pt | 2D / realistic person | 0.824 (bbox)
0.809 (mask) | 0.605 (bbox)
0.508 (mask) | +| mediapipe_face_full | realistic face | - | - | +| mediapipe_face_short | realistic face | - | - | +| mediapipe_face_mesh | realistic face | - | - | + +The yolo models can be found on huggingface [Bingsu/adetailer](https://huggingface.co/Bingsu/adetailer). + +### Additional Model + +Put your [ultralytics](https://github.com/ultralytics/ultralytics) yolo model in `webui/models/adetailer`. The model name should end with `.pt` or `.pth`. + +It must be a bbox detection or segment model and use all label. + +## Example + + + + +[](https://ko-fi.com/F1F1L7V2N) diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/Taskfile.yml b/TestingforfaivsixEx/content/eichellis/extensions/adetai/Taskfile.yml new file mode 100644 index 0000000000000000000000000000000000000000..70e0d357d11f0b8b908fb31f62ce0c104042d2c2 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/Taskfile.yml @@ -0,0 +1,27 @@ +# https://taskfile.dev + +version: "3" + +dotenv: + - .env + +tasks: + default: + cmds: + - echo "$PYTHON" + - echo "$WEBUI" + silent: true + + launch: + dir: "{{.WEBUI}}" + cmds: + - "{{.PYTHON}} launch.py --xformers --api" + silent: true + + lint: + cmds: + - pre-commit run -a + + update: + cmds: + - "{{.PYTHON}} -m pip install -U ultralytics mediapipe ruff pre-commit black" diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/__init__.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..dae6181e4042e320e7854837ed8e96b1da583c83 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/__init__.py @@ -0,0 +1,20 @@ +from .__version__ import __version__ +from .args import AD_ENABLE, ALL_ARGS, ADetailerArgs, EnableChecker +from .common import PredictOutput, get_models +from .mediapipe import mediapipe_predict +from .ultralytics import ultralytics_predict + +AFTER_DETAILER = "ADetailer" + +__all__ = [ + "__version__", + "AD_ENABLE", + "ADetailerArgs", + "AFTER_DETAILER", + "ALL_ARGS", + "EnableChecker", + "PredictOutput", + "get_models", + "mediapipe_predict", + "ultralytics_predict", +] diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/__version__.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/__version__.py new file mode 100644 index 0000000000000000000000000000000000000000..0a9df8e09313bc01841f7abafe5720e5d7ef1131 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/__version__.py @@ -0,0 +1 @@ +__version__ = "23.9.2" diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/args.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/args.py new file mode 100644 index 0000000000000000000000000000000000000000..ea22b99001633eaba5d388049715d3d53853582b --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/args.py @@ -0,0 +1,251 @@ +from __future__ import annotations + +from collections import UserList +from functools import cached_property, partial +from typing import Any, Literal, NamedTuple, Optional, Union + +import pydantic +from pydantic import ( + BaseModel, + Extra, + NonNegativeFloat, + NonNegativeInt, + PositiveInt, + confloat, + conint, + constr, + root_validator, + validator, +) + +cn_model_regex = r".*(inpaint|tile|scribble|lineart|openpose).*|^None$" + + +class Arg(NamedTuple): + attr: str + name: str + + +class ArgsList(UserList): + @cached_property + def attrs(self) -> tuple[str]: + return tuple(attr for attr, _ in self) + + @cached_property + def names(self) -> tuple[str]: + return tuple(name for _, name in self) + + +class ADetailerArgs(BaseModel, extra=Extra.forbid): + ad_model: str = "None" + ad_prompt: str = "" + ad_negative_prompt: str = "" + ad_confidence: confloat(ge=0.0, le=1.0) = 0.3 + ad_mask_k_largest: NonNegativeInt = 0 + ad_mask_min_ratio: confloat(ge=0.0, le=1.0) = 0.0 + ad_mask_max_ratio: confloat(ge=0.0, le=1.0) = 1.0 + ad_dilate_erode: int = 4 + ad_x_offset: int = 0 + ad_y_offset: int = 0 + ad_mask_merge_invert: Literal["None", "Merge", "Merge and Invert"] = "None" + ad_mask_blur: NonNegativeInt = 4 + ad_denoising_strength: confloat(ge=0.0, le=1.0) = 0.4 + ad_inpaint_only_masked: bool = True + ad_inpaint_only_masked_padding: NonNegativeInt = 32 + ad_use_inpaint_width_height: bool = False + ad_inpaint_width: PositiveInt = 512 + ad_inpaint_height: PositiveInt = 512 + ad_use_steps: bool = False + ad_steps: PositiveInt = 28 + ad_use_cfg_scale: bool = False + ad_cfg_scale: NonNegativeFloat = 7.0 + ad_use_checkpoint: bool = False + ad_checkpoint: Optional[str] = None + ad_use_vae: bool = False + ad_vae: Optional[str] = None + ad_use_sampler: bool = False + ad_sampler: str = "DPM++ 2M Karras" + ad_use_noise_multiplier: bool = False + ad_noise_multiplier: confloat(ge=0.5, le=1.5) = 1.0 + ad_use_clip_skip: bool = False + ad_clip_skip: conint(ge=1, le=12) = 1 + ad_restore_face: bool = False + ad_controlnet_model: constr(regex=cn_model_regex) = "None" + ad_controlnet_module: Optional[constr(regex=r".*inpaint.*|^None$")] = None + ad_controlnet_weight: confloat(ge=0.0, le=1.0) = 1.0 + ad_controlnet_guidance_start: confloat(ge=0.0, le=1.0) = 0.0 + ad_controlnet_guidance_end: confloat(ge=0.0, le=1.0) = 1.0 + is_api: bool = True + + @root_validator(skip_on_failure=True) + def ad_controlnt_module_validator(cls, values): # noqa: N805 + cn_model = values.get("ad_controlnet_model", "None") + cn_module = values.get("ad_controlnet_module", None) + if "inpaint" not in cn_model or cn_module == "None": + values["ad_controlnet_module"] = None + return values + + @validator("is_api", pre=True) + def is_api_validator(cls, v: Any): # noqa: N805 + "tuple is json serializable but cannot be made with json deserialize." + return type(v) is not tuple + + @staticmethod + def ppop( + p: dict[str, Any], + key: str, + pops: list[str] | None = None, + cond: Any = None, + ) -> None: + if pops is None: + pops = [key] + if key not in p: + return + value = p[key] + cond = (not bool(value)) if cond is None else value == cond + + if cond: + for k in pops: + p.pop(k, None) + + def extra_params(self, suffix: str = "") -> dict[str, Any]: + if self.ad_model == "None": + return {} + + p = {name: getattr(self, attr) for attr, name in ALL_ARGS} + ppop = partial(self.ppop, p) + + ppop("ADetailer prompt") + ppop("ADetailer negative prompt") + ppop("ADetailer mask only top k largest", cond=0) + ppop("ADetailer mask min ratio", cond=0.0) + ppop("ADetailer mask max ratio", cond=1.0) + ppop("ADetailer x offset", cond=0) + ppop("ADetailer y offset", cond=0) + ppop("ADetailer mask merge/invert", cond="None") + ppop("ADetailer inpaint only masked", ["ADetailer inpaint padding"]) + ppop( + "ADetailer use inpaint width/height", + [ + "ADetailer use inpaint width/height", + "ADetailer inpaint width", + "ADetailer inpaint height", + ], + ) + ppop( + "ADetailer use separate steps", + ["ADetailer use separate steps", "ADetailer steps"], + ) + ppop( + "ADetailer use separate CFG scale", + ["ADetailer use separate CFG scale", "ADetailer CFG scale"], + ) + ppop( + "ADetailer use separate checkpoint", + ["ADetailer use separate checkpoint", "ADetailer checkpoint"], + ) + ppop( + "ADetailer use separate VAE", + ["ADetailer use separate VAE", "ADetailer VAE"], + ) + ppop( + "ADetailer use separate sampler", + ["ADetailer use separate sampler", "ADetailer sampler"], + ) + ppop( + "ADetailer use separate noise multiplier", + ["ADetailer use separate noise multiplier", "ADetailer noise multiplier"], + ) + + ppop( + "ADetailer use separate CLIP skip", + ["ADetailer use separate CLIP skip", "ADetailer CLIP skip"], + ) + + ppop("ADetailer restore face") + ppop( + "ADetailer ControlNet model", + [ + "ADetailer ControlNet model", + "ADetailer ControlNet module", + "ADetailer ControlNet weight", + "ADetailer ControlNet guidance start", + "ADetailer ControlNet guidance end", + ], + cond="None", + ) + ppop("ADetailer ControlNet module") + ppop("ADetailer ControlNet weight", cond=1.0) + ppop("ADetailer ControlNet guidance start", cond=0.0) + ppop("ADetailer ControlNet guidance end", cond=1.0) + + if suffix: + p = {k + suffix: v for k, v in p.items()} + + return p + + +class EnableChecker(BaseModel): + enable: bool + arg_list: list + + def is_enabled(self) -> bool: + ad_model = ALL_ARGS[0].attr + if not self.enable: + return False + return any(arg.get(ad_model, "None") != "None" for arg in self.arg_list) + + +_all_args = [ + ("ad_enable", "ADetailer enable"), + ("ad_model", "ADetailer model"), + ("ad_prompt", "ADetailer prompt"), + ("ad_negative_prompt", "ADetailer negative prompt"), + ("ad_confidence", "ADetailer confidence"), + ("ad_mask_k_largest", "ADetailer mask only top k largest"), + ("ad_mask_min_ratio", "ADetailer mask min ratio"), + ("ad_mask_max_ratio", "ADetailer mask max ratio"), + ("ad_x_offset", "ADetailer x offset"), + ("ad_y_offset", "ADetailer y offset"), + ("ad_dilate_erode", "ADetailer dilate/erode"), + ("ad_mask_merge_invert", "ADetailer mask merge/invert"), + ("ad_mask_blur", "ADetailer mask blur"), + ("ad_denoising_strength", "ADetailer denoising strength"), + ("ad_inpaint_only_masked", "ADetailer inpaint only masked"), + ("ad_inpaint_only_masked_padding", "ADetailer inpaint padding"), + ("ad_use_inpaint_width_height", "ADetailer use inpaint width/height"), + ("ad_inpaint_width", "ADetailer inpaint width"), + ("ad_inpaint_height", "ADetailer inpaint height"), + ("ad_use_steps", "ADetailer use separate steps"), + ("ad_steps", "ADetailer steps"), + ("ad_use_cfg_scale", "ADetailer use separate CFG scale"), + ("ad_cfg_scale", "ADetailer CFG scale"), + ("ad_use_checkpoint", "ADetailer use separate checkpoint"), + ("ad_checkpoint", "ADetailer checkpoint"), + ("ad_use_vae", "ADetailer use separate VAE"), + ("ad_vae", "ADetailer VAE"), + ("ad_use_sampler", "ADetailer use separate sampler"), + ("ad_sampler", "ADetailer sampler"), + ("ad_use_noise_multiplier", "ADetailer use separate noise multiplier"), + ("ad_noise_multiplier", "ADetailer noise multiplier"), + ("ad_use_clip_skip", "ADetailer use separate CLIP skip"), + ("ad_clip_skip", "ADetailer CLIP skip"), + ("ad_restore_face", "ADetailer restore face"), + ("ad_controlnet_model", "ADetailer ControlNet model"), + ("ad_controlnet_module", "ADetailer ControlNet module"), + ("ad_controlnet_weight", "ADetailer ControlNet weight"), + ("ad_controlnet_guidance_start", "ADetailer ControlNet guidance start"), + ("ad_controlnet_guidance_end", "ADetailer ControlNet guidance end"), +] + +AD_ENABLE = Arg(*_all_args[0]) +_args = [Arg(*args) for args in _all_args[1:]] +ALL_ARGS = ArgsList(_args) + +BBOX_SORTBY = [ + "None", + "Position (left to right)", + "Position (center to edge)", + "Area (large to small)", +] +MASK_MERGE_INVERT = ["None", "Merge", "Merge and Invert"] diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/common.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/common.py new file mode 100644 index 0000000000000000000000000000000000000000..a29bdd5beb2ac3e1f027398dc176e70fb77d082d --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/common.py @@ -0,0 +1,127 @@ +from __future__ import annotations + +from collections import OrderedDict +from dataclasses import dataclass, field +from pathlib import Path +from typing import Optional, Union + +from huggingface_hub import hf_hub_download +from PIL import Image, ImageDraw +from rich import print + +repo_id = "Bingsu/adetailer" + + +@dataclass +class PredictOutput: + bboxes: list[list[int | float]] = field(default_factory=list) + masks: list[Image.Image] = field(default_factory=list) + preview: Optional[Image.Image] = None + + +def hf_download(file: str): + try: + path = hf_hub_download(repo_id, file) + except Exception: + msg = f"[-] ADetailer: Failed to load model {file!r} from huggingface" + print(msg) + path = "INVALID" + return path + + +def get_models( + model_dir: Union[str, Path], huggingface: bool = True +) -> OrderedDict[str, Optional[str]]: + model_dir = Path(model_dir) + if model_dir.is_dir(): + model_paths = [ + p + for p in model_dir.rglob("*") + if p.is_file() and p.suffix in (".pt", ".pth") + ] + else: + model_paths = [] + + models = OrderedDict() + if huggingface: + models.update( + { + "face_yolov8n.pt": hf_download("face_yolov8n.pt"), + "face_yolov8s.pt": hf_download("face_yolov8s.pt"), + "hand_yolov8n.pt": hf_download("hand_yolov8n.pt"), + "person_yolov8n-seg.pt": hf_download("person_yolov8n-seg.pt"), + "person_yolov8s-seg.pt": hf_download("person_yolov8s-seg.pt"), + } + ) + models.update( + { + "mediapipe_face_full": None, + "mediapipe_face_short": None, + "mediapipe_face_mesh": None, + "mediapipe_face_mesh_eyes_only": None, + } + ) + + invalid_keys = [k for k, v in models.items() if v == "INVALID"] + for key in invalid_keys: + models.pop(key) + + for path in model_paths: + if path.name in models: + continue + models[path.name] = str(path) + + return models + + +def create_mask_from_bbox( + bboxes: list[list[float]], shape: tuple[int, int] +) -> list[Image.Image]: + """ + Parameters + ---------- + bboxes: list[list[float]] + list of [x1, y1, x2, y2] + bounding boxes + shape: tuple[int, int] + shape of the image (width, height) + + Returns + ------- + masks: list[Image.Image] + A list of masks + + """ + masks = [] + for bbox in bboxes: + mask = Image.new("L", shape, 0) + mask_draw = ImageDraw.Draw(mask) + mask_draw.rectangle(bbox, fill=255) + masks.append(mask) + return masks + + +def create_bbox_from_mask( + masks: list[Image.Image], shape: tuple[int, int] +) -> list[list[int]]: + """ + Parameters + ---------- + masks: list[Image.Image] + A list of masks + shape: tuple[int, int] + shape of the image (width, height) + + Returns + ------- + bboxes: list[list[float]] + A list of bounding boxes + + """ + bboxes = [] + for mask in masks: + mask = mask.resize(shape) + bbox = mask.getbbox() + if bbox is not None: + bboxes.append(list(bbox)) + return bboxes diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/mask.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/mask.py new file mode 100644 index 0000000000000000000000000000000000000000..913ea7e1d780581b3950d5e2d285b66cfc5dd553 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/mask.py @@ -0,0 +1,255 @@ +from __future__ import annotations + +from enum import IntEnum +from functools import partial, reduce +from math import dist + +import cv2 +import numpy as np +from PIL import Image, ImageChops + +from adetailer.args import MASK_MERGE_INVERT +from adetailer.common import PredictOutput + + +class SortBy(IntEnum): + NONE = 0 + LEFT_TO_RIGHT = 1 + CENTER_TO_EDGE = 2 + AREA = 3 + + +class MergeInvert(IntEnum): + NONE = 0 + MERGE = 1 + MERGE_INVERT = 2 + + +def _dilate(arr: np.ndarray, value: int) -> np.ndarray: + kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (value, value)) + return cv2.dilate(arr, kernel, iterations=1) + + +def _erode(arr: np.ndarray, value: int) -> np.ndarray: + kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (value, value)) + return cv2.erode(arr, kernel, iterations=1) + + +def dilate_erode(img: Image.Image, value: int) -> Image.Image: + """ + The dilate_erode function takes an image and a value. + If the value is positive, it dilates the image by that amount. + If the value is negative, it erodes the image by that amount. + + Parameters + ---------- + img: PIL.Image.Image + the image to be processed + value: int + kernel size of dilation or erosion + + Returns + ------- + PIL.Image.Image + The image that has been dilated or eroded + """ + if value == 0: + return img + + arr = np.array(img) + arr = _dilate(arr, value) if value > 0 else _erode(arr, -value) + + return Image.fromarray(arr) + + +def offset(img: Image.Image, x: int = 0, y: int = 0) -> Image.Image: + """ + The offset function takes an image and offsets it by a given x(→) and y(↑) value. + + Parameters + ---------- + mask: Image.Image + Pass the mask image to the function + x: int + → + y: int + ↑ + + Returns + ------- + PIL.Image.Image + A new image that is offset by x and y + """ + return ImageChops.offset(img, x, -y) + + +def is_all_black(img: Image.Image) -> bool: + arr = np.array(img) + return cv2.countNonZero(arr) == 0 + + +def bbox_area(bbox: list[float]): + return (bbox[2] - bbox[0]) * (bbox[3] - bbox[1]) + + +def mask_preprocess( + masks: list[Image.Image], + kernel: int = 0, + x_offset: int = 0, + y_offset: int = 0, + merge_invert: int | MergeInvert | str = MergeInvert.NONE, +) -> list[Image.Image]: + """ + The mask_preprocess function takes a list of masks and preprocesses them. + It dilates and erodes the masks, and offsets them by x_offset and y_offset. + + Parameters + ---------- + masks: list[Image.Image] + A list of masks + kernel: int + kernel size of dilation or erosion + x_offset: int + → + y_offset: int + ↑ + + Returns + ------- + list[Image.Image] + A list of processed masks + """ + if not masks: + return [] + + if x_offset != 0 or y_offset != 0: + masks = [offset(m, x_offset, y_offset) for m in masks] + + if kernel != 0: + masks = [dilate_erode(m, kernel) for m in masks] + masks = [m for m in masks if not is_all_black(m)] + + return mask_merge_invert(masks, mode=merge_invert) + + +# Bbox sorting +def _key_left_to_right(bbox: list[float]) -> float: + """ + Left to right + + Parameters + ---------- + bbox: list[float] + list of [x1, y1, x2, y2] + """ + return bbox[0] + + +def _key_center_to_edge(bbox: list[float], *, center: tuple[float, float]) -> float: + """ + Center to edge + + Parameters + ---------- + bbox: list[float] + list of [x1, y1, x2, y2] + image: Image.Image + the image + """ + bbox_center = ((bbox[0] + bbox[2]) / 2, (bbox[1] + bbox[3]) / 2) + return dist(center, bbox_center) + + +def _key_area(bbox: list[float]) -> float: + """ + Large to small + + Parameters + ---------- + bbox: list[float] + list of [x1, y1, x2, y2] + """ + return -bbox_area(bbox) + + +def sort_bboxes( + pred: PredictOutput, order: int | SortBy = SortBy.NONE +) -> PredictOutput: + if order == SortBy.NONE or len(pred.bboxes) <= 1: + return pred + + if order == SortBy.LEFT_TO_RIGHT: + key = _key_left_to_right + elif order == SortBy.CENTER_TO_EDGE: + width, height = pred.preview.size + center = (width / 2, height / 2) + key = partial(_key_center_to_edge, center=center) + elif order == SortBy.AREA: + key = _key_area + else: + raise RuntimeError + + items = len(pred.bboxes) + idx = sorted(range(items), key=lambda i: key(pred.bboxes[i])) + pred.bboxes = [pred.bboxes[i] for i in idx] + pred.masks = [pred.masks[i] for i in idx] + return pred + + +# Filter by ratio +def is_in_ratio(bbox: list[float], low: float, high: float, orig_area: int) -> bool: + area = bbox_area(bbox) + return low <= area / orig_area <= high + + +def filter_by_ratio(pred: PredictOutput, low: float, high: float) -> PredictOutput: + if not pred.bboxes: + return pred + + w, h = pred.preview.size + orig_area = w * h + items = len(pred.bboxes) + idx = [i for i in range(items) if is_in_ratio(pred.bboxes[i], low, high, orig_area)] + pred.bboxes = [pred.bboxes[i] for i in idx] + pred.masks = [pred.masks[i] for i in idx] + return pred + + +def filter_k_largest(pred: PredictOutput, k: int = 0) -> PredictOutput: + if not pred.bboxes or k == 0: + return pred + areas = [bbox_area(bbox) for bbox in pred.bboxes] + idx = np.argsort(areas)[-k:] + pred.bboxes = [pred.bboxes[i] for i in idx] + pred.masks = [pred.masks[i] for i in idx] + return pred + + +# Merge / Invert +def mask_merge(masks: list[Image.Image]) -> list[Image.Image]: + arrs = [np.array(m) for m in masks] + arr = reduce(cv2.bitwise_or, arrs) + return [Image.fromarray(arr)] + + +def mask_invert(masks: list[Image.Image]) -> list[Image.Image]: + return [ImageChops.invert(m) for m in masks] + + +def mask_merge_invert( + masks: list[Image.Image], mode: int | MergeInvert | str +) -> list[Image.Image]: + if isinstance(mode, str): + mode = MASK_MERGE_INVERT.index(mode) + + if mode == MergeInvert.NONE or not masks: + return masks + + if mode == MergeInvert.MERGE: + return mask_merge(masks) + + if mode == MergeInvert.MERGE_INVERT: + merged = mask_merge(masks) + return mask_invert(merged) + + raise RuntimeError diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/mediapipe.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/mediapipe.py new file mode 100644 index 0000000000000000000000000000000000000000..e52b620883b6e7360a3ad5e332cdd03f24d6daab --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/mediapipe.py @@ -0,0 +1,179 @@ +from __future__ import annotations + +from functools import partial + +import mediapipe as mp +import numpy as np +from PIL import Image, ImageDraw + +from adetailer import PredictOutput +from adetailer.common import create_bbox_from_mask, create_mask_from_bbox + + +def mediapipe_predict( + model_type: str, image: Image.Image, confidence: float = 0.3 +) -> PredictOutput: + mapping = { + "mediapipe_face_short": partial(mediapipe_face_detection, 0), + "mediapipe_face_full": partial(mediapipe_face_detection, 1), + "mediapipe_face_mesh": mediapipe_face_mesh, + "mediapipe_face_mesh_eyes_only": mediapipe_face_mesh_eyes_only, + } + if model_type in mapping: + func = mapping[model_type] + return func(image, confidence) + msg = f"[-] ADetailer: Invalid mediapipe model type: {model_type}, Available: {list(mapping.keys())!r}" + raise RuntimeError(msg) + + +def mediapipe_face_detection( + model_type: int, image: Image.Image, confidence: float = 0.3 +) -> PredictOutput: + img_width, img_height = image.size + + mp_face_detection = mp.solutions.face_detection + draw_util = mp.solutions.drawing_utils + + img_array = np.array(image) + + with mp_face_detection.FaceDetection( + model_selection=model_type, min_detection_confidence=confidence + ) as face_detector: + pred = face_detector.process(img_array) + + if pred.detections is None: + return PredictOutput() + + preview_array = img_array.copy() + + bboxes = [] + for detection in pred.detections: + draw_util.draw_detection(preview_array, detection) + + bbox = detection.location_data.relative_bounding_box + x1 = bbox.xmin * img_width + y1 = bbox.ymin * img_height + w = bbox.width * img_width + h = bbox.height * img_height + x2 = x1 + w + y2 = y1 + h + + bboxes.append([x1, y1, x2, y2]) + + masks = create_mask_from_bbox(bboxes, image.size) + preview = Image.fromarray(preview_array) + + return PredictOutput(bboxes=bboxes, masks=masks, preview=preview) + + +def get_convexhull(points: np.ndarray) -> list[tuple[int, int]]: + """ + Parameters + ---------- + points: An ndarray of shape (n, 2) containing the 2D points. + + Returns + ------- + list[tuple[int, int]]: Input for the draw.polygon function + """ + from scipy.spatial import ConvexHull + + hull = ConvexHull(points) + vertices = hull.vertices + return list(zip(points[vertices, 0], points[vertices, 1])) + + +def mediapipe_face_mesh(image: Image.Image, confidence: float = 0.3) -> PredictOutput: + mp_face_mesh = mp.solutions.face_mesh + draw_util = mp.solutions.drawing_utils + drawing_styles = mp.solutions.drawing_styles + + w, h = image.size + + with mp_face_mesh.FaceMesh( + static_image_mode=True, max_num_faces=20, min_detection_confidence=confidence + ) as face_mesh: + arr = np.array(image) + pred = face_mesh.process(arr) + + if pred.multi_face_landmarks is None: + return PredictOutput() + + preview = arr.copy() + masks = [] + + for landmarks in pred.multi_face_landmarks: + draw_util.draw_landmarks( + image=preview, + landmark_list=landmarks, + connections=mp_face_mesh.FACEMESH_TESSELATION, + landmark_drawing_spec=None, + connection_drawing_spec=drawing_styles.get_default_face_mesh_tesselation_style(), + ) + + points = np.array([(land.x * w, land.y * h) for land in landmarks.landmark]) + outline = get_convexhull(points) + + mask = Image.new("L", image.size, "black") + draw = ImageDraw.Draw(mask) + draw.polygon(outline, fill="white") + masks.append(mask) + + bboxes = create_bbox_from_mask(masks, image.size) + preview = Image.fromarray(preview) + return PredictOutput(bboxes=bboxes, masks=masks, preview=preview) + + +def mediapipe_face_mesh_eyes_only( + image: Image.Image, confidence: float = 0.3 +) -> PredictOutput: + mp_face_mesh = mp.solutions.face_mesh + + left_idx = np.array(list(mp_face_mesh.FACEMESH_LEFT_EYE)).flatten() + right_idx = np.array(list(mp_face_mesh.FACEMESH_RIGHT_EYE)).flatten() + + w, h = image.size + + with mp_face_mesh.FaceMesh( + static_image_mode=True, max_num_faces=20, min_detection_confidence=confidence + ) as face_mesh: + arr = np.array(image) + pred = face_mesh.process(arr) + + if pred.multi_face_landmarks is None: + return PredictOutput() + + preview = image.copy() + masks = [] + + for landmarks in pred.multi_face_landmarks: + points = np.array([(land.x * w, land.y * h) for land in landmarks.landmark]) + left_eyes = points[left_idx] + right_eyes = points[right_idx] + left_outline = get_convexhull(left_eyes) + right_outline = get_convexhull(right_eyes) + + mask = Image.new("L", image.size, "black") + draw = ImageDraw.Draw(mask) + for outline in (left_outline, right_outline): + draw.polygon(outline, fill="white") + masks.append(mask) + + bboxes = create_bbox_from_mask(masks, image.size) + preview = draw_preview(preview, bboxes, masks) + return PredictOutput(bboxes=bboxes, masks=masks, preview=preview) + + +def draw_preview( + preview: Image.Image, bboxes: list[list[int]], masks: list[Image.Image] +) -> Image.Image: + red = Image.new("RGB", preview.size, "red") + for mask in masks: + masked = Image.composite(red, preview, mask) + preview = Image.blend(preview, masked, 0.25) + + draw = ImageDraw.Draw(preview) + for bbox in bboxes: + draw.rectangle(bbox, outline="red", width=2) + + return preview diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/traceback.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/traceback.py new file mode 100644 index 0000000000000000000000000000000000000000..e3fcd9e8cd8a56281fbc02c83406f636c1c590c0 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/traceback.py @@ -0,0 +1,161 @@ +from __future__ import annotations + +import io +import platform +import sys +from importlib.metadata import version +from typing import Any, Callable + +from rich.console import Console, Group +from rich.panel import Panel +from rich.table import Table +from rich.traceback import Traceback + +from adetailer.__version__ import __version__ + + +def processing(*args: Any) -> dict[str, Any]: + try: + from modules.processing import ( + StableDiffusionProcessingImg2Img, + StableDiffusionProcessingTxt2Img, + ) + except ImportError: + return {} + + p = None + for arg in args: + if isinstance( + arg, (StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img) + ): + p = arg + break + + if p is None: + return {} + + info = { + "prompt": p.prompt, + "negative_prompt": p.negative_prompt, + "n_iter": p.n_iter, + "batch_size": p.batch_size, + "width": p.width, + "height": p.height, + "sampler_name": p.sampler_name, + "enable_hr": getattr(p, "enable_hr", False), + "hr_upscaler": getattr(p, "hr_upscaler", ""), + } + + info.update(sd_models()) + return info + + +def sd_models() -> dict[str, str]: + try: + from modules import shared + + opts = shared.opts + except Exception: + return {} + + return { + "checkpoint": getattr(opts, "sd_model_checkpoint", "------"), + "vae": getattr(opts, "sd_vae", "------"), + "unet": getattr(opts, "sd_unet", "------"), + } + + +def ad_args(*args: Any) -> dict[str, Any]: + ad_args = [ + arg + for arg in args + if isinstance(arg, dict) and arg.get("ad_model", "None") != "None" + ] + if not ad_args: + return {} + + arg0 = ad_args[0] + is_api = arg0.get("is_api", True) + return { + "version": __version__, + "ad_model": arg0["ad_model"], + "ad_prompt": arg0.get("ad_prompt", ""), + "ad_negative_prompt": arg0.get("ad_negative_prompt", ""), + "ad_controlnet_model": arg0.get("ad_controlnet_model", "None"), + "is_api": type(is_api) is not tuple, + } + + +def library_version(): + libraries = ["torch", "torchvision", "ultralytics", "mediapipe"] + d = {} + for lib in libraries: + try: + d[lib] = version(lib) + except Exception: + d[lib] = "Unknown" + return d + + +def sys_info() -> dict[str, Any]: + try: + import launch + + version = launch.git_tag() + commit = launch.commit_hash() + except Exception: + version = "Unknown (too old or vladmandic)" + commit = "Unknown" + + return { + "Platform": platform.platform(), + "Python": sys.version, + "Version": version, + "Commit": commit, + "Commandline": sys.argv, + "Libraries": library_version(), + } + + +def get_table(title: str, data: dict[str, Any]) -> Table: + table = Table(title=title, highlight=True) + table.add_column(" ", justify="right", style="dim") + table.add_column("Value") + for key, value in data.items(): + if not isinstance(value, str): + value = repr(value) + table.add_row(key, value) + + return table + + +def rich_traceback(func: Callable) -> Callable: + def wrapper(*args, **kwargs): + string = io.StringIO() + width = Console().width + width = width - 4 if width > 4 else None + console = Console(file=string, width=width) + try: + return func(*args, **kwargs) + except Exception as e: + tables = [ + get_table(title, data) + for title, data in [ + ("System info", sys_info()), + ("Inputs", processing(*args)), + ("ADetailer", ad_args(*args)), + ] + if data + ] + tables.append(Traceback()) + + console.print(Panel(Group(*tables))) + output = "\n" + string.getvalue() + + try: + error = e.__class__(output) + except Exception: + error = RuntimeError(output) + raise error from None + + return wrapper diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/ui.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/ui.py new file mode 100644 index 0000000000000000000000000000000000000000..b05e3aa351ecdaf804943324eb6d53219b142cf8 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/ui.py @@ -0,0 +1,605 @@ +from __future__ import annotations + +from dataclasses import dataclass +from functools import partial +from types import SimpleNamespace +from typing import Any, Callable + +import gradio as gr + +from adetailer import AFTER_DETAILER, __version__ +from adetailer.args import AD_ENABLE, ALL_ARGS, MASK_MERGE_INVERT +from controlnet_ext import controlnet_exists, get_cn_models + +cn_module_choices = [ + "inpaint_global_harmonious", + "inpaint_only", + "inpaint_only+lama", +] + + +class Widgets(SimpleNamespace): + def tolist(self): + return [getattr(self, attr) for attr in ALL_ARGS.attrs] + + +@dataclass +class WebuiInfo: + ad_model_list: list[str] + sampler_names: list[str] + t2i_button: gr.Button + i2i_button: gr.Button + checkpoints_list: list[str] + vae_list: list[str] + + +def gr_interactive(value: bool = True): + return gr.update(interactive=value) + + +def ordinal(n: int) -> str: + d = {1: "st", 2: "nd", 3: "rd"} + return str(n) + ("th" if 11 <= n % 100 <= 13 else d.get(n % 10, "th")) + + +def suffix(n: int, c: str = " ") -> str: + return "" if n == 0 else c + ordinal(n + 1) + + +def on_widget_change(state: dict, value: Any, *, attr: str): + state[attr] = value + return state + + +def on_generate_click(state: dict, *values: Any): + for attr, value in zip(ALL_ARGS.attrs, values): + state[attr] = value + state["is_api"] = () + return state + + +def on_cn_model_update(cn_model: str): + if "inpaint" in cn_model: + return gr.update( + visible=True, choices=cn_module_choices, value=cn_module_choices[0] + ) + return gr.update(visible=False, choices=["None"], value="None") + + +def elem_id(item_id: str, n: int, is_img2img: bool) -> str: + tap = "img2img" if is_img2img else "txt2img" + suf = suffix(n, "_") + return f"script_{tap}_adetailer_{item_id}{suf}" + + +def adui( + num_models: int, + is_img2img: bool, + webui_info: WebuiInfo, +): + states = [] + infotext_fields = [] + eid = partial(elem_id, n=0, is_img2img=is_img2img) + + with gr.Accordion(AFTER_DETAILER, open=False, elem_id=eid("ad_main_accordion")): + with gr.Row(): + with gr.Column(scale=6): + ad_enable = gr.Checkbox( + label="Enable ADetailer", + value=False, + visible=True, + elem_id=eid("ad_enable"), + ) + + with gr.Column(scale=1, min_width=180): + gr.Markdown( + f"v{__version__}", + elem_id=eid("ad_version"), + ) + + infotext_fields.append((ad_enable, AD_ENABLE.name)) + + with gr.Group(), gr.Tabs(): + for n in range(num_models): + with gr.Tab(ordinal(n + 1)): + state, infofields = one_ui_group( + n=n, + is_img2img=is_img2img, + webui_info=webui_info, + ) + + states.append(state) + infotext_fields.extend(infofields) + + # components: [bool, dict, dict, ...] + components = [ad_enable, *states] + return components, infotext_fields + + +def one_ui_group(n: int, is_img2img: bool, webui_info: WebuiInfo): + w = Widgets() + state = gr.State({}) + eid = partial(elem_id, n=n, is_img2img=is_img2img) + + with gr.Row(): + model_choices = ( + [*webui_info.ad_model_list, "None"] + if n == 0 + else ["None", *webui_info.ad_model_list] + ) + + w.ad_model = gr.Dropdown( + label="ADetailer model" + suffix(n), + choices=model_choices, + value=model_choices[0], + visible=True, + type="value", + elem_id=eid("ad_model"), + ) + + with gr.Group(): + with gr.Row(elem_id=eid("ad_toprow_prompt")): + w.ad_prompt = gr.Textbox( + label="ad_prompt" + suffix(n), + show_label=False, + lines=3, + placeholder="ADetailer prompt" + + suffix(n) + + "\nIf blank, the main prompt is used.", + elem_id=eid("ad_prompt"), + ) + + with gr.Row(elem_id=eid("ad_toprow_negative_prompt")): + w.ad_negative_prompt = gr.Textbox( + label="ad_negative_prompt" + suffix(n), + show_label=False, + lines=2, + placeholder="ADetailer negative prompt" + + suffix(n) + + "\nIf blank, the main negative prompt is used.", + elem_id=eid("ad_negative_prompt"), + ) + + with gr.Group(): + with gr.Accordion( + "Detection", open=False, elem_id=eid("ad_detection_accordion") + ): + detection(w, n, is_img2img) + + with gr.Accordion( + "Mask Preprocessing", + open=False, + elem_id=eid("ad_mask_preprocessing_accordion"), + ): + mask_preprocessing(w, n, is_img2img) + + with gr.Accordion( + "Inpainting", open=False, elem_id=eid("ad_inpainting_accordion") + ): + inpainting(w, n, is_img2img, webui_info) + + with gr.Group(): + controlnet(w, n, is_img2img) + + all_inputs = [state, *w.tolist()] + target_button = webui_info.i2i_button if is_img2img else webui_info.t2i_button + target_button.click( + fn=on_generate_click, inputs=all_inputs, outputs=state, queue=False + ) + + infotext_fields = [(getattr(w, attr), name + suffix(n)) for attr, name in ALL_ARGS] + + return state, infotext_fields + + +def detection(w: Widgets, n: int, is_img2img: bool): + eid = partial(elem_id, n=n, is_img2img=is_img2img) + + with gr.Row(): + with gr.Column(variant="compact"): + w.ad_confidence = gr.Slider( + label="Detection model confidence threshold" + suffix(n), + minimum=0.0, + maximum=1.0, + step=0.01, + value=0.3, + visible=True, + elem_id=eid("ad_confidence"), + ) + w.ad_mask_k_largest = gr.Slider( + label="Mask only the top k largest (0 to disable)" + suffix(n), + minumum=0, + maximum=10, + step=1, + value=0, + visible=True, + elem_id=eid("ad_mask_k_largest"), + ) + + with gr.Column(variant="compact"): + w.ad_mask_min_ratio = gr.Slider( + label="Mask min area ratio" + suffix(n), + minimum=0.0, + maximum=1.0, + step=0.001, + value=0.0, + visible=True, + elem_id=eid("ad_mask_min_ratio"), + ) + w.ad_mask_max_ratio = gr.Slider( + label="Mask max area ratio" + suffix(n), + minimum=0.0, + maximum=1.0, + step=0.001, + value=1.0, + visible=True, + elem_id=eid("ad_mask_max_ratio"), + ) + + +def mask_preprocessing(w: Widgets, n: int, is_img2img: bool): + eid = partial(elem_id, n=n, is_img2img=is_img2img) + + with gr.Group(): + with gr.Row(): + with gr.Column(variant="compact"): + w.ad_x_offset = gr.Slider( + label="Mask x(→) offset" + suffix(n), + minimum=-200, + maximum=200, + step=1, + value=0, + visible=True, + elem_id=eid("ad_x_offset"), + ) + w.ad_y_offset = gr.Slider( + label="Mask y(↑) offset" + suffix(n), + minimum=-200, + maximum=200, + step=1, + value=0, + visible=True, + elem_id=eid("ad_y_offset"), + ) + + with gr.Column(variant="compact"): + w.ad_dilate_erode = gr.Slider( + label="Mask erosion (-) / dilation (+)" + suffix(n), + minimum=-128, + maximum=128, + step=4, + value=4, + visible=True, + elem_id=eid("ad_dilate_erode"), + ) + + with gr.Row(): + w.ad_mask_merge_invert = gr.Radio( + label="Mask merge mode" + suffix(n), + choices=MASK_MERGE_INVERT, + value="None", + elem_id=eid("ad_mask_merge_invert"), + ) + + +def inpainting(w: Widgets, n: int, is_img2img: bool, webui_info: WebuiInfo): + eid = partial(elem_id, n=n, is_img2img=is_img2img) + + with gr.Group(): + with gr.Row(): + w.ad_mask_blur = gr.Slider( + label="Inpaint mask blur" + suffix(n), + minimum=0, + maximum=64, + step=1, + value=4, + visible=True, + elem_id=eid("ad_mask_blur"), + ) + + w.ad_denoising_strength = gr.Slider( + label="Inpaint denoising strength" + suffix(n), + minimum=0.0, + maximum=1.0, + step=0.01, + value=0.4, + visible=True, + elem_id=eid("ad_denoising_strength"), + ) + + with gr.Row(): + with gr.Column(variant="compact"): + w.ad_inpaint_only_masked = gr.Checkbox( + label="Inpaint only masked" + suffix(n), + value=True, + visible=True, + elem_id=eid("ad_inpaint_only_masked"), + ) + w.ad_inpaint_only_masked_padding = gr.Slider( + label="Inpaint only masked padding, pixels" + suffix(n), + minimum=0, + maximum=256, + step=4, + value=32, + visible=True, + elem_id=eid("ad_inpaint_only_masked_padding"), + ) + + w.ad_inpaint_only_masked.change( + gr_interactive, + inputs=w.ad_inpaint_only_masked, + outputs=w.ad_inpaint_only_masked_padding, + queue=False, + ) + + with gr.Column(variant="compact"): + w.ad_use_inpaint_width_height = gr.Checkbox( + label="Use separate width/height" + suffix(n), + value=False, + visible=True, + elem_id=eid("ad_use_inpaint_width_height"), + ) + + w.ad_inpaint_width = gr.Slider( + label="inpaint width" + suffix(n), + minimum=64, + maximum=2048, + step=4, + value=512, + visible=True, + elem_id=eid("ad_inpaint_width"), + ) + + w.ad_inpaint_height = gr.Slider( + label="inpaint height" + suffix(n), + minimum=64, + maximum=2048, + step=4, + value=512, + visible=True, + elem_id=eid("ad_inpaint_height"), + ) + + w.ad_use_inpaint_width_height.change( + lambda value: (gr_interactive(value), gr_interactive(value)), + inputs=w.ad_use_inpaint_width_height, + outputs=[w.ad_inpaint_width, w.ad_inpaint_height], + queue=False, + ) + + with gr.Row(): + with gr.Column(variant="compact"): + w.ad_use_steps = gr.Checkbox( + label="Use separate steps" + suffix(n), + value=False, + visible=True, + elem_id=eid("ad_use_steps"), + ) + + w.ad_steps = gr.Slider( + label="ADetailer steps" + suffix(n), + minimum=1, + maximum=150, + step=1, + value=28, + visible=True, + elem_id=eid("ad_steps"), + ) + + w.ad_use_steps.change( + gr_interactive, + inputs=w.ad_use_steps, + outputs=w.ad_steps, + queue=False, + ) + + with gr.Column(variant="compact"): + w.ad_use_cfg_scale = gr.Checkbox( + label="Use separate CFG scale" + suffix(n), + value=False, + visible=True, + elem_id=eid("ad_use_cfg_scale"), + ) + + w.ad_cfg_scale = gr.Slider( + label="ADetailer CFG scale" + suffix(n), + minimum=0.0, + maximum=30.0, + step=0.5, + value=7.0, + visible=True, + elem_id=eid("ad_cfg_scale"), + ) + + w.ad_use_cfg_scale.change( + gr_interactive, + inputs=w.ad_use_cfg_scale, + outputs=w.ad_cfg_scale, + queue=False, + ) + + with gr.Row(): + with gr.Column(variant="compact"): + w.ad_use_checkpoint = gr.Checkbox( + label="Use separate checkpoint (experimental)" + suffix(n), + value=False, + visible=True, + elem_id=eid("ad_use_checkpoint"), + ) + + ckpts = ["Use same checkpoint", *webui_info.checkpoints_list] + + w.ad_checkpoint = gr.Dropdown( + label="ADetailer checkpoint" + suffix(n), + choices=ckpts, + value=ckpts[0], + visible=True, + elem_id=eid("ad_checkpoint"), + ) + + with gr.Column(variant="compact"): + w.ad_use_vae = gr.Checkbox( + label="Use separate VAE (experimental)" + suffix(n), + value=False, + visible=True, + elem_id=eid("ad_use_vae"), + ) + + vaes = ["Use same VAE", *webui_info.vae_list] + + w.ad_vae = gr.Dropdown( + label="ADetailer VAE" + suffix(n), + choices=vaes, + value=vaes[0], + visible=True, + elem_id=eid("ad_vae"), + ) + + with gr.Row(), gr.Column(variant="compact"): + w.ad_use_sampler = gr.Checkbox( + label="Use separate sampler" + suffix(n), + value=False, + visible=True, + elem_id=eid("ad_use_sampler"), + ) + + w.ad_sampler = gr.Dropdown( + label="ADetailer sampler" + suffix(n), + choices=webui_info.sampler_names, + value=webui_info.sampler_names[0], + visible=True, + elem_id=eid("ad_sampler"), + ) + + w.ad_use_sampler.change( + gr_interactive, + inputs=w.ad_use_sampler, + outputs=w.ad_sampler, + queue=False, + ) + + with gr.Row(): + with gr.Column(variant="compact"): + w.ad_use_noise_multiplier = gr.Checkbox( + label="Use separate noise multiplier" + suffix(n), + value=False, + visible=True, + elem_id=eid("ad_use_noise_multiplier"), + ) + + w.ad_noise_multiplier = gr.Slider( + label="Noise multiplier for img2img" + suffix(n), + minimum=0.5, + maximum=1.5, + step=0.01, + value=1.0, + visible=True, + elem_id=eid("ad_noise_multiplier"), + ) + + w.ad_use_noise_multiplier.change( + gr_interactive, + inputs=w.ad_use_noise_multiplier, + outputs=w.ad_noise_multiplier, + queue=False, + ) + + with gr.Column(variant="compact"): + w.ad_use_clip_skip = gr.Checkbox( + label="Use separate CLIP skip" + suffix(n), + value=False, + visible=True, + elem_id=eid("ad_use_clip_skip"), + ) + + w.ad_clip_skip = gr.Slider( + label="ADetailer CLIP skip" + suffix(n), + minimum=1, + maximum=12, + step=1, + value=1, + visible=True, + elem_id=eid("ad_clip_skip"), + ) + + w.ad_use_clip_skip.change( + gr_interactive, + inputs=w.ad_use_clip_skip, + outputs=w.ad_clip_skip, + queue=False, + ) + + with gr.Row(), gr.Column(variant="compact"): + w.ad_restore_face = gr.Checkbox( + label="Restore faces after ADetailer" + suffix(n), + value=False, + elem_id=eid("ad_restore_face"), + ) + + +def controlnet(w: Widgets, n: int, is_img2img: bool): + eid = partial(elem_id, n=n, is_img2img=is_img2img) + cn_models = ["None", *get_cn_models()] + + with gr.Row(variant="panel"): + with gr.Column(variant="compact"): + w.ad_controlnet_model = gr.Dropdown( + label="ControlNet model" + suffix(n), + choices=cn_models, + value="None", + visible=True, + type="value", + interactive=controlnet_exists, + elem_id=eid("ad_controlnet_model"), + ) + + w.ad_controlnet_module = gr.Dropdown( + label="ControlNet module" + suffix(n), + choices=cn_module_choices, + value="inpaint_global_harmonious", + visible=False, + type="value", + interactive=controlnet_exists, + elem_id=eid("ad_controlnet_module"), + ) + + w.ad_controlnet_weight = gr.Slider( + label="ControlNet weight" + suffix(n), + minimum=0.0, + maximum=1.0, + step=0.01, + value=1.0, + visible=True, + interactive=controlnet_exists, + elem_id=eid("ad_controlnet_weight"), + ) + + w.ad_controlnet_model.change( + on_cn_model_update, + inputs=w.ad_controlnet_model, + outputs=w.ad_controlnet_module, + queue=False, + ) + + with gr.Column(variant="compact"): + w.ad_controlnet_guidance_start = gr.Slider( + label="ControlNet guidance start" + suffix(n), + minimum=0.0, + maximum=1.0, + step=0.01, + value=0.0, + visible=True, + interactive=controlnet_exists, + elem_id=eid("ad_controlnet_guidance_start"), + ) + + w.ad_controlnet_guidance_end = gr.Slider( + label="ControlNet guidance end" + suffix(n), + minimum=0.0, + maximum=1.0, + step=0.01, + value=1.0, + visible=True, + interactive=controlnet_exists, + elem_id=eid("ad_controlnet_guidance_end"), + ) diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/ultralytics.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/ultralytics.py new file mode 100644 index 0000000000000000000000000000000000000000..addd0f5c01435a03d899a562d5e5e8b113fabcbc --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/adetailer/ultralytics.py @@ -0,0 +1,50 @@ +from __future__ import annotations + +from pathlib import Path + +import cv2 +from PIL import Image +from torchvision.transforms.functional import to_pil_image +from ultralytics import YOLO + +from adetailer import PredictOutput +from adetailer.common import create_mask_from_bbox + + +def ultralytics_predict( + model_path: str | Path, + image: Image.Image, + confidence: float = 0.3, + device: str = "", +) -> PredictOutput: + model = YOLO(model_path) + pred = model(image, conf=confidence, device=device) + + bboxes = pred[0].boxes.xyxy.cpu().numpy() + if bboxes.size == 0: + return PredictOutput() + bboxes = bboxes.tolist() + + if pred[0].masks is None: + masks = create_mask_from_bbox(bboxes, image.size) + else: + masks = mask_to_pil(pred[0].masks.data, image.size) + preview = pred[0].plot() + preview = cv2.cvtColor(preview, cv2.COLOR_BGR2RGB) + preview = Image.fromarray(preview) + + return PredictOutput(bboxes=bboxes, masks=masks, preview=preview) + + +def mask_to_pil(masks, shape: tuple[int, int]) -> list[Image.Image]: + """ + Parameters + ---------- + masks: torch.Tensor, dtype=torch.float32, shape=(N, H, W). + The device can be CUDA, but `to_pil_image` takes care of that. + + shape: tuple[int, int] + (width, height) of the original image + """ + n = masks.shape[0] + return [to_pil_image(masks[i], mode="L").resize(shape) for i in range(n)] diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/controlnet_ext/__init__.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/controlnet_ext/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..0ab666835157561426d684d798735e724a5a4dbe --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/controlnet_ext/__init__.py @@ -0,0 +1,7 @@ +from .controlnet_ext import ControlNetExt, controlnet_exists, get_cn_models + +__all__ = [ + "ControlNetExt", + "controlnet_exists", + "get_cn_models", +] diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/controlnet_ext/controlnet_ext.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/controlnet_ext/controlnet_ext.py new file mode 100644 index 0000000000000000000000000000000000000000..ec86d46b713c2ee39d9bdd615411494cfe05aba3 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/controlnet_ext/controlnet_ext.py @@ -0,0 +1,140 @@ +from __future__ import annotations + +import importlib +import re +from functools import lru_cache +from pathlib import Path + +from modules import extensions, sd_models, shared +from modules.paths import data_path, models_path, script_path + +ext_path = Path(data_path, "extensions") +ext_builtin_path = Path(script_path, "extensions-builtin") +controlnet_exists = False +controlnet_path = None +cn_base_path = "" + +for extension in extensions.active(): + if not extension.enabled: + continue + # For cases like sd-webui-controlnet-master + if "sd-webui-controlnet" in extension.name: + controlnet_exists = True + controlnet_path = Path(extension.path) + cn_base_path = ".".join(controlnet_path.parts[-2:]) + break + +cn_model_module = { + "inpaint": "inpaint_global_harmonious", + "scribble": "t2ia_sketch_pidi", + "lineart": "lineart_coarse", + "openpose": "openpose_full", + "tile": None, +} +cn_model_regex = re.compile("|".join(cn_model_module.keys())) + + +class ControlNetExt: + def __init__(self): + self.cn_models = ["None"] + self.cn_available = False + self.external_cn = None + + def init_controlnet(self): + import_path = cn_base_path + ".scripts.external_code" + + self.external_cn = importlib.import_module(import_path, "external_code") + self.cn_available = True + models = self.external_cn.get_models() + self.cn_models.extend(m for m in models if cn_model_regex.search(m)) + + def update_scripts_args( + self, + p, + model: str, + module: str | None, + weight: float, + guidance_start: float, + guidance_end: float, + ): + if (not self.cn_available) or model == "None": + return + + if module is None: + for m, v in cn_model_module.items(): + if m in model: + module = v + break + + cn_units = [ + self.external_cn.ControlNetUnit( + model=model, + weight=weight, + control_mode=self.external_cn.ControlMode.BALANCED, + module=module, + guidance_start=guidance_start, + guidance_end=guidance_end, + pixel_perfect=True, + ) + ] + + self.external_cn.update_cn_script_in_processing(p, cn_units) + + +def get_cn_model_dirs() -> list[Path]: + cn_model_dir = Path(models_path, "ControlNet") + if controlnet_path is not None: + cn_model_dir_old = controlnet_path.joinpath("models") + else: + cn_model_dir_old = None + ext_dir1 = shared.opts.data.get("control_net_models_path", "") + ext_dir2 = getattr(shared.cmd_opts, "controlnet_dir", "") + + dirs = [cn_model_dir] + for ext_dir in [cn_model_dir_old, ext_dir1, ext_dir2]: + if ext_dir: + dirs.append(Path(ext_dir)) + + return dirs + + +@lru_cache +def _get_cn_models() -> list[str]: + """ + Since we can't import ControlNet, we use a function that does something like + controlnet's `list(global_state.cn_models_names.values())`. + """ + cn_model_exts = (".pt", ".pth", ".ckpt", ".safetensors") + dirs = get_cn_model_dirs() + name_filter = shared.opts.data.get("control_net_models_name_filter", "") + name_filter = name_filter.strip(" ").lower() + + model_paths = [] + + for base in dirs: + if not base.exists(): + continue + + for p in base.rglob("*"): + if ( + p.is_file() + and p.suffix in cn_model_exts + and cn_model_regex.search(p.name) + ): + if name_filter and name_filter not in p.name.lower(): + continue + model_paths.append(p) + model_paths.sort(key=lambda p: p.name) + + models = [] + for p in model_paths: + model_hash = sd_models.model_hash(p) + name = f"{p.stem} [{model_hash}]" + models.append(name) + return models + + +def get_cn_models() -> list[str]: + if controlnet_exists: + return _get_cn_models() + return [] diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/controlnet_ext/restore.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/controlnet_ext/restore.py new file mode 100644 index 0000000000000000000000000000000000000000..152ffd0a9732b5512e4440b795a1ad89bcb27cab --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/controlnet_ext/restore.py @@ -0,0 +1,43 @@ +from __future__ import annotations + +from contextlib import contextmanager + +from modules import img2img, processing, shared + + +class CNHijackRestore: + def __init__(self): + self.process = hasattr(processing, "__controlnet_original_process_images_inner") + self.img2img = hasattr(img2img, "__controlnet_original_process_batch") + + def __enter__(self): + if self.process: + self.orig_process = processing.process_images_inner + processing.process_images_inner = getattr( + processing, "__controlnet_original_process_images_inner" + ) + if self.img2img: + self.orig_img2img = img2img.process_batch + img2img.process_batch = getattr( + img2img, "__controlnet_original_process_batch" + ) + + def __exit__(self, *args, **kwargs): + if self.process: + processing.process_images_inner = self.orig_process + if self.img2img: + img2img.process_batch = self.orig_img2img + + +@contextmanager +def cn_allow_script_control(): + orig = False + if "control_net_allow_script_control" in shared.opts.data: + try: + orig = shared.opts.data["control_net_allow_script_control"] + shared.opts.data["control_net_allow_script_control"] = True + yield + finally: + shared.opts.data["control_net_allow_script_control"] = orig + else: + yield diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/install.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/install.py new file mode 100644 index 0000000000000000000000000000000000000000..426d8fe27249ffbb5512d91ff2865d61f337be28 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/install.py @@ -0,0 +1,76 @@ +from __future__ import annotations + +import importlib.util +import subprocess +import sys +from importlib.metadata import version # python >= 3.8 + +from packaging.version import parse + +import_name = {"py-cpuinfo": "cpuinfo", "protobuf": "google.protobuf"} + + +def is_installed( + package: str, min_version: str | None = None, max_version: str | None = None +): + name = import_name.get(package, package) + try: + spec = importlib.util.find_spec(name) + except ModuleNotFoundError: + return False + + if spec is None: + return False + + if not min_version and not max_version: + return True + + if not min_version: + min_version = "0.0.0" + if not max_version: + max_version = "99999999.99999999.99999999" + + try: + pkg_version = version(package) + return parse(min_version) <= parse(pkg_version) <= parse(max_version) + except Exception: + return False + + +def run_pip(*args): + subprocess.run([sys.executable, "-m", "pip", "install", *args]) + + +def install(): + deps = [ + # requirements + ("ultralytics", "8.0.173", None), + ("mediapipe", "0.10.3", None), + ("rich", "13.0.0", None), + # mediapipe + ("protobuf", "3.20", "3.9999"), + ] + + for pkg, low, high in deps: + if not is_installed(pkg, low, high): + if low and high: + cmd = f"{pkg}>={low},<={high}" + elif low: + cmd = f"{pkg}>={low}" + elif high: + cmd = f"{pkg}<={high}" + else: + cmd = pkg + + run_pip("-U", cmd) + + +try: + import launch + + skip_install = launch.args.skip_install +except Exception: + skip_install = False + +if not skip_install: + install() diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/preload.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/preload.py new file mode 100644 index 0000000000000000000000000000000000000000..10be161f22b0a5ef7083609829a21b547eae9aea --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/preload.py @@ -0,0 +1,9 @@ +import argparse + + +def preload(parser: argparse.ArgumentParser): + parser.add_argument( + "--ad-no-huggingface", + action="store_true", + help="Don't use adetailer models from huggingface", + ) diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/pyproject.toml b/TestingforfaivsixEx/content/eichellis/extensions/adetai/pyproject.toml new file mode 100644 index 0000000000000000000000000000000000000000..b2407ed41791a535b5c43de6644d1a1eb6fb2d44 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/pyproject.toml @@ -0,0 +1,40 @@ +[project] +name = "adetailer" +description = "An object detection and auto-mask extension for stable diffusion webui." +authors = [{ name = "dowon", email = "ks2515@naver.com" }] +requires-python = ">=3.8,<3.12" +readme = "README.md" +license = { text = "AGPL-3.0" } + +[project.urls] +repository = "https://github.com/Bing-su/adetailer" + +[tool.isort] +profile = "black" +known_first_party = ["launch", "modules"] + +[tool.ruff] +select = [ + "A", + "B", + "C4", + "C90", + "E", + "EM", + "F", + "FA", + "I001", + "ISC", + "N", + "PIE", + "PT", + "RET", + "RUF", + "SIM", + "UP", + "W", +] +ignore = ["B008", "B905", "E501", "F401", "UP007"] + +[tool.ruff.isort] +known-first-party = ["launch", "modules"] diff --git a/TestingforfaivsixEx/content/eichellis/extensions/adetai/scripts/!adetailer.py b/TestingforfaivsixEx/content/eichellis/extensions/adetai/scripts/!adetailer.py new file mode 100644 index 0000000000000000000000000000000000000000..544e7b2775cd656089f4e42987a94e22235ed3e6 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/adetai/scripts/!adetailer.py @@ -0,0 +1,834 @@ +from __future__ import annotations + +import os +import platform +import re +import sys +import traceback +from contextlib import contextmanager +from copy import copy, deepcopy +from functools import partial +from pathlib import Path +from textwrap import dedent +from typing import Any + +import gradio as gr +import torch +from rich import print + +import modules +from adetailer import ( + AFTER_DETAILER, + __version__, + get_models, + mediapipe_predict, + ultralytics_predict, +) +from adetailer.args import ALL_ARGS, BBOX_SORTBY, ADetailerArgs, EnableChecker +from adetailer.common import PredictOutput +from adetailer.mask import ( + filter_by_ratio, + filter_k_largest, + mask_preprocess, + sort_bboxes, +) +from adetailer.traceback import rich_traceback +from adetailer.ui import WebuiInfo, adui, ordinal, suffix +from controlnet_ext import ControlNetExt, controlnet_exists, get_cn_models +from controlnet_ext.restore import ( + CNHijackRestore, + cn_allow_script_control, +) +from modules import images, safe, script_callbacks, scripts, shared +from modules.devices import NansException +from modules.paths import data_path, models_path +from modules.processing import ( + Processed, + StableDiffusionProcessingImg2Img, + create_infotext, + process_images, +) +from modules.sd_samplers import all_samplers +from modules.shared import cmd_opts, opts, state + +no_huggingface = getattr(cmd_opts, "ad_no_huggingface", False) +adetailer_dir = Path(models_path, "adetailer") +model_mapping = get_models(adetailer_dir, huggingface=not no_huggingface) +txt2img_submit_button = img2img_submit_button = None +SCRIPT_DEFAULT = "dynamic_prompting,dynamic_thresholding,wildcard_recursive,wildcards,lora_block_weight" + +if ( + not adetailer_dir.exists() + and adetailer_dir.parent.exists() + and os.access(adetailer_dir.parent, os.W_OK) +): + adetailer_dir.mkdir() + +print( + f"[-] ADetailer initialized. version: {__version__}, num models: {len(model_mapping)}" +) + + +@contextmanager +def change_torch_load(): + orig = torch.load + try: + torch.load = safe.unsafe_torch_load + yield + finally: + torch.load = orig + + +@contextmanager +def pause_total_tqdm(): + orig = opts.data.get("multiple_tqdm", True) + try: + opts.data["multiple_tqdm"] = False + yield + finally: + opts.data["multiple_tqdm"] = orig + + +@contextmanager +def preseve_prompts(p): + all_pt = copy(p.all_prompts) + all_ng = copy(p.all_negative_prompts) + try: + yield + finally: + p.all_prompts = all_pt + p.all_negative_prompts = all_ng + + +class AfterDetailerScript(scripts.Script): + def __init__(self): + super().__init__() + self.ultralytics_device = self.get_ultralytics_device() + + self.controlnet_ext = None + + def __repr__(self): + return f"{self.__class__.__name__}(version={__version__})" + + def title(self): + return AFTER_DETAILER + + def show(self, is_img2img): + return scripts.AlwaysVisible + + def ui(self, is_img2img): + num_models = opts.data.get("ad_max_models", 2) + ad_model_list = list(model_mapping.keys()) + sampler_names = [sampler.name for sampler in all_samplers] + + try: + checkpoint_list = modules.sd_models.checkpoint_tiles(use_shorts=True) + except TypeError: + checkpoint_list = modules.sd_models.checkpoint_tiles() + vae_list = modules.shared_items.sd_vae_items() + + webui_info = WebuiInfo( + ad_model_list=ad_model_list, + sampler_names=sampler_names, + t2i_button=txt2img_submit_button, + i2i_button=img2img_submit_button, + checkpoints_list=checkpoint_list, + vae_list=vae_list, + ) + + components, infotext_fields = adui(num_models, is_img2img, webui_info) + + self.infotext_fields = infotext_fields + return components + + def init_controlnet_ext(self) -> None: + if self.controlnet_ext is not None: + return + self.controlnet_ext = ControlNetExt() + + if controlnet_exists: + try: + self.controlnet_ext.init_controlnet() + except ImportError: + error = traceback.format_exc() + print( + f"[-] ADetailer: ControlNetExt init failed:\n{error}", + file=sys.stderr, + ) + + def update_controlnet_args(self, p, args: ADetailerArgs) -> None: + if self.controlnet_ext is None: + self.init_controlnet_ext() + + if ( + self.controlnet_ext is not None + and self.controlnet_ext.cn_available + and args.ad_controlnet_model != "None" + ): + self.controlnet_ext.update_scripts_args( + p, + model=args.ad_controlnet_model, + module=args.ad_controlnet_module, + weight=args.ad_controlnet_weight, + guidance_start=args.ad_controlnet_guidance_start, + guidance_end=args.ad_controlnet_guidance_end, + ) + + def is_ad_enabled(self, *args_) -> bool: + arg_list = [arg for arg in args_ if isinstance(arg, dict)] + if not args_ or not arg_list or not isinstance(args_[0], (bool, dict)): + message = f""" + [-] ADetailer: Invalid arguments passed to ADetailer. + input: {args_!r} + ADetailer disabled. + """ + print(dedent(message), file=sys.stderr) + return False + enable = args_[0] if isinstance(args_[0], bool) else True + checker = EnableChecker(enable=enable, arg_list=arg_list) + return checker.is_enabled() + + def get_args(self, p, *args_) -> list[ADetailerArgs]: + """ + `args_` is at least 1 in length by `is_ad_enabled` immediately above + """ + args = [arg for arg in args_ if isinstance(arg, dict)] + + if not args: + message = f"[-] ADetailer: Invalid arguments passed to ADetailer: {args_!r}" + raise ValueError(message) + + if hasattr(p, "adetailer_xyz"): + args[0] = {**args[0], **p.adetailer_xyz} + + all_inputs = [] + + for n, arg_dict in enumerate(args, 1): + try: + inp = ADetailerArgs(**arg_dict) + except ValueError as e: + msgs = [ + f"[-] ADetailer: ValidationError when validating {ordinal(n)} arguments: {e}\n" + ] + for attr in ALL_ARGS.attrs: + arg = arg_dict.get(attr) + dtype = type(arg) + arg = "DEFAULT" if arg is None else repr(arg) + msgs.append(f" {attr}: {arg} ({dtype})") + raise ValueError("\n".join(msgs)) from e + + all_inputs.append(inp) + + return all_inputs + + def extra_params(self, arg_list: list[ADetailerArgs]) -> dict: + params = {} + for n, args in enumerate(arg_list): + params.update(args.extra_params(suffix=suffix(n))) + params["ADetailer version"] = __version__ + return params + + @staticmethod + def get_ultralytics_device() -> str: + if "adetailer" in shared.cmd_opts.use_cpu: + return "cpu" + + if platform.system() == "Darwin": + return "" + + vram_args = ["lowvram", "medvram", "medvram_sdxl"] + if any(getattr(cmd_opts, vram, False) for vram in vram_args): + return "cpu" + + return "" + + def prompt_blank_replacement( + self, all_prompts: list[str], i: int, default: str + ) -> str: + if not all_prompts: + return default + if i < len(all_prompts): + return all_prompts[i] + j = i % len(all_prompts) + return all_prompts[j] + + def _get_prompt( + self, ad_prompt: str, all_prompts: list[str], i: int, default: str + ) -> list[str]: + prompts = re.split(r"\s*\[SEP\]\s*", ad_prompt) + blank_replacement = self.prompt_blank_replacement(all_prompts, i, default) + for n in range(len(prompts)): + if not prompts[n]: + prompts[n] = blank_replacement + elif "[PROMPT]" in prompts[n]: + prompts[n] = prompts[n].replace("[PROMPT]", f" {blank_replacement} ") + return prompts + + def get_prompt(self, p, args: ADetailerArgs) -> tuple[list[str], list[str]]: + i = p._ad_idx + + prompt = self._get_prompt(args.ad_prompt, p.all_prompts, i, p.prompt) + negative_prompt = self._get_prompt( + args.ad_negative_prompt, p.all_negative_prompts, i, p.negative_prompt + ) + + return prompt, negative_prompt + + def get_seed(self, p) -> tuple[int, int]: + i = p._ad_idx + + if not p.all_seeds: + seed = p.seed + elif i < len(p.all_seeds): + seed = p.all_seeds[i] + else: + j = i % len(p.all_seeds) + seed = p.all_seeds[j] + + if not p.all_subseeds: + subseed = p.subseed + elif i < len(p.all_subseeds): + subseed = p.all_subseeds[i] + else: + j = i % len(p.all_subseeds) + subseed = p.all_subseeds[j] + + return seed, subseed + + def get_width_height(self, p, args: ADetailerArgs) -> tuple[int, int]: + if args.ad_use_inpaint_width_height: + width = args.ad_inpaint_width + height = args.ad_inpaint_height + else: + width = p.width + height = p.height + + return width, height + + def get_steps(self, p, args: ADetailerArgs) -> int: + return args.ad_steps if args.ad_use_steps else p.steps + + def get_cfg_scale(self, p, args: ADetailerArgs) -> float: + return args.ad_cfg_scale if args.ad_use_cfg_scale else p.cfg_scale + + def get_sampler(self, p, args: ADetailerArgs) -> str: + return args.ad_sampler if args.ad_use_sampler else p.sampler_name + + def get_override_settings(self, p, args: ADetailerArgs) -> dict[str, Any]: + d = {} + + if args.ad_use_clip_skip: + d["CLIP_stop_at_last_layers"] = args.ad_clip_skip + + if ( + args.ad_use_checkpoint + and args.ad_checkpoint + and args.ad_checkpoint not in ("None", "Use same checkpoint") + ): + d["sd_model_checkpoint"] = args.ad_checkpoint + + if ( + args.ad_use_vae + and args.ad_vae + and args.ad_vae not in ("None", "Use same VAE") + ): + d["sd_vae"] = args.ad_vae + return d + + def get_initial_noise_multiplier(self, p, args: ADetailerArgs) -> float | None: + return args.ad_noise_multiplier if args.ad_use_noise_multiplier else None + + @staticmethod + def infotext(p) -> str: + return create_infotext( + p, p.all_prompts, p.all_seeds, p.all_subseeds, None, 0, 0 + ) + + def write_params_txt(self, p) -> None: + infotext = self.infotext(p) + params_txt = Path(data_path, "params.txt") + params_txt.write_text(infotext, encoding="utf-8") + + def script_filter(self, p, args: ADetailerArgs): + script_runner = copy(p.scripts) + script_args = deepcopy(p.script_args) + self.disable_controlnet_units(script_args) + + ad_only_seleted_scripts = opts.data.get("ad_only_seleted_scripts", True) + if not ad_only_seleted_scripts: + return script_runner, script_args + + ad_script_names = opts.data.get("ad_script_names", SCRIPT_DEFAULT) + script_names_set = { + name + for script_name in ad_script_names.split(",") + for name in (script_name, script_name.strip()) + } + + if args.ad_controlnet_model != "None": + script_names_set.add("controlnet") + + filtered_alwayson = [] + for script_object in script_runner.alwayson_scripts: + filepath = script_object.filename + filename = Path(filepath).stem + if filename in script_names_set: + filtered_alwayson.append(script_object) + + script_runner.alwayson_scripts = filtered_alwayson + return script_runner, script_args + + def disable_controlnet_units(self, script_args: list[Any]) -> None: + for obj in script_args: + if "controlnet" in obj.__class__.__name__.lower(): + if hasattr(obj, "enabled"): + obj.enabled = False + if hasattr(obj, "input_mode"): + obj.input_mode = getattr(obj.input_mode, "SIMPLE", "simple") + + elif isinstance(obj, dict) and "module" in obj: + obj["enabled"] = False + + def get_i2i_p(self, p, args: ADetailerArgs, image): + seed, subseed = self.get_seed(p) + width, height = self.get_width_height(p, args) + steps = self.get_steps(p, args) + cfg_scale = self.get_cfg_scale(p, args) + initial_noise_multiplier = self.get_initial_noise_multiplier(p, args) + sampler_name = self.get_sampler(p, args) + override_settings = self.get_override_settings(p, args) + + i2i = StableDiffusionProcessingImg2Img( + init_images=[image], + resize_mode=0, + denoising_strength=args.ad_denoising_strength, + mask=None, + mask_blur=args.ad_mask_blur, + inpainting_fill=1, + inpaint_full_res=args.ad_inpaint_only_masked, + inpaint_full_res_padding=args.ad_inpaint_only_masked_padding, + inpainting_mask_invert=0, + initial_noise_multiplier=initial_noise_multiplier, + sd_model=p.sd_model, + outpath_samples=p.outpath_samples, + outpath_grids=p.outpath_grids, + prompt="", # replace later + negative_prompt="", + styles=p.styles, + seed=seed, + subseed=subseed, + subseed_strength=p.subseed_strength, + seed_resize_from_h=p.seed_resize_from_h, + seed_resize_from_w=p.seed_resize_from_w, + sampler_name=sampler_name, + batch_size=1, + n_iter=1, + steps=steps, + cfg_scale=cfg_scale, + width=width, + height=height, + restore_faces=args.ad_restore_face, + tiling=p.tiling, + extra_generation_params=p.extra_generation_params, + do_not_save_samples=True, + do_not_save_grid=True, + override_settings=override_settings, + ) + + i2i.cached_c = [None, None] + i2i.cached_uc = [None, None] + i2i.scripts, i2i.script_args = self.script_filter(p, args) + i2i._ad_disabled = True + + if args.ad_controlnet_model != "None": + self.update_controlnet_args(i2i, args) + else: + i2i.control_net_enabled = False + + return i2i + + def save_image(self, p, image, *, condition: str, suffix: str) -> None: + i = p._ad_idx + if p.all_prompts: + i %= len(p.all_prompts) + save_prompt = p.all_prompts[i] + else: + save_prompt = p.prompt + seed, _ = self.get_seed(p) + + if opts.data.get(condition, False): + images.save_image( + image=image, + path=p.outpath_samples, + basename="", + seed=seed, + prompt=save_prompt, + extension=opts.samples_format, + info=self.infotext(p), + p=p, + suffix=suffix, + ) + + def get_ad_model(self, name: str): + if name not in model_mapping: + msg = f"[-] ADetailer: Model {name!r} not found. Available models: {list(model_mapping.keys())}" + raise ValueError(msg) + return model_mapping[name] + + def sort_bboxes(self, pred: PredictOutput) -> PredictOutput: + sortby = opts.data.get("ad_bbox_sortby", BBOX_SORTBY[0]) + sortby_idx = BBOX_SORTBY.index(sortby) + return sort_bboxes(pred, sortby_idx) + + def pred_preprocessing(self, pred: PredictOutput, args: ADetailerArgs): + pred = filter_by_ratio( + pred, low=args.ad_mask_min_ratio, high=args.ad_mask_max_ratio + ) + pred = filter_k_largest(pred, k=args.ad_mask_k_largest) + pred = self.sort_bboxes(pred) + return mask_preprocess( + pred.masks, + kernel=args.ad_dilate_erode, + x_offset=args.ad_x_offset, + y_offset=args.ad_y_offset, + merge_invert=args.ad_mask_merge_invert, + ) + + @staticmethod + def ensure_rgb_image(image: Any): + if hasattr(image, "mode") and image.mode != "RGB": + image = image.convert("RGB") + return image + + @staticmethod + def i2i_prompts_replace( + i2i, prompts: list[str], negative_prompts: list[str], j: int + ) -> None: + i1 = min(j, len(prompts) - 1) + i2 = min(j, len(negative_prompts) - 1) + prompt = prompts[i1] + negative_prompt = negative_prompts[i2] + i2i.prompt = prompt + i2i.negative_prompt = negative_prompt + + @staticmethod + def compare_prompt(p, processed, n: int = 0): + if p.prompt != processed.all_prompts[0]: + print( + f"[-] ADetailer: applied {ordinal(n + 1)} ad_prompt: {processed.all_prompts[0]!r}" + ) + + if p.negative_prompt != processed.all_negative_prompts[0]: + print( + f"[-] ADetailer: applied {ordinal(n + 1)} ad_negative_prompt: {processed.all_negative_prompts[0]!r}" + ) + + @staticmethod + def need_call_process(p) -> bool: + i = p._ad_idx + bs = p.batch_size + return i % bs == bs - 1 + + @staticmethod + def need_call_postprocess(p) -> bool: + i = p._ad_idx + bs = p.batch_size + return i % bs == 0 + + @rich_traceback + def process(self, p, *args_): + if getattr(p, "_ad_disabled", False): + return + + if self.is_ad_enabled(*args_): + arg_list = self.get_args(p, *args_) + extra_params = self.extra_params(arg_list) + p.extra_generation_params.update(extra_params) + + def _postprocess_image_inner( + self, p, pp, args: ADetailerArgs, *, n: int = 0 + ) -> bool: + """ + Returns + ------- + bool + + `True` if image was processed, `False` otherwise. + """ + if state.interrupted or state.skipped: + return False + + i = p._ad_idx + + i2i = self.get_i2i_p(p, args, pp.image) + seed, subseed = self.get_seed(p) + ad_prompts, ad_negatives = self.get_prompt(p, args) + + is_mediapipe = args.ad_model.lower().startswith("mediapipe") + + kwargs = {} + if is_mediapipe: + predictor = mediapipe_predict + ad_model = args.ad_model + else: + predictor = ultralytics_predict + ad_model = self.get_ad_model(args.ad_model) + kwargs["device"] = self.ultralytics_device + + with change_torch_load(): + pred = predictor(ad_model, pp.image, args.ad_confidence, **kwargs) + + masks = self.pred_preprocessing(pred, args) + + if not masks: + print( + f"[-] ADetailer: nothing detected on image {i + 1} with {ordinal(n + 1)} settings." + ) + return False + + self.save_image( + p, + pred.preview, + condition="ad_save_previews", + suffix="-ad-preview" + suffix(n, "-"), + ) + + steps = len(masks) + processed = None + state.job_count += steps + + if is_mediapipe: + print(f"mediapipe: {steps} detected.") + + p2 = copy(i2i) + for j in range(steps): + p2.image_mask = masks[j] + p2.init_images[0] = self.ensure_rgb_image(p2.init_images[0]) + self.i2i_prompts_replace(p2, ad_prompts, ad_negatives, j) + + if re.match(r"^\s*\[SKIP\]\s*$", p2.prompt): + continue + + p2.seed = seed + j + p2.subseed = subseed + j + + try: + processed = process_images(p2) + except NansException as e: + msg = f"[-] ADetailer: 'NansException' occurred with {ordinal(n + 1)} settings.\n{e}" + print(msg, file=sys.stderr) + continue + finally: + p2.close() + + self.compare_prompt(p2, processed, n=n) + p2 = copy(i2i) + p2.init_images = [processed.images[0]] + + if processed is not None: + pp.image = processed.images[0] + return True + + return False + + @rich_traceback + def postprocess_image(self, p, pp, *args_): + if getattr(p, "_ad_disabled", False): + return + + if not self.is_ad_enabled(*args_): + return + + p._ad_idx = getattr(p, "_ad_idx", -1) + 1 + init_image = copy(pp.image) + arg_list = self.get_args(p, *args_) + + if p.scripts is not None and self.need_call_postprocess(p): + dummy = Processed(p, [], p.seed, "") + with preseve_prompts(p): + p.scripts.postprocess(copy(p), dummy) + + is_processed = False + with CNHijackRestore(), pause_total_tqdm(), cn_allow_script_control(): + for n, args in enumerate(arg_list): + if args.ad_model == "None": + continue + is_processed |= self._postprocess_image_inner(p, pp, args, n=n) + + if is_processed: + self.save_image( + p, init_image, condition="ad_save_images_before", suffix="-ad-before" + ) + + if p.scripts is not None and self.need_call_process(p): + with preseve_prompts(p): + p.scripts.process(copy(p)) + + try: + ia = p._ad_idx + lenp = len(p.all_prompts) + if ia % lenp == lenp - 1: + self.write_params_txt(p) + except Exception: + pass + + +def on_after_component(component, **_kwargs): + global txt2img_submit_button, img2img_submit_button + if getattr(component, "elem_id", None) == "txt2img_generate": + txt2img_submit_button = component + return + + if getattr(component, "elem_id", None) == "img2img_generate": + img2img_submit_button = component + + +def on_ui_settings(): + section = ("ADetailer", AFTER_DETAILER) + shared.opts.add_option( + "ad_max_models", + shared.OptionInfo( + default=2, + label="Max models", + component=gr.Slider, + component_args={"minimum": 1, "maximum": 10, "step": 1}, + section=section, + ), + ) + + shared.opts.add_option( + "ad_save_previews", + shared.OptionInfo(False, "Save mask previews", section=section), + ) + + shared.opts.add_option( + "ad_save_images_before", + shared.OptionInfo(False, "Save images before ADetailer", section=section), + ) + + shared.opts.add_option( + "ad_only_seleted_scripts", + shared.OptionInfo( + True, "Apply only selected scripts to ADetailer", section=section + ), + ) + + textbox_args = { + "placeholder": "comma-separated list of script names", + "interactive": True, + } + + shared.opts.add_option( + "ad_script_names", + shared.OptionInfo( + default=SCRIPT_DEFAULT, + label="Script names to apply to ADetailer (separated by comma)", + component=gr.Textbox, + component_args=textbox_args, + section=section, + ), + ) + + shared.opts.add_option( + "ad_bbox_sortby", + shared.OptionInfo( + default="None", + label="Sort bounding boxes by", + component=gr.Radio, + component_args={"choices": BBOX_SORTBY}, + section=section, + ), + ) + + +# xyz_grid + + +def make_axis_on_xyz_grid(): + xyz_grid = None + for script in scripts.scripts_data: + if script.script_class.__module__ == "xyz_grid.py": + xyz_grid = script.module + break + + if xyz_grid is None: + return + + model_list = ["None", *model_mapping.keys()] + samplers = [sampler.name for sampler in all_samplers] + + def set_value(p, x, xs, *, field: str): + if not hasattr(p, "adetailer_xyz"): + p.adetailer_xyz = {} + p.adetailer_xyz[field] = x + + axis = [ + xyz_grid.AxisOption( + "[ADetailer] ADetailer model 1st", + str, + partial(set_value, field="ad_model"), + choices=lambda: model_list, + ), + xyz_grid.AxisOption( + "[ADetailer] ADetailer prompt 1st", + str, + partial(set_value, field="ad_prompt"), + ), + xyz_grid.AxisOption( + "[ADetailer] ADetailer negative prompt 1st", + str, + partial(set_value, field="ad_negative_prompt"), + ), + xyz_grid.AxisOption( + "[ADetailer] Mask erosion / dilation 1st", + int, + partial(set_value, field="ad_dilate_erode"), + ), + xyz_grid.AxisOption( + "[ADetailer] Inpaint denoising strength 1st", + float, + partial(set_value, field="ad_denoising_strength"), + ), + xyz_grid.AxisOption( + "[ADetailer] Inpaint only masked 1st", + str, + partial(set_value, field="ad_inpaint_only_masked"), + choices=lambda: ["True", "False"], + ), + xyz_grid.AxisOption( + "[ADetailer] Inpaint only masked padding 1st", + int, + partial(set_value, field="ad_inpaint_only_masked_padding"), + ), + xyz_grid.AxisOption( + "[ADetailer] ADetailer sampler 1st", + str, + partial(set_value, field="ad_sampler"), + choices=lambda: samplers, + ), + xyz_grid.AxisOption( + "[ADetailer] ControlNet model 1st", + str, + partial(set_value, field="ad_controlnet_model"), + choices=lambda: ["None", *get_cn_models()], + ), + ] + + if not any(x.label.startswith("[ADetailer]") for x in xyz_grid.axis_options): + xyz_grid.axis_options.extend(axis) + + +def on_before_ui(): + try: + make_axis_on_xyz_grid() + except Exception: + error = traceback.format_exc() + print( + f"[-] ADetailer: xyz_grid error:\n{error}", + file=sys.stderr, + ) + + +script_callbacks.on_ui_settings(on_ui_settings) +script_callbacks.on_after_component(on_after_component) +script_callbacks.on_before_ui(on_before_ui) diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/.github/workflows/pytest.yml b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/.github/workflows/pytest.yml new file mode 100644 index 0000000000000000000000000000000000000000..1c57d08221beaffd96176da66384661b4c4c899a --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/.github/workflows/pytest.yml @@ -0,0 +1,31 @@ +name: Tests + +on: + pull_request: + push: + branches: [main] + +jobs: + test: + runs-on: ubuntu-latest + + strategy: + matrix: + python-version: ['3.11', '3.10'] + + steps: + - name: Checkout code + uses: actions/checkout@v2 + + - name: Set up Python ${{ matrix.python-version }} + uses: actions/setup-python@v2 + with: + python-version: ${{ matrix.python-version }} + + - name: Install dependencies + run: | + python -m pip install --upgrade pip + pip install tox + + - name: Run tox + run: tox diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/.gitignore b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/.gitignore new file mode 100644 index 0000000000000000000000000000000000000000..689264bad5776046b9d205383c4983c599db144a --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/.gitignore @@ -0,0 +1,2 @@ +.idea +.pytest_cache diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/.pre-commit-config.yaml b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/.pre-commit-config.yaml new file mode 100644 index 0000000000000000000000000000000000000000..0e1a718fa24808c7f30521d606ab5251ddc2d284 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/.pre-commit-config.yaml @@ -0,0 +1,35 @@ +repos: + - repo: https://github.com/pre-commit/pre-commit-hooks + rev: v4.4.0 + hooks: + - id: trailing-whitespace + - id: end-of-file-fixer + - id: check-yaml + - id: debug-statements + - id: double-quote-string-fixer + - id: name-tests-test + - repo: https://github.com/asottile/reorder_python_imports + rev: v3.9.0 + hooks: + - id: reorder-python-imports + args: [--py37-plus, --add-import, "from __future__ import annotations"] + - repo: https://github.com/asottile/add-trailing-comma + rev: v2.4.0 + hooks: + - id: add-trailing-comma + args: [--py36-plus] + - repo: https://github.com/asottile/pyupgrade + rev: v3.3.2 + hooks: + - id: pyupgrade + args: [--py37-plus] + - repo: https://github.com/pre-commit/mirrors-autopep8 + rev: v2.0.2 + hooks: + - id: autopep8 + - repo: https://github.com/PyCQA/flake8 + rev: 6.0.0 + hooks: + - id: flake8 + args: + - --max-line-length=120 diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/README.md b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/README.md new file mode 100644 index 0000000000000000000000000000000000000000..6236dd14fadf3e670d365aae1557562322d13123 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/README.md @@ -0,0 +1,91 @@ +# Aspect Ratio Helper [](https://github.com/thomasasfk/sd-webui-aspect-ratio-helper/actions/workflows/pytest.yml) + +Simple extension to easily maintain aspect ratio while changing dimensions. + +Install via the extensions tab on the [AUTOMATIC1111 webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui). + +## Features + +**(note this list is a little out of date, will need to find time to update it)** + +- JavaScript aspect ratio controls + - Adds a dropdown of configurable aspect ratios, to which the dimensions will auto-scale + - When selected, you will only be able to modify the higher dimension + - The smaller or equivalent dimension will scale accordingly + - If "Lock/🔒" is selected, the aspect ratio of the current dimensions will be kept + - If "Image/🖼️" is selected, the aspect ratio of the current image will be kept (img2img only) + - If you click the "Swap/⇅" button, the current dimensions will swap + - Configurable aspect ratios will also flip, reducing the need for duplication of config + +https://user-images.githubusercontent.com/22506439/227396634-7a63671a-fd38-419a-b734-a3d26647cc1d.mp4 + +
+ +- Scale to maximum dimension + - Upon clicking, the width and height will scale according to the configured maximum value + - Aspect ratio will be retained, the smaller or equivalent dimension will be scaled to match +- Scale to aspect ratio + - Upon clicking, the current dimensions will be scaled to the given aspect ratio, using the highest width or height + - i.e `4:3 of 256x512 = 512x384` `9:16 of 512x256 = 288x512` `1:1 of 256x300 = 300x300` + - You can optionally toggle this to use the "Maximum dimension" slider value + - i.e `4:3 of 512 = 512x384` `9:16 of 512 = 288x512` `1:1 of 300 = 300x300` +- Scale by percentage + - Upon clicking, the current dimensions will be multiplied by the given percentage, with aspect ratio maintained + - i.e `-25% of 512x256 = 384x192` `+50% of 512x512 = 768x768` + - You can also change the display of these if you find it more intuitive + - i.e `75% of 512x256 = 384x192` `150% of 512x512 = 768x768` + - i.e `x0.75 of 512x256 = 384x192` `x1.5 of 512x512 = 768x768` + + + +## Settings + +- Hide accordion by default (`False`) +- Expand accordion by default (`False`) + - Determines whether the 'Aspect Ratio Helper' accordion expands by default +- UI Component order (`MaxDimensionScaler, PredefinedAspectRatioButtons, PredefinedPercentageButtons`) + - Determines the order in which the UI components will render +- Enable JavaScript aspect ratio controls +- JavaScript aspect ratio buttons `(1:1, 4:3, 16:9, 9:16, 21:9)` + - i.e `1:1, 4:3, 16:9, 9:16, 21:9` `2:3, 1:5, 3:5` +- Show maximum dimension button (`True`) +- Maximum dimension default (`1024`) +- Show pre-defined aspect ratio buttons (`True`) +- Use "Maximum dimension" for aspect ratio buttons (`False`) +- Pre-defined aspect ratio buttons (`1:1, 4:3, 16:9, 9:16, 21:9`) + - i.e `1:1, 4:3, 16:9, 9:16, 21:9` `2:3, 1:5, 3:5` +- Show pre-defined percentage buttons (`True`) +- Pre-defined percentage buttons (`25, 50, 75, 125, 150, 175, 200`) + - i.e `25, 50, 75, 125, 150, 175, 200` `50, 125, 300` +- Pre-defined percentage display format (`Incremental/decremental percentage (-50%, +50%)`) + - `Incremental/decremental percentage (-50%, +50%)` + - `Raw percentage (50%, 150%)` + - `Multiplication (x0.5, x1.5)` + + + + +JavaScript & accordion aspect ratios _might_ not play well together - I don't think many users will use both simultaneously, but we'll see. + +## Contributing + +- Open an issue for suggestions +- Raise a pull request + +## Dependencies + +Developed using existing [AUTOMATIC1111 webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui) dependencies. + +It's recommended to use the python version specified by A1111 webui. + +However - for running unit tests, we use pytest. + +```bash +pip install pytest +``` + +## Testing +From the root of the repository. +```bash +pytest +``` diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_components.py b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_components.py new file mode 100644 index 0000000000000000000000000000000000000000..2fae4e7f0e258024c3bfb868f88fa3ae040513dc --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_components.py @@ -0,0 +1,360 @@ +from __future__ import annotations + +from abc import ABC +from abc import abstractmethod +from functools import partial +from typing import Callable + +import gradio as gr + +import aspect_ratio_helper._constants as _constants +import aspect_ratio_helper._settings as _settings +import aspect_ratio_helper._util as _util + + +class ArhUIComponent(ABC): + + def __init__(self, script): + self.script = script + + @abstractmethod + def render(self): ... + + @staticmethod + @abstractmethod + def should_show() -> bool: ... + + @staticmethod + @abstractmethod + def add_options(): ... + + +class MaxDimensionScaler(ArhUIComponent): + + def render(self): + max_dim_default = _settings.safe_opt( + _constants.ARH_MAX_WIDTH_OR_HEIGHT_KEY, + ) + self.script.max_dimension = float(max_dim_default) + + inputs = outputs = [self.script.wc, self.script.hc] + + with gr.Row( + visible=self.should_show(), + ): + max_dim_default = _settings.safe_opt( + _constants.ARH_MAX_WIDTH_OR_HEIGHT_KEY, + ) + # todo: when using gr.Slider (not deprecated), the default value + # is somehow always 270?... can't figure out why. + # using legacy inputs.Slider for now as it doesn't have the issue. + max_dimension_slider = gr.inputs.Slider( + minimum=_constants.MIN_DIMENSION, + maximum=_constants.MAX_DIMENSION, + step=8, + default=max_dim_default, + label='Maximum dimension', + ) + + def _update_max_dimension(_max_dimension): + self.script.max_dimension = _max_dimension + + max_dimension_slider.change( + _update_max_dimension, + inputs=[max_dimension_slider], + show_progress=False, + ) + + gr.Button( + value='Scale to maximum dimension', + visible=self.should_show(), + ).click( + fn=_util.scale_dimensions_to_max_dim, + inputs=[*inputs, max_dimension_slider], + outputs=outputs, + ) + + @staticmethod + def should_show() -> bool: + return _settings.safe_opt(_constants.ARH_SHOW_MAX_WIDTH_OR_HEIGHT_KEY) + + @staticmethod + def add_options(shared): + shared.opts.add_option( + key=_constants.ARH_SHOW_MAX_WIDTH_OR_HEIGHT_KEY, + info=shared.OptionInfo( + default=_settings.OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_SHOW_MAX_WIDTH_OR_HEIGHT_KEY, + ), + label='Show maximum dimension button', + section=_constants.SECTION, + ), + ) + shared.opts.add_option( + key=_constants.ARH_MAX_WIDTH_OR_HEIGHT_KEY, + info=shared.OptionInfo( + default=_settings.OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_MAX_WIDTH_OR_HEIGHT_KEY, + ), + label='Maximum dimension default', + component=gr.Slider, + component_args={ + 'minimum': _constants.MIN_DIMENSION, + 'maximum': _constants.MAX_DIMENSION, + 'step': 8, + }, + section=_constants.SECTION, + ), + ) + + +class MinDimensionScaler(ArhUIComponent): + + def render(self): + min_dim_default = _settings.safe_opt( + _constants.ARH_MIN_WIDTH_OR_HEIGHT_KEY, + ) + self.script.min_dimension = float(min_dim_default) + + inputs = outputs = [self.script.wc, self.script.hc] + + with gr.Row( + visible=self.should_show(), + ): + min_dim_default = _settings.safe_opt( + _constants.ARH_MIN_WIDTH_OR_HEIGHT_KEY, + ) + # todo: when using gr.Slider (not deprecated), the default value + # is somehow always 270?... can't figure out why. + # using legacy inputs.Slider for now as it doesn't have the issue. + min_dimension_slider = gr.inputs.Slider( + minimum=_constants.MIN_DIMENSION, + maximum=_constants.MAX_DIMENSION, + step=8, + default=min_dim_default, + label='Minimum dimension', + ) + + def _update_min_dimension(_min_dimension): + self.script.min_dimension = _min_dimension + + min_dimension_slider.change( + _update_min_dimension, + inputs=[min_dimension_slider], + show_progress=False, + ) + + gr.Button( + value='Scale to minimum dimension', + visible=self.should_show(), + ).click( + fn=_util.scale_dimensions_to_min_dim, + inputs=[*inputs, min_dimension_slider], + outputs=outputs, + ) + + @staticmethod + def should_show() -> bool: + return _settings.safe_opt(_constants.ARH_SHOW_MIN_WIDTH_OR_HEIGHT_KEY) + + @staticmethod + def add_options(shared): + shared.opts.add_option( + key=_constants.ARH_SHOW_MIN_WIDTH_OR_HEIGHT_KEY, + info=shared.OptionInfo( + default=_settings.OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_SHOW_MIN_WIDTH_OR_HEIGHT_KEY, + ), + label='Show minimum dimension button', + section=_constants.SECTION, + ), + ) + shared.opts.add_option( + key=_constants.ARH_MIN_WIDTH_OR_HEIGHT_KEY, + info=shared.OptionInfo( + default=_settings.OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_MIN_WIDTH_OR_HEIGHT_KEY, + ), + label='Minimum dimension default', + component=gr.Slider, + component_args={ + 'minimum': _constants.MIN_DIMENSION, + 'maximum': _constants.MAX_DIMENSION, + 'step': 8, + }, + section=_constants.SECTION, + ), + ) + + +class PredefinedAspectRatioButtons(ArhUIComponent): + + def render(self): + use_max_dim_op = _settings.safe_opt( + _constants.ARH_PREDEFINED_ASPECT_RATIO_USE_MAX_DIM_KEY, + ) + aspect_ratios = _settings.safe_opt( + _constants.ARH_PREDEFINED_ASPECT_RATIOS_KEY, + ).split(',') + + with gr.Column( + variant='panel', + visible=self.should_show(), + ), gr.Row( + variant='compact', + visible=self.should_show(), + elem_classes='arh-btn-row', + ): + for ar_str in aspect_ratios: + w, h, *_ = (abs(float(d)) for d in ar_str.split(':')) + + inputs = [] + if use_max_dim_op: + ar_func = partial( + _util.scale_dimensions_to_max_dim_func, + width=w, height=h, + max_dim=lambda: self.script.max_dimension, + ) + else: + inputs.extend([self.script.wc, self.script.hc]) + ar_func = partial( + _util.scale_dimensions_to_ui_width_or_height, + arw=w, arh=h, + ) + + gr.Button( + value=self.display_func(ar_str) or ar_str, + visible=self.should_show(), + ).click( + fn=ar_func, + inputs=inputs, + outputs=[self.script.wc, self.script.hc], + ) + + @staticmethod + def should_show() -> bool: + return _settings.safe_opt( + _constants.ARH_SHOW_PREDEFINED_ASPECT_RATIOS_KEY, + ) + + @staticmethod + def add_options(shared): + shared.opts.add_option( + key=_constants.ARH_SHOW_PREDEFINED_ASPECT_RATIOS_KEY, + info=shared.OptionInfo( + default=_settings.OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_SHOW_PREDEFINED_ASPECT_RATIOS_KEY, + ), + label='Show pre-defined aspect ratio buttons', + section=_constants.SECTION, + ), + ) + shared.opts.add_option( + key=_constants.ARH_PREDEFINED_ASPECT_RATIO_USE_MAX_DIM_KEY, + info=shared.OptionInfo( + default=_settings.OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_PREDEFINED_ASPECT_RATIO_USE_MAX_DIM_KEY, + ), + label='Use "Maximum dimension" for aspect ratio buttons (by ' + 'default we use the max width or height)', + section=_constants.SECTION, + ), + ) + shared.opts.add_option( + key=_constants.ARH_PREDEFINED_ASPECT_RATIOS_KEY, + info=shared.OptionInfo( + default=_settings.OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_PREDEFINED_ASPECT_RATIOS_KEY, + ), + label='Pre-defined aspect ratio buttons ' + '(1:1, 4:3, 16:9, 9:16, 21:9)', + section=_constants.SECTION, + ), + ) + + @property + def display_func(self) -> Callable[[str], str]: + return lambda _: None # todo: different displays for aspect ratios. + + +class PredefinedPercentageButtons(ArhUIComponent): + + def render(self): + inputs = outputs = [self.script.wc, self.script.hc] + with gr.Column( + variant='panel', + visible=self.should_show(), + ), gr.Row( + variant='compact', + visible=self.should_show(), + elem_classes='arh-btn-row', + ): + pps = _settings.safe_opt(_constants.ARH_PREDEFINED_PERCENTAGES_KEY) + percentages = [abs(int(x)) for x in pps.split(',')] + + for percentage in percentages: + display = self.display_func(percentage) + gr.Button( + value=display, + visible=self.should_show(), + ).click( + fn=partial( + _util.scale_by_percentage, + pct=percentage / 100, + ), + inputs=inputs, + outputs=outputs, + ) + + @staticmethod + def should_show() -> bool: + return _settings.safe_opt( + _constants.ARH_SHOW_PREDEFINED_PERCENTAGES_KEY, + ) + + @staticmethod + def add_options(shared): + shared.opts.add_option( + key=_constants.ARH_SHOW_PREDEFINED_PERCENTAGES_KEY, + info=shared.OptionInfo( + default=_settings.OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_SHOW_PREDEFINED_PERCENTAGES_KEY, + ), + label='Show pre-defined percentage buttons', + section=_constants.SECTION, + ), + ) + shared.opts.add_option( + key=_constants.ARH_PREDEFINED_PERCENTAGES_KEY, + info=shared.OptionInfo( + default=_settings.OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_PREDEFINED_PERCENTAGES_KEY, + ), + label='Pre-defined percentage buttons (75, 125, 150)', + section=_constants.SECTION, + ), + ) + shared.opts.add_option( + key=_constants.ARH_PREDEFINED_PERCENTAGES_DISPLAY_KEY, + info=shared.OptionInfo( + default=_settings.OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_PREDEFINED_PERCENTAGES_DISPLAY_KEY, + ), + label='Pre-defined percentage display format', + component=gr.Dropdown, + component_args=lambda: { + 'choices': tuple( + _settings.PREDEFINED_PERCENTAGES_DISPLAY_MAP.keys(), + ), + }, + section=_constants.SECTION, + ), + ) + + @property + def display_func(self) -> Callable[[str], str]: + return _settings.PREDEFINED_PERCENTAGES_DISPLAY_MAP.get( + _settings.safe_opt( + _constants.ARH_PREDEFINED_PERCENTAGES_DISPLAY_KEY, + ), + ) diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_constants.py b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_constants.py new file mode 100644 index 0000000000000000000000000000000000000000..a9a7c89c51415c63a579beaa959794469096dd4b --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_constants.py @@ -0,0 +1,27 @@ +from __future__ import annotations +EXTENSION_NAME = 'Aspect Ratio Helper' + +MAX_DIMENSION = 2048 +MIN_DIMENSION = 64 + +ARH_EXPAND_BY_DEFAULT_KEY = 'arh_expand_by_default' +ARH_HIDE_ACCORDION_BY_DEFAULT_KEY = 'arh_hide_accordion_by_default' +ARH_UI_COMPONENT_ORDER_KEY = 'arh_ui_component_order_key' +ARH_UI_JAVASCRIPT_SELECTION_METHOD = 'arh_ui_javascript_selection_method' +ARH_JAVASCRIPT_ASPECT_RATIO_SHOW_KEY = 'arh_javascript_aspect_ratio_show' +ARH_JAVASCRIPT_ASPECT_RATIOS_KEY = 'arh_javascript_aspect_ratio' +ARH_SHOW_MAX_WIDTH_OR_HEIGHT_KEY = 'arh_show_max_width_or_height' +ARH_SHOW_MIN_WIDTH_OR_HEIGHT_KEY = 'arh_show_min_width_or_height' +ARH_MAX_WIDTH_OR_HEIGHT_KEY = 'arh_max_width_or_height' +ARH_MIN_WIDTH_OR_HEIGHT_KEY = 'arh_min_width_or_height' +ARH_SHOW_PREDEFINED_PERCENTAGES_KEY = 'arh_show_predefined_percentages' +ARH_PREDEFINED_PERCENTAGES_KEY = 'arh_predefined_percentages' +ARH_SHOW_PREDEFINED_ASPECT_RATIOS_KEY = 'arh_show_predefined_aspect_ratios' +ARH_PREDEFINED_ASPECT_RATIOS_KEY = 'arh_predefined_aspect_ratios' +ARH_PREDEFINED_ASPECT_RATIO_USE_MAX_DIM_KEY \ + = 'arh_predefined_aspect_ratio_use_max_dim' +ARH_PREDEFINED_PERCENTAGES_DISPLAY_KEY \ + = 'arh_predefined_percentages_display_key' +DEFAULT_PERCENTAGES_DISPLAY_KEY \ + = 'Incremental/decremental percentage (-50%, +50%)' +SECTION = 'aspect_ratio_helper', EXTENSION_NAME diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_settings.py b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_settings.py new file mode 100644 index 0000000000000000000000000000000000000000..075d8bf85f84b04868c3b52a9e1f596e810a5c93 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_settings.py @@ -0,0 +1,178 @@ +from __future__ import annotations + +import itertools + +import gradio as gr +from modules import shared + +import aspect_ratio_helper._components as _components +import aspect_ratio_helper._constants as _constants +import aspect_ratio_helper._util as _util + +PREDEFINED_PERCENTAGES_DISPLAY_MAP = { + _constants.DEFAULT_PERCENTAGES_DISPLAY_KEY: _util.display_minus_and_plus, + 'Raw percentage (50%, 150%)': _util.display_raw_percentage, + 'Multiplication (x0.5, x1.5)': _util.display_multiplication, +} + +COMPONENTS = ( + _components.MaxDimensionScaler, + _components.MinDimensionScaler, + _components.PredefinedAspectRatioButtons, + _components.PredefinedPercentageButtons, +) + +DEFAULT_UI_COMPONENT_ORDER_KEY_LIST = [e.__name__ for e in COMPONENTS] +DEFAULT_UI_COMPONENT_ORDER_KEY = ', '.join( + DEFAULT_UI_COMPONENT_ORDER_KEY_LIST, +) +OPT_KEY_TO_DEFAULT_MAP = { + _constants.ARH_EXPAND_BY_DEFAULT_KEY: False, + _constants.ARH_HIDE_ACCORDION_BY_DEFAULT_KEY: True, + _constants.ARH_UI_COMPONENT_ORDER_KEY: + DEFAULT_UI_COMPONENT_ORDER_KEY, + _constants.ARH_UI_JAVASCRIPT_SELECTION_METHOD: 'Aspect Ratios Dropdown', + _constants.ARH_JAVASCRIPT_ASPECT_RATIO_SHOW_KEY: True, + _constants.ARH_JAVASCRIPT_ASPECT_RATIOS_KEY: + '1:1, 3:2, 4:3, 5:4, 16:9', + _constants.ARH_SHOW_MAX_WIDTH_OR_HEIGHT_KEY: False, + _constants.ARH_MAX_WIDTH_OR_HEIGHT_KEY: + _constants.MAX_DIMENSION / 2, + _constants.ARH_SHOW_MIN_WIDTH_OR_HEIGHT_KEY: False, + _constants.ARH_MIN_WIDTH_OR_HEIGHT_KEY: + _constants.MAX_DIMENSION / 2, + _constants.ARH_SHOW_PREDEFINED_PERCENTAGES_KEY: False, + _constants.ARH_PREDEFINED_PERCENTAGES_KEY: + '25, 50, 75, 125, 150, 175, 200', + _constants.ARH_PREDEFINED_PERCENTAGES_DISPLAY_KEY: + _constants.DEFAULT_PERCENTAGES_DISPLAY_KEY, + _constants.ARH_SHOW_PREDEFINED_ASPECT_RATIOS_KEY: False, + _constants.ARH_PREDEFINED_ASPECT_RATIO_USE_MAX_DIM_KEY: False, + _constants.ARH_PREDEFINED_ASPECT_RATIOS_KEY: + '1:1, 4:3, 16:9, 9:16, 21:9', +} + + +def safe_opt(key): + return _util.safe_opt_util(shared.opts, key, OPT_KEY_TO_DEFAULT_MAP) + + +def sort_components_by_keys( + components: list[_components.ArhUIComponent], +) -> list[_components.ArhUIComponent]: + ordered_component_keys = safe_opt( + _constants.ARH_UI_COMPONENT_ORDER_KEY, + ).split(',') + + # this can happen if we add new components, but the user has old settings. + # if this happens, we find the missing components, and append them. + if len(ordered_component_keys) != len(COMPONENTS): + all_components = set(DEFAULT_UI_COMPONENT_ORDER_KEY_LIST) + missing_components = all_components - set(ordered_component_keys) + ordered_component_keys.extend(missing_components) + + try: + component_key_to_order_dict = { + key: order for order, key in enumerate( + [k.strip() for k in ordered_component_keys], + ) + } + return sorted( + components, + key=lambda c: component_key_to_order_dict.get( + c.__class__.__name__, + ), + ) + except ValueError: + print( + f'{_constants.EXTENSION_NAME} UI component order is erroneous. ' + f'Defaulting to regular order, to fix this, please use' + f'the intended syntax for the setting, i.e ' + f'"{DEFAULT_UI_COMPONENT_ORDER_KEY}"', + ) + return components + + +def on_ui_settings(): + # javascript ui options + shared.opts.add_option( + key=_constants.ARH_JAVASCRIPT_ASPECT_RATIO_SHOW_KEY, + info=shared.OptionInfo( + default=OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_JAVASCRIPT_ASPECT_RATIO_SHOW_KEY, + ), + label='Enable JavaScript aspect ratio controls', + section=_constants.SECTION, + ), + ) + shared.opts.add_option( + key=_constants.ARH_JAVASCRIPT_ASPECT_RATIOS_KEY, + info=shared.OptionInfo( + default=OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_JAVASCRIPT_ASPECT_RATIOS_KEY, + ), + label='JavaScript aspect ratio buttons' + ' (1:1, 4:3, 16:9, 9:16, 21:9)', + section=_constants.SECTION, + ), + ) + shared.opts.add_option( + key=_constants.ARH_UI_JAVASCRIPT_SELECTION_METHOD, + info=shared.OptionInfo( + default=OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_UI_JAVASCRIPT_SELECTION_METHOD, + ), + label='JavaScript selection method', + component=gr.Dropdown, + component_args=lambda: { + 'choices': [ + 'Aspect Ratios Dropdown', + 'Default Options Button', + ], + }, + section=_constants.SECTION, + ), + ) + + # accordion options + shared.opts.add_option( + key=_constants.ARH_HIDE_ACCORDION_BY_DEFAULT_KEY, + info=shared.OptionInfo( + default=OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_HIDE_ACCORDION_BY_DEFAULT_KEY, + ), + label='Hide accordion by default', + section=_constants.SECTION, + ), + ) + shared.opts.add_option( + key=_constants.ARH_EXPAND_BY_DEFAULT_KEY, + info=shared.OptionInfo( + default=OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_EXPAND_BY_DEFAULT_KEY, + ), + label='Expand accordion by default', + section=_constants.SECTION, + ), + ) + shared.opts.add_option( + key=_constants.ARH_UI_COMPONENT_ORDER_KEY, + info=shared.OptionInfo( + default=OPT_KEY_TO_DEFAULT_MAP.get( + _constants.ARH_UI_COMPONENT_ORDER_KEY, + ), + label='UI Component order', + component=gr.Dropdown, + component_args=lambda: { + 'choices': [ + ', '.join(p) for p in itertools.permutations( # TODO: Rethink this, exponential growth... + DEFAULT_UI_COMPONENT_ORDER_KEY_LIST, + ) + ], + }, + section=_constants.SECTION, + ), + ) + + for component in COMPONENTS: + component.add_options(shared) diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_util.py b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_util.py new file mode 100644 index 0000000000000000000000000000000000000000..b10be805845ec58a21eafedc0f26842f13399ffc --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/_util.py @@ -0,0 +1,119 @@ +from __future__ import annotations + +import contextlib + +import aspect_ratio_helper._constants as _const + + +def safe_opt_util(shared_opts, key, default_key_map: dict[str, object]): + # attempt to retrieve key from shared options + with contextlib.suppress(AttributeError): + value = shared_opts.__getattr__(key) + if value is not None: + return value + + # attempt to retrieve default, and last resort the constant default + return shared_opts.get_default(key) or default_key_map.get(key) + + +def display_multiplication(num) -> str: + return f'x{round(num / 100, 3)}' + + +def display_raw_percentage(num) -> str: + return f'{num}%' + + +def display_minus_and_plus(num) -> str: + num -= 100 + if num > 0: + return f'+{num}%' + return f'{num}%' + + +def scale_by_percentage(width, height, pct) -> tuple[int, int]: + aspect_ratio = float(width) / float(height) + step = (pct - 1.0) + new_width = int(round(width * (1.0 + step))) + new_height = int(round(new_width / aspect_ratio)) + return clamp_to_boundaries(new_width, new_height, aspect_ratio) + + +def scale_dimensions_to_ui_width_or_height( + width, height, arw, arh, +) -> tuple[int, int]: + return scale_dimensions_to_max_dim(arw, arh, max(width, height)) + + +def scale_dimensions_to_max_dim_func( + width, height, max_dim: callable, +) -> tuple[int, int]: + return scale_dimensions_to_max_dim(width, height, max_dim()) + + +def scale_dimensions_to_max_dim( + width, height, max_dim, +) -> tuple[int, int]: + aspect_ratio = float(width) / float(height) + return scale_dimensions_to_ar(width, height, max_dim, aspect_ratio) + + +def scale_dimensions_to_min_dim( + width, height, min_dim, +) -> tuple[int, int]: + aspect_ratio = float(width) / float(height) + if width >= height: + max_dim = min_dim * aspect_ratio + else: + max_dim = min_dim / aspect_ratio + return scale_dimensions_to_ar(width, height, max_dim, aspect_ratio) + + +def scale_dimensions_to_ar( + width, height, max_dim, aspect_ratio, +) -> tuple[int, int]: + if width > height: + new_width = max_dim + new_height = int(round(max_dim / aspect_ratio)) + else: + new_height = max_dim + new_width = int(round(max_dim * aspect_ratio)) + return clamp_to_boundaries(new_width, new_height, aspect_ratio) + + +def round_to_multiple_of_8(value): + return int(round(value / 8.0)) * 8 + + +def clamp_to_boundaries(owidth, oheight, aspect_ratio) -> tuple[int, int]: + width, height = owidth, oheight + if width > _const.MAX_DIMENSION: + width = _const.MAX_DIMENSION + height = int(round(width / aspect_ratio)) + if height > _const.MAX_DIMENSION: + height = _const.MAX_DIMENSION + width = int(round(height * aspect_ratio)) + if width < _const.MIN_DIMENSION: + width = _const.MIN_DIMENSION + height = int(round(width / aspect_ratio)) + if height < _const.MIN_DIMENSION: + height = _const.MIN_DIMENSION + width = int(round(height * aspect_ratio)) + + width = round_to_multiple_of_8(width) + height = round_to_multiple_of_8(height) + # for insane aspect ratios we don't support... i.e 1:100 + # 64:6400 when run through this function, so we clamp to 64:2048 (‾◡◝) + # also. when the user does this it breaks the "scale to max" function + # on the ui until they change to another reasonable aspect ratio. + # todo: figure out why ^ and handle it better. + if width < _const.MIN_DIMENSION: + width = _const.MIN_DIMENSION + elif width > _const.MAX_DIMENSION: + width = _const.MAX_DIMENSION + if height < _const.MIN_DIMENSION: + height = _const.MIN_DIMENSION + elif height > _const.MAX_DIMENSION: + height = _const.MAX_DIMENSION + + return width, height diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/main.py b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/main.py new file mode 100644 index 0000000000000000000000000000000000000000..814569e123b3d9c3e6d83cdece9b7f6b7523b468 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/aspect_ratio_helper/main.py @@ -0,0 +1,72 @@ +from __future__ import annotations + +import gradio as gr +from modules import script_callbacks +from modules import scripts + +import aspect_ratio_helper._constants as _constants +import aspect_ratio_helper._settings as _settings + + +class AspectRatioStepScript(scripts.Script): + + def __init__(self): + self.t2i_w: gr.components.Slider | None = None + self.t2i_h: gr.components.Slider | None = None + self.i2i_w: gr.components.Slider | None = None + self.i2i_h: gr.components.Slider | None = None + self.wc: gr.components.Slider + self.hc: gr.components.Slider + self.max_dimension: float + + def title(self) -> str: + return _constants.EXTENSION_NAME + + def show(self, _) -> scripts.AlwaysVisible: + return scripts.AlwaysVisible + + def ui(self, is_img2img): + if is_img2img: + self.wc, self.hc = self.i2i_w, self.i2i_h + else: + self.wc, self.hc = self.t2i_w, self.t2i_h # noqa + + components = _settings.sort_components_by_keys( + [component(self) for component in _settings.COMPONENTS], + ) + + hide_accordion: bool = _settings.safe_opt( + _constants.ARH_HIDE_ACCORDION_BY_DEFAULT_KEY, + ) + + if hide_accordion or not any(c.should_show() for c in components): + return # no components should render, so just return. + + start_expanded: bool = _settings.safe_opt( + _constants.ARH_EXPAND_BY_DEFAULT_KEY, + ) + with gr.Group(), gr.Accordion( + _constants.EXTENSION_NAME, + open=start_expanded, + ): + for component in components: + # we deliberately DON'T check component.should_show() here. + # we need to call render to instantiate the components, we use + # the visible property on each component to hide them. + component.render() + + def after_component(self, component: gr.components.Component, **kwargs): + element_id = kwargs.get('elem_id') + + if isinstance(component, gr.components.Slider): + if element_id == 'txt2img_width': + self.t2i_w: gr.components.Slider = component + elif element_id == 'txt2img_height': + self.t2i_h: gr.components.Slider = component + elif element_id == 'img2img_width': + self.i2i_w: gr.components.Slider = component + elif element_id == 'img2img_height': + self.i2i_h: gr.components.Slider = component + + +script_callbacks.on_ui_settings(_settings.on_ui_settings) diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/docs/opts.png b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/docs/opts.png new file mode 100644 index 0000000000000000000000000000000000000000..76037916f8a8501e0d8209eac0b6ec8ac4f759e0 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/docs/opts.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/docs/ui.png b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/docs/ui.png new file mode 100644 index 0000000000000000000000000000000000000000..51b1371f094efa2875be1559d009dd9fbabfb23e Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/docs/ui.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/javascript/aspectRatioController.js b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/javascript/aspectRatioController.js new file mode 100644 index 0000000000000000000000000000000000000000..c699fb708506d49b69e38bf0983e6da3a7cf3fe1 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/javascript/aspectRatioController.js @@ -0,0 +1,474 @@ +// constants +const _OFF = "Off"; // Don't use the functionality at all +const _LOCK = '🔒'; // Aspect ratio is "locked" +const _IMAGE = '🖼️'; // Aspect ratio is "locked" to that of the image + +const _MAXIMUM_DIMENSION = 2048; +const _MINIMUM_DIMENSION = 64; + +const _IMAGE_INPUT_CONTAINER_IDS = [ + 'img2img_image', + 'img2img_sketch', + 'img2maskimg', + 'inpaint_sketch', + 'img_inpaint_base', +]; + +const getSelectedImage2ImageTab = () => { + const selectedButton = gradioApp().getElementById('mode_img2img').querySelector('button.selected'); + const allButtons = gradioApp().getElementById('mode_img2img').querySelectorAll('button'); + const selectedIndex = Array.prototype.indexOf.call(allButtons, selectedButton); + return selectedIndex; +} + +const getCurrentImage = () => { + const currentTabIndex = getSelectedImage2ImageTab(); + const currentTabImageId = _IMAGE_INPUT_CONTAINER_IDS[currentTabIndex]; + return document.getElementById(currentTabImageId).querySelector('img'); +} + +const roundToClosestMultiple = (num, multiple) => { + const rounded = Math.round(Number(num) / multiple) * multiple; + return rounded; +} + +const aspectRatioFromStr = (ar) => { + if (!ar.includes(':')) return; + return ar.split(':').map(x => Number(x)); +} + +const reverseAspectRatio = (ar) => { + if (!ar.includes(':')) return; + const [width, height] = ar.split(':'); + return `${height}:${width}`; +} + +const clampToBoundaries = (width, height) => { + const aspectRatio = width / height; + width = Math.max(Math.min(width, _MAXIMUM_DIMENSION), _MINIMUM_DIMENSION); + height = Math.max(Math.min(height, _MAXIMUM_DIMENSION), _MINIMUM_DIMENSION); + if (width / height > aspectRatio) { + height = Math.round(width / aspectRatio); + } else if (width / height < aspectRatio) { + width = Math.round(height * aspectRatio); + } + + if (width > _MAXIMUM_DIMENSION) { + width = _MAXIMUM_DIMENSION; + } else if (width < _MINIMUM_DIMENSION) { + width = _MINIMUM_DIMENSION; + } + + if (height < _MINIMUM_DIMENSION) { + height = _MINIMUM_DIMENSION; + } else if (height > _MAXIMUM_DIMENSION) { + height = _MAXIMUM_DIMENSION; + } + + return [width, height]; +} + +const getOptions = () => { + return window.opts.arh_javascript_aspect_ratio.split(',').map(o => o.trim()); +} + +const reverseAllOptions = () => { + const allAspectRatioOptions = Array.from(gradioApp().querySelectorAll('.ar-option')); + allAspectRatioOptions.forEach(el => { + const reversed = reverseAspectRatio(el.value); + if (reversed) { + el.value = reversed; + el.textContent = reversed; + } + }); +} + +class OptionPickingController { + constructor(page, defaultOptions, controller) { + this.page = page; + this.options = this.getOptions(defaultOptions); + this.switchButton = gradioApp().getElementById(page + '_res_switch_btn'); + + const wrapperDiv = document.createElement('div'); + wrapperDiv.setAttribute("id", `${this.page}_size_toolbox`); + wrapperDiv.setAttribute("class", "flex flex-col relative col gap-4"); + wrapperDiv.setAttribute("style", "min-width: min(320px, 100%); flex-grow: 0"); + wrapperDiv.innerHTML = this.getElementInnerHTML(); + + const parent = this.switchButton.parentNode; + parent.removeChild(this.switchButton); + wrapperDiv.appendChild(this.switchButton); + parent.insertBefore(wrapperDiv, parent.lastChild.previousElementSibling); + + this.getPickerElement().onchange = this.pickerChanged(controller); + this.switchButton.onclick = this.switchButtonOnclick(controller); + } + + getOptions(defaultOptions) { + return [...new Set([...defaultOptions, ...getOptions()])]; + } + + pickerChanged(controller) { + return () => { + const picked = this.getCurrentOption(); + if (_IMAGE === picked) { + // this.switchButton.disabled = true; + } else { + this.switchButton.removeAttribute('disabled') + } + + controller.setAspectRatio(picked); + }; + } + + switchButtonOnclick(controller) { + return () => { + reverseAllOptions(); + const picked = this.getCurrentOption(); + if (_LOCK === picked) { + controller.setAspectRatio(`${controller.heightRatio}:${controller.widthRatio}`); + } else { + controller.setAspectRatio(picked); + } + }; + } + + getElementInnerHTML() { + throw new Error('Not implemented'); + } + + getPickerElement() { + throw new Error('Not implemented'); + } + + getCurrentOption() { + throw new Error('Not implemented'); + } +} + + +class SelectOptionPickingController extends OptionPickingController { + constructor(page, defaultOptions, controller) { + super(page, defaultOptions, controller); + } + + getElementInnerHTML() { + return ` +
+
+
+ `;
+ }
+
+ getPickerElement() {
+ return gradioApp().getElementById(`${this.page}_select_aspect_ratio`);
+ }
+
+ getCurrentOption() {
+ const selectElement = this.getPickerElement();
+ const options = Array.from(selectElement);
+ return options[selectElement.selectedIndex].value;
+ }
+}
+
+class DefaultOptionsButtonOptionPickingController extends OptionPickingController {
+ constructor(page, defaultOptions, controller) {
+ super(page, defaultOptions, controller);
+ this.currentIndex = 0;
+ this.getPickerElement().onclick = this.pickerChanged(controller);
+ }
+
+ pickerChanged(controller) {
+ return () => {
+ this.currentIndex = (this.currentIndex + 1) % this.options.length;
+ this.getPickerElement().querySelector('button').textContent = this.getCurrentOption();
+ super.pickerChanged(controller)();
+ }
+ }
+
+ getElementInnerHTML() {
+ const classes = Array.from(this.switchButton.classList);
+ return `
+
+ `;
+ }
+
+ getPickerElement() {
+ return gradioApp().getElementById(`${this.page}_ar_default_options_button`);
+ }
+
+ getOptions(defaultOptions) {
+ return defaultOptions;
+ }
+
+ getCurrentOption() {
+ return this.options[this.currentIndex || 0];
+ }
+}
+
+
+class SliderController {
+ constructor(element) {
+ this.element = element;
+ this.numberInput = this.element.querySelector('input[type=number]');
+ this.rangeInput = this.element.querySelector('input[type=range]');
+ this.inputs = [this.numberInput, this.rangeInput];
+ this.inputs.forEach(input => {
+ input.isWidth = element.isWidth;
+ });
+ }
+
+ getVal() {
+ return Number(this.numberInput.value);
+ }
+
+ updateVal(value) {
+ this.inputs.forEach(input => {
+ input.value = Number(value)
+ })
+ }
+
+ updateMin(value) {
+ this.inputs.forEach(input => {
+ input.min = roundToClosestMultiple(Number(value), 8);
+ })
+ }
+
+ updateMax(value) {
+ this.inputs.forEach(input => {
+ input.max = roundToClosestMultiple(Number(value), 8);
+ })
+ }
+
+ triggerEvent(event) {
+ this.numberInput.dispatchEvent(event)
+ }
+
+ setVal(value) {
+ value = Number(value)
+ const newValue = roundToClosestMultiple(value, 8)
+ this.updateVal(newValue);
+ }
+
+}
+
+class AspectRatioController {
+ constructor(page, widthContainer, heightContainer, defaultOptions) {
+ widthContainer.isWidth = true;
+ heightContainer.isWidth = false;
+ this.widthContainer = new SliderController(widthContainer);
+ this.heightContainer = new SliderController(heightContainer);
+ this.inputs = [...this.widthContainer.inputs, ...this.heightContainer.inputs];
+ this.inputs.forEach(input => {
+ input.addEventListener('change', (e) => {
+ e.preventDefault()
+ this.maintainAspectRatio(input);
+ });
+ })
+
+ if (window.opts.arh_ui_javascript_selection_method === 'Default Options Button') {
+ this.optionPickingControler = new DefaultOptionsButtonOptionPickingController(page, defaultOptions, this);
+ } else {
+ this.optionPickingControler = new SelectOptionPickingController(page, defaultOptions, this);
+ }
+
+ this.setAspectRatio(_OFF);
+ }
+
+ updateInputStates() {
+ if (this.isLandscapeOrSquare()) {
+ const AR = this.widthRatio / this.heightRatio;
+
+ const minWidthByAr = Math.round(_MINIMUM_DIMENSION * AR);
+ const minWidth = Math.max(minWidthByAr, _MINIMUM_DIMENSION);
+ this.widthContainer.updateMin(minWidth);
+ this.heightContainer.updateMin(_MINIMUM_DIMENSION);
+
+ const maxHeightByAr = Math.round(_MAXIMUM_DIMENSION / AR)
+ const maxHeight = Math.min(_MAXIMUM_DIMENSION, maxHeightByAr);
+ this.heightContainer.updateMax(maxHeight);
+ this.widthContainer.updateMax(_MAXIMUM_DIMENSION);
+ } else {
+ const AR = this.heightRatio / this.widthRatio;
+
+ const minHeightByAr = Math.round(_MINIMUM_DIMENSION * AR)
+ const minHeight = Math.max(minHeightByAr, _MINIMUM_DIMENSION);
+ this.heightContainer.updateMin(minHeight);
+ this.widthContainer.updateMin(_MINIMUM_DIMENSION);
+
+ const maxWidthByAr = Math.round(_MAXIMUM_DIMENSION / AR)
+ const maxWidth = Math.min(_MAXIMUM_DIMENSION, maxWidthByAr);
+ this.widthContainer.updateMax(maxWidth);
+ this.heightContainer.updateMax(_MAXIMUM_DIMENSION);
+ }
+ }
+
+ disable() {
+ this.widthContainer.updateMin(_MINIMUM_DIMENSION);
+ this.heightContainer.updateMin(_MINIMUM_DIMENSION);
+ this.widthContainer.updateMax(_MAXIMUM_DIMENSION);
+ this.heightContainer.updateMax(_MAXIMUM_DIMENSION);
+ }
+
+ isLandscapeOrSquare() {
+ return this.widthRatio >= this.heightRatio;
+ }
+
+ setAspectRatio(aspectRatio) {
+ this.aspectRatio = aspectRatio;
+
+ let wR, hR;
+ if (aspectRatio === _OFF) {
+ return this.disable();
+ } else if (aspectRatio === _IMAGE) {
+ const img = getCurrentImage();
+ wR = img && img.naturalWidth || 1;
+ hR = img && img.naturalHeight || 1;
+ } else if (aspectRatio === _LOCK) {
+ wR = this.widthContainer.getVal();
+ hR = this.heightContainer.getVal();
+ } else {
+ [wR, hR] = aspectRatioFromStr(aspectRatio);
+ }
+
+ [wR, hR] = clampToBoundaries(wR, hR);
+
+ this.widthRatio = wR;
+ this.heightRatio = hR;
+ this.updateInputStates();
+ this.maintainAspectRatio();
+ }
+
+ maintainAspectRatio(changedElement) {
+ if (this.aspectRatio === _OFF) return;
+ if (!changedElement) {
+ const allValues = Object.values(this.inputs).map(x => Number(x.value));
+ changedElement = {value: Math.max(...allValues)};
+ }
+
+ const aspectRatio = this.widthRatio / this.heightRatio;
+ let w, h;
+
+ if (changedElement.isWidth === undefined) {
+ if (this.isLandscapeOrSquare()) {
+ if (changedElement.isWidth) {}
+ w = Math.round(changedElement.value);
+ h = Math.round(changedElement.value / aspectRatio);
+ } else {
+ h = Math.round(changedElement.value);
+ w = Math.round(changedElement.value * aspectRatio);
+ }
+ } else {
+ if (changedElement.isWidth) {
+ w = Math.round(changedElement.value);
+ h = Math.round(changedElement.value / aspectRatio);
+ } else {
+ h = Math.round(changedElement.value);
+ w = Math.round(changedElement.value * aspectRatio);
+ }
+ }
+
+ const [width, height] = clampToBoundaries(w, h)
+
+ const inputEvent = new Event("input", {bubbles: true});
+ this.widthContainer.setVal(width);
+ this.widthContainer.triggerEvent(inputEvent);
+ this.heightContainer.setVal(height);
+ this.heightContainer.triggerEvent(inputEvent);
+ this.heightContainer.inputs.forEach(input => {
+ dimensionChange({target: input}, false, true);
+ });
+ this.widthContainer.inputs.forEach(input => {
+ dimensionChange({target: input}, true, false);
+ });
+ }
+
+ static observeStartup(key, page, defaultOptions, postSetup = (_) => {
+ }) {
+ let observer = new MutationObserver(() => {
+ const widthContainer = gradioApp().querySelector(`#${page}_width`);
+ const heightContainer = gradioApp().querySelector(`#${page}_height`);
+
+ // wait for width and height containers to exist.
+ if (widthContainer && heightContainer && window.opts && window.opts.arh_javascript_aspect_ratio_show !== undefined) {
+ observer.disconnect();
+ if (!window.opts.arh_javascript_aspect_ratio_show) {
+ return;
+ }
+
+ const controller = new AspectRatioController(
+ page,
+ widthContainer,
+ heightContainer,
+ defaultOptions,
+ );
+
+ postSetup(controller);
+ window[key] = controller;
+ }
+ });
+
+ observer.observe(gradioApp(), {childList: true, subtree: true});
+ }
+
+}
+
+const addImg2ImgTabSwitchClickListeners = (controller) => {
+ const img2imgTabButtons = Array.from(document.querySelectorAll('#img2img_settings > div > div > button:not(.selected):not(.hasTabSwitchListener)'));
+ img2imgTabButtons.forEach(button => {
+ button.addEventListener('click', (_) => {
+ // set aspect ratio is RECALLED to change to the image specific to the newly selected tab.
+ if (controller.optionPickingControler.getCurrentOption() === _IMAGE) {
+ controller.setAspectRatio(_IMAGE);
+ }
+
+ addImg2ImgTabSwitchClickListeners(controller);
+ });
+
+ button.classList.add('hasTabSwitchListener');
+ });
+}
+
+const postImageControllerSetupFunction = (controller) => {
+ const scaleToImg2ImgImage = (e) => {
+ const picked = controller.optionPickingControler.getCurrentOption();
+ if (picked !== _IMAGE) return;
+ const files = e.dataTransfer ? e.dataTransfer.files : e.target.files;
+ const img = new Image();
+ img.src = URL.createObjectURL(files[0]);
+ img.onload = () => {
+ controller.setAspectRatio(`${img.naturalWidth}:${img.naturalHeight}`)
+ };
+ }
+
+ _IMAGE_INPUT_CONTAINER_IDS.forEach(imageContainerId => {
+ const imageContainer = document.getElementById(imageContainerId);
+ const inputElement = imageContainer.querySelector('input');
+ inputElement.parentElement.addEventListener('drop', scaleToImg2ImgImage);
+ inputElement.addEventListener('change', scaleToImg2ImgImage);
+ })
+
+ addImg2ImgTabSwitchClickListeners(controller);
+}
+
+document.addEventListener("DOMContentLoaded", () => {
+ AspectRatioController.observeStartup(
+ "__txt2imgAspectRatioController",
+ "txt2img",
+ [_OFF, _LOCK]
+ );
+ AspectRatioController.observeStartup(
+ "__img2imgAspectRatioController",
+ "img2img",
+ [_OFF, _LOCK, _IMAGE],
+ postImageControllerSetupFunction
+ );
+});
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/scripts/sd_webui_aspect_ratio_helper.py b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/scripts/sd_webui_aspect_ratio_helper.py
new file mode 100644
index 0000000000000000000000000000000000000000..ed78b20d6a33f4f180b0f04392dfbb08f72ec1a8
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/scripts/sd_webui_aspect_ratio_helper.py
@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+from aspect_ratio_helper.main import AspectRatioStepScript
+
+__all__ = ['AspectRatioStepScript']
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/style.css b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/style.css
new file mode 100644
index 0000000000000000000000000000000000000000..d6a3378cdb00d7d716570823ad5e49d8d091f39e
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/style.css
@@ -0,0 +1,36 @@
+#txt2img_size_toolbox, #img2img_size_toolbox {
+ min-width: unset !important;
+ gap: 0;
+}
+
+#txt2img_ratio, #img2img_ratio {
+ padding: 0px;
+ min-width: unset;
+ max-width: fit-content;
+}
+
+#txt2img_ratio select, #img2img_ratio select {
+ background: var(--background-fill-primary);
+ border: 1px solid var(--block-border-color);
+ color: var(--block-label-text-color);
+ -webkit-appearance: none;
+ -moz-appearance: none;
+ appearance: none;
+ background-image: none;
+ tab-size: 4;
+ font-size: 14px;
+ line-height: 1.4;
+ box-sizing: border-box;
+ position: relative;
+ border-radius: 8px;
+ box-shadow: 0 1px 2px 0 rgba(0, 0, 0, 0.05);
+ padding-right: 1px;
+ margin-bottom: 10px;
+ padding-left: 2px;
+ text-align: center;
+ min-width: 2.5em;
+}
+
+.arh-btn-row button {
+ min-width: fit-content;
+}
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/tests/__init__.py b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/tests/__init__.py
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/tests/_util_test.py b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/tests/_util_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..4d736c15376638b53e564b0d8449e49d00f0bcb1
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/tests/_util_test.py
@@ -0,0 +1,291 @@
+from __future__ import annotations
+
+import pytest
+
+import aspect_ratio_helper._constants as _constants
+import aspect_ratio_helper._util as _util
+
+
+@pytest.mark.parametrize(
+ 'num, expected',
+ [
+ (50, 'x0.5'),
+ (150, 'x1.5'),
+ (175, 'x1.75'),
+ (250, 'x2.5'),
+ ],
+)
+def test_display_multiplication(num, expected):
+ assert _util.display_multiplication(num) == expected
+
+
+@pytest.mark.parametrize(
+ 'num, expected',
+ [
+ (50, '50%'),
+ (75, '75%'),
+ (100, '100%'),
+ (150, '150%'),
+ (250, '250%'),
+ ],
+)
+def test_display_raw_percentage(num, expected):
+ assert _util.display_raw_percentage(num) == expected
+
+
+@pytest.mark.parametrize(
+ 'num, expected_output', [
+ (150, '+50%'),
+ (100, '0%'),
+ (50, '-50%'),
+ (0, '-100%'),
+ (200, '+100%'),
+ (75, '-25%'),
+ ],
+)
+def test_display_minus_and_plus(num, expected_output):
+ assert _util.display_minus_and_plus(num) == expected_output
+
+
+@pytest.mark.parametrize(
+ 'width, height, pct, expected',
+ [
+ pytest.param(200, 400, 0.5, (96, 200), id='50_percent_scale_down'),
+ pytest.param(100, 200, 2.0, (200, 400), id='200_percent_scale_up'),
+ pytest.param(100, 200, 1.1, (112, 224), id='10_percent_scale_up'),
+ pytest.param(100, 200, 0.9, (88, 176), id='10_percent_scale_down'),
+ pytest.param(100, 200, 0.0, (64, 128), id='scale_full_down'),
+ pytest.param(
+ _constants.MIN_DIMENSION - 1,
+ _constants.MIN_DIMENSION - 1,
+ 0.5,
+ (_constants.MIN_DIMENSION, _constants.MIN_DIMENSION),
+ id='scale_belowMIN_DIMENSION',
+ ),
+ pytest.param(
+ _constants.MAX_DIMENSION + 1,
+ _constants.MAX_DIMENSION + 1,
+ 2.0,
+ (_constants.MAX_DIMENSION, _constants.MAX_DIMENSION),
+ id='scale_aboveMAX_DIMENSION',
+ ),
+ ],
+)
+def test_scale_by_percentage(
+ width, height, pct, expected,
+):
+ assert _util.scale_by_percentage(
+ width, height, pct,
+ ) == expected
+
+
+@pytest.mark.parametrize(
+ 'width, height, max_dim, expected',
+ [
+ # pytest.param(
+ # 100, 200, 400, (200, 400),
+ # id='scale_up_to_max_dimension_horizontally',
+ # ),
+ # pytest.param(
+ # 200, 100, 400, (400, 200),
+ # id='scale_up_to_max_dimension_vertically',
+ # ),
+ # pytest.param(
+ # 400, 64, 400, (400, 64),
+ # id='no_scale_up_needed_with_max_dimension_width',
+ # ),
+ # pytest.param(
+ # 64, 400, 400, (64, 400),
+ # id='no_scale_up_needed_with_max_dimension_height',
+ # ),
+ # pytest.param(
+ # _constants.MIN_DIMENSION,
+ # _constants.MIN_DIMENSION,
+ # _constants.MAX_DIMENSION,
+ # (_constants.MAX_DIMENSION, _constants.MAX_DIMENSION),
+ # id='scale_from_min_to_max',
+ # ),
+ # pytest.param(
+ # _constants.MAX_DIMENSION,
+ # _constants.MAX_DIMENSION,
+ # _constants.MIN_DIMENSION,
+ # (_constants.MIN_DIMENSION, _constants.MIN_DIMENSION),
+ # id='scale_from_max_to_min',
+ # ),
+ pytest.param(
+ _constants.MIN_DIMENSION, 32, _constants.MIN_DIMENSION,
+ (128, _constants.MIN_DIMENSION),
+ id='scale_below_min_height_dimension_clamps_retains_ar',
+ ),
+ pytest.param(
+ 32, _constants.MIN_DIMENSION, _constants.MIN_DIMENSION,
+ (_constants.MIN_DIMENSION, 128),
+ id='scale_below_min_width_dimension_clamps_retains_ar',
+ ),
+ # pytest.param(
+ # _constants.MAX_DIMENSION, 4096, _constants.MAX_DIMENSION,
+ # (1024, _constants.MAX_DIMENSION),
+ # id='scale_above_max_height_dimension_clamps_retains_ar',
+ # ),
+ # pytest.param(
+ # 4096, _constants.MAX_DIMENSION, _constants.MAX_DIMENSION,
+ # (_constants.MAX_DIMENSION, 1024),
+ # id='scale_above_max_width_dimension_clamps_retains_ar',
+ # ),
+ # pytest.param(
+ # 64, 64, _constants.MIN_DIMENSION - 1,
+ # (_constants.MIN_DIMENSION, _constants.MIN_DIMENSION),
+ # id='scale_dimension_below_min_dimension_clamps_retains_ar',
+ # ),
+ # pytest.param(
+ # 64, 64, _constants.MAX_DIMENSION + 1,
+ # (_constants.MAX_DIMENSION, _constants.MAX_DIMENSION),
+ # id='scale_dimension_above_max_dimension_clamps_retains_ar',
+ # ),
+ ],
+)
+def test_scale_dimensions_to_max_dim(
+ width, height, max_dim, expected,
+):
+ assert _util.scale_dimensions_to_max_dim(
+ width, height, max_dim,
+ ) == expected
+
+
+@pytest.mark.parametrize(
+ 'width, height, min_dim, expected',
+ [
+ pytest.param(
+ 100, 200, 400, (400, 800),
+ id='scale_up_to_min_dimension_with_ar_preservation',
+ ),
+ pytest.param(
+ 200, 100, 400, (800, 400),
+ id='scale_up_to_min_dimension_with_ar_preservation',
+ ),
+ pytest.param(
+ 100, 100, 400, (400, 400),
+ id='no_scale_up_needed_with_min_dimension',
+ ),
+ pytest.param(
+ _constants.MIN_DIMENSION, _constants.MIN_DIMENSION, _constants.MAX_DIMENSION,
+ (_constants.MAX_DIMENSION, _constants.MAX_DIMENSION),
+ id='scale_up_to_max_dimension_with_ar_preservation',
+ ),
+ pytest.param(
+ _constants.MAX_DIMENSION, _constants.MAX_DIMENSION, _constants.MIN_DIMENSION,
+ (_constants.MIN_DIMENSION, _constants.MIN_DIMENSION),
+ id='scale_down_to_min_dimension_with_ar_preservation',
+ ),
+ pytest.param(
+ 100, 100, _constants.MAX_DIMENSION,
+ (_constants.MAX_DIMENSION, _constants.MAX_DIMENSION),
+ id='scale_up_to_max_dimension_with_ar_preservation',
+ ),
+ ],
+)
+def test_scale_dimensions_to_min_dim(
+ width, height, min_dim, expected,
+):
+ assert _util.scale_dimensions_to_min_dim(
+ width, height, min_dim,
+ ) == expected
+
+
+class SharedOpts:
+ def __init__(self, options=None, defaults=None):
+ self.options = options or {}
+ self.defaults = defaults or {}
+
+ def __getattr__(self, key):
+ try:
+ return self.options[key]
+ except KeyError:
+ raise AttributeError()
+
+ def get_default(self, key):
+ return self.defaults.get(key, None)
+
+
+def test_safe_opt_util():
+ shared_opts = SharedOpts(options={'key': 'value'})
+ assert _util.safe_opt_util(shared_opts, 'key', {}) == 'value'
+
+
+def test_safe_opt_util_none():
+ shared_opts = SharedOpts(options={'key': None}, defaults={'key': 'value'})
+ assert _util.safe_opt_util(shared_opts, 'key', {}) == 'value'
+
+
+def testsafe_opt_util_default_a():
+ shared_opts = SharedOpts(
+ defaults={'key': 'default_a'},
+ )
+ assert _util.safe_opt_util(
+ shared_opts, 'key', {'key': 'default_b'},
+ ) == 'default_a'
+
+
+def test_safe_opt_util_default_b():
+ shared_opts = SharedOpts(defaults={'key': None})
+ assert _util.safe_opt_util(
+ shared_opts, 'key', {'key': 'default_b'},
+ ) == 'default_b'
+
+
+@pytest.mark.parametrize(
+ 'options', ({'key': None}, {}),
+)
+def test_safe_opt_safe_return_no_defaults_b(options):
+ shared_opts = SharedOpts(options=options)
+ assert _util.safe_opt_util(shared_opts, 'unknown_key', {}) is None
+
+
+@pytest.mark.parametrize(
+ 'value, expected', [
+ (0, 0),
+ (7, 8),
+ (10, 8),
+ (16, 16),
+ (23, 24),
+ (32, 32),
+ (33, 32),
+ (100, 96),
+ (10.5, 8),
+ (15.3, 16),
+ (21.8, 24),
+ (33.9, 32),
+ (98.7, 96),
+ ],
+)
+def test_round_to_multiple_of_8(value, expected):
+ assert _util.round_to_multiple_of_8(value) == expected
+
+
+@pytest.mark.parametrize(
+ 'width, height, aspect_ratio, expected', [
+ (100, 100, 1.0, (96, 96)),
+ (3000, 2000, 1.5, (2048, 1368)),
+ (500, 8000, 0.5, (1024, 2048)),
+ (500, 300, 2.0, (496, 304)),
+ (100, 200, 0.5, (96, 200)),
+ (500, 500, 1.2, (496, 496)),
+ (2048, 2048, 1.0, (2048, 2048)),
+ (2049, 2048, 1.0, (2048, 2048)),
+ (2048, 2049, 1.0, (2048, 2048)),
+ (2049, 2049, 1.0, (2048, 2048)),
+ (63, 63, 1.0, (64, 64)),
+ (2050, 2050, 1.0, (2048, 2048)),
+ (63, 64, 1.0, (64, 64)),
+ (64, 63, 1.0, (64, 64)),
+ (64, 64, 1.0, (64, 64)),
+ (63, 63, 1.0, (64, 64)),
+ (2050, 63, 1.0, (2048, 2048)),
+ (63, 2050, 1.0, (2048, 2048)),
+ (2050, 2050, 0.01, (64, 2048)),
+ (100.5, 100.5, 1.0, (104, 104)),
+ (200.3, 100.7, 0.5, (200, 104)),
+ ],
+)
+def test_clamp_to_boundaries(width, height, aspect_ratio, expected):
+ assert _util.clamp_to_boundaries(width, height, aspect_ratio) == expected
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/tox.ini b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/tox.ini
new file mode 100644
index 0000000000000000000000000000000000000000..4c6cac4b77bc3e9d6f668da6c3ce61e3c1551929
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/aspect-ratio/tox.ini
@@ -0,0 +1,6 @@
+[tox]
+envlist = py310, py311
+
+[testenv]
+deps = pytest
+commands = pytest
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/.github/CODEOWNERS b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/.github/CODEOWNERS
new file mode 100644
index 0000000000000000000000000000000000000000..ea30e3119f3bb1dbefb1ecddcf5e2c2af997a482
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/.github/CODEOWNERS
@@ -0,0 +1,2 @@
+# Global rule:
+* @etherealxx
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/.gitignore b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/.gitignore
new file mode 100644
index 0000000000000000000000000000000000000000..ef06d19b1f72bc93f56af94b78a7d52724a9daf3
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/.gitignore
@@ -0,0 +1,168 @@
+*.bak
+*.backup
+imageratio.*
+*.lnk
+tempCodeRunnerFile.python
+.DS_Store
+gradiovenv/
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+# Usually these files are written by a python script from a template
+# before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+# For a library or package, you might want to ignore these files since the code is
+# intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+# However, in case of collaboration, if having platform-specific dependencies or dependencies
+# having no cross-platform support, pipenv may install dependencies that don't work, or not
+# install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+# This is especially recommended for binary packages to ensure reproducibility, and is more
+# commonly ignored for libraries.
+# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+# in version control.
+# https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+# and can be added to the global gitignore or merged into this file. For a more nuclear
+# option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/LICENSE b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/LICENSE
new file mode 100644
index 0000000000000000000000000000000000000000..dfd7367e611d85047cdf8e09109a9b77f08eb861
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/LICENSE
@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 Etherealxx
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/README.md b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/README.md
new file mode 100644
index 0000000000000000000000000000000000000000..8538b40fdd03f1dc08e19ced931461f592aee879
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/README.md
@@ -0,0 +1,491 @@
++ +
+
+ 
+
+
+
+
+
+
+
+
+
+
+
+BatchLinks Downloader
+ +
+ Batch-downloading models and stuff in stable-diffusion-webui colab made simple and fast.
+
+
+
+ Report Bug
+ ·
+ Request Feature
+
+
+
+[does this repo still maintained?](https://github.com/etherealxx/batchlinks-webui/blob/main/regardingthisrepo.md)
+
+## Installation
+
+Copy this line into your colab installation cell. Or into a new cell if you already launched the webui.
+
+```
+!git clone https://github.com/etherealxx/batchlinks-webui /content/stable-diffusion-webui/extensions/batchlinks-webui
+```
+
+or, you can copy the url of this repo and install it via webui and restart the UI.Table of Contents
+-
+
- Installation +
- About +
- Example +
- Syntax + + + + + + + +
- Additional Syntax + + + + + +
- Gradio Queue + + + + +
- Other Features + + + + + +
- Release Notes +
- Roadmap +
- Known Bugs + +
- Contributing +
- Contact +
- Acknowledgments + +
+
+(If `gradio no interface is running` or `bad gateway` shows up when restarting the UI, that means you need to restart the cell anyway 😅) + + + +Using `--gradio-queue` on the launch.py argument is highly recommended, as it enables this extension to show download progress bar on the UI and a cancel button. The option itself has no negative effect on the webui. [Read more here.](https://github.com/etherealxx/batchlinks-webui#gradio-queue)
+
+ +You can also run this extension in [SDless mode](https://github.com/etherealxx/batchlinks-webui#sdless-mode) btw. + +While it's not recommended to use this extension on your local installation, you can use this extension on Windows. [More here.](https://github.com/etherealxx/batchlinks-webui#local-installation-support) + +## About + +This extension will streamline your downloads on your [stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui) colab session. Paste the links you need to download (or you can upload a txt file containing the links), use the hashstag syntax to choose the download location (see below), and hit the `Download All` button to download them! + +## Example + +Look at this example
+
+ +``` +#model +https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix3/AOM3_orangemixs.safetensors +#vae +https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/vae/kl-f8-anime2.ckpt +#embed +https://huggingface.co/datasets/Nerfgun3/bad_prompt/resolve/main/bad_prompt_version2.pt +https://huggingface.co/etherealxx/whoseisthis/resolve/main/bad_artist.pt +#lora +https://civitai.com/api/download/models/8840 ##1-mb-lora-trained-in-5-mins-that-does-the-same-thing-as-25-gb-model-but-better +https://civitai.com/api/download/models/6891 ##sans undertale +https://mega.nz/file/gAxTWBAI#uL7EZay-OND5G6ELJlGfUNG0s7Q4TynZKqFdvs0v0tc ##agent8-finetune +https://mega.nz/file/oU43FSTY#vwAfsAb7RKJ4xtsSP7uzrKpWhh1y8BdpIBFurwsVP2o ##agent8-dreambooth +#hypernet +https://cdn.discordapp.com/attachments/1070489470127841381/1070489471964954684/MomopocoV3.pt +``` + +This piece of lines will be read from top to bottom. Every hashtag, it will change the current output directory to said directory (see below). So what this example do is it will download `AOM3` model to the model folder, then it will download the vae and put it to the Vae folder. Next it will download two embed, `bad prompt` and `bad artist`. Next it will download several LoRAs from CivitAI and MEGA, and put it to the Lora folder. Lastly, it changes the directory to hypernet directory, then + +You can also copy that example and paste it to a `.txt` file to use later. You can load a `.txt` file containing that piece of lines directly from the UI. + +When the items is downloading, you can inspect the running code on the colab cell, or just take a coffee and chill☕. If you activate logging, you can inspect the download progress from the UI, more [here](https://github.com/etherealxx/batchlinks-webui#logging) + +When the download is complete, the downloaded file will be listed
+
+ +## Syntax + +#### Hashtag +- Hashtag means change current output directory to this directory. `#model` means every links below this hashtag, will be downloaded to _/content/stable-diffusion-webui/models/Stable-diffusion_, until it hits another hashtag, which will change the output directory again. See [below](https://github.com/etherealxx/batchlinks-webui#valid-hashtags) for valid hashtags. + + Note: If you use some colab that purposefully uses `sd-webui-additional-networks` extension to load Lora, use `#addnetlora` instead of `#lora`. It will download the lora to where it supposed to be. + +#### Links +- Links are the main things you wants to be downloaded. Current supported links are from: + - Huggingface (https://huggingface.co/) + - MEGA (https://mega.nz/) {using `mega-cmd`} + - CivitAI (https://civitai.com/) + - Discord attachments (https://cdn.discordapp.com/attachments/) + - catbox (https://files.catbox.moe) + - Github (https://github.com or https://raw.githubusercontent.com) + - Google Drive (https://drive.google.com) {using `gdown`} + - Pixeldrain (https://pixeldrain.com/u/) + - Mediafire (https://www.mediafire.com/file) {using `mediafire-dl`} + - anonfiles (https://anonfiles.com) + + Every links other than that will be ignored. Keep in mind the only supported links are direct download links (see [here](https://github.com/etherealxx/batchlinks-downloader/blob/main/howtogetthedirectlinks.md)). For Huggingface, Civitai (model page link method), Discord attachments, catbox, pixeldrain, and anonfiles, there will be four different method of downloading offered (see [below](https://github.com/etherealxx/batchlinks-webui#huggingfaces-download-method)). For MEGA, it will use `mega-cmd` to download. For Github, if, the link is a raw file, it will download the file. Else, it will use `git clone`, useful to clone extension repo into the webui extension folder. + + More about CivitAI download method [here](https://github.com/etherealxx/batchlinks-webui#civitais-download-method). + +#### Double Hashtag +- Double hashtag means comment. You can put double hashtag in the same line of the link and it will be ignored (keep in mind to put the link first then the double hashtag) + +#### Others +- Other texts will be ignored. + +### Valid Hashtags + +`#model`, `#models`, `#checkpoint`, or `#checkpoints` will put the downloaded file to _/content/stable-diffusion-webui/models/Stable-diffusion_ + +`#embedding`, `#embeddings`, `#embed`, `#embeds`, `#textualinversion`, or `#ti` will put the downloaded file to _/content/stable-diffusion-webui/embeddings_ + +`#vae` or `#vaes` will put the downloaded file to _/content/stable-diffusion-webui/models/VAE_ + +`#hypernetwork`, `#hypernetworks`, `#hypernet`, `#hypernets`, `#hynet`, or `#hynets` will put the downloaded file to _/content/stable-diffusion-webui/models/hypernetworks_ + +`#lora` or `#loras` will put the downloaded file to _/content/stable-diffusion-webui/models/Lora_ + +`#addnetlora`, `#loraaddnet`, `#additionalnetworks`, or `#addnet` will put the downloaded file to _/content/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora_ + +`#aestheticembedding` or `#aestheticembed` will put the downloaded file to _content/stable-diffusion-webui/extensions/stable-diffusion-webui-aesthetic-gradients/aesthetic_embeddings_ + +`#controlnet` or `#cnet` will put the downloaded file to _/content/stable-diffusion-webui/extensions/sd-webui-controlnet/models_ + +`#upscale` or `#upscaler` will put the downloaded file to _/content/stable-diffusion-webui/models/ESRGAN_ + +`#lycoris`, `#locon`, or `#loha` will put the downloaded file to _/content/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora/lycoris_ + +`#altmodel` or `#altmodels` will put the downloaded file to the path you choose when using `--ckptdir` argument on `launch.py` line. If you didn't use that argument, this hashtag will points to the same directory as `#model` + +_Lycoris/Locon/Loha_ will works just fine if you use `#addnetlora` instead, as long as you have both [addnet extension](https://github.com/kohya-ss/sd-webui-additional-networks) and [locon extension](https://github.com/KohakuBlueleaf/a1111-sd-webui-locon) installed. It's unclear if it's can be loaded by native auto1111 lora. (Tell me if you know more about this.) + +Github links (if it does not contain `/raw/` or `/release/download/` in it) doesn't need hashtag. It will always considered as webui extension, and the repository will be cloned to _/content/stable-diffusion-webui/extensions/(reponame)_ + +### How to get the direct links (Important!) + +See [here](https://github.com/etherealxx/batchlinks-downloader/blob/main/howtogetthedirectlinks.md) + +### Huggingface's download method + +So there's four supported method: `gdown`, `wget`, `curl` and `aria2`. Use whatever, really. The difference between them are actually little. Myself love using `gdown` since the output is cleaner than the others. `aria2` has the fastest download speed though. + + + +## Additional Syntax +### Rename Downloaded Files +Using `>` symbol, you can rename files. Take this for example
+
+Mostly catbox file has random name, by using `>` symbol after the link, you can type the desired name on the right. (Don't forget the extension) + +### Running Shell Commands +You can run shell commands by using `!` in front of the command you want, just like in google colab cells. Then press the `Download All!` button. (Sure, it doesn't download anything, but, well😅)
+
+
+You can run many lines at once too! + +### Extract Everyting (`@extract`) +You can use `@extract` to extract every `*.7z`, `*.rar`, and `*.zip` on current directory). +``` +#lora +
For example, you type this and press the `Download All!` button: +``` +@new #private /content/stable-diffusion-webui/outputs +#private +
+
+ +### Cancel +Pressing `cancel` button while download in progress will stops the current session. Useful when at one time the download speeds is too slow. If you're currently downloading a single item, that item will be cancelled, but the other downloaded one will remain intact.
+
+ +### Progress Bar +There will be an additional progress bar that tells you the current activities.
+
+ +### Changes if `--gradio-queue` is off +Feature listed above will dissapear, and your only option for download is just `aria2` (speed is priority).
Note that when you pressed the `Download All!` button, **nothing will shows up on the UI**. You need to check the colab console.
+
+Another thing to note is your download session will always be cutted every 70 seconds (to prevent hangs/desync).
+
+Don't worry, you can continue your session by pressing the `Resume Download` button. It will refresh the links with the one you haven't downloaded yet, then pressing `Download All!` will download the remaining links.
+
+ +## Other Features + +### Notification + +This extension will play sound effect when the download process is completed. It's somewhat buggy at the moment though. + +### Copy From Pastebin + +You can put your links on pastebin (write it the way you wrote on Batchlinks), put the pastebin link (https://pastebin.com/xxxx) on the link input, and press the 'Copy from Pastebin' button. It will copy all the the links on the pastebin to the UI. Useful to avoid hassle copy and pasting a bunch of text while on mobile.
+
+ +### SDless mode +This mode will run the extension without the need of stable-diffusion-webui. Good for my own purpose of debugging it😁.
+
+ +Copy-paste this on a new colab cell then run it: +``` +from IPython.display import clear_output +!pip3 show virtualenv >/dev/null || pip install -q virtualenv +![ -d gradiovenv ] || virtualenv gradiovenv +!git clone https://github.com/etherealxx/batchlinks-webui \ +/content/stable-diffusion-webui/extensions/batchlinks-webui +pip3 show gradio >/dev/null || pip install -q gradio==3.16.2; \ +pip3 show tqdm >/dev/null || pip3 install -q tqdm +clear_output(wait=True) +!source gradiovenv/bin/activate; \ +python /content/stable-diffusion-webui/extensions/batchlinks-webui/scripts/batchlinks-downloader.py +``` +or here's the quicker version without venv: +``` +!pip install gradio==3.16.2 +!git clone https://github.com/etherealxx/batchlinks-webui /content/stable-diffusion-webui/extensions/batchlinks-webui +!python /content/stable-diffusion-webui/extensions/batchlinks-webui/scripts/batchlinks-downloader.py +``` + +For Windows, run the sdless-windows.bat. Make sure you have `python3` and `gradio` package version 3.16.2 or above. It won't use venv. + +This one for MacOS. Tested on Mojave. Create a shell script and give execute access to it. +``` +#!/usr/bin/env bash +pip3 show virtualenv >/dev/null || pip3 install virtualenv +[ -d gradiovenv ] || virtualenv gradiovenv +git clone https://github.com/etherealxx/batchlinks-webui \ +$HOME/Downloads/stable-diffusion-webui/extensions/batchlinks-webui +source gradiovenv/bin/activate; \ +pip3 show gradio >/dev/null || pip3 install gradio==3.16.2; \ +pip3 show tqdm >/dev/null || pip3 install -q tqdm +source gradiovenv/bin/activate; \ +python3 $HOME/Downloads/stable-diffusion-webui/extensions/batchlinks-webui/scripts/batchlinks-downloader.py +``` + +### Local Installation Support + +This extension was tested to work on Windows 11. But ultimately, the number one priority for me is colab use, so there's might be bugs on Windows 11 one. Maybe also works on Debian-based linux (but you better inspect the source code first). +On Windows, this extension will install [MEGAcmd](https://github.com/meganz/MEGAcmd) for MEGA file download, [wget-windows](https://github.com/webfolderio/wget-windows) for download using `wget`, [aria2-static-builds](https://github.com/q3aql/aria2-static-builds) for download using `aria2`, [7zr and 7z](https://www.7-zip.org) for extracting `aria2` installation, and for `@extract` command. +MacOS is just partially supported. + +Also remember, in order for this extension to work properly, you need make sure that you tick that `add Python 3.10 to PATH` checkbox when you installed Python (Which is already recommended when installing stable-diffusion-webui)
+
+ +### Debug Mode (Developer only) + +By manually switching `globaldebug = False` to `True` on the source code, or using `--debug` argument when running the sdless mode, the Debug Mode will be activated. It shows every `print` calls i use to track variables. It will also tracks every files on the models (etc.) paths into a `snapshot.txt` file (if there isn't one). + +There are some batchlinks syntax features that only available on debug mode (putting it on the textbox and click `Download All!`): + +`@debugresetdownload` - This command will check every model (etc.) directory, and remove every file that isn't on `snapshot.txt`. Useful for me to removes file quickly when testing. **Be careful using this on your local installation.** + +`@debugresetdownload` - This command is (supposed to) download a single link with every method available (gdown, curl, wget, aria2), but now i rarely use this command, and haven't updated since. Might be buggy. + +## Release Notes +Moved [here](https://github.com/etherealxx/batchlinks-webui/blob/main/releasenotes.md). Check it out for latest features & bug fixes. + +## Roadmap + +- [ ] Add checker for downloaded models (so that it won't download again after the model is downloaded) +- [ ] Different UI for mobile +- [ ] Gradio progress bar +- [ ] Integrating refresh button (`create_refresh_button`) to refresh models etc. all at once +- [ ] Logo change +- [ ] Moving most of the content of this `Readme.md` to Wiki instead +- [ ] Other download sites (s-ul.eu, gitgud, bunkr.ru, icedrive) +- [ ] Separate 'download' and 'run command' function with a subtab +- [ ] Support download MEGA and Google Drive folder +- [ ] Using `yield` instead of gradio's `every` +- [ ] Youtube video (Tutorial) +- [ ] (Crawler) Download every sd-webui related file from a huggingface repo +- [ ] (Crawler) Download every model from a civitai user page +
_ +- [x] aria2 for huggingface download method +- [x] Cleaning the code from unnecesarry comments +- [x] Completed download will use the webui's notification.mp3 +- [x] Support customizable hashtag from the UI +- [x] Supports Windows local installation +- [x] UI overhaul +- [x] Using threading/subprocess instead of os.system to download files +- [x] (Windows) wget & aria2 support + +## Known Bugs + +- Progress bar (the yellow bar) doesn't progress as expected +- Sometimes notification sound doesn't play when downloading same file twice in a row +- Sometimes notification sound shows up when starting download, instead of when the download process is completed. +- There's still a chance that the UI of non `--gradio-queue` session and/or onedotsix freezes after a download session +- Windows: The delay between file is downloaded and the output shows is pretty long, and even sometimes the notification comes at the wrong time. +
+ +### Known Fixed Bugs + +
+
+
+## Contributing
+
+
+
+[I no longer actively maintain this project](https://github.com/etherealxx/batchlinks-webui/blob/main/regardingthisrepo.md). if you have a suggestion or code-fixing that would make this better, please notify me in the issue tab, fork the repo and create a pull request. 👈Fixed Bugs
+-
+
Links that has bracket in it needs to be 'escaped' (For example,Baka-DiffusionV1(Fp16).safetensorsmust be typedBaka-DiffusionV1\(Fp16\).safetensors)
☝️Fixed in v2.0.0 +The delay between file is downloaded and the output shows is really long (1min+) on camenduru's v1.6 colab (Gradio related?)
☝️Seems like fixed in v1.1.0 +File downloaded from MEGA will not listed on the output, as it use different download method. There is some delay between the transfare() function complete until it writes the file. I don't know how long the delay is.
☝️Fixed in v1.1.0 +Sometimes colab cannot be shut down with a single click on the stop button. Hitting the button several times will raise a KeyboardInterrupt and forcely stopping the cell.
☝️Seems like fixed somewhere on v3.0.0 and after +
-
+
+But still, a star on this project would be nice. Thanks again. + + + +## Contact + +My Email - gwathon3@gmail.com
+My Youtube - [MJ Devlog](https://www.youtube.com/@mjdevlog) + +## Acknowledgments + +- [Camenduru's Webui Huggingface](https://github.com/camenduru/stable-diffusion-webui-huggingface) - I use his extension as my base (my gradio skill sucks T.T) +- [SD Civitai Browser](https://github.com/Vetchems/sd-civitai-browser) - Civit download script +- [Mega-to-Google-Drive](https://github.com/menukaonline/Mega-to-Google-Drive) - MEGA download script +- [MEGAcmd](https://github.com/meganz/MEGAcmd) +- [mediafire-dl](https://github.com/Juvenal-Yescas/mediafire-dl) +- [Pixeldrain downloader](https://github.com/FayasNoushad/Pixeldrain) - Inspiration for Pixeldrain download method +- [wget-windows](https://github.com/webfolderio/wget-windows) +- [aria2-static-builds](https://github.com/q3aql/aria2-static-builds) +- [7zr, 7z](https://www.7-zip.org) diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/howtogetthedirectlinks.md b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/howtogetthedirectlinks.md new file mode 100644 index 0000000000000000000000000000000000000000..f7387a794f4dbde713f74f1e5e7f45c92000cc4e --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/howtogetthedirectlinks.md @@ -0,0 +1,104 @@ +# How to get the direct links + +Jump to: [Huggingface](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#huggingface) - [Civit AI](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#civit-ai) - [MEGA](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#mega) - [Discord](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#discord-attachments) - [Github (Extension)](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#webui-extension-github-repo-link) - [Github (Raw File)](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#raw-file-from-github) - [Github (Release File)](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#release-file-from-github) - [Google Drive](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#google-drive) - [Pixeldrain](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#pixeldrain) - - [Mediafire](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#mediafire) - [Anonfiles](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#anonfiles) - [Dropbox](https://github.com/etherealxx/batchlinks-webui/blob/main/howtogetthedirectlinks.md#dropbox) + +## Huggingface + +Locate the model page, click on `files` tab.
+
+choose the model you want, **right click** the arrow button beside the LFS logo, then click `copy link address`.
+
+ +## Civit AI + +There are two ways to download links from Civit. The model page link method, and the direct link method. The model page link method **will automatically** choose the directory of the saved model without even using hashtag, **except** if you want to put Lora into `#addnetlora` directory, you can use that hashtag. You still need to define the output directory when using the direct link method. + +### Model Page Link Method + +Locate the desired model on the main page of Civitai, **right click** the model thumbnail, then click `copy link address`.
+
+You can also click the thumbnail first then copy the link.
+
+The advantage of this method is you will also get the model preview images that can shows up on the latest version of [stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui). + +### Direct Link Method + +Locate the model page, **right click** the huge blue button, then click `copy link address`.
+
+With this method, you can also choose model variation, by clicking the dropdown icon on the blue button, then **right click** the desired model an.
+
+ +## MEGA + +Locate the model you wanted, **right click** the item, click `copy link address`.
+
+Click the huge green copy button.
+
+ +## Discord attachments + +Locate the file you wanted, **right click** the download icon, then click `get link`.
+
+Then click `copy link`.
+
+ +## webui Extension (Github repo link) + +Locate the extension's repository, then copy the repository link.
+
+ +Extension **will automatically** cloned/downloaded on `extensions` directory of webui without the need of hashtag. + +## Raw file from Github + +Locate the file's repository, then click the desired file.
+
+Then **right click** the text that says `raw` **or** `download`, and click `copy link address`
+
+
+ +## Release file from Github + +Locate the file's repository, then click release section.
+
+Then **right click** the desired assets (the blue text), and click `copy link address`
+
+ +## Google Drive + +Whether you own the file or not, make sure the file is public. If you owns the file, right click on the file on Google Drive, then click `get link` (or `share`).
+
+Then change the access to `anyone with thelink`, and copy the link by clicking the button
+
+ +These are some valid google drive link patterns that works: +``` +https://drive.google.com/file/d/1-UbwFtSzKTQkvjdPiKLCejH0Ztq9UJFO/view +https://drive.google.com/file/d/1-UbwFtSzKTQkvjdPiKLCejH0Ztq9UJFO/view?usp=share_link +https://drive.google.com/u/0/uc?id=1-UbwFtSzKTQkvjdPiKLCejH0Ztq9UJFO&export=download +https://drive.google.com/u/0/uc?id=1-UbwFtSzKTQkvjdPiKLCejH0Ztq9UJFO&export=download&confirm=t +``` +## Pixeldrain + +Just click the `copy link` button, or just copy the link.
+
+ +## Anonfiles + +**Don't** right click the download button. Instead, copy the link on the url bar
+
+ +## Mediafire + +When you arrived on this page, click the chain link icon to copy link, or just copy the link above
+
+ +Don't forget that when being pasted, the link must starts with `https://www.mediafire.com/file/` (notice the triple W) + +## Dropbox + +When the item is yours, go to the homepage, and click copy link, the link will be copied to your clipboard
+
+ +Or, if you got into this screen, just copy the link above, with the link ends with either `?dl=0` or `?dl=1`
+
\ No newline at end of file diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/anonfiles.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/anonfiles.png new file mode 100644 index 0000000000000000000000000000000000000000..35c0c1f21d0b215c4194e32dfb92fd949f81b0d7 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/anonfiles.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/batchlinks_logo.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/batchlinks_logo.png new file mode 100644 index 0000000000000000000000000000000000000000..dc6e5d11ab78ca77993a96f338397199f5af94d3 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/batchlinks_logo.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/cancel.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/cancel.jpg new file mode 100644 index 0000000000000000000000000000000000000000..89dc5a6831d9966aa2045a1fca8730672dee802d Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/cancel.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/checkconsole1.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/checkconsole1.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5a6e095623449bd460c2d272c4dfb5631b1297f9 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/checkconsole1.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/checkconsole2.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/checkconsole2.jpg new file mode 100644 index 0000000000000000000000000000000000000000..59f7d9d1704fdea277bcd20cade11b87d6a461f0 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/checkconsole2.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civit1.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civit1.jpg new file mode 100644 index 0000000000000000000000000000000000000000..a87cccfdde87e026b0c9f602d405a7dbbafdc4da Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civit1.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civit2.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civit2.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5eb77dc1ea9fc5b1749a0712a77b3fbd6eea203a Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civit2.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civitnew1.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civitnew1.jpg new file mode 100644 index 0000000000000000000000000000000000000000..7e9847b166479d3429bd99cedf1a524a90764346 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civitnew1.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civitnew2.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civitnew2.jpg new file mode 100644 index 0000000000000000000000000000000000000000..800ef89144fbab0e4cdf7e6e2037c8fdf181a55c Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/civitnew2.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/discord1.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/discord1.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c7ba80743b7721bc3e436e209d1e67c661f61722 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/discord1.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/discord2.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/discord2.jpg new file mode 100644 index 0000000000000000000000000000000000000000..42948c286a483dab68d7ef82afadbdcb2912d91e Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/discord2.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/downloaded.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/downloaded.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2dd5e1518347369bff9afb163d46ba98a9726123 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/downloaded.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/drive1.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/drive1.png new file mode 100644 index 0000000000000000000000000000000000000000..98dbc750e1768d867f8127ba353fa0783bbf3630 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/drive1.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/drive2.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/drive2.png new file mode 100644 index 0000000000000000000000000000000000000000..5ac6901e4b2516d5fb8d846b26fe46ab656a7197 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/drive2.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/dropbox1.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/dropbox1.png new file mode 100644 index 0000000000000000000000000000000000000000..8090d94242da605587870612dcfdb6d092c1e5f3 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/dropbox1.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/dropbox2.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/dropbox2.png new file mode 100644 index 0000000000000000000000000000000000000000..7ff64657e17618e9b319e190f20faa50018ee503 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/dropbox2.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/example.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/example.jpg new file mode 100644 index 0000000000000000000000000000000000000000..f1a58c7c05d81f564c4c3573000d7e1b739b7ae0 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/example.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/ext_installer.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/ext_installer.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8e5553ed6a76166a80a75ac3b8f46f6aa65ea59d Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/ext_installer.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/github.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/github.jpg new file mode 100644 index 0000000000000000000000000000000000000000..32b6a9f145e39b31aff169add332acacb86f317a Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/github.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubdirect1.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubdirect1.png new file mode 100644 index 0000000000000000000000000000000000000000..a6242bb23c1d8c5a0e3d4d620ab93d8d1f4a8ec4 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubdirect1.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubdirect2.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubdirect2.png new file mode 100644 index 0000000000000000000000000000000000000000..c07cdc55f6337c61e46a4bbb3e2c00e45e1272b5 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubdirect2.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubdirect3.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubdirect3.png new file mode 100644 index 0000000000000000000000000000000000000000..a1dbd9e15856c31cde956130b6062aab90265646 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubdirect3.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubrelease1.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubrelease1.png new file mode 100644 index 0000000000000000000000000000000000000000..9e54a3f008551eaf60ff37c5f6f1e8be390ca528 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubrelease1.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubrelease2.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubrelease2.png new file mode 100644 index 0000000000000000000000000000000000000000..bf07b1f04b9811d844eb73e53ba08cfc7c292bbc Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/githubrelease2.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/gradiooff1.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/gradiooff1.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8baff0f3fd4c79e74b5f6cac3bc3c8018ac2a4a5 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/gradiooff1.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/gradiooff2.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/gradiooff2.jpg new file mode 100644 index 0000000000000000000000000000000000000000..82ad2ced202f2b7eca44b4a2a82c9f53e57ece1f Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/gradiooff2.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/huggingface1.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/huggingface1.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9dc10e0c6d4119d4b5b1253a0c0e35c31dbabd63 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/huggingface1.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/huggingface2.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/huggingface2.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cb8facfd8051f4254c0ea2c4235ab0c202d23985 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/huggingface2.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/logging.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/logging.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9d7082231c85d3eee336a4713b76cc15ed0cbba9 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/logging.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/mediafire.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/mediafire.png new file mode 100644 index 0000000000000000000000000000000000000000..ed2b630d531a19c74aabd2f885b869da1c0449a5 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/mediafire.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/mega1.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/mega1.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9a127dd23c545fb05edf589f5ade548fad7c5577 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/mega1.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/mega2.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/mega2.jpg new file mode 100644 index 0000000000000000000000000000000000000000..b8124ad533f50ad86ee39fcee34d4b30a858a257 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/mega2.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/pastebin.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/pastebin.png new file mode 100644 index 0000000000000000000000000000000000000000..a3b3238104845fafb7a99179daad44258a7f0bc7 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/pastebin.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/pixeldrain.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/pixeldrain.png new file mode 100644 index 0000000000000000000000000000000000000000..b9f73a7d8a1826bb4ca67ae985a3e8a348e01b81 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/pixeldrain.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/progress.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/progress.jpg new file mode 100644 index 0000000000000000000000000000000000000000..cb1b3eab1e56f92c4bf2710637ae227a5008e2a3 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/progress.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/python.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/python.png new file mode 100644 index 0000000000000000000000000000000000000000..e036f95d9458918ff384ac380a7b07c35e4edd13 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/python.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/queue.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/queue.jpg new file mode 100644 index 0000000000000000000000000000000000000000..a577d4dbdd8eda051b4fc8abd254febc4d774671 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/queue.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/rename.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/rename.jpg new file mode 100644 index 0000000000000000000000000000000000000000..9060050b466e906cade8ee7f315602c04cf96bf6 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/rename.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/sdless.png b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/sdless.png new file mode 100644 index 0000000000000000000000000000000000000000..0b1cf2795830dd0f88039d46d2a41d8cf31e4609 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/sdless.png differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/shell1.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/shell1.jpg new file mode 100644 index 0000000000000000000000000000000000000000..2893bd9f4c38c6fa730f782c335ceb20a8a08d53 Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/shell1.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/shell2.jpg b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/shell2.jpg new file mode 100644 index 0000000000000000000000000000000000000000..00896f0699ffd8f1d6b266abd0d31e21b2ec29af Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/images/shell2.jpg differ diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/install.py b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/install.py new file mode 100644 index 0000000000000000000000000000000000000000..e7b31a25c27a092c849474ab35aa3ab1130ab210 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/install.py @@ -0,0 +1,9 @@ +import launch +import platform + +if platform.system() == "Windows": + if not launch.is_installed("gdown"): + launch.run_pip("install gdown", "requirements for Batchlinks Download extension") + + if not launch.is_installed("wget"): + launch.run_pip("install wget", "requirements for Batchlinks Download extension") \ No newline at end of file diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/javascript/notification.js b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/javascript/notification.js new file mode 100644 index 0000000000000000000000000000000000000000..8339d3093e0d307db66de30999803862cca01275 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/javascript/notification.js @@ -0,0 +1,19 @@ +let prevTextAreaValue = ''; +let isTextAreaVisible = true; + +onUiUpdate(function() { + const textArea = gradioApp().querySelectorAll('#tab_batchlinks textarea')?.[2]; + const textAreaValue = textArea ? textArea.value : ''; + + if (isTextAreaVisible && textAreaValue.startsWith('All done!') && prevTextAreaValue !== textAreaValue) { + gradioApp().querySelector('#finish_audio audio')?.play(); + } + + if (!textArea) { + isTextAreaVisible = false; + } else if (textArea.style.display !== 'none') { + isTextAreaVisible = true; + } + + prevTextAreaValue = textAreaValue; +}); diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/misc/markydynamictohtml.py b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/misc/markydynamictohtml.py new file mode 100644 index 0000000000000000000000000000000000000000..09ea8b1299a9d8ad5e4d2ba4e60748e9b829c5b9 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/misc/markydynamictohtml.py @@ -0,0 +1,86 @@ +#tableofcontentmaker +intro = """ +
+
+"""
+
+tableofcontents = """
+- [Installation](#installation)
+- [About](#about)
+- [Example](#example)
+- [Syntax](#syntax)
+ - [Hashtag](#hashtag)
+ - [Links](#links)
+ - [Double Hashtag](#double-hashtag)
+ - [Others](#others)
+ - [Valid Hashtags](#valid-hashtags)
+ - [How to get the direct links (Important!)](#how-to-get-the-direct-links-important)
+ - [Huggingface's download method](#huggingfaces-download-method)
+- [Additional Syntax](#additional-syntax)
+ - [Rename Downloaded Files](#rename-downloaded-files)
+ - [Running Shell Commands](#running-shell-commands)
+ - [Extract Everyting (`@extract`)](#extract-everyting-extract)
+ - [Custom Hashtag Path (`@new`)](#custom-hashtag-path-new)
+ - [Custom Direct Link Download with `aria2` (`@aria2`)](#custom-direct-link-download-with-aria2-aria2)
+- [Gradio Queue](#gradio-queue)
+ - [Logging](#logging)
+ - [Cancel](#cancel)
+ - [Progress Bar](#progress-bar)
+ - [Changes if `--gradio-queue` is off](#changes-if---gradio-queue-is-off)
+- [Other Features](#other-features)
+ - [Notification](#notification)
+ - [Copy From Pastebin](#copy-from-pastebin)
+ - [SDless mode](#sdless-mode)
+ - [Local Installation Support](#local-installation-support)
+ - [Debug Mode (Developer only)](#debug-mode-developer-only)
+- [Release Notes](#release-notes)
+- [Roadmap](#roadmap)
+- [Known Bugs](#known-bugs)
+ - [Known Fixed Bugs](#known-fixed-bugs)
+- [Contributing](#contributing)
+- [Contact](#contact)
+- [Acknowledgments](#acknowledgments)
+"""
+
+def mlineprint(toprint):
+ if toprint[-1] == '\n':
+ toprint = toprint[:-1]
+ print(toprint.replace('\n', '', 1))
+
+
+def writehtmllist(space, namehead, hashtaghead):
+ spaces = ""
+ for _ in range(space):
+ spaces += " "
+ if initialspace >= 2:
+ return spaces + f''
+ else:
+ return spaces + f'Table of Contents
+-
+"""
+
+outro = """
+
+
+MJ \ No newline at end of file diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/releasenotes.md b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/releasenotes.md new file mode 100644 index 0000000000000000000000000000000000000000..d17521bd5222b0b006d419f6b376f3a271be7bf9 --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/releasenotes.md @@ -0,0 +1,138 @@ +# Release Notes + +### Latest Patch: v3.2.1 +- Changed "pruned" option to "fp precision" due to CivitAI API changes +- Fixed bug: No matter when the download process is cancelled or not, CivitAI download will give print out notification that the download process is successful +- Fixed bug: Preventing the UI errored out when storage is full and trying to convert jpg image while downloading civitai stuff (PIL.UnidentifiedImageError) +- UI overhaul + +#### Patch: v3.2.0a +- Added support for [vladmandic's fork of automatic1111's webui](https://github.com/vladmandic/automatic) + +## Latest Release: v3.2.0 +- Bug fixed when downloading with aria2 on Windows +- Bug fixed on gradio update checker on sdless batch file for windows +- CivitAi error prevention (when the website is down) +- Cleaned the code of the hashtag system +- `Copy from Pastebin` feature +- Custom aria command with `@aria` +- Making sure custom hashtag doesn't start with number +- New calmer notification sound +- Readme page cleanup +- Some help link changed to link into wiki instead +- UI fix to adapt with latest gradio (v3.23.0) +- Various bug fixes related to CivitAI download + +#### Release v3.1.1b +- Hotfix: Fixed CivitAI link refuses to be downloaded properly. Seems like CivitAI changes its API. +- Fixed several bugs on Windows, especially SDless + +#### Release v3.1.1a +- Hotfix: Fixed CivitAI 'model type chooser' bug and fixed `@extract` bug on colab + +#### Release v3.1.1 +- Added some fix in case CivitAI website is down +- Added message when user pressed "Download All!" but the textbox is empty +- Fixed bug where Lycoris folder always shows on downloaded files the first time user download something + +#### Release v3.1.0a +- Hotfix: Indented block on line 1497 fix + +### Release v3.1.0 +- New hashtag: `#altmodel`, when you use `--ckptdir` argument on `launch.py` line, this hashtag will points to that directory. Otherwise, it'll point to the same directory as `#model` +- New hashtag: `#lycoris`, change current save directory to _/content/stable-diffusion-webui/extensions/sd-webui-additional-networks/models/lora/lycoris_
(Side note: _Lycoris/Locon/Loha_ will works just fine if you use `#addnetlora` instead, as long as you have both [addnet extension](https://github.com/kohya-ss/sd-webui-additional-networks) and [locon extension](https://github.com/KohakuBlueleaf/a1111-sd-webui-locon) installed) +- New hashtag: `#upscaler`, change current save directory to _/content/stable-diffusion-webui/models/ESRGAN_ (This was added few commits ago, but i forgot to write it on the release notes) + +Fixes: +- Fixed bug when installing wget/aria2 on Windows +- Fixed sdless scripts so that it also installs `tqdm` +- Fixed bug where custom paths doesn't put CivitAi download (Model Path method) when custom hashtag is used +- Ongoing downloads will correctly stops when Cancel button is pressed on Windows +- CivitAi (Model Path method) now will not download training dataset by accident + +#### Release v3.0.2 +- Hotfix: removed a comment that messed up the non queue mode. Now the extension works again without `--gradio-queue`. + +#### Release v3.0.1 +- Hotfix: using `urllib.request` instead of `curl -sI` to get the model name on CivitAI direct link method, since it's more reliable (and the curl method always fails somehow). The `requests` method is returned as a fallback. + +### Release v3.0.0 +- Added `@extract` syntax +- (Almost) Full Windows support +- Auto-download config file if available when downloading from CivitAI (SD 2.0+) +- Auto-renaming for downloading ckpt/safetensors and pruned model from CivitAI using direct link method +- CivitAI direct link now use `curl` to get the filename, and use the chosen download method (from the four) to download. Huge download speed boost. `requests` is no longer needed. +- Supports download from Anonfiles, Dropbox, Google Drive, Mediafire, Pixeldrain +- Supports download from Github (raw and release files) +- Supports for custom hashtags with `@new` syntax +- Supports for SDless mode (read more [here](https://github.com/etherealxx/batchlinks-webui#sdless-mode)) +- UI overhaul: + - Now there's a table that shows where does the hashtags points into + - Option to stretch the UI, if your monitor is small, or using colab on mobile + - Option to hide help text + - Option to choose preferred CivitAI models. This will works if you download the model via model page link (https://civitai.com/models/) + - Upload txt now use a little button instead of covering half of the screen + +Fixes: +- CivitAI `model page link` no longer randomly download the first model on the json list. +- Most of Windows bugs +- Renaming problem when using CivitAI model page link method +- Warning message when CivitAI download isn't possible (server down) + +### Older Release +
+
\ No newline at end of file
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/scripts/a_gradiocheck.py b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/scripts/a_gradiocheck.py
new file mode 100644
index 0000000000000000000000000000000000000000..3250a3156bb4b1db2c1cd937e6c5c7519788ee3c
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/scripts/a_gradiocheck.py
@@ -0,0 +1,36 @@
+from modules.shared import cmd_opts
+import os, inspect, re #, sys, shlex
+
+script_dir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
+batchlinks_dir = os.path.join(script_dir, "batchlinks-downloader.py")
+
+# remove the progress bar if queue off
+with open(batchlinks_dir, 'r', encoding='utf-8') as f:
+ contents = f.read()
+
+gradio_queue_on = False
+
+try:
+ if cmd_opts.gradio_queue: #automatic1111
+ gradio_queue_on = True
+except AttributeError:
+ if not cmd_opts.disable_queue: #vladmandic
+ gradio_queue_on = True
+
+if gradio_queue_on:
+ #os.system(f"sed -i 's/def run(command, choosedowner):/def run(command, choosedowner, progress=gr.Progress()):/g' {batchlinks_dir}")
+ # os.system(f"sed -i '/^#progress/s/^#//' {batchlinks_dir}")
+ new_contents = re.sub(r'^(\s*)#progress\(', r'\1progress(', contents, flags=re.MULTILINE)
+ contents = re.sub(r'civitvae\):', 'civitvae, progress=gr.Progress()):', new_contents, flags=re.MULTILINE)
+
+ with open(batchlinks_dir, 'w', encoding='utf-8') as f:
+ f.write(contents)
+
+else:
+ # os.system(f"sed -i 's/def run(command, choosedowner, progress=gr.Progress()):/def run(command, choosedowner):/g' {batchlinks_dir}")
+ # os.system(f"sed -i '/^progress(/ s/^/#/' {batchlinks_dir}")
+ new_contents = re.sub(r'^(\s*)progress\(', r'\1#progress(', contents, flags=re.MULTILINE)
+ contents = re.sub(r'civitvae,\s*progress=gr\.Progress\(\)\):$', 'civitvae):', new_contents, flags=re.MULTILINE)
+
+ with open(batchlinks_dir, 'w', encoding='utf-8') as f:
+ f.write(contents)
\ No newline at end of file
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/batch-link/scripts/batchlinks-downloader.py b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/scripts/batchlinks-downloader.py
new file mode 100644
index 0000000000000000000000000000000000000000..c278cedb87caf56bb5802a6f515d2f0baa1e3a69
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/batch-link/scripts/batchlinks-downloader.py
@@ -0,0 +1,2289 @@
+#github.com/etherealxx
+import os
+import time
+import gradio as gr
+import urllib.request, subprocess, contextlib #these handle mega.nz
+import http.client
+import requests #this handle civit
+import re
+from tqdm import tqdm
+#from IPython.display import display, clear_output
+import pathlib
+import inspect
+import platform
+import shlex
+import signal
+import sys
+sdless = False
+try:
+ from modules import script_callbacks #,scripts
+ from modules.paths import models_path, script_path #, data_path
+ from modules.shared import cmd_opts #check for gradio queue
+except ImportError: #sdless
+ if platform.system() == "Windows":
+ userprofile = os.environ['USERPROFILE']
+ downloadpath = os.path.join(userprofile, "Downloads")
+ script_path = os.path.join(downloadpath, "stable-diffusion-webui")
+ elif platform.system() == "Darwin":
+ userhome = os.environ['HOME']
+ downloadpath = os.path.join(userhome, "Downloads")
+ script_path = os.path.join(downloadpath, "stable-diffusion-webui")
+ else:
+ script_path = '/content/stable-diffusion-webui'
+ models_path = os.path.join(script_path, 'models')
+ gradio_queue = True
+ ckpt_dir = None
+ import sys
+ import types
+ module = types.ModuleType('cmd_opts')
+ module.gradio_queue = gradio_queue
+ module.ckpt_dir = ckpt_dir
+ sys.modules['cmd_opts'] = module
+ import cmd_opts
+ sdless = True
+
+script_dir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
+extension_dir = os.path.abspath(os.path.join(script_dir, "../"))
+#Version checking{
+version_dir = os.path.join(extension_dir, "version.txt")
+with open(version_dir, 'r', encoding='utf-8') as file:
+ curverall = file.readlines()
+currentversion = curverall[0].strip()
+
+try:
+ versionurl = "https://raw.githubusercontent.com/etherealxx/batchlinks-webui/main/version.txt"
+ versionresp = requests.get(versionurl)
+ version_lines = versionresp.text.splitlines()
+ latestversion = version_lines[0].strip()
+except requests.exceptions.RequestException:
+ latestversion = '??'
+
+if latestversion != '??':
+ if currentversion == latestversion:
+ latestversiontext = ""
+ else:
+ latestversiontext = f"[Latest version: {latestversion}]"
+else:
+ latestversiontext = ""
+
+currentverforlink = latestversion.replace('.', '')
+#}
+
+vladmandic = False
+try:
+ global gradiostate
+ if cmd_opts.gradio_queue: #automatic1111
+ gradiostate = True
+ else:
+ gradiostate = False
+except AttributeError:
+ try:
+ if not cmd_opts.disable_queue: #vladmandic
+ gradiostate = True
+ else:
+ gradiostate = False
+ vladmandic = True
+ except AttributeError:
+ gradiostate = False #at this point just use onedotsix
+ pass
+
+typechecker = [
+ "embedding", "embeddings", "embed", "embeds", "textualinversion", "ti",
+ "model", "models", "checkpoint", "checkpoints",
+ "vae", "vaes",
+ "lora", "loras",
+ "hypernetwork", "hypernetworks", "hypernet", "hypernets", "hynet", "hynets",
+ "addnetlora", "loraaddnet", "additionalnetworks", "addnet",
+ "aestheticembedding", "aestheticembed",
+ "controlnet", "cnet",
+ "extension", "extensions", "ext", #obsolete
+ "upscaler", "upscale",
+ "altmodel", "altmodels",
+ "lycoris", "locon", "loha"
+ ]
+
+typemain = [
+ "model", "altmodel", "vae",
+ "embed", "hynet", "lora",
+ "addnetlora", "aestheticembed", "cnet",
+ "ext", "upscaler", "lycoris"
+]
+countofdefaulthashtags = len(typemain)
+
+supportedlinks = [
+ "https://mega.nz",
+ "https://huggingface.co",
+ "https://civitai.com/api/download/models",
+ "https://civitai.com/models/",
+ "https://cdn.discordapp.com/attachments",
+ "https://github.com",
+ "https://raw.githubusercontent.com",
+ "https://files.catbox.moe",
+ "https://drive.google.com",
+ "https://pixeldrain.com",
+ "https://www.mediafire.com/file",
+ "https://anonfiles.com",
+ "https://www.dropbox.com/s"
+]
+
+modelpath = os.path.join(script_path, "models/Stable-diffusion")
+embedpath = os.path.join(script_path, "embeddings")
+vaepath = os.path.join(script_path, "models/VAE")
+lorapath = os.path.join(script_path, "models/Lora")
+addnetlorapath = os.path.join(script_path, "extensions/sd-webui-additional-networks/models/lora")
+hynetpath = os.path.join(script_path, "models/hypernetworks")
+aestheticembedpath = os.path.join(script_path, "extensions/stable-diffusion-webui-aesthetic-gradients/aesthetic_embeddings")
+cnetpath = os.path.join(script_path, "extensions/sd-webui-controlnet/models")
+extpath = os.path.join(script_path, "extensions") #obsolete
+upscalerpath = os.path.join(script_path, "models/ESRGAN")
+lycorispath = os.path.join(addnetlorapath, "lycoris")
+
+if vladmandic:
+ cnetpath = os.path.join(script_path, "models/ControlNet")
+ lycorispath = os.path.join(script_path, "models/LyCORIS")
+
+if cmd_opts.ckpt_dir:
+ altmodelpath = cmd_opts.ckpt_dir
+ currentfolder = altmodelpath
+else:
+ altmodelpath = modelpath
+ currentfolder = modelpath
+
+if platform.system() == "Windows":
+ for x in typemain:
+ # exec(f"{x}path = os.path.normpath({x}path)")
+ exec(f"{x}path = {x}path.replace('/', '\\\\')")
+ #exec(f"print({x}path)")
+if os.path.exists(addnetlorapath) and not os.path.exists(lycorispath):
+ os.makedirs(lycorispath, exist_ok=True)
+
+newlines = ['\n', '\r\n', '\r']
+currentlink = ''
+finalwrite = []
+currentcondition = ''
+currentsuboutput = ''
+processid = ''
+logging = False
+#currentiterfolder = ''
+prockilled = False
+currentfoldertrack = []
+everyprocessid = []
+remaininglinks = []
+gdownupgraded = False
+mediafireinstalled = False
+prevciviterror = False
+# addedcustompath = []
+# ariamode = False
+
+globaldebug = False #set this to true to activate every debug features
+if len(sys.argv) > 1:
+ if sys.argv[1] == '--debug':
+ globaldebug = True
+
+process = ''
+
+def stopwatch(func): #unused
+ """
+ A function that acts as a stopwatch for another function.
+
+ Args:
+ func (function): The function to be timed.
+
+ Returns:
+ The return value of the timed function.
+ """
+ def wrapper(*args, **kwargs):
+ if not globaldebug:
+ return func(*args, **kwargs)
+ start_time = time.time()
+ result = func(*args, **kwargs)
+ end_time = time.time()
+ elapsed_time = end_time - start_time
+ minutes = int(elapsed_time // 60)
+ seconds = elapsed_time % 60
+ print(f"Time taken by {func.__name__}: {minutes:.0f}m {seconds:.2f}s")
+ return result
+ return wrapper
+
+#debuggingpurpose{
+ #Hello debuggers! This will track every files when the extension is launched, and
+ #you can remove every downloaded files after with hashtag '@debugresetdownloads', for debugging purposes on colab
+ #(Note: You need to fill the textbox with only a single line of @debugresetdownloads and nothing more)
+ #There's also another feature, '#debugeverymethod', which will download a single link with all four possible methods available.
+import shutil
+snapshot = []
+paths_to_scan = []
+
+# take a snapshot of the directories
+def take_snapshot():
+ snapshotdir = os.path.join(script_path, 'snapshot.txt')
+ global snapshot
+ global paths_to_scan
+ paths_to_scan = []
+ for x in typemain:
+ exec(f"paths_to_scan.append({x}path)")
+ if os.path.exists(snapshotdir):
+ with open(snapshotdir, 'r') as f:
+ snapshot = [line.strip() for line in f.readlines()]
+ print("Batchlinks extension: snapshot already exist.")
+ return
+ else:
+ snapshot = []
+ for path in paths_to_scan:
+ if os.path.exists(path):
+ pathtemp = os.listdir(path)
+ for file in pathtemp:
+ pathoffile = os.path.join(path, file)
+ snapshot.append(pathoffile)
+ with open(snapshotdir, 'w') as f:
+ for item in snapshot:
+ f.write(f"{item}\n")
+ print("Batchlinks extension: snapshot taken.")
+
+# rewind everything to a fresh state
+def global_rewind():
+ global paths_to_scan
+ global path
+ global currentsuboutput
+ removedall, removed_files, removed_dirs, new_snapshot = [], [], [], []
+ print('[1;32mDebug rewind initiated...')
+ print('[0m')
+ for path in paths_to_scan:
+ if os.path.exists(path):
+ pathtemp = os.listdir(path)
+ for file in pathtemp:
+ pathoffile = os.path.join(path, file)
+ new_snapshot.append(pathoffile)
+ toremoves = [x for x in new_snapshot if x not in snapshot]
+ for fileordir in toremoves:
+ if os.path.exists(fileordir):
+ if os.path.isdir(fileordir):
+ try:
+ shutil.rmtree(fileordir)
+ removed_dirs.append(fileordir)
+ except PermissionError as e:
+ print("Error: " + str(e))
+ else:
+ os.remove(fileordir)
+ removed_files.append(fileordir)
+ if removed_files or removed_dirs:
+ print("Removed files:")
+ removedall.append("Removed files:")
+ for file in removed_files:
+ print(file)
+ removedall.append(file)
+ print("Removed directories:")
+ removedall.append("Removed directories:")
+ for dir in removed_dirs:
+ print(dir)
+ removedall.append(dir)
+ print('[1;32mrewind completed')
+ print('[0m')
+ return removedall
+
+# Take a snapshot of the directories
+if globaldebug == True:
+ take_snapshot()
+# }
+
+def printdebug(toprint):
+ if globaldebug == True:
+ print(toprint)
+
+def printvardebug(toprint):
+ if globaldebug == True:
+ import inspect
+ caller = inspect.currentframe().f_back
+ try:
+ name = [k for k, v in caller.f_locals.items() if v is toprint][0]
+ except IndexError:
+ name = "idk"
+ if not isinstance(toprint, str):
+ toprint = str(toprint)
+ print(f"debug {name}: {toprint}")
+
+def runwithsubprocess(rawcommand, folder=None, justrun=False, additionalcontext=''): #@note runwithsubprocess
+
+ commandtorun = shlex.split(rawcommand) #construct_command(rawcommand)
+ printdebug(f"raw command: {rawcommand}")
+ printdebug(f"merged command: {commandtorun}")
+ global prockilled
+ global everyprocessid
+ # if gradiostate == False and not rawcommand.startswith("aria"):
+ # subprocess.run(commandtorun, stderr=subprocess.STDOUT, universal_newlines=True)
+ # else:
+ if folder != None:
+ printdebug("debug folderforsavestate: " + str(folder))
+ savestate_folder(folder)
+ printdebug("debug currentfoldertrack: " + str(currentfoldertrack))
+
+ global process
+ if justrun:
+ process = subprocess.Popen(rawcommand, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True, shell=True)
+ else:
+ process = subprocess.Popen(commandtorun, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True)
+ global processid
+ processid = process.pid
+ everyprocessid.append(processid)
+ printdebug("debug processid: " + str(processid))
+
+ ariacomplete = False
+ global currentsuboutput
+ while True:
+ # Read the output from the process
+ nextline = process.stdout.readline()
+ if nextline == '' and process.poll() is not None:
+ break
+ # Check if the line contains progress information
+ elif additionalcontext == 'aria2':
+ if 'Download Results' in nextline:
+ ariacomplete = True
+ print('\n')
+ print(nextline, end='')
+ currentsuboutput = nextline
+ else:
+ if not ariacomplete:
+ match = re.search(r'\[[^\]]+\]', nextline)
+ if match:
+ stripnext = match.group().strip()
+ print("\r", end="")
+ print(f"\r{stripnext}", end='')
+ currentsuboutput = stripnext
+ else:
+ print(nextline, end='')
+ currentsuboutput = nextline
+ elif additionalcontext == '7z':
+ sevenzmessage = [
+ "Extracting", "Everything", "ERROR"
+ ]
+ if nextline.strip().startswith(tuple(sevenzmessage)):
+ print(nextline, end=' ')
+ currentsuboutput = nextline
+ else:
+ stripnext = nextline.strip()
+ print("\r", end="")
+ print(f"\r{stripnext}", end='')
+ currentsuboutput = stripnext
+ else:
+ if justrun:
+ print(nextline, end='')
+ else:
+ if "%" in nextline.strip() or rawcommand.startswith("curl"):
+ stripnext = nextline.strip()
+ print("\r", end="")
+ print(f"\r{stripnext}", end='')
+ else:
+ print(nextline, end='')
+ if additionalcontext == 'mega':
+ process.kill()
+ if additionalcontext == 'mega':
+ if nextline.strip().startswith("Download finished:"):
+ _ = subprocess.getoutput("taskkill /f /t /im MEGAcmdServer.exe")
+ printdebug("MEGA server killed")
+ currentsuboutput = nextline
+
+
+ process.wait()
+ currentsuboutput = ''
+ processid = ''
+ if prockilled == True:
+ printdebug("before rewind folder")
+ printvardebug(folder)
+ rewind_folder(folder)
+ print('[1;31mOperation Cancelled')
+ print('[0m')
+ global currentcondition
+ currentcondition = 'Operation Cancelled'
+ return
+
+#these code below handle mega.nz
+def unbuffered(proc, stream='stdout'):
+ stream = getattr(proc, stream)
+ with contextlib.closing(stream):
+ while prockilled == False:
+ out = []
+ last = stream.read(1)
+ # Don't loop forever
+ if last == '' and proc.poll() is not None:
+ break
+ while last not in newlines:
+ # Don't loop forever
+ if last == '' and proc.poll() is not None:
+ break
+ out.append(last)
+ last = stream.read(1)
+ out = ''.join(out)
+ yield out
+
+def transfare(todownload, folder, torename=''):
+ #import codecs
+ #decoder = codecs.getincrementaldecoder("UTF-8")()
+ todownload_s = shlex.quote(todownload)
+ folder_s = shlex.quote(folder)
+ if platform.system() == "Windows":
+ savestate_folder(folder)
+ localappdata = os.environ['LOCALAPPDATA']
+ megagetloc = os.path.join(localappdata, "MEGAcmd", "mega-get.bat")
+ global process
+ process = runwithsubprocess(f"{shlex.quote(megagetloc)} {todownload_s} {folder_s}", folder, False, 'mega')
+ else:
+ savestate_folder(folder_s)
+ cmd = ["mega-get", todownload_s, folder_s]
+ proc = subprocess.Popen(
+ cmd,
+ stdout=subprocess.PIPE,
+ stderr=subprocess.STDOUT,
+ universal_newlines=True,
+ )
+ global processid
+ global everyprocessid
+ processid = proc.pid
+ everyprocessid.append(processid)
+
+ global currentsuboutput
+ for line in unbuffered(proc):
+ if prockilled == False:
+ if not line.startswith("Download"):
+ currentsuboutput = line
+ print(f"\r{line}", end="")
+ else:
+ print(f"\n{line}")
+ _ = subprocess.getoutput("pkill -f \"mega-cmd-server\"")
+ printdebug("MEGA server killed")
+
+ else:
+ currentsuboutput = ''
+ print('[1;31mOperation Cancelled')
+ print('[0m')
+ global currentcondition
+ currentcondition = 'Operation Cancelled'
+ return
+ currentsuboutput = ''
+ if torename:
+ listfilenew = os.listdir(folder)
+ newerfoldertrack = []
+ for file in listfilenew:
+ pathoffile = os.path.join(folder, file)
+ newerfoldertrack.append(pathoffile)
+ checkrename = [x for x in newerfoldertrack if x not in currentfoldertrack]
+ if checkrename:
+ # renamedfile = os.path.basename(checkrename[0])
+ # pathtorename = os.path.join(folder, renamedfile)
+ os.rename(checkrename[0], os.path.join(folder, torename))
+
+
+def installmega():
+ HOME = os.path.expanduser("~")
+ ocr_file = pathlib.Path(f"{HOME}/.ipython/ocr.py")
+ if not ocr_file.exists():
+ hCode = "https://raw.githubusercontent.com/biplobsd/" \
+ "OneClickRun/master/res/ocr.py"
+ urllib.request.urlretrieve(hCode, str(ocr_file))
+
+ from importlib.util import module_from_spec, spec_from_file_location
+ ocr_spec = spec_from_file_location("ocr", str(ocr_file))
+ ocr = module_from_spec(ocr_spec)
+ ocr_spec.loader.exec_module(ocr)
+
+ if not os.path.exists("/usr/bin/mega-cmd"):
+ #ocr.loadingAn()
+ print('[1;32mInstalling MEGA ...')
+ print('[0m')
+ ocr.runSh('sudo apt-get -y update')
+ ocr.runSh('sudo apt-get -y install libmms0 libc-ares2 libc6 libcrypto++6 libgcc1 libmediainfo0v5 libpcre3 libpcrecpp0v5 libssl1.1 libstdc++6 libzen0v5 zlib1g apt-transport-https')
+ ocr.runSh('sudo curl -sL -o /var/cache/apt/archives/MEGAcmd.deb https://mega.nz/linux/MEGAsync/Debian_9.0/amd64/megacmd-Debian_9.0_amd64.deb', output=True)
+ ocr.runSh('sudo dpkg -i /var/cache/apt/archives/MEGAcmd.deb', output=True)
+ print('[1;32mMEGA is installed.')
+ print('[0m')
+
+def installmegawin():
+ userprofile = os.environ['USERPROFILE']
+ localappdata = os.environ['LOCALAPPDATA']
+ megagetloc = os.path.join(localappdata, "MEGAcmd", "mega-get.bat")
+ megacmdloc = os.path.join(userprofile, "Downloads", "MEGAcmdSetup64.exe")
+ if not os.path.exists(megagetloc):
+ print('[1;32mInstalling MEGA ...')
+ print('[0m')
+ runwithsubprocess(f"curl -o {shlex.quote(megacmdloc)} https://mega.nz/MEGAcmdSetup64.exe")
+ time.sleep(1)
+ runwithsubprocess(f"{shlex.quote(megacmdloc)} /S")
+ time.sleep(4)
+ print('[1;32mMEGA is installed.')
+ print('[0m')
+ #clear_output()
+#these code above handle mega.nz
+
+def civitdown(url, folder, torename=''):
+ filename = url.split('?')[0].rsplit('/', 1)[-1] + ".bdgh"
+ pathtodown = os.path.join(folder, filename)
+ max_retries = 5
+ retry_delay = 10
+ # url_s = quote(url)
+
+ while prockilled == False:
+
+ downloaded_size = 0
+ headers = {}
+
+ progress = tqdm(total=1000000000, unit="B", unit_scale=True, desc=f"Downloading {filename}. (will be renamed correctly after downloading)", initial=downloaded_size, leave=False)
+ global currentsuboutput
+ global currentcondition
+ with open(pathtodown, "ab") as f:
+ while prockilled == False:
+ try:
+ response = requests.get(url, headers=headers, stream=True)
+ total_size = int(response.headers.get("Content-Length", 0))
+ # if total_size == 0:
+ # total_size = downloaded_size
+ # progress.total = total_size
+
+
+ for chunk in response.iter_content(chunk_size=1024):
+ if chunk and prockilled == False:
+ f.write(chunk)
+ progress.update(len(chunk))
+ currentsuboutput = str(progress)
+ else:
+ break
+
+ downloaded_size = os.path.getsize(pathtodown)
+ currentsuboutput = ''
+ break
+ except ConnectionError as e:
+ max_retries -= 1
+
+ if max_retries == 0:
+ raise e
+
+ time.sleep(retry_delay)
+
+ progress.close()
+ if prockilled == True:
+ if os.path.exists(pathtodown):
+ os.remove(pathtodown)
+ print('[1;31mOperation Cancelled')
+ print('[0m')
+ currentcondition = 'Operation Cancelled'
+ currentsuboutput = ''
+ return "Operation Cancelled"
+
+ if torename:
+ actualfilename = torename
+ else:
+ try:
+ actualfilename = response.headers['Content-Disposition'].split("filename=")[1].strip('"')
+ except KeyError:
+ print('[1;31mLink Error')
+ print('[0m')
+ break
+ #%cd {folder}
+ actualpath = os.path.join(folder, actualfilename)
+ os.rename(pathtodown, actualpath)
+ downloaded_size = os.path.getsize(actualpath)
+ # Check if the download was successful
+ if downloaded_size >= total_size:
+ print(f"{actualfilename} successfully downloaded.")
+ break
+ else:
+ print(f"Error: File download failed. Retrying...")
+
+def getcivitname(link, frommodeltypechooser=False): #@note getcivitname
+ # searcher = "findstr" if platform.system() == "Windows" else "grep"
+ printdebug("getcivitname:")
+ printvardebug(link)
+ global currentcondition
+ tempcondition = currentcondition
+ currentcondition = "Connecting to CivitAI..."
+ try:
+ req = urllib.request.Request(link, headers={'User-Agent': 'Mozilla/5.0'})
+ response = urllib.request.urlopen(req, timeout=7)
+ contentdis = response.geturl()
+ # contentdis = [line for line in subprocess.getoutput(f"curl -sI {link} | {searcher} -i content-disposition").splitlines() if line.startswith('location')][0]
+ except Exception as e:
+ errortype = str(type(e))
+ printdebug("Error: " + errortype)
+ if 'TimeoutError' in errortype:
+ print("Unable to connect to CivitAI in 7 seconds. Skipping the link...")
+ currentcondition = tempcondition
+ return 'batchlinksskip'
+ else:
+ if frommodeltypechooser:
+ currentcondition = tempcondition
+ return ''
+ else:
+ print("Cannot get the filename. Download will continue with the old slow method.\nReason:" + str(e))
+ currentcondition = tempcondition
+ return 'batchlinksold'
+ if contentdis == "https://civitai.com/":
+ print('[1;31mCivitAI website is currently down ツ')
+ print('[0m')
+ currentcondition = tempcondition
+ return 'batchlinksskip'
+ cuttedcontent = contentdis.find('response-content-disposition=attachment%3B%20filename%3D%22') + 59
+ # filename = str(contentdis[cuttedcontent:]).replace('%22&x-id=GetObject', '') #obsolete since 30-03-2023
+ filename = str(contentdis[cuttedcontent:]).partition('%22&')[0]
+ # filename = contentdis.split('filename=')[-1].split('%22')[1] #this also works i guess
+ filename = civitmodeltypename(filename, link)
+ currentcondition = tempcondition
+ return filename
+
+def getcivitname2(link):
+ searcher = "findstr" if platform.system() == "Windows" else "grep"
+ try:
+ contentdis = [line for line in subprocess.getoutput(f"curl -sI {link} | {searcher} -i content-disposition").splitlines() if line.startswith('location')][0]
+ cuttedcontent = contentdis.find('response-content-disposition=attachment%3B%20filename%3D%22') + 59
+ filename = str(contentdis[cuttedcontent:]).replace('%22&x-id=GetObject', '')
+ if filename:
+ return [0, link, filename]
+ except:
+ return []
+
+def civitmodeltypename(name, filelink):
+ nameonly, extension = os.path.splitext(name)
+ if 'type=Pruned' in filelink:
+ nameonly += "_pruned"
+
+ if 'format=Safetensor' in filelink:
+ extension = '.safetensors'
+ elif 'format=PickleTensor' in filelink:
+ extension = '.ckpt'
+
+ nameoffile = nameonly + extension
+ return nameoffile
+
+def checkcivitconfig(link): #check if the current civit link has a config file (v2.0+) @note checkcivitconfig
+ def getrequest(link2):
+ response = requests.get(link2)
+ return response.status_code
+
+ params = ['?type=Config','?type=Config&format=Other']
+
+ for param in params:
+ response = getrequest(link + param)
+ if response == 200:
+ #print('The link exists')
+ return link + param
+ return ''
+
+def civitmodeltypechooser(modeljson, prunedmodel, torchortensor, linkandnames):
+
+# prunedorfull = ['?type=Model', '?type=Pruned%20Model']
+# if prunedmodel == 'True':
+ prunedorfull = ['&fp=fp32', '&fp=fp16']
+ if prunedmodel == 'fp16':
+ prunedorfull.reverse()
+ pickleorsafe = ['&format=SafeTensor', '&format=PickleTensor']
+ if torchortensor=='ckpt':
+ pickleorsafe.reverse()
+
+ defaultlinkurl = [link.get('downloadUrl') for link in modeljson['modelVersions'][0]['files'] if not '?type=' in link.get('downloadUrl')][0]
+
+ indexlinkname = list()
+ checklater = ''
+ for index, (link, name) in enumerate(linkandnames.items()):
+ if prunedorfull[0] in link:
+ if pickleorsafe[0] in link:
+ indexlinkname.extend([index, link, name])
+ break
+ else:
+ checklater = link
+ continue
+ else:
+ continue
+
+ if checklater and not indexlinkname:
+ for index, (link, name) in enumerate(linkandnames.items()):
+ if checklater == link:
+ indexlinkname.extend([index, link, name])
+ checklater = ''
+ break
+
+ if not indexlinkname:
+ indexlinkname = getcivitname(defaultlinkurl+prunedorfull[0]+pickleorsafe[0], True) #@note indexlinkname
+ if indexlinkname == "batchlinksskip" or not indexlinkname:
+ indexlinkname = list()
+
+ if not indexlinkname:
+ for index, (link, name) in enumerate(linkandnames.items()):
+ if defaultlinkurl == link:
+ indexlinkname.extend([index, link, name])
+ break
+
+ return indexlinkname
+
+#thank you @rti7743 for this part {
+def civitdown2_get_json(url):
+ import re
+ m = re.search(r'https://civitai.com/models/(\d+)', url)
+ model_id = m.group(1)
+
+ api_url = "https://civitai.com/api/v1/models/" + model_id
+
+ import requests
+ txt = requests.get(api_url).text
+
+ import json
+ try:
+ return json.loads(txt)
+ except json.decoder.JSONDecodeError:
+ return 'error'
+
+def civitdown2_get_save_directory(model_type, default_folder):
+ printdebug("model_type: " + model_type)
+ customtypemains = list(typemain[countofdefaulthashtags:])
+ printdebug("customtypemains: " + str(customtypemains))
+ for type in customtypemains:
+ xpath = eval(type + "path")
+ if default_folder == xpath:
+ printdebug("customtype returned")
+ return default_folder
+ if model_type == "Checkpoint":
+ if cmd_opts.ckpt_dir and default_folder == altmodelpath:
+ return altmodelpath
+ else:
+ return modelpath
+ elif model_type == "Hypernetwork":
+ return hynetpath
+ elif model_type == "TextualInversion":
+ return embedpath
+ elif model_type == "AestheticGradient":
+ return aestheticembedpath
+ elif model_type == "VAE":
+ return vaepath
+ elif model_type == "LORA":
+ if default_folder == addnetlorapath:
+ return addnetlorapath
+ else:
+ return lorapath
+ elif model_type == "LoCon":
+ return lycorispath
+ else:
+ printdebug("defaultfolder returned")
+ return default_folder
+
+def civitdown2_convertimage(imagejpg_save_path, imagepng_save_path):
+ from PIL import Image
+ try:
+ img = Image.open(imagejpg_save_path)
+ img_resized = img.resize((img.width // 2, img.height // 2))
+ img_resized.save(imagepng_save_path)
+ if os.path.exists(imagejpg_save_path):
+ os.remove(imagejpg_save_path)
+ except Exception as e:
+ print("Civitai image convertion failed. Reason: " + str(e))
+ pass
+
+def civitdown2(url, folder, downloader, renamedfilename, isdebugevery, modeldefaulttype, isprunedmodel, isdownvae): #@note civitdown2
+ def civitlinkandnamer(model):
+ linkandname = dict()
+ for i, link in enumerate(model['modelVersions'][0]['files']):
+ name = link.get('name')
+ url = link.get('downloadUrl')
+ if name and url:
+ linkandname[url] = name
+ # print(i, url, name)
+ else:
+ pass
+ return linkandname
+
+ model = civitdown2_get_json(url)
+ if model == 'error':
+ print('[1;31mFailed retrieving the model data.')
+ print('[1;31mCivitAI website might going down currently.')
+ print('[0m')
+ return
+
+ if model == {'error': 'Model not found'}:
+ print('[1;31mModel ' + url + ' is not available anymore')
+ print('[0m')
+ return
+
+ save_directory = civitdown2_get_save_directory(model['type'], folder)
+ if not os.path.exists(save_directory):
+ os.makedirs(save_directory, exist_ok=True)
+ try:
+ parameter = url.split("?")[-1] + "?"
+ except:
+ parameter = ''
+
+ civitlinkandnames = civitlinkandnamer(model)
+
+ if model['type'] == "Checkpoint":
+ data_index, data_url, data_filename = civitmodeltypechooser(model, isprunedmodel, modeldefaulttype, civitlinkandnames) #@note
+ data_filename = civitmodeltypename(data_filename, data_url)
+ image_filename = model['modelVersions'][0]['files'][0]['name']
+ else:
+ data_url = model['modelVersions'][0]['files'][0]['downloadUrl']
+ data_filename = model['modelVersions'][0]['files'][0]['name']
+ if "?type=Training%20Data" in data_url:
+ data_url = data_url.partition("?")[0]
+ wannabename = getcivitname(data_url)
+ if wannabename:
+ if wannabename != 'batchlinksskip':
+ data_filename = wannabename
+ else:
+ return ''
+ image_filename = data_filename
+
+ if renamedfilename:
+ data_filename = renamedfilename
+ image_filename = data_filename
+
+ image_url = model['modelVersions'][0]['images'][0]['url']
+ printvardebug(data_url)
+ printvardebug(data_filename)
+ printvardebug(image_url)
+ if model['type'] == "TextualInversion":
+ image_filename_jpg = pathlib.PurePath(image_filename).stem + ".preview.jpg"
+ image_filename_png = pathlib.PurePath(image_filename).stem + ".preview.png"
+ else:
+ image_filename_jpg = pathlib.PurePath(image_filename).stem + ".jpg"
+ image_filename_png = pathlib.PurePath(image_filename).stem + ".png"
+
+ data_save_path = os.path.join(save_directory, data_filename)
+ imagejpg_save_path = os.path.join(save_directory, image_filename_jpg)
+ imagepng_save_path = os.path.join(save_directory, image_filename_png)
+
+ currentmode = 'civit'
+ if isdebugevery:
+ currentmode = 'civitdebugevery'
+
+ printdebug(f"debug download_url({data_url}, {data_save_path}, {downloader})")
+ if prockilled == False:
+ hfdown(data_url, data_save_path, downloader, currentmode)
+ printdebug("normal download done, now check for the config")
+ config_url = checkcivitconfig(data_url)
+ vae_url, vae_name = '',''
+ for link, name in civitlinkandnames.items():
+ if '?type=VAE' in link:
+ vae_url, vae_name = link, name
+ break
+ if model['type'] == "Checkpoint":
+ if prockilled == False and isdownvae and vae_url:
+ namefile = os.path.splitext(data_filename)[0]
+ vae_name = namefile + '.vae.pt'
+ data_save_path = os.path.join(os.path.dirname(data_save_path), vae_name)
+ hfdown(vae_url, data_save_path, downloader, currentmode)
+ if prockilled == False and config_url:
+ namefile = os.path.splitext(data_filename)[0]
+ config_name = namefile + '.yaml'
+ data_save_path = os.path.join(os.path.dirname(data_save_path), config_name)
+ hfdown(config_url, data_save_path, downloader, currentmode)
+
+ if prockilled == False:
+ hfdown(image_url, imagejpg_save_path, downloader, currentmode)
+ if os.path.exists(imagejpg_save_path):
+ civitdown2_convertimage(imagejpg_save_path, imagepng_save_path)
+ else:
+ print(imagejpg_save_path + " doesn't exist")
+ if prockilled == False:
+ print(f"{data_save_path} successfully downloaded.")
+
+ if isdebugevery:
+ additionalname = '-' + downloader
+
+ if model['type'] == "TextualInversion":
+ image_filename_jpg = pathlib.PurePath(data_filename).stem + additionalname + ".preview.jpg"
+ image_filename_png = pathlib.PurePath(data_filename).stem + additionalname + ".preview.png"
+ else:
+ image_filename_jpg = pathlib.PurePath(data_filename).stem + additionalname + ".jpg"
+ image_filename_png = pathlib.PurePath(data_filename).stem + additionalname + ".png"
+
+ imagejpg_save_path = os.path.join(save_directory, image_filename_jpg)
+ imagepng_save_path = os.path.join(save_directory, image_filename_png)
+
+#}
+def mediadrivedown(todownload, folder, mode, torename=''): #@note mediadrivedown
+ todownload_s = shlex.quote(todownload)
+ if platform.system() == "Windows":
+ folder_s = folder
+ else:
+ folder_s = shlex.quote(folder)
+ prevcurdir = os.getcwd()
+ os.chdir(folder)
+ savestate_folder(folder_s)
+
+ if mode=='gdrive':
+ runwithsubprocess(f"gdown --fuzzy {todownload_s}", folder_s)
+ elif mode=='mediafire':
+ runwithsubprocess(f"mediafire-dl {todownload_s}", folder_s)
+
+ os.chdir(prevcurdir)
+ if torename:
+ listfilenew = os.listdir(folder)
+ newerfoldertrack = []
+ for file in listfilenew:
+ pathoffile = os.path.join(folder, file)
+ newerfoldertrack.append(pathoffile)
+ checkrename = [x for x in newerfoldertrack if x not in currentfoldertrack]
+ if checkrename:
+ os.rename(checkrename[0], os.path.join(folder, torename))
+
+def hfdown(todownload, folder, downloader, mode='default', torename=''): #@note hfdown
+ global currentcondition
+ global prevciviterror
+ prevciviterror = False
+ # global ariamode
+ if mode=='civit' or mode=='civitdebugevery':
+ filename = pathlib.Path(folder).name
+ filename_s = shlex.quote(filename)
+ filepath = folder
+ filepath_s = shlex.quote(folder)
+ todownload_s = todownload
+ folder = os.path.dirname(filepath)
+ # folder_s = pathlib.Path(folder).parent.resolve()
+ folder_s = shlex.quote(folder)
+ folder_win = folder
+ else:
+ if mode=='pixeldrain' or mode=='dropbox':
+ filename = torename
+ else:
+ filename = todownload.rsplit('/', 1)[-1]
+ filename_s = shlex.quote(filename)
+ filepath = os.path.join(folder, filename)
+ filepath_s = shlex.quote(filepath)
+ todownload_s = shlex.quote(todownload)
+ folder_s = shlex.quote(folder)
+ folder_win = folder
+ ariafilename_s = filename_s
+ if torename:
+ ariafilename_s = shlex.quote(torename)
+
+ printvardebug(folder)
+ printvardebug(todownload)
+ printvardebug(filename)
+ printvardebug(filename_s)
+ printvardebug(filepath)
+ printvardebug(filepath_s)
+ printvardebug(todownload_s)
+ printvardebug(folder_s)
+ printvardebug(folder_win)
+ #savestate_folder(folder_s)
+
+ if platform.system() == "Windows": #@note windows downloader
+ localappdata = os.environ['LOCALAPPDATA']
+ batchlinksinstallpath = os.path.join(localappdata, "batchlinks")
+ printvardebug(batchlinksinstallpath)
+ wgetpath = os.path.join(batchlinksinstallpath, "wget-gnutls-x64.exe")
+
+ ariazippath = os.path.join(batchlinksinstallpath, "aria2-1.36.0-win-64bit-build2.7z")
+ ariapath = os.path.join(batchlinksinstallpath, "aria2-1.36.0-win-64bit-build2", "aria2c.exe")
+ os.makedirs(batchlinksinstallpath, exist_ok=True)
+ if os.path.exists(batchlinksinstallpath):
+ printdebug("batchlinksinstallpath created")
+ if downloader=='gdown':
+ global gdownupgraded
+ if gdownupgraded == False:
+ tempcondition = currentcondition
+ currentcondition = "Upgrading gdown..."
+ print('[1;32mUpgrading gdown ...')
+ print('[0m')
+ runwithsubprocess(f"pip3 install -q --upgrade --no-cache-dir gdown")
+ print('[1;32mgdown upgraded!')
+ print('[0m')
+ currentcondition = tempcondition
+ gdownupgraded = True
+ try:
+ runwithsubprocess(f"gdown {todownload_s} -O {filepath_s}", folder_win)
+ except FileNotFoundError:
+ import gdown
+ gdown.download(todownload, filepath, quiet=False)
+ elif downloader=='wget':
+ #os.system("python -m wget -o " + os.path.join(folder, filename) + " " + todownload)
+ # import wget
+ # wget.download(todownload, filepath)
+ # checkwget = subprocess.getoutput(wgetpath)
+ # if "is not recognized" in checkwget:
+ if not os.path.exists(wgetpath):
+ tempcondition = currentcondition
+ currentcondition = "Installing wget Windows..."
+ print('[1;32mInstalling wget Windows...')
+ print('[0m')
+ wgetwinlink = "https://github.com/webfolderio/wget-windows/releases/download/v1.21.3.june.19.2022/wget-gnutls-x64.exe"
+ print(wgetwinlink)
+ runwithsubprocess("curl -Lo " + shlex.quote(wgetpath) + " " + wgetwinlink, batchlinksinstallpath)
+ print('[1;32mwget Windows installed!')
+ print('[0m')
+ currentcondition = tempcondition
+ runwithsubprocess(f"{shlex.quote(wgetpath)} -O {filepath_s} {todownload_s} --progress=bar:force", folder_win)
+ elif downloader=='curl':
+ runwithsubprocess(f"curl -Lo {filepath_s} {todownload_s}", folder_win)
+ elif downloader=='aria2':
+ # checkaria = subprocess.getoutput(ariapath)
+ # if "is not recognized" in checkaria or "cannot find the path" in checkaria:
+ if not os.path.exists(ariapath):
+ tempcondition = currentcondition
+ currentcondition = "Installing aria2 Windows..."
+ print('[1;32mInstalling aria2 Windows...')
+ print('[0m')
+ ariaziplink = "https://github.com/q3aql/aria2-static-builds/releases/download/v1.36.0/aria2-1.36.0-win-64bit-build2.7z"
+ print(ariaziplink)
+ runwithsubprocess("curl -Lo " + shlex.quote(ariazippath) + " " + ariaziplink, batchlinksinstallpath)
+ sevenzpath = install7zWin()
+ runwithsubprocess(f"{shlex.quote(sevenzpath)} x {shlex.quote(ariazippath)} -o{shlex.quote(batchlinksinstallpath)}", batchlinksinstallpath, False, '7z')
+ print()
+ print('[1;32maria2 Windows installed!')
+ print('[0m')
+ currentcondition = tempcondition
+ # ariamode = True
+ runwithsubprocess(f"{shlex.quote(ariapath)} --summary-interval=1 --console-log-level=error --check-certificate=false -c -x 16 -s 16 -k 1M {todownload_s} -d {folder_s} -o {ariafilename_s}", folder_win, False, 'aria2')
+ else: #non-windows
+ if downloader=='gdown':
+ printdebug(f"debug gdown {todownload_s} -O {filepath_s}")
+ runwithsubprocess(f"gdown {todownload_s} -O {filepath_s}", folder_s)
+ elif downloader=='wget':
+ runwithsubprocess(f"wget -O {filepath_s} {todownload_s} --progress=bar:force", folder_s)
+ elif downloader=='curl':
+ runwithsubprocess(f"curl -Lo {filepath_s} {todownload_s}", folder_s)
+ # curdir = os.getcwd()
+ # os.rename(os.path.join(curdir, filename), filepath)
+ elif downloader=='aria2':
+ ariachecker = "dpkg-query -W -f='${Status}' aria2"
+ checkaria = subprocess.getoutput(ariachecker)
+ if "no packages found matching aria2" in checkaria:
+ tempcondition = currentcondition
+ currentcondition = "Installing aria2..."
+ print('[1;32mInstalling aria2 ...')
+ print('[0m')
+ runwithsubprocess(f"apt -y update -qq")
+ runwithsubprocess(f"apt -y install -qq aria2")
+ print('[1;32maria2 installed!')
+ print('[0m')
+ currentcondition = tempcondition
+ # ariamode = True
+ runwithsubprocess(f"aria2c --summary-interval=1 --console-log-level=error -c -x 16 -s 16 -k 1M {todownload_s} -d {folder_s} -o {ariafilename_s}", folder_s, False, 'aria2')
+ printdebug("\nmode: " + mode)
+ if mode=='debugevery':
+ time.sleep(2)
+ try:
+ os.rename(filepath, os.path.join(folder, f"{os.path.splitext(filename)[0]}-{downloader}{os.path.splitext(filename)[1]}"))
+ printdebug("renamed to " + f"{os.path.splitext(filename)[0]}-{downloader}{os.path.splitext(filename)[1]}")
+ except FileNotFoundError or FileExistsError:
+ printdebug("rename failed somehow")
+ pass
+ elif mode=='civitdebugevery':
+ time.sleep(2)
+ printdebug("debug filename: " + str(filename))
+ printdebug("debug filename_s: " + str(filename_s))
+ printdebug("debug filepath: " + str(filepath_s))
+ printdebug("debug todownload_s: " + str(todownload_s))
+ printdebug("debug folder_s: " + str(folder_s))
+ try:
+ os.rename(folder, os.path.join(folder_s, f"{os.path.splitext(filename)[0]}-{downloader}{os.path.splitext(filename)[1]}"))
+ printdebug("renamed to " + f"{os.path.splitext(filename)[0]}-{downloader}{os.path.splitext(filename)[1]}")
+ except FileNotFoundError or FileExistsError:
+ printdebug("rename failed somehow")
+ pass
+ if torename and prockilled == False and filename != torename and downloader != 'aria2':
+ def saferename(oldpath, newpath):
+ try:
+ os.rename(oldpath, newpath)
+ except Exception as e:
+ print("Rename failed. Reason: " + str(e))
+ pass
+
+ saferename(filepath, os.path.join(folder, shlex.quote(torename)))
+ if mode=='civit0' and prockilled == False:
+ # if torename:
+ # filepath = os.path.join(folder, torename)
+ try:
+ if os.path.exists(filepath):
+ if os.path.getsize(filepath) <= 5 * 1024:
+ try:
+ with open(filepath, "r", encoding="utf-8") as file:
+ if "👈v2.0.0 - v2.1.1
+-
+
wgetdisabled on windows currently, until it fixed +gdown&curlbug fixed +utf-8as default encoding for queue checker (fix bug in Windows) +- Supports renaming downloaded file with
>(for example:https://files.catbox.moe/uarze8.safetensors > neurosama.safetensors) + - Supports extension usage without
--gradio-queue(ported from onedotsix) + - Supports running shell command from the UI with
!(for example: type!pip freeze, then hit theDownload all!button and see the colab console) + - Progress bar for
--gradio-queue+ aria2as the only download method when using without--gradio-queue+- Download session will be cut every 80 seconds on when using without
--gradio-queue(just like onedotsix) + - Debug stopwatch (decorator) won't run automatically when
globaldebug = True, must be uncommented manually (it disrupt the progress bar) + - Dropped support for webui based on Gradio 3.9 (update your installation, or use onedotsix instead) +
- UI tweak (Smaller font size) +
aria2as download method. +- Cancel button for cancelling download process (
--gradio-queuerequired) + - Detection if a CivitAI links no longer exist +
- New hashtags:
#textualinversion,#ti,#aestheticembedding,#aestheticembed,#controlnet, and#cnet+ - Toggle logging on/off +
shlex.quoteto properly quote links (Thanks @rti7743!) +- Supports cloning webui extensions +
- Supports download from catbox.moe +
- Supports download from CivitAI model links (Thanks @rti7743!) +
- Supports download from Github (repository and raw files) +
- Supports for aesthetic gradients, controlnet model, and extensions path. +
- UI font scaled down +
- Uses
subprocess.Popeninstead ofos.system+
_ + - Debug snapshot
+ Whenglobaldebug = True, the moment this extension launch, it saves the current state of the webui on various location (intosnapshot.txt), and when you type#debugresetdownloadson the textbox, it will compare the current state and the last saved state, and removes every new file/folder. This will be useful for debugging and testing. + - Debug every download method
+ Whenglobaldebug = Trueand you type#debugevery methodon the textbox, every link that has 4 different method of download (Huggingface etc.) will be downloaded with every method, regardless of the radio button choice. The result is 4 file being downloaded. + - Debug stopwatch
+ Whenglobaldebug = True, it will give an output for how long a single download session lasts +
Release v2.1.1
+ Partial Windows support is back + Changes: +-
+
-
+
Release v2.1.0
+ Features: +-
+
-
+
Release v2.0.0
+ Features: +-
+
⬇️ Batchlinks Downloader ({currentversion}) {latestversiontext}
""" +introductiontext = f""" +{titletext} +(what's new?)
+This tool will read the textbox and download every links from top to bottom one by one
+Put your links down below. Supported link: Huggingface, CivitAI, MEGA, Discord, Github, Catbox, Google Drive, Pixeldrain, Mediafire, Anonfiles, Dropbox
+Use hashtag to separate downloaded items based on their download location
+Valid hashtags: #embed, #model, #hypernet, #lora, #vae, #addnetlora, etc.
+(For colab that uses sd-webui-additional-networks extension to load LoRA, use #addnetlora instead)
+Use double hashtag (##) after links for comment. Useful to mark which links downloads what.
+Remember to always press the 🔄️ refresh button on the UI after downloading models etc. in order for them to show up on the list.
Click these links for more:
+Readme Page
+Example
+Syntax
+Valid Hashtags
+Here's how you can get the direct links
+Report Bug
After clicking the Download All button, it's recommended to inspect the
+ colab console, as every information about the download progress is there.
If you feel the UI is too cramped, click the Stretch UI button above.
+ # """) +# with gr.Accordion("List of Every Hashtags and its Path", open=False, visible=True) as rightlist: +# righttable = gr.DataFrame( +# buildarrayofhashtags('right'), +# headers=["hashtag", "path"], +# datatype=["str", "str"], +# interactive=False +# ) + # sidetext = gr.Markdown(knowmoretext, visible=True) + def passvaluebetweenradio(newval): + return gr.Radio.update(value=newval) + + def passvaluebetweencheckbox(newval): + return gr.Checkbox.update(value=newval) + + choose_downloader2.change(passvaluebetweenradio, inputs=choose_downloader2, outputs=choose_downloader) + civit_default2.change(passvaluebetweenradio, inputs=civit_default2, outputs=civit_default) + civit_ispruned2.change(passvaluebetweenradio, inputs=civit_ispruned2, outputs=civit_ispruned) + civit_alsodownvae2.change(passvaluebetweencheckbox, inputs=civit_alsodownvae2, outputs=civit_alsodownvae) + finish_audio = gr.Audio(interactive=False, value=os.path.join(extension_dir, "notification.mp3"), elem_id="finish_audio", visible=False) + with gr.Accordion("List of Every Hashtags and its Path", open=False, visible=True) as bottomlist: + bottomtable = gr.DataFrame( + buildarrayofhashtags('bottom'), + headers=["hashtag", "path"], + datatype=["str", "str"], + interactive=False, + ) + gr.Markdown( + f""" +Made with ❤️ by etherealxx
(Side note: _Lycoris/Locon/Loha_ will works just fine if you use `#addnetlora` instead, as long as you have both [addnet extension](https://github.com/kohya-ss/sd-webui-additional-networks) and [locon extension](https://github.com/KohakuBlueleaf/a1111-sd-webui-locon) installed) +- New hashtag: `#upscaler`, change current save directory to _/content/stable-diffusion-webui/models/ESRGAN_ (This was added few commits ago, but i forgot to write it on the release notes) + +Fixes: +- Fixed bug when installing wget/aria2 on Windows +- Fixed bug where custom paths doesn't put CivitAi download (Model Path method) when custom hashtag is used +- Ongoing downloads will correctly stops when Cancel button is pressed on Windows +- CivitAi (Model Path method) now will not download training dataset by accident + +## v3.0.2 +- Hotfix: removed a comment that messed up the non queue mode. Now the extension works again without `--gradio-queue`. + +## v3.0.1 +- Hotfix: using `urllib.request` instead of `curl -sI` to get the model name on CivitAI direct link method, since it's more reliable (and the curl method always fails somehow). The `requests` method is returned as a fallback. + +## v3.0.0 +- Added `@extract` syntax +- Supports for custom hashtags with `@new` syntax +- (Almost) Full Windows support +- Auto-download config file if available when downloading from CivitAI (SD 2.0+) +- Auto-renaming for downloading ckpt/safetensors and pruned model from CivitAI using direct link method +- CivitAI direct link now use `curl` to get the filename, and use the chosen download method (from the four) to download. Huge download speed boost. `requests` is no longer needed. +- Supports download from Anonfiles, Dropbox, Google Drive, Mediafire, Pixeldrain +- Supports download from Github (raw and release files) +- Supports for SDless mode (read more [here](https://github.com/etherealxx/batchlinks-webui#sdless-mode)) +- UI overhaul: + - Now there's a table that shows where does the hashtags points into + - Option to stretch the UI, if your monitor is small, or using colab on mobile + - Option to hide help text + - Option to choose preferred CivitAI models. This will works if you download the model via model page link (https://civitai.com/models/) + - Upload txt now use a little button instead of covering half of the screen + +Fixes: +- CivitAI `model page link` no longer randomly download the first model on the json list. +- Most of Windows bugs +- Renaming problem when using CivitAI model page link method +- Warning message when CivitAI download isn't possible (server down) + +## v2.1.2 +- CivitAI direct link now use curl to get the filename, and use the chosen download method to download. Requests is no longer needed. + +## v2.1.1 +- Partial Windows support is back + +Changes: +- wget disabled on windows currently, until it fixed + +Fixes: +- gdown & curl bug fixed +- utf-8 as default encoding for queue checker (fix bug in Windows) + +## v2.1.0 +Features: +- Supports renaming downloaded file with `>` (for example: `https://files.catbox.moe/uarze8.safetensors > neurosama.safetensors`) +- Supports extension usage without `--gradio-queue` (ported from [onedotsix](https://github.com/etherealxx/batchlinks-webui/tree/onedotsix)) +- Supports running shell command from the UI with `!` (for example: type `!pip freeze`, then hit the `Download all!` button and see the colab console) +- Progress bar for `--gradio-queue` + +Changes: +- `aria2` as *the only* download method when using without `--gradio-queue` +- Download session will be cut every 80 seconds on when using without `--gradio-queue` (just like [onedotsix](https://github.com/etherealxx/batchlinks-webui/tree/onedotsix)) +- _Debug stopwatch (decorator)_ won't run automatically when `globaldebug = True`, must be uncommented manually (it disrupt the progress bar) +- Dropped support for webui based on Gradio 3.9 (update your installation, or use [onedotsix](https://github.com/etherealxx/batchlinks-webui/tree/onedotsix) instead) +- UI tweak (Smaller font size) + +## v2.0.0 +Features: +- `aria2` as download method. +- Cancel button for cancelling download process (`--gradio-queue` required) +- Debug snapshot.
+When `global_debug = True`, the moment this extension launch, it saves the current state of the webui on various location (into `snapshot.txt`), and when you type `#debugresetdownloads` on the textbox, it will compare the current state and the last saved state, and removes every new file/folder. This will be useful for debugging and testing. +- Debug every download method.
+When `global_debug = True` and you type `#debugevery method` on the textbox, every link that has 4 different method of download (Huggingface etc.) will be downloaded with every method, regardless of the radio button choice. The result is 4 file being downloaded. +- Detection if a CivitAI links no longer exist +- New hashtags: `#textualinversion`, `#ti`, `#aestheticembedding`, `#aestheticembed`, `#controlnet`, and `#cnet` +- Toggle logging on/off +- `shlex.quote` to properly quote links (Thanks **[@rti7743](https://github.com/rti7743)**!) +- Supports cloning webui extensions +- Supports download from CivitAI model links (Thanks **[@rti7743](https://github.com/rti7743)**!) +- Supports download from Github (repository and raw files) +- Supports for aesthetic gradients, controlnet model, and extensions path. +- UI font scaled down +- Uses `subprocess.Popen` instead of `os.system` + + +Older releases: + +- commit "alpha" is v0.1.0 (09b1d81806e5bfc4d1783ccf644359fabc372fb7) +- commit "Civit support, i hope it works." is v0.2.0 (2420ea4dd670ec71481a2ed43ff22c7c1a959543) +- commit "🪲Bug fix and 🖼️stuff" is v0.2.1 (c7a91b6e14ca6f35438e29ab7ad3d2e00d62c99b) +- commit "Merge pull request #1 from etherealxx/gradioevery" is v1.0.0 (a89e50fdfb5008f703750cf0e7878fc6c25f9219) diff --git a/TestingforfaivsixEx/content/eichellis/extensions/block-weight/README.md b/TestingforfaivsixEx/content/eichellis/extensions/block-weight/README.md new file mode 100644 index 0000000000000000000000000000000000000000..3ed96f746120612db88589402722ae40894c008f --- /dev/null +++ b/TestingforfaivsixEx/content/eichellis/extensions/block-weight/README.md @@ -0,0 +1,350 @@ +# LoRA Block Weight +- custom script for [AUTOMATIC1111's stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui) +- When applying Lora, strength can be set block by block. + +- [AUTOMATIC1111's stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui) 用のスクリプトです +- Loraを適用する際、強さを階層ごとに設定できます + +### Updates/更新情報 +2023.5.24.2000(JST) +- changed directory for presets(extentions/sd-webui-lora-block-weight/scripts/) +- プリセットの保存フォルダがextentions/sd-webui-lora-block-weight/scripts/に変更になりました。 + +2023.5.12.2100(JST) +- changed syntax of lycoris +- lycorisの書式を変更しました + +2023.04.14.2000(JST) +- support LyCORIS(a1111-sd-webui-lycoris) +- LyCORIS(a1111-sd-webui-lycoris)に対応 + +2023.03.20.2030(JST) +- Comment lines can now be added to presets +- プリセットにコメント行を追加できるようになりました +- support XYZ plot hires.fix +- XYZプロットがhires.fixに対応しました + +2023.03.16.2030(JST) +- [LyCORIS](https://github.com/KohakuBlueleaf/LyCORIS)に対応しました +- Support [LyCORIS](https://github.com/KohakuBlueleaf/LyCORIS) + +別途[LyCORIS Extention](https://github.com/KohakuBlueleaf/a1111-sd-webui-locon)が必要です。 +For use LyCORIS, [Extension](https://github.com/KohakuBlueleaf/a1111-sd-webui-locon) for LyCORIS needed. + +日本語説明は[後半](#概要)後半にあります。 + +# Overview +Lora is a powerful tool, but it is sometimes difficult to use and can affect areas that you do not want it to affect. This script allows you to set the weights block-by-block. Using this script, you may be able to get the image you want. + +## Usage +Place lora_block_weight.py in the script folder. +Or you can install from Extentions tab in web-ui. When installing, please restart web-ui.bat. + +### Active +Check this box to activate it. + +### Prompt +In the prompt box, enter the Lora you wish to use as usual. Enter the weight or identifier by typing ":" after the strength value. The identifier can be edited in the Weights setting. +``` +
changeable blocks : BASE,IN00,IN01,IN02,IN03,IN04,IN05,IN06,IN07,IN08,IN09,IN10,IN11,M00,OUT00,OUT01,OUT02,OUT03,OUT04,OUT05,OUT06,OUT07,OUT08,OUT09,OUT10,OUT11
') + xyzsetting = gr.Radio(label = "Active",choices = ["Disable","XYZ plot","Effective Block Analyzer"], value ="Disable",type = "index") + with gr.Row(visible = False) as esets: + diffcol = gr.Radio(label = "diff image color",choices = ["black","white"], value ="black",type = "value",interactive =True) + revxy = gr.Checkbox(value = False,label="change X-Y",interactive =True,elem_id="lbw_changexy") + thresh = gr.Textbox(label="difference threshold",lines=1,value="20",interactive =True,elem_id="diff_thr") + xtype = gr.Dropdown(label="X Types ", choices=[x for x in ATYPES], value=ATYPES [2],interactive =True,elem_id="lbw_xtype") + xmen = gr.Textbox(label="X Values ",lines=1,value="0,0.25,0.5,0.75,1",interactive =True,elem_id="lbw_xmen") + ytype = gr.Dropdown(label="Y Types ", choices=[y for y in ATYPES], value=ATYPES [1],interactive =True,elem_id="lbw_ytype") + ymen = gr.Textbox(label="Y Values " ,lines=1,value="IN05-OUT05",interactive =True,elem_id="lbw_ymen") + ztype = gr.Dropdown(label="Z type ", choices=[z for z in ATYPES], value=ATYPES[0],interactive =True,elem_id="lbw_ztype") + zmen = gr.Textbox(label="Z values ",lines=1,value="",interactive =True,elem_id="lbw_zmen") + + exmen = gr.Textbox(label="Range",lines=1,value="0.5,1",interactive =True,elem_id="lbw_exmen",visible = False) + eymen = gr.Textbox(label="Blocks" ,lines=1,value="BASE,IN00,IN01,IN02,IN03,IN04,IN05,IN06,IN07,IN08,IN09,IN10,IN11,M00,OUT00,OUT01,OUT02,OUT03,OUT04,OUT05,OUT06,OUT07,OUT08,OUT09,OUT10,OUT11",interactive =True,elem_id="lbw_eymen",visible = False) + ecount = gr.Number(value=1, label="number of seed", interactive=True, visible = True) + + with gr.Accordion("Weights setting",open = True): + with gr.Row(): + reloadtext = gr.Button(value="Reload Presets",variant='primary',elem_id="lbw_reload") + reloadtags = gr.Button(value="Reload Tags",variant='primary',elem_id="lbw_reload") + savetext = gr.Button(value="Save Presets",variant='primary',elem_id="lbw_savetext") + openeditor = gr.Button(value="Open TextEditor",variant='primary',elem_id="lbw_openeditor") + lbw_loraratios = gr.TextArea(label="",value=lbwpresets,visible =True,interactive = True,elem_id="lbw_ratiospreset") + + with gr.Accordion("Elemental",open = False): + with gr.Row(): + e_reloadtext = gr.Button(value="Reload Presets",variant='primary',elem_id="lbw_reload") + e_savetext = gr.Button(value="Save Presets",variant='primary',elem_id="lbw_savetext") + e_openeditor = gr.Button(value="Open TextEditor",variant='primary',elem_id="lbw_openeditor") + elemsets = gr.Checkbox(value = False,label="print change",interactive =True,elem_id="lbw_print_change") + elemental = gr.TextArea(label="Identifer:BlockID:Elements:Ratio,...,separated by empty line ",value = elempresets,interactive =True,elem_id="element") + + d_true = gr.Checkbox(value = True,visible = False) + d_false = gr.Checkbox(value = False,visible = False) + + import subprocess + def openeditors(b): + path = extpath if b else extpathe + subprocess.Popen(['start', path], shell=True) + + def reloadpresets(isweight): + if isweight: + try: + with open(extpath,encoding="utf-8") as f: + return f.read() + except OSError as e: + pass + else: + try: + with open(extpath,encoding="utf-8") as f: + return f.read() + except OSError as e: + pass + + def tagdicter(presets): + presets=presets.splitlines() + wdict={} + for l in presets: + if ":" not in l or not (l.count(",") == 16 or l.count(",") == 25) : continue + w=[] + if ":" in l : + key = l.split(":",1)[0] + w = l.split(":",1)[1] + if len([w for w in w.split(",")]) == 17 or len([w for w in w.split(",")]) ==26: + wdict[key.strip()]=w + return ",".join(list(wdict.keys())) + + def savepresets(text,isweight): + if isweight: + with open(extpath,mode = 'w',encoding="utf-8") as f: + f.write(text) + else: + with open(extpathe,mode = 'w',encoding="utf-8") as f: + f.write(text) + + reloadtext.click(fn=reloadpresets,inputs=[d_true],outputs=[lbw_loraratios]) + reloadtags.click(fn=tagdicter,inputs=[lbw_loraratios],outputs=[bw_ratiotags]) + savetext.click(fn=savepresets,inputs=[lbw_loraratios,d_true],outputs=[]) + openeditor.click(fn=openeditors,inputs=[d_true],outputs=[]) + + e_reloadtext.click(fn=reloadpresets,inputs=[d_false],outputs=[elemental]) + e_savetext.click(fn=savepresets,inputs=[elemental,d_false],outputs=[]) + e_openeditor.click(fn=openeditors,inputs=[d_false],outputs=[]) + + def urawaza(active): + if active > 0: + for obj in scripts.scripts_txt2img.alwayson_scripts: + if "lora_block_weight" in obj.filename: + scripts.scripts_txt2img.selectable_scripts.append(obj) + scripts.scripts_txt2img.titles.append("LoRA Block Weight") + for obj in scripts.scripts_img2img.alwayson_scripts: + if "lora_block_weight" in obj.filename: + scripts.scripts_img2img.selectable_scripts.append(obj) + scripts.scripts_img2img.titles.append("LoRA Block Weight") + scripts.scripts_txt2img.run = newrun + scripts.scripts_img2img.run = newrun + if active == 1:return [*[gr.update(visible = True) for x in range(6)],*[gr.update(visible = False) for x in range(4)]] + else:return [*[gr.update(visible = False) for x in range(6)],*[gr.update(visible = True) for x in range(4)]] + else: + scripts.scripts_txt2img.run = runorigin + scripts.scripts_img2img.run = runorigini + return [*[gr.update(visible = True) for x in range(6)],*[gr.update(visible = False) for x in range(4)]] + + xyzsetting.change(fn=urawaza,inputs=[xyzsetting],outputs =[xtype,xmen,ytype,ymen,ztype,zmen,exmen,eymen,ecount,esets]) + + return lbw_loraratios,lbw_useblocks,xyzsetting,xtype,xmen,ytype,ymen,ztype,zmen,exmen,eymen,ecount,diffcol,thresh,revxy,elemental,elemsets + + def process(self, p, loraratios,useblocks,xyzsetting,xtype,xmen,ytype,ymen,ztype,zmen,exmen,eymen,ecount,diffcol,thresh,revxy,elemental,elemsets): + #print("self =",self,"p =",p,"presets =",loraratios,"useblocks =",useblocks,"xyzsettings =",xyzsetting,"xtype =",xtype,"xmen =",xmen,"ytype =",ytype,"ymen =",ymen,"ztype =",ztype,"zmen =",zmen) + #Note that this does not use the default arg syntax because the default args are supposed to be at the end of the function + if(loraratios == None): + loraratios = DEF_WEIGHT_PRESET + if(useblocks == None): + useblocks = True + + if useblocks: + loraratios=loraratios.splitlines() + elemental = elemental.split("\n\n") if elemental is not None else [] + lratios={} + elementals={} + for l in loraratios: + if ":" not in l or not (l.count(",") == 16 or l.count(",") == 25) : continue + l0=l.split(":",1)[0] + lratios[l0.strip()]=l.split(":",1)[1] + for e in elemental: + e0=e.split(":",1)[0] + elementals[e0.strip()]=e.split(":",1)[1] + if elemsets : print(xyelem) + if xyzsetting and "XYZ" in p.prompt: + lratios["XYZ"] = lxyz + lratios["ZYX"] = lzyx + if xyelem != "": + if "XYZ" in elementals.keys(): + elementals["XYZ"] = elementals["XYZ"] + ","+ xyelem + else: + elementals["XYZ"] = xyelem + self.lratios = lratios + self.elementals = elementals + global princ + princ = elemsets + return + + def before_process_batch(self, p, loraratios,useblocks,xyzsetting,xtype,xmen,ytype,ymen,ztype,zmen,exmen,eymen,ecount,diffcol,thresh,revxy,elemental,elemsets,**kwargs): + if useblocks: + global prompts + prompts = kwargs["prompts"].copy() + + def process_batch(self, p, loraratios,useblocks,xyzsetting,xtype,xmen,ytype,ymen,ztype,zmen,exmen,eymen,ecount,diffcol,thresh,revxy,elemental,elemsets,**kwargs): + if useblocks: + o_prompts = [p.prompt] + for prompt in prompts: + if "'
+ for pic in model['images']:
+ img_html = img_html + f'![]({pic["url"]})
'
+ img_html = img_html + '
'
+ output_html = f"Model: {model_name}
Version: {model_version}
Uploaded by: {model_uploader}
Download Here
{model_desc}
{img_html}
"
+
+ return gr.HTML.update(value=output_html), gr.Textbox.update(value=output_training), gr.Dropdown.update(choices=[PLACEHOLDER] + [k for k, v in dl_dict.items()], value=PLACEHOLDER)
+ else:
+ return gr.HTML.update(value=None), gr.Textbox.update(value=None), gr.Dropdown.update(choices=[], value=None)
+
+
+def request_civit_api(api_url=None):
+ # Make a GET request to the API
+ response = requests.get(api_url)
+
+ # Check the status code of the response
+ if response.status_code != 200:
+ print("Request failed with status code: {}".format(response.status_code))
+ exit()
+
+ data = json.loads(response.text)
+ return data
+
+#from https://github.com/thetrebor/sd-civitai-browser/blob/add-download-images/scripts/civitai-api.py
+def update_everything(list_models, list_versions, model_filename, dl_url):
+ (a, d, f) = update_model_info(list_models, list_versions)
+ dl_url = update_dl_url(list_models, list_versions, f['value'])
+ return (a, d, f, list_versions, list_models, dl_url)
+
+def save_image_files(preview_image_html, model_filename, content_type, use_new_folder, list_models, lora_old):
+ print("Save Images Clicked")
+ model_folder = make_new_folder(content_type, use_new_folder, list_models, lora_old)
+
+ img_urls = re.findall(r'src=[\'"]?([^\'" >]+)', preview_image_html)
+
+ name = os.path.splitext(model_filename)[0]
+ assert(name != "不仅仅是图像浏览器，更是一个强大的图像管理器。精确的图像搜索配合多选操作进行筛选/归档/打包，成倍提高效率。更是支持以独立模式运行，无需SD-Webui
+ + + +https://github.com/zanllp/sd-webui-infinite-image-browsing/assets/25872019/807b890b-7be8-4816-abba-f3ad340a2232 + + +## 主要特性 + +### 🔥 极佳性能 +- 存在缓存的情况下后，图像可以在几毫秒内显示。 +- 默认使用缩略图显示图像，默认大小为512像素，您可以在全局设置页中调整缩略图分辨率。 +- 你还可以控制网格图像的宽度，允许以64px到1024px的宽度范围进行显示 + +### 🔍 图像搜索和收藏 +- 将会把Prompt、Model、Lora等信息转成标签，将根据使用频率排序以供进行精确的搜索。 +- 支持标签自动完成、[翻译](https://github.com/zanllp/sd-webui-infinite-image-browsing/issues/39)和自定义。 +- 可通过在右键菜单切换自定义标签来实现图像收藏。 +- 支持类似谷歌的高级搜索。 +- 同样支持模糊搜索，您可以使用文件名或生成信息的一部分进行搜索。 +- 支持添加自定义搜索路径，方便管理自己创建的文件夹集合。 + +### 🖼️ 查看图像/视频和“发送到” +- 支持查看图像生成信息。全屏预览下同样支持。 +- 支持将图像发送到其他选项卡和其他插件，例如 ControlNet, openOutpaint。 +- 支持全屏预览，并且支持在全屏预览下使用自定义快捷键进行操作 +- 支持在全屏预览模式下通过按下方向键或点击按钮移动到前一个或后一个图像。 +- 支持播放远程服务器上的视频文件 + +### 💻 多种使用方法 +- 您可以将其作为 SD-webui 的扩展安装。 +- 您可以使用 Python 独立运行它。 +- 还提供桌面应用程序版本。 + +### 🚶♀️ Walk模式 +- 自动加载下一个文件夹 `(类似于 os.walk)`，可让您无需分页浏览所有图像。 +- 已测试可正常处理超过 27,000 个文件。 + +### 🌳 基于文件树结构的预览和文件操作 +- 支持基于文件树结构的预览。 +- 支持自动刷新。 +- 支持基本文件操作以及多选删除/移动/复制，新建文件夹等。 +- 按住 Ctrl、Shift 或 Cmd 键可选择多个项目。 + +### 🆚 图像对比 (类似ImgSli) +- 提供两张图片的并排比较 + +### 🌐 多语言支持 +- 目前支持简体中文/繁体中文/英文/德语。 +- 如果您希望添加新的语言，请参考 [i18n.ts](https://github.com/zanllp/sd-webui-infinite-image-browsing/blob/main/vue/src/i18n/zh-hans.ts) 并提交相关的代码。 + + +### 🔐 隐私和安全 +- 支持自定义secret key来进行身份验证 +- 支持配置对文件系统的访问控制，默认将在服务允许公开访问时启用 +- 支持自定义访问控制允许的路径。 +- 支持控制访问权限。你可以让IIB以只读模式运行 +- [点击这里查看详情](.env.example) + +### ⌨️ 快捷键 +- 支持删除和添加/移除Tag，在全局设置页进行自定义触发按钮 + +### 📦 打包 / 批量下载 +- 允许你一次性打包下载多个图像 +- 数据来源可以是搜索结果/普通的图像网格查看页面/walk模式等。使用拖拽或者“发送到”都可将图片添加待处理列表 + + +如果您喜欢这个项目并且觉得它对您有帮助，请考虑给我点个⭐️。这将对我持续开发和维护这个项目非常重要。如果您有任何建议或者想法，请随时在issue中提出，我会尽快回复。再次感谢您的支持！ + + +
+
+
+[视频演示可以在Bilibili上观看](https://space.bilibili.com/27227392/channel/series)
+# 安装/运行
+
+## 作为SD-webui的扩展程序:
+1. 在SD-webui中打开`扩展`选项卡。
+2. 选择`从URL安装`选项。
+3. 输入 `https://github.com/zanllp/sd-webui-infinite-image-browsing`。
+4. 点击`安装`按钮。
+5. 等待安装完成，然后点击`应用并重启UI`。
+
+## 作为使用Python运行的独立程序（不需要SD-webui）:
+请参考[Can the extension function without the web UI?](https://github.com/zanllp/sd-webui-infinite-image-browsing/issues/47)
+
+如果需要查看ComfyUI生成的图片相关，请先参考 https://github.com/zanllp/sd-webui-infinite-image-browsing/issues/202
+
+如果你需要dockerfile 参考 https://github.com/zanllp/sd-webui-infinite-image-browsing/discussions/366
+
+## 作为桌面应用程序（不需要SD-webui和Python）:
+
+从仓库页面右侧的`releases`部分下载并安装程序。如果提升检测到病毒忽略即可这是误报。
+# 预览
+
+
+
+## 图像搜索
+
+在第一次使用时，你需要点击等待索引的生成，我2万张图像的情况下大概需要15秒（配置是amd 5600x和pcie ssd）。后续使用他会检查文件夹是否发生变化，如果发生变化则需要重新生成索引,通常这个过程极快。
+
+图像搜索支持翻译，具体看这个 https://github.com/zanllp/sd-webui-infinite-image-browsing/issues/39 。
+
+## 图像比较
+
+
+
+## 全屏预览
+
+
+在全屏预览下同样可以查看图片信息和进行上下文菜单上的的操作，支持拖拽/调整/展开收起
+
+https://user-images.githubusercontent.com/25872019/235327735-bfb50ea7-7682-4e50-b303-38159456e527.mp4
+
+
+如果你和我一样不需要查看生成信息，你可以选择直接缩小这个面板，所有上下文操作仍然可用
+
+
+
+也可以通过右上角的图标来触发
+
+
+
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/README.md b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/README.md
new file mode 100644
index 0000000000000000000000000000000000000000..ba3846598043a72a012a59ec37dcce081054c2f3
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/README.md
@@ -0,0 +1,141 @@
+[中文文档](./README-zh.md)
+
+[View recent updates (Change log)](https://github.com/zanllp/sd-webui-infinite-image-browsing/wiki/Change-log)
+
+# Stable Diffusion webui Infinite Image Browsing
+
+It's not just an image browser, but also a powerful image manager. Precise image search combined with multi-selection operations allows for filtering/archiving/packaging, greatly increasing efficiency. It also supports running in standalone mode, without the need for SD-Webui.
+ +https://github.com/zanllp/sd-webui-infinite-image-browsing/assets/25872019/807b890b-7be8-4816-abba-f3ad340a2232 +## Key Features + +### 🔥 Excellent Performance +- Once caching is generated, images can be displayed in just a few milliseconds. +- Images are displayed with thumbnails by default, with a default size of 512 pixels. You can adjust the thumbnail resolution on the global settings page. +- You can also control the width of the grid images, allowing them to be displayed in widths ranging from 64px to 1024px. + +### 🔍 Image Search & Favorite +- The prompt, model, Lora, and other information will be converted into tags and sorted by frequency of use for precise searching. +- Supports tag autocomplete, [auto-translation](https://github.com/zanllp/sd-webui-infinite-image-browsing/issues/39), and customization. +- Image favorite can be achieved by toggling custom tags for images in the right-click menu. +- Support for advanced search similar to Google +- Also supports fuzzy search, you can search by a part of the filename or generated information. +- Support adding custom search paths for easy management of folders created by the user. + +### 🖼️ View Images/Videos & `Send To` +- Supports viewing image generation information. Also supported in full-screen preview mode. +- Supports sending images to other tabs and third-party extensions such as ControlNet , openOutpaint. +- Support full-screen preview and enable custom shortcut key operations while in full-screen preview mode. +- Support navigating to the previous or next image in full-screen preview mode by pressing arrow keys or clicking buttons. +- Support playing video files from a remote server. + +### 💻 Multiple Usage Methods +- You can install it as an extension on SD-webui. +- You can run it independently using Python. +- The desktop app version is also available. + + +### 🚶♀️ Walk Mode +- Automatically load the next folder `(similar to os.walk)`, allowing you to browse all images without paging. +- Tested to work properly with over 27,000 files. + +### 🌳 Preview based on File Tree Structure & File operations +- Supports file tree-based preview. +- Supports automatic refreshing. +- Supports basic file operations, such as multiple selection for deleting/moving/copying, and creating new folders. +- Hold down the Ctrl, Shift, or Cmd key to select multiple items. + +### 🆚 image comparison (similar to Imgsli) +- Provides a side-by-side comparison of two images. + +### 🌐 Multilingual Support +- Currently supports Simplified Chinese/Traditional Chinese/English/German. +- If you would like to add a new language, please refer to [i18n.ts](https://github.com/zanllp/sd-webui-infinite-image-browsing/blob/main/vue/src/i18n/zh-hans.ts) and submit the relevant code. + +### 🔐 Privacy and Security +- Supports custom secret key for authentication. +- Supports configuring access control for the file system, which will be enabled by default when the service allows public access. +- Supports customizing the allowed paths for access control. +- Supports controlling access permissions. You can run IIB in read-only mode. +- [Click here to see details](.env.example) + + +### 📦 Packaging/Batch Download +- Allows you to download multiple images at once. +- The data source can be search results, a regular image grid view page, walk mode, etc. Images can be added to the processing list through drag-and-drop or "Send To". +### ⌨️ Keyboard Shortcuts +- Allows for deleting and adding/removing tags, with customizable trigger buttons in the global settings page. + + +If you like this project and find it helpful, please consider giving it a ⭐️. This would be very important for me to continue developing and maintaining this project. If you have any suggestions or ideas, please feel free to raise them in the issue section, and I will respond as soon as possible. Thank you again for your support! + + +
+
+
+
+# Installation / Running
+## As an extension for SD-webui:
+1. Open the `Extensions` tab in SD-webui.
+2. Select the `Install from URL` option.
+3. Enter `https://github.com/zanllp/sd-webui-infinite-image-browsing`.
+4. Click on the `Install` button.
+5. Wait for the installation to complete and click on `Apply and restart UI`.
+
+## As a standalone program that runs using Python. (without SD-webui):
+
+Refer to [Can the extension function without the web UI?](https://github.com/zanllp/sd-webui-infinite-image-browsing/issues/47)
+
+If you need to view images generated by ComfyUI, please refer to https://github.com/zanllp/sd-webui-infinite-image-browsing/issues/202.
+
+If you need a Dockerfile, you can refer to this link. https://github.com/zanllp/sd-webui-infinite-image-browsing/discussions/366
+
+## As a desktop application (without SD-webui and Python):
+Download and install the program from the `releases` section on the right-hand side of the repository page.
+If the antivirus detects a virus, it can be ignored as a false positive.
+# Preview
+
+
+
+## Image Search
+
+During the first use, you need to click and wait for the index generation. For my case with 20,000 images, it took about 15 seconds (with an AMD 5600X CPU and PCIe SSD). For subsequent uses, it will check whether there are changes in the folder, and if so, it needs to regenerate the index. Usually, this process is very fast.
+
+Image search supports translation, see https://github.com/zanllp/sd-webui-infinite-image-browsing/issues/39 for more detail. Feel free to share files for other languages to facilitate everyone's use.
+
+
+## Full Screen Preview
+
+
+
+In full-screen preview mode, you can also view image information and perform operations on the context menu. It supports dragging, resizing and expanding/collapsing .
+
+https://user-images.githubusercontent.com/25872019/235327735-bfb50ea7-7682-4e50-b303-38159456e527.mp4
+
+If you, like me, don't need to view the generation information, you can choose to simply minimize this panel, and all contextual operations will still be available.
+
+
+
+You can also trigger it by hovering your mouse over the icon in the top right corner.
+
+
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/app.py b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/app.py
new file mode 100644
index 0000000000000000000000000000000000000000..c6a6c7e331abbae29f6434005d23ba130b1959ce
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/app.py
@@ -0,0 +1,221 @@
+from typing import List
+from fastapi import FastAPI
+from fastapi.responses import FileResponse
+import uvicorn
+import os
+from scripts.iib.api import infinite_image_browsing_api, index_html_path
+from scripts.iib.tool import get_sd_webui_conf, get_valid_img_dirs, sd_img_dirs, normalize_paths
+from scripts.iib.db.datamodel import DataBase, Image
+from scripts.iib.db.update_image_data import update_image_data
+import argparse
+from typing import Optional, Coroutine
+import json
+
+tag = "\033[31m[warn]\033[0m"
+
+default_port = 8000
+default_host = "127.0.0.1"
+
+
+def sd_webui_paths_check(sd_webui_config: str, relative_to_config: bool):
+ conf = {}
+ with open(sd_webui_config, "r") as f:
+ conf = json.loads(f.read())
+ if relative_to_config:
+ for dir in sd_img_dirs:
+ if not os.path.isabs(conf[dir]):
+ conf[dir] = os.path.normpath(
+ os.path.join(sd_webui_config, "../", conf[dir])
+ )
+ paths = [conf.get(key) for key in sd_img_dirs]
+ paths_check(paths)
+
+
+def paths_check(paths):
+ for path in paths:
+ if not path or len(path.strip()) == 0:
+ continue
+ if os.path.isabs(path):
+ abs_path = path
+ else:
+ abs_path = os.path.join(os.getcwd(), path)
+ if not os.path.exists(abs_path):
+ print(f"{tag} The path '{abs_path}' will be ignored (value: {path}).")
+
+
+def do_update_image_index(sd_webui_config: str, relative_to_config=False):
+ dirs = get_valid_img_dirs(
+ get_sd_webui_conf(
+ sd_webui_config=sd_webui_config,
+ sd_webui_path_relative_to_config=relative_to_config,
+ )
+ )
+ if not len(dirs):
+ return print(f"{tag} no valid image directories, skipped")
+ conn = DataBase.get_conn()
+ update_image_data(dirs)
+ if Image.count(conn=conn) == 0:
+ return print(f"{tag} it appears that there is some issue")
+ print("update image index completed. ✨")
+
+
+class AppUtils:
+ def __init__(
+ self,
+ sd_webui_config: Optional[str] = None,
+ update_image_index: bool = False,
+ extra_paths: List[str] = [],
+ sd_webui_path_relative_to_config=False,
+ allow_cors=False,
+ enable_shutdown=False,
+ sd_webui_dir: Optional[str] = None,
+ ):
+ """
+ Parameter definitions can be found by running the `python app.py -h `command or by examining the setup_parser() function.
+ """
+ self.sd_webui_config = sd_webui_config
+ self.update_image_index = update_image_index
+ self.extra_paths = extra_paths
+ self.sd_webui_path_relative_to_config = sd_webui_path_relative_to_config
+ self.allow_cors = allow_cors
+ self.enable_shutdown = enable_shutdown
+ self.sd_webui_dir = sd_webui_dir
+ if sd_webui_dir:
+ DataBase.path = os.path.join(
+ sd_webui_dir, "extensions/sd-webui-infinite-image-browsing/iib.db"
+ )
+ self.sd_webui_config = os.path.join(sd_webui_dir, "config.json")
+ self.sd_webui_path_relative_to_config = True
+
+ def set_params(self, *args, **kwargs) -> None:
+ """改变参数，与__init__的行为一致"""
+ self.__init__(*args, **kwargs)
+
+ @staticmethod
+ def async_run(app: FastAPI, port: int = default_port, host = default_host) -> Coroutine:
+ """
+ 用于从异步运行的 FastAPI，在 Jupyter Notebook 环境中非常有用
+ """
+ # 不建议改成 async def，并且用 await 替换 return，
+ # 因为这样会失去对 server.serve() 的控制。
+ config = uvicorn.Config(app, host=host, port=port)
+ server = uvicorn.Server(config)
+ return server.serve()
+
+ def wrap_app(self, app: FastAPI) -> None:
+ """
+ 为传递的app挂载上infinite_image_browsing后端
+ """
+ sd_webui_config = self.sd_webui_config
+ update_image_index = self.update_image_index
+ extra_paths = self.extra_paths
+
+ if sd_webui_config:
+ sd_webui_paths_check(sd_webui_config, self.sd_webui_path_relative_to_config)
+ if update_image_index:
+ do_update_image_index(
+ sd_webui_config, self.sd_webui_path_relative_to_config
+ )
+ paths_check(extra_paths)
+
+ infinite_image_browsing_api(
+ app,
+ sd_webui_config=sd_webui_config,
+ extra_paths_cli=normalize_paths(extra_paths, os.getcwd()),
+ sd_webui_path_relative_to_config=self.sd_webui_path_relative_to_config,
+ allow_cors=self.allow_cors,
+ enable_shutdown=self.enable_shutdown,
+ launch_mode="server",
+ )
+
+ def get_root_browser_app(self) -> FastAPI:
+ """
+ 获取首页挂载在"/"上的infinite_image_browsing FastAPI实例
+ """
+ app = FastAPI()
+
+ # 用于在首页显示
+ @app.get("/")
+ def index():
+ return FileResponse(index_html_path)
+
+ self.wrap_app(app)
+ return app
+
+
+def setup_parser() -> argparse.ArgumentParser:
+ parser = argparse.ArgumentParser(
+ description="A fast and powerful image browser for Stable Diffusion webui."
+ )
+ parser.add_argument(
+ "--host", type=str, default=default_host, help="The host to use"
+ )
+ parser.add_argument(
+ "--port", type=int, help="The port to use", default=default_port
+ )
+ parser.add_argument(
+ "--sd_webui_config", type=str, default=None, help="The path to the config file"
+ )
+ parser.add_argument(
+ "--update_image_index", action="store_true", help="Update the image index"
+ )
+ parser.add_argument(
+ "--extra_paths",
+ nargs="+",
+ help="Extra paths to use, will be added to Quick Move.",
+ default=[],
+ )
+ parser.add_argument(
+ "--sd_webui_path_relative_to_config",
+ action="store_true",
+ help="Use the file path of the sd_webui_config file as the base for all relative paths provided within the sd_webui_config file.",
+ )
+ parser.add_argument(
+ "--allow_cors",
+ action="store_true",
+ help="Allow Cross-Origin Resource Sharing (CORS) for the API.",
+ )
+ parser.add_argument(
+ "--enable_shutdown",
+ action="store_true",
+ help="Enable the shutdown endpoint.",
+ )
+ parser.add_argument(
+ "--sd_webui_dir",
+ type=str,
+ default=None,
+ help="The path to the sd_webui folder. When specified, the sd_webui's configuration will be used and the extension must be installed within the sd_webui. Data will be shared between the two.",
+ )
+ return parser
+
+
+def launch_app(port: int = default_port, host: str = default_host, *args, **kwargs: dict) -> None:
+ """
+ Launches the application on the specified port.
+
+ Args:
+ **kwargs (dict): Optional keyword arguments that can be used to configure the application.
+ These can be viewed by running 'python app.py -h' or by checking the setup_parser() function.
+ """
+ app_utils = AppUtils(*args, **kwargs)
+ app = app_utils.get_root_browser_app()
+ uvicorn.run(app, host=host, port=port)
+
+
+async def async_launch_app(port: int = default_port, host: str = default_host, *args, **kwargs: dict) -> None:
+ """
+ Asynchronously launches the application on the specified port.
+
+ Args:
+ **kwargs (dict): Optional keyword arguments that can be used to configure the application.
+ These can be viewed by running 'python app.py -h' or by checking the setup_parser() function.
+ """
+ app_utils = AppUtils(*args, **kwargs)
+ app = app_utils.get_root_browser_app()
+ await app_utils.async_run(app, host=host, port=port)
+
+
+if __name__ == "__main__":
+ parser = setup_parser()
+ args = parser.parse_args()
+ launch_app(**vars(args))
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/install.py b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/install.py
new file mode 100644
index 0000000000000000000000000000000000000000..308c9570c8f614b94ec7334b18f0bf419420ca48
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/install.py
@@ -0,0 +1,28 @@
+import launch
+import os
+import pkg_resources
+
+req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "requirements.txt")
+
+def dist2package(dist: str):
+ return ({
+ "pyfunctional": "functional",
+ "python-dotenv": "dotenv",
+ "Pillow": "PIL"
+ }).get(dist, dist)
+
+# copy from controlnet, thanks
+with open(req_file) as file:
+ for package in file:
+ try:
+ package = package.strip()
+ if '==' in package:
+ package_name, package_version = package.split('==')
+ installed_version = pkg_resources.get_distribution(package_name).version
+ if installed_version != package_version:
+ launch.run_pip(f"install {package}", f"sd-webui-infinite-image-browsing requirement: changing {package_name} version from {installed_version} to {package_version}")
+ elif not launch.is_installed(dist2package(package)):
+ launch.run_pip(f"install {package}", f"sd-webui-infinite-image-browsing requirement: {package}")
+ except Exception as e:
+ print(e)
+ print(f'Warning: Failed to install {package}, something may not work.')
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/javascript/index.js b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/javascript/index.js
new file mode 100644
index 0000000000000000000000000000000000000000..94113a7534102f7a1c58783dce4b8ca4e52c3980
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/javascript/index.js
@@ -0,0 +1,176 @@
+Promise.resolve().then(async () => {
+ /**
+ * This is a file generated using `yarn build`.
+ * If you want to make changes, please modify `index.tpl.js` and run the command to generate it again.
+ */
+ const html = `
+
+
+
+
+
+
+
+
+ It seems to have failed to load. You can try refreshing the page.
If that doesn't work, click on FAQ for help
+
+
+
+
+`
+ let containerSelector = '#infinite_image_browsing_container_wrapper'
+ let shouldMaximize = true
+
+ try {
+ containerSelector = __iib_root_container__
+ shouldMaximize = __iib_should_maximize__
+ } catch (e) {}
+
+ const delay = (timeout = 0) => new Promise((resolve) => setTimeout(resolve, timeout))
+ const asyncCheck = async (getter, checkSize = 100, timeout = 1000) => {
+ let target = getter()
+ let num = 0
+ while (checkSize * num < timeout && (target === undefined || target === null)) {
+ await delay(checkSize)
+ target = getter()
+ num++
+ }
+ return target
+ }
+ const getTabIdxById = (id) => {
+ const tabList = gradioApp().querySelectorAll('#tabs > .tabitem[id^=tab_]')
+ return Array.from(tabList).findIndex((v) => v.id.includes(id))
+ }
+
+ const switch2targetTab = (idx) => {
+ try {
+ gradioApp().querySelector('#tabs').querySelectorAll('button')[idx].click()
+ } catch (error) {
+ console.error(error)
+ }
+ }
+
+ /**
+ * @type {HTMLDivElement}
+ */
+ const wrap = await asyncCheck(() => gradioApp().querySelector(containerSelector), 500, Infinity)
+ wrap.childNodes.forEach((v) => wrap.removeChild(v))
+ const iframe = document.createElement('iframe')
+ iframe.srcdoc = html
+ iframe.style = `width: 100%;height:100vh`
+ wrap.appendChild(iframe)
+
+ if (shouldMaximize) {
+ onUiTabChange(() => {
+ const el = get_uiCurrentTabContent()
+ if (el?.id.includes('infinite-image-browsing')) {
+ try {
+ const iibTop = gradioApp().querySelector('#iib_top')
+ if (!iibTop) {
+ throw new Error("element '#iib_top' is not found")
+ }
+ const topRect = iibTop.getBoundingClientRect()
+ wrap.style = `
+ top:${Math.max(48, topRect.top) - 10}px;
+ position: fixed;
+ left: 10px;
+ right: 10px;
+ z-index: 100;
+ width: unset;
+ bottom: 10px;`
+ iframe.style = `width: 100%;height:100%`
+ } catch (error) {
+ console.error('Error mounting IIB. Running fallback.', error)
+ wrap.style = ''
+ iframe.style = `width: 100%;height:100vh`
+ }
+ }
+ })
+ }
+
+ const IIB_container_id = [Date.now(), Math.random()].join()
+ window.IIB_container_id = IIB_container_id
+ const imgTransferBus = new BroadcastChannel('iib-image-transfer-bus')
+ imgTransferBus.addEventListener('message', async (ev) => {
+ const data = ev.data
+ if (
+ typeof data !== 'object' ||
+ (typeof data.IIB_container_id === 'string' && data.IIB_container_id !== IIB_container_id)
+ ) {
+ return
+ }
+ console.log(`iib-message:`, data)
+ const appDoc = gradioApp()
+ switch (data.event) {
+ case 'click_hidden_button': {
+ const btn = gradioApp().querySelector(`#${data.btnEleId}`)
+ btn.click()
+ break
+ }
+ case 'send_to_control_net': {
+ data.type === 'img2img' ? window.switch_to_img2img() : window.switch_to_txt2img()
+ await delay(100)
+ const cn = appDoc.querySelector(`#${data.type}_controlnet`)
+ const wrap = cn.querySelector('.label-wrap')
+ if (!wrap.className.includes('open')) {
+ wrap.click()
+ await delay(100)
+ }
+ wrap.scrollIntoView()
+ wrap.dispatchEvent(await createPasteEvent(data.url))
+ break
+ }
+ case 'send_to_outpaint': {
+ switch2targetTab(getTabIdxById('openOutpaint'))
+ await delay(100)
+ const iframe = appDoc.querySelector('#openoutpaint-iframe')
+ openoutpaint_send_image(await imgUrl2DataUrl(data.url))
+ iframe.contentWindow.postMessage({
+ key: appDoc.querySelector('#openoutpaint-key').value,
+ type: 'openoutpaint/set-prompt',
+ prompt: data.prompt,
+ negPrompt: data.negPrompt
+ })
+ break
+ }
+ }
+
+ function imgUrl2DataUrl(imgUrl) {
+ return new Promise((resolve, reject) => {
+ fetch(imgUrl)
+ .then((response) => response.blob())
+ .then((blob) => {
+ const reader = new FileReader()
+ reader.readAsDataURL(blob)
+ reader.onloadend = function () {
+ const dataURL = reader.result
+ resolve(dataURL)
+ }
+ })
+ .catch((error) => reject(error))
+ })
+ }
+
+ async function createPasteEvent(imgUrl) {
+ const response = await fetch(imgUrl)
+ const imageBlob = await response.blob()
+ const imageFile = new File([imageBlob], 'image.jpg', {
+ type: imageBlob.type,
+ lastModified: Date.now()
+ })
+ const dataTransfer = new DataTransfer()
+ dataTransfer.items.add(imageFile)
+ const pasteEvent = new ClipboardEvent('paste', {
+ clipboardData: dataTransfer,
+ bubbles: true
+ })
+ return pasteEvent
+ }
+ })
+})
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/requirements.txt b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/requirements.txt
new file mode 100644
index 0000000000000000000000000000000000000000..3b2c752f436e287ebc298e73aa1b052b414390c7
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/requirements.txt
@@ -0,0 +1,6 @@
+fastapi
+uvicorn
+piexif
+python-dotenv
+Pillow
+pyfunctional
\ No newline at end of file
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/api.py b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/api.py
new file mode 100644
index 0000000000000000000000000000000000000000..f0e93780ce1c028e28a445f4cbb433a39c31ef51
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/api.py
@@ -0,0 +1,768 @@
+from datetime import datetime, timedelta
+import os
+import shutil
+import sqlite3
+from scripts.iib.tool import (
+ comfyui_exif_data_to_str,
+ get_comfyui_exif_data,
+ human_readable_size,
+ is_img_created_by_comfyui,
+ is_img_created_by_comfyui_with_webui_gen_info,
+ is_valid_image_path,
+ temp_path,
+ read_sd_webui_gen_info_from_image,
+ get_formatted_date,
+ is_win,
+ cwd,
+ locale,
+ enable_access_control,
+ get_windows_drives,
+ get_sd_webui_conf,
+ get_valid_img_dirs,
+ open_folder,
+ get_img_geninfo_txt_path,
+ unique_by,
+ create_zip_file,
+ normalize_paths,
+ to_abs_path,
+ is_secret_key_required
+)
+from fastapi import FastAPI, HTTPException
+from fastapi.staticfiles import StaticFiles
+import asyncio
+from typing import List, Optional
+from pydantic import BaseModel
+from fastapi.responses import FileResponse
+from PIL import Image
+from fastapi import Depends, FastAPI, HTTPException, Request
+from fastapi.middleware.cors import CORSMiddleware
+import hashlib
+from scripts.iib.db.datamodel import (
+ DataBase,
+ ExtraPathType,
+ Image as DbImg,
+ Tag,
+ Floder,
+ ImageTag,
+ ExtraPath,
+ FileInfoDict,
+)
+from scripts.iib.db.update_image_data import update_image_data
+from scripts.iib.logger import logger
+from functional import seq
+import urllib.parse
+
+index_html_path = os.path.join(cwd, "vue/dist/index.html") # 在app.py也被使用
+
+
+send_img_path = {"value": ""}
+mem = {"secret_key_hash": None, "extra_paths": [], "all_scanned_paths": []}
+secret_key = os.getenv("IIB_SECRET_KEY")
+if secret_key:
+ print("Secret key loaded successfully. ")
+
+WRITEABLE_PERMISSIONS = ["read-write", "write-only"]
+
+is_api_writeable = not (os.getenv("IIB_ACCESS_CONTROL_PERMISSION")) or (
+ os.getenv("IIB_ACCESS_CONTROL_PERMISSION") in WRITEABLE_PERMISSIONS
+)
+
+
+async def write_permission_required():
+ if not is_api_writeable:
+ error_msg = (
+ "User is not authorized to perform this action. Required permission: "
+ + ", ".join(WRITEABLE_PERMISSIONS)
+ )
+ raise HTTPException(status_code=403, detail=error_msg)
+
+
+async def verify_secret(request: Request):
+ if not secret_key:
+ if is_secret_key_required:
+ raise HTTPException(status_code=400, detail={ "type": "secret_key_required" })
+ return
+ token = request.cookies.get("IIB_S")
+ if not token:
+ raise HTTPException(status_code=401, detail="Unauthorized")
+ if not mem["secret_key_hash"]:
+ mem["secret_key_hash"] = hashlib.sha256(
+ (secret_key + "_ciallo").encode("utf-8")
+ ).hexdigest()
+ if mem["secret_key_hash"] != token:
+ raise HTTPException(status_code=401, detail="Unauthorized")
+
+
+def infinite_image_browsing_api(app: FastAPI, **kwargs):
+ pre = "/infinite_image_browsing"
+
+ if kwargs.get("allow_cors"):
+ app.add_middleware(
+ CORSMiddleware,
+ allow_origin_regex="^[\w./:-]+$",
+ allow_methods=["*"],
+ allow_headers=["*"],
+ )
+
+ def get_img_search_dirs():
+ try:
+ return get_valid_img_dirs(get_sd_webui_conf(**kwargs))
+ except Exception as e:
+ print(e)
+ return []
+
+ def update_all_scanned_paths():
+ allowed_paths = os.getenv("IIB_ACCESS_CONTROL_ALLOWED_PATHS")
+ if allowed_paths:
+ sd_webui_conf = get_sd_webui_conf(**kwargs)
+ path_config_key_map = {
+ "save": "outdir_save",
+ "extra": "outdir_extras_samples",
+ "txt2img": "outdir_txt2img_samples",
+ "img2img": "outdir_img2img_samples",
+ }
+
+ def path_map(path: str):
+ path = path.strip()
+ if path in path_config_key_map:
+ return sd_webui_conf.get(path_config_key_map.get(path))
+ return path
+
+ paths = normalize_paths(
+ seq(allowed_paths.split(","))
+ .map(path_map)
+ .filter(lambda x: x)
+ .to_list(),
+ os.getcwd()
+ )
+ else:
+ paths = (
+ get_img_search_dirs() + mem["extra_paths"] + kwargs.get("extra_paths_cli", [])
+ )
+ mem["all_scanned_paths"] = unique_by(paths)
+
+ update_all_scanned_paths()
+
+ def update_extra_paths(conn: sqlite3.Connection):
+ r = ExtraPath.get_extra_paths(conn)
+ mem["extra_paths"] = [x.path for x in r]
+ update_all_scanned_paths()
+
+ def safe_commonpath(seq):
+ try:
+ return os.path.commonpath(seq)
+ except Exception as e:
+ # logger.error(e)
+ return ""
+
+ def is_path_under_parents(path, parent_paths: List[str] = []):
+ """
+ Check if the given path is under one of the specified parent paths.
+ :param path: The path to check.
+ :param parent_paths: By default, all scanned paths are included in the list of parent paths
+ :return: True if the path is under one of the parent paths, False otherwise.
+ """
+ try:
+ if not parent_paths:
+ parent_paths = mem["all_scanned_paths"]
+ path = to_abs_path(path)
+ for parent_path in parent_paths:
+ if safe_commonpath([path, parent_path]) == parent_path:
+ return True
+ except Exception as e:
+ logger.error(e)
+ return False
+
+ def is_path_trusted(path: str):
+ if not enable_access_control:
+ return True
+ try:
+ parent_paths = mem["all_scanned_paths"]
+ path = to_abs_path(path)
+ for parent_path in parent_paths:
+ if len(path) <= len(parent_path):
+ if parent_path.startswith(path):
+ return True
+ else:
+ if path.startswith(parent_path):
+ return True
+ except:
+ pass
+ return False
+
+ def check_path_trust(path: str):
+ if not is_path_trusted(path):
+ raise HTTPException(status_code=403)
+
+ def filter_allowed_files(files: List[FileInfoDict]):
+ return [x for x in files if is_path_trusted(x["fullpath"])]
+
+ static_dir = f"{cwd}/vue/dist"
+ if os.path.exists(static_dir):
+ app.mount(
+ f"{pre}/fe-static",
+ StaticFiles(directory=static_dir),
+ name="infinite_image_browsing-fe-static",
+ )
+
+ @app.get(f"{pre}/hello")
+ async def greeting():
+ return "hello"
+
+ @app.get(f"{pre}/global_setting", dependencies=[Depends(verify_secret)])
+ async def global_setting():
+ all_custom_tags = []
+
+ extra_paths = []
+ try:
+ conn = DataBase.get_conn()
+ all_custom_tags = Tag.get_all_custom_tag(conn)
+ extra_paths = ExtraPath.get_extra_paths(conn) + [
+ ExtraPath(path, ExtraPathType.cli_only)
+ for path in kwargs.get("extra_paths_cli", [])
+ ]
+ update_extra_paths(conn)
+ except Exception as e:
+ print(e)
+ return {
+ "global_setting": get_sd_webui_conf(**kwargs),
+ "cwd": cwd,
+ "is_win": is_win,
+ "home": os.environ.get("USERPROFILE") if is_win else os.environ.get("HOME"),
+ "sd_cwd": os.getcwd(),
+ "all_custom_tags": all_custom_tags,
+ "extra_paths": extra_paths,
+ "enable_access_control": enable_access_control,
+ "launch_mode": kwargs.get("launch_mode", "sd"),
+ }
+
+ class DeleteFilesReq(BaseModel):
+ file_paths: List[str]
+
+ @app.post(
+ pre + "/delete_files",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ async def delete_files(req: DeleteFilesReq):
+ conn = DataBase.get_conn()
+ for path in req.file_paths:
+ check_path_trust(path)
+ try:
+ if os.path.isdir(path):
+ if len(os.listdir(path)):
+ error_msg = (
+ "When a folder is not empty, it is not allowed to be deleted."
+ if locale == "en"
+ else "文件夹不为空时不允许删除。"
+ )
+ raise HTTPException(400, detail=error_msg)
+ shutil.rmtree(path)
+ else:
+ os.remove(path)
+ txt_path = get_img_geninfo_txt_path(path)
+ if txt_path:
+ os.remove(txt_path)
+ img = DbImg.get(conn, os.path.normpath(path))
+ if img:
+ logger.info("delete file: %s", path)
+ ImageTag.remove(conn, img.id)
+ DbImg.remove(conn, img.id)
+ except OSError as e:
+ # 处理删除失败的情况
+ logger.error("delete failed")
+ error_msg = (
+ f"Error deleting file {path}: {e}"
+ if locale == "en"
+ else f"删除文件 {path} 时出错：{e}"
+ )
+ raise HTTPException(400, detail=error_msg)
+
+ class CreateFoldersReq(BaseModel):
+ dest_folder: str
+
+ @app.post(
+ pre + "/mkdirs",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ async def create_folders(req: CreateFoldersReq):
+ if enable_access_control:
+ if not is_path_under_parents(req.dest_folder):
+ raise HTTPException(status_code=403)
+ os.makedirs(req.dest_folder, exist_ok=True)
+
+ class MoveFilesReq(BaseModel):
+ file_paths: List[str]
+ dest: str
+ create_dest_folder: Optional[bool] = False
+
+ @app.post(
+ pre + "/copy_files",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ async def copy_files(req: MoveFilesReq):
+ for path in req.file_paths:
+ check_path_trust(path)
+ try:
+ shutil.copy(path, req.dest)
+ txt_path = get_img_geninfo_txt_path(path)
+ if txt_path:
+ shutil.copy(txt_path, req.dest)
+ except OSError as e:
+ error_msg = (
+ f"Error copying file {path} to {req.dest}: {e}"
+ if locale == "en"
+ else f"复制文件 {path} 到 {req.dest} 时出错：{e}"
+ )
+ raise HTTPException(400, detail=error_msg)
+
+ @app.post(
+ pre + "/move_files",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ async def move_files(req: MoveFilesReq):
+ if req.create_dest_folder:
+ os.makedirs(req.dest, exist_ok=True)
+ elif not os.path.isdir(req.dest):
+ error_msg = (
+ f"Destination folder {req.dest} does not exist."
+ if locale == "en"
+ else f"目标文件夹 {req.dest} 不存在。"
+ )
+ raise HTTPException(400, detail=error_msg)
+
+ conn = DataBase.get_conn()
+ for path in req.file_paths:
+ check_path_trust(path)
+ try:
+ ret_path = shutil.move(path, req.dest)
+ txt_path = get_img_geninfo_txt_path(path)
+ if txt_path:
+ shutil.move(txt_path, req.dest)
+ img = DbImg.get(conn, os.path.normpath(path))
+ if img:
+ img.update_path(conn, ret_path)
+ conn.commit()
+ except OSError as e:
+ error_msg = (
+ f"Error moving file {path} to {req.dest}: {e}"
+ if locale == "en"
+ else f"移动文件 {path} 到 {req.dest} 时出错：{e}"
+ )
+ raise HTTPException(400, detail=error_msg)
+
+ @app.get(pre + "/files", dependencies=[Depends(verify_secret)])
+ async def get_target_floder_files(folder_path: str):
+ files: List[FileInfoDict] = []
+ try:
+ if is_win and folder_path == "/":
+ for item in get_windows_drives():
+ files.append(
+ {"type": "dir", "size": "-", "name": item, "fullpath": item}
+ )
+ else:
+ if not os.path.exists(folder_path):
+ return {"files": []}
+ folder_path = to_abs_path(folder_path)
+ check_path_trust(folder_path)
+ folder_listing: List[os.DirEntry] = os.scandir(folder_path)
+ is_under_scanned_path = is_path_under_parents(folder_path)
+ for item in folder_listing:
+ if not os.path.exists(item.path):
+ continue
+ fullpath = os.path.normpath(item.path)
+ name = os.path.basename(item.path)
+ date = get_formatted_date(item.stat().st_mtime)
+ created_time = get_formatted_date(item.stat().st_ctime)
+ if item.is_file():
+ bytes = item.stat().st_size
+ size = human_readable_size(bytes)
+ files.append(
+ {
+ "type": "file",
+ "date": date,
+ "size": size,
+ "name": name,
+ "bytes": bytes,
+ "created_time": created_time,
+ "fullpath": fullpath,
+ "is_under_scanned_path": is_under_scanned_path,
+ }
+ )
+ elif item.is_dir():
+ files.append(
+ {
+ "type": "dir",
+ "date": date,
+ "created_time": created_time,
+ "size": "-",
+ "name": name,
+ "is_under_scanned_path": is_under_scanned_path,
+ "fullpath": fullpath,
+ }
+ )
+ except Exception as e:
+ # logger.error(e)
+ raise HTTPException(status_code=400, detail=str(e))
+
+ return {"files": filter_allowed_files(files)}
+
+ @app.get(pre + "/image-thumbnail", dependencies=[Depends(verify_secret)])
+ async def thumbnail(path: str, t: str, size: str = "256x256"):
+ check_path_trust(path)
+ if not temp_path:
+ return
+ # 生成缓存文件的路径
+ hash_dir = hashlib.md5((path + t).encode("utf-8")).hexdigest()
+ hash = hash_dir + size
+ cache_dir = os.path.join(temp_path, "iib_cache", hash_dir)
+ cache_path = os.path.join(cache_dir, f"{size}.webp")
+
+ # 如果缓存文件存在，则直接返回该文件
+ if os.path.exists(cache_path):
+ return FileResponse(
+ cache_path,
+ media_type="image/webp",
+ headers={"Cache-Control": "max-age=31536000", "ETag": hash},
+ )
+
+ # 如果缓存文件不存在，则生成缩略图并保存
+ with Image.open(path) as img:
+ w, h = size.split("x")
+ img.thumbnail((int(w), int(h)))
+ os.makedirs(cache_dir, exist_ok=True)
+ img.save(cache_path, "webp")
+
+ # 返回缓存文件
+ return FileResponse(
+ cache_path,
+ media_type="image/webp",
+ headers={"Cache-Control": "max-age=31536000", "ETag": hash},
+ )
+
+ @app.get(pre + "/file", dependencies=[Depends(verify_secret)])
+ async def get_file(path: str, t: str, disposition: Optional[str] = None):
+ filename = path
+ import mimetypes
+
+ check_path_trust(path)
+ if not os.path.exists(filename):
+ raise HTTPException(status_code=404)
+ if not os.path.isfile(filename):
+ raise HTTPException(status_code=400, detail=f"{filename} is not a file")
+ # 根据文件后缀名获取媒体类型
+ media_type, _ = mimetypes.guess_type(filename)
+ headers = {}
+ if disposition:
+ encoded_filename = urllib.parse.quote(disposition.encode('utf-8'))
+ headers['Content-Disposition'] = f"attachment; filename*=UTF-8''{encoded_filename}"
+
+ if is_path_under_parents(filename) and is_valid_image_path(
+ filename
+ ): # 认为永远不变,不要协商缓存了试试
+ headers[
+ "Cache-Control"
+ ] = "public, max-age=31536000" # 针对同样名字文件但实际上不同内容的文件要求必须传入创建时间来避免浏览器缓存
+ headers["Expires"] = (datetime.now() + timedelta(days=365)).strftime(
+ "%a, %d %b %Y %H:%M:%S GMT"
+ )
+
+ return FileResponse(
+ filename,
+ media_type=media_type,
+ headers=headers,
+ )
+
+ @app.post(pre + "/send_img_path", dependencies=[Depends(verify_secret)])
+ async def api_set_send_img_path(path: str):
+ send_img_path["value"] = path
+
+ # 等待图片信息生成完成
+ @app.get(pre + "/gen_info_completed", dependencies=[Depends(verify_secret)])
+ async def api_set_send_img_path():
+ for _ in range(30): # timeout 3s
+ if send_img_path["value"] == "": # 等待setup里面生成完成
+ return True
+ v = send_img_path["value"]
+ # is_dev and logger.info("gen_info_completed %s %s", _, v)
+ await asyncio.sleep(0.1)
+ return send_img_path["value"] == ""
+
+ @app.get(pre + "/image_geninfo", dependencies=[Depends(verify_secret)])
+ async def image_geninfo(path: str):
+ with Image.open(path) as img:
+ if is_img_created_by_comfyui(img):
+ if is_img_created_by_comfyui_with_webui_gen_info(img):
+ return read_sd_webui_gen_info_from_image(img, path)
+ else:
+ try:
+ params = get_comfyui_exif_data(img)
+ return comfyui_exif_data_to_str(params)
+ except:
+ logger.error('parse comfyui image failed. prompt:')
+ logger.error(img.info.get('prompt'))
+ return ''
+ else:
+ return read_sd_webui_gen_info_from_image(img, path)
+
+ class CheckPathExistsReq(BaseModel):
+ paths: List[str]
+
+ @app.post(pre + "/check_path_exists", dependencies=[Depends(verify_secret)])
+ async def check_path_exists(req: CheckPathExistsReq):
+ update_all_scanned_paths()
+ res = {}
+ for path in req.paths:
+ res[path] = os.path.exists(path) and is_path_trusted(path)
+ return res
+
+ @app.get(pre)
+ def index_bd():
+ return FileResponse(index_html_path)
+
+ class PathsReq(BaseModel):
+ paths: List[str]
+
+ class OpenFolderReq(BaseModel):
+ path: str
+
+ @app.post(
+ pre + "/open_folder",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ def open_folder_using_explore(req: OpenFolderReq):
+ if not is_path_trusted(req.path):
+ raise HTTPException(status_code=403)
+ open_folder(*os.path.split(req.path))
+
+ @app.post(pre + "/shutdown")
+ async def shutdown_app():
+ # This API endpoint is mainly used as a sidecar in Tauri applications to shut down the application
+ if not kwargs.get("enable_shutdown"):
+ raise HTTPException(status_code=403, detail="Shutdown is disabled.")
+ os.kill(os.getpid(), 9)
+ return {"message": "Application is shutting down."}
+
+ @app.post(
+ pre + "/zip",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ def zip_files(req: PathsReq):
+ now = datetime.now()
+ timestamp = now.strftime("%Y-%m-%d-%H-%M-%S")
+ zip_temp_dir = os.path.join(cwd, "zip_temp")
+ os.makedirs(zip_temp_dir, exist_ok=True)
+ file_path = os.path.join(zip_temp_dir, f"iib_batch_download_{timestamp}.zip")
+ create_zip_file(req.paths, file_path)
+ return FileResponse(file_path, media_type="application/zip")
+
+ db_pre = pre + "/db"
+
+ @app.get(db_pre + "/basic_info", dependencies=[Depends(verify_secret)])
+ async def get_db_basic_info():
+ conn = DataBase.get_conn()
+ img_count = DbImg.count(conn)
+ tags = Tag.get_all(conn)
+ expired_dirs = Floder.get_expired_dirs(conn)
+ return {
+ "img_count": img_count,
+ "tags": tags,
+ "expired": len(expired_dirs) != 0,
+ "expired_dirs": expired_dirs,
+ }
+
+ @app.get(db_pre + "/expired_dirs", dependencies=[Depends(verify_secret)])
+ async def get_db_expired():
+ conn = DataBase.get_conn()
+ expired_dirs = Floder.get_expired_dirs(conn)
+ return {
+ "expired": len(expired_dirs) != 0,
+ "expired_dirs": expired_dirs,
+ }
+
+ @app.post(
+ db_pre + "/update_image_data",
+ dependencies=[Depends(verify_secret)],
+ )
+ async def update_image_db_data():
+ try:
+ DataBase._initing = True
+ conn = DataBase.get_conn()
+ img_count = DbImg.count(conn)
+ update_extra_paths(conn)
+ dirs = (
+ get_img_search_dirs() if img_count == 0 else Floder.get_expired_dirs(conn)
+ ) + mem["extra_paths"]
+
+ update_image_data(dirs)
+ finally:
+ DataBase._initing = False
+
+ class MatchImagesByTagsReq(BaseModel):
+ and_tags: List[int]
+ or_tags: List[int]
+ not_tags: List[int]
+ cursor: str
+ size: Optional[int] = 200
+
+ @app.post(db_pre + "/match_images_by_tags", dependencies=[Depends(verify_secret)])
+ async def match_image_by_tags(req: MatchImagesByTagsReq):
+ conn = DataBase.get_conn()
+ imgs, next_cursor = ImageTag.get_images_by_tags(
+ conn=conn,
+ tag_dict={"and": req.and_tags, "or": req.or_tags, "not": req.not_tags},
+ cursor=req.cursor,
+ limit=req.size
+ )
+ return {
+ "files": filter_allowed_files([x.to_file_info() for x in imgs]),
+ "cursor": next_cursor
+ }
+
+ @app.get(db_pre + "/img_selected_custom_tag", dependencies=[Depends(verify_secret)])
+ async def get_img_selected_custom_tag(path: str):
+ path = os.path.normpath(path)
+ if not is_valid_image_path(path):
+ return []
+ conn = DataBase.get_conn()
+ update_extra_paths(conn)
+ if not is_path_under_parents(path):
+ return []
+ img = DbImg.get(conn, path)
+ if not img:
+ if DbImg.count(conn) == 0:
+ return []
+ update_image_data([os.path.dirname(path)])
+ img = DbImg.get(conn, path)
+ assert img
+ # tags = Tag.get_all_custom_tag()
+ return ImageTag.get_tags_for_image(conn, img.id, type="custom")
+
+ @app.post(db_pre + "/get_image_tags", dependencies=[Depends(verify_secret)])
+ async def get_img_tags(req: PathsReq):
+ conn = DataBase.get_conn()
+ return ImageTag.batch_get_tags_by_path(conn, req.paths)
+
+ class ToggleCustomTagToImgReq(BaseModel):
+ img_path: str
+ tag_id: int
+
+ @app.post(
+ db_pre + "/toggle_custom_tag_to_img",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ async def toggle_custom_tag_to_img(req: ToggleCustomTagToImgReq):
+ conn = DataBase.get_conn()
+ path = os.path.normpath(req.img_path)
+ update_extra_paths(conn)
+ if not is_path_under_parents(path):
+ raise HTTPException(
+ 400,
+ '当前文件不在搜索路径内，你可以将它添加到扫描路径再尝试。在右上角的"更多"里面'
+ if locale == "zh"
+ else 'The current file is not within the scan path. You can add it to the scan path and try again. In the top right corner, click on "More".',
+ )
+ img = DbImg.get(conn, path)
+ if not img:
+ if DbImg.count(conn):
+ update_image_data([os.path.dirname(path)])
+ img = DbImg.get(conn, path)
+ else:
+ raise HTTPException(
+ 400,
+ "你需要先通过图像搜索页生成索引"
+ if locale == "zh"
+ else "You need to generate an index through the image search page first.",
+ )
+ tags = ImageTag.get_tags_for_image(
+ conn=conn, image_id=img.id, type="custom", tag_id=req.tag_id
+ )
+ is_remove = len(tags)
+ if is_remove:
+ ImageTag.remove(conn, img.id, tags[0].id)
+ else:
+ ImageTag(img.id, req.tag_id).save(conn)
+ conn.commit()
+ return {"is_remove": is_remove}
+
+ class AddCustomTagReq(BaseModel):
+ tag_name: str
+
+ @app.post(
+ db_pre + "/add_custom_tag",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ async def add_custom_tag(req: AddCustomTagReq):
+ conn = DataBase.get_conn()
+ tag = Tag.get_or_create(conn, name=req.tag_name, type="custom")
+ conn.commit()
+ return tag
+
+ class RemoveCustomTagReq(BaseModel):
+ tag_id: str
+
+ @app.post(
+ db_pre + "/remove_custom_tag",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ async def remove_custom_tag(req: RemoveCustomTagReq):
+ conn = DataBase.get_conn()
+ ImageTag.remove(conn, tag_id=req.tag_id)
+ Tag.remove(conn, req.tag_id)
+
+ class RemoveCustomTagFromReq(BaseModel):
+ img_id: int
+ tag_id: str
+
+ @app.post(
+ db_pre + "/remove_custom_tag_from_img",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ async def remove_custom_tag_from_img(req: RemoveCustomTagFromReq):
+ conn = DataBase.get_conn()
+ ImageTag.remove(conn, image_id=req.img_id, tag_id=req.tag_id)
+
+ @app.get(db_pre + "/search_by_substr", dependencies=[Depends(verify_secret)])
+ async def search_by_substr(substr: str, cursor: str = '', size = 200):
+ conn = DataBase.get_conn()
+ imgs, next_cursor = DbImg.find_by_substring(conn=conn, substring=substr, cursor=cursor, limit=size)
+ return {
+ "files": filter_allowed_files([x.to_file_info() for x in imgs]),
+ "cursor": next_cursor
+ }
+
+
+ class ExtraPathModel(BaseModel):
+ path: str
+ type: Optional[ExtraPathType]
+
+ @app.post(
+ f"{db_pre}/extra_paths",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ async def create_extra_path(extra_path: ExtraPathModel):
+ if enable_access_control:
+ if not is_path_under_parents(extra_path.path):
+ raise HTTPException(status_code=403)
+ conn = DataBase.get_conn()
+ path = ExtraPath(extra_path.path, extra_path.type)
+ try:
+ path.save(conn)
+ finally:
+ conn.commit()
+
+ @app.get(
+ f"{db_pre}/extra_paths",
+ dependencies=[Depends(verify_secret)],
+ )
+ async def read_extra_paths():
+ conn = DataBase.get_conn()
+ return ExtraPath.get_extra_paths(conn)
+
+ @app.delete(
+ f"{db_pre}/extra_paths",
+ dependencies=[Depends(verify_secret), Depends(write_permission_required)],
+ )
+ async def delete_extra_path(extra_path: ExtraPathModel):
+ path = to_abs_path(extra_path.path)
+ conn = DataBase.get_conn()
+ ExtraPath.remove(conn, path, extra_path.type, img_search_dirs=get_img_search_dirs())
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/db/datamodel.py b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/db/datamodel.py
new file mode 100644
index 0000000000000000000000000000000000000000..361dcbd24103dcacebcde2ea7ac8acef26a7c892
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/db/datamodel.py
@@ -0,0 +1,621 @@
+from sqlite3 import Connection, connect
+from enum import Enum
+from typing import Dict, List, Optional, TypedDict
+from scripts.iib.tool import (
+ cwd,
+ get_modified_date,
+ human_readable_size,
+ tags_translate,
+ is_dev,
+ find,
+ unique_by
+)
+from contextlib import closing
+import os
+import threading
+
+class FileInfoDict(TypedDict):
+ type: str
+ date: float
+ size: int
+ name: str
+ bytes: bytes
+ created_time: float
+ fullpath: str
+
+class Cursor:
+ def __init__(self, has_next = True, next = ''):
+ self.has_next = has_next
+ self.next = next
+
+
+class DataBase:
+ local = threading.local()
+
+ _initing = False
+
+ num = 0
+
+ path = "iib.db"
+
+ @classmethod
+ def get_conn(clz) -> Connection:
+ # for : sqlite3.ProgrammingError: SQLite objects created in a thread can only be used in that same thread
+ if hasattr(clz.local, "conn"):
+ return clz.local.conn
+ else:
+ conn = clz.init()
+ clz.local.conn = conn
+ return conn
+
+ @classmethod
+ def init(clz):
+ # 创建连接并打开数据库
+ conn = connect(clz.path if os.path.isabs(clz.path) else os.path.join(cwd, clz.path))
+ try:
+ Floder.create_table(conn)
+ ImageTag.create_table(conn)
+ Tag.create_table(conn)
+ Image.create_table(conn)
+ ExtraPath.create_table(conn)
+ finally:
+ conn.commit()
+ clz.num += 1
+ if is_dev:
+ print(f"当前连接数{clz.num}")
+ return conn
+
+
+class Image:
+ def __init__(self, path, exif=None, size=0, date="", id=None):
+ self.path = path
+ self.exif = exif
+ self.id = id
+ self.size = size
+ self.date = date
+
+ def to_file_info(self) -> FileInfoDict:
+ return {
+ "type": "file",
+ "id": self.id,
+ "date": self.date,
+ "created_date": self.date,
+ "size": human_readable_size(self.size),
+ "is_under_scanned_path": True,
+ "bytes": self.size,
+ "name": os.path.basename(self.path),
+ "fullpath": self.path,
+ }
+
+ def save(self, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ "INSERT OR REPLACE INTO image (path, exif, size, date) VALUES (?, ?, ?, ?)",
+ (self.path, self.exif, self.size, self.date),
+ )
+ self.id = cur.lastrowid
+
+ def update_path(self, conn: Connection, new_path: str):
+ self.path = os.path.normpath(new_path)
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ "UPDATE image SET path = ? WHERE id = ?",
+ (self.path, self.id)
+ )
+
+ @classmethod
+ def get(cls, conn: Connection, id_or_path):
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ "SELECT * FROM image WHERE id = ? OR path = ?", (id_or_path, id_or_path)
+ )
+ row = cur.fetchone()
+ if row is None:
+ return None
+ else:
+ return cls.from_row(row)
+
+ @classmethod
+ def get_by_ids(cls, conn: Connection, ids: List[int]) -> List["Image"]:
+ if not ids:
+ return []
+
+ query = """
+ SELECT * FROM image
+ WHERE id IN ({})
+ """.format(
+ ",".join("?" * len(ids))
+ )
+
+ with closing(conn.cursor()) as cur:
+ cur.execute(query, ids)
+ rows = cur.fetchall()
+
+ images = []
+ for row in rows:
+ images.append(cls.from_row(row))
+ return images
+
+ @classmethod
+ def create_table(cls, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ """CREATE TABLE IF NOT EXISTS image (
+ id INTEGER PRIMARY KEY AUTOINCREMENT,
+ path TEXT UNIQUE,
+ exif TEXT,
+ size INTEGER,
+ date TEXT
+ )"""
+ )
+ cur.execute("CREATE INDEX IF NOT EXISTS image_idx_path ON image(path)")
+
+ @classmethod
+ def count(cls, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute("SELECT COUNT(*) FROM image")
+ count = cur.fetchone()[0]
+ return count
+
+ @classmethod
+ def from_row(cls, row: tuple):
+ image = cls(path=row[1], exif=row[2], size=row[3], date=row[4])
+ image.id = row[0]
+ return image
+
+ @classmethod
+ def remove(cls, conn: Connection, image_id: int) -> None:
+ with closing(conn.cursor()) as cur:
+ cur.execute("DELETE FROM image WHERE id = ?", (image_id,))
+ conn.commit()
+
+ @classmethod
+ def safe_batch_remove(cls, conn: Connection, image_ids: List[int]) -> None:
+ if not(image_ids):
+ return
+ with closing(conn.cursor()) as cur:
+ try:
+ placeholders = ",".join("?" * len(image_ids))
+ cur.execute(f"DELETE FROM image_tag WHERE image_id IN ({placeholders})", image_ids)
+ cur.execute(f"DELETE FROM image WHERE id IN ({placeholders})", image_ids)
+ except BaseException as e:
+ print(e)
+ finally:
+ conn.commit()
+
+ @classmethod
+ def find_by_substring(cls, conn: Connection, substring: str, limit: int = 500, cursor = '') -> tuple[List["Image"], Cursor]:
+ api_cur = Cursor()
+ with closing(conn.cursor()) as cur:
+ if cursor:
+ sql = f"SELECT * FROM image WHERE (path LIKE ? OR exif LIKE ?) AND (date < ?) ORDER BY date DESC LIMIT ?"
+ cur.execute(sql, (f"%{substring}%", f"%{substring}%", cursor, limit))
+ else:
+ sql = "SELECT * FROM image WHERE path LIKE ? OR exif LIKE ? ORDER BY date DESC LIMIT ?"
+ cur.execute(sql, (f"%{substring}%", f"%{substring}%", limit))
+ rows = cur.fetchall()
+
+ api_cur.has_next = len(rows) >= limit
+ images = []
+ deleted_ids = []
+ for row in rows:
+ img = cls.from_row(row)
+ if os.path.exists(img.path):
+ images.append(img)
+ else:
+ deleted_ids.append(img.id)
+ cls.safe_batch_remove(conn, deleted_ids)
+ if images:
+ api_cur.next = str(images[-1].date)
+ return images, api_cur
+
+
+class Tag:
+ def __init__(self, name: str, score: int, type: str, count=0):
+ self.name = name
+ self.score = score
+ self.type = type
+ self.count = count
+ self.id = None
+ self.display_name = tags_translate.get(name)
+
+ def save(self, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ "INSERT OR REPLACE INTO tag (id, name, score, type, count) VALUES (?, ?, ?, ?, ?)",
+ (self.id, self.name, self.score, self.type, self.count),
+ )
+ self.id = cur.lastrowid
+
+ @classmethod
+ def remove(cls, conn, tag_id):
+ with closing(conn.cursor()) as cur:
+ cur.execute("DELETE FROM tag WHERE id = ?", (tag_id,))
+ conn.commit()
+
+ @classmethod
+ def get(cls, conn: Connection, id):
+ with closing(conn.cursor()) as cur:
+ cur.execute("SELECT * FROM tag WHERE id = ?", (id,))
+ row = cur.fetchone()
+ if row is None:
+ return None
+ else:
+ return cls.from_row(row)
+
+ @classmethod
+ def get_all_custom_tag(cls, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute("SELECT * FROM tag where type = 'custom'")
+ rows = cur.fetchall()
+ tags: list[Tag] = []
+ for row in rows:
+ tags.append(cls.from_row(row))
+ return tags
+
+ @classmethod
+ def get_all(cls, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute("SELECT * FROM tag")
+ rows = cur.fetchall()
+ tags: list[Tag] = []
+ for row in rows:
+ tags.append(cls.from_row(row))
+ return tags
+
+ @classmethod
+ def get_or_create(cls, conn: Connection, name: str, type: str):
+ assert name and type
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ "SELECT tag.* FROM tag WHERE name = ? and type = ?", (name, type)
+ )
+ row = cur.fetchone()
+ if row is None:
+ tag = cls(name=name, score=0, type=type)
+ tag.save(conn)
+ return tag
+ else:
+ return cls.from_row(row)
+
+ @classmethod
+ def from_row(cls, row: tuple):
+ tag = cls(name=row[1], score=row[2], type=row[3], count=row[4])
+ tag.id = row[0]
+ return tag
+
+ @classmethod
+ def create_table(cls, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ """CREATE TABLE IF NOT EXISTS tag (
+ id INTEGER PRIMARY KEY AUTOINCREMENT,
+ name TEXT,
+ score INTEGER,
+ type TEXT,
+ count INTEGER,
+ UNIQUE(name, type) ON CONFLICT REPLACE
+ );
+ """
+ )
+ cur.execute("CREATE INDEX IF NOT EXISTS tag_idx_name ON tag(name)")
+ cur.execute(
+ """INSERT OR IGNORE INTO tag(name, score, type, count)
+ VALUES ("like", 0, "custom", 0);
+ """
+ )
+
+
+
+
+class ImageTag:
+ def __init__(self, image_id: int, tag_id: int):
+ assert tag_id and image_id
+ self.image_id = image_id
+ self.tag_id = tag_id
+
+ def save(self, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ "INSERT INTO image_tag (image_id, tag_id) VALUES (?, ?)",
+ (self.image_id, self.tag_id),
+ )
+
+ @classmethod
+ def get_tags_for_image(
+ cls,
+ conn: Connection,
+ image_id: int,
+ tag_id: Optional[int] = None,
+ type: Optional[str] = None,
+ ):
+ with closing(conn.cursor()) as cur:
+ query = "SELECT tag.* FROM tag INNER JOIN image_tag ON tag.id = image_tag.tag_id WHERE image_tag.image_id = ?"
+ params = [image_id]
+ if tag_id:
+ query += " AND image_tag.tag_id = ?"
+ params.append(tag_id)
+ if type:
+ query += " AND tag.type = ?"
+ params.append(type)
+ cur.execute(query, tuple(params))
+ rows = cur.fetchall()
+ return [Tag.from_row(x) for x in rows]
+
+ @classmethod
+ def get_images_for_tag(cls, conn: Connection, tag_id):
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ "SELECT image.* FROM image INNER JOIN image_tag ON image.id = image_tag.image_id WHERE image_tag.tag_id = ?",
+ (tag_id,),
+ )
+ rows = cur.fetchall()
+ images = []
+ for row in rows:
+ images.append(Image.from_row(row))
+ return images
+
+ @classmethod
+ def create_table(cls, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ """CREATE TABLE IF NOT EXISTS image_tag (
+ image_id INTEGER,
+ tag_id INTEGER,
+ FOREIGN KEY (image_id) REFERENCES image(id),
+ FOREIGN KEY (tag_id) REFERENCES tag(id),
+ PRIMARY KEY (image_id, tag_id)
+ )"""
+ )
+
+ @classmethod
+ def get_images_by_tags(
+ cls, conn: Connection, tag_dict: Dict[str, List[int]], limit: int = 500, cursor = ''
+ ) -> tuple[List[Image], Cursor]:
+ query = """
+ SELECT image.id, image.path, image.size,image.date
+ FROM image
+ INNER JOIN image_tag ON image.id = image_tag.image_id
+ """
+
+ where_clauses = []
+ params = []
+
+ for operator, tag_ids in tag_dict.items():
+ if operator == "and":
+ where_clauses.append(
+ "tag_id IN ({})".format(",".join("?" * len(tag_ids)))
+ )
+ params.extend(tag_ids)
+ elif operator == "not" and tag_dict["not"]:
+ where_clauses.append(
+ """(image_id NOT IN (
+ SELECT image_id
+ FROM image_tag
+ WHERE tag_id IN ({})
+))""".format(
+ ",".join("?" * len(tag_ids))
+ )
+ )
+ params.extend(tag_ids)
+ elif operator == "or" and tag_dict["or"]:
+ where_clauses.append(
+ """(image_id IN (
+ SELECT image_id
+ FROM image_tag
+ WHERE tag_id IN ({})
+ GROUP BY image_id
+ HAVING COUNT(DISTINCT tag_id) >= 1
+)
+)""".format(",".join("?" * len(tag_ids)))
+ )
+ params.extend(tag_ids)
+ if cursor:
+ where_clauses.append("(image.date < ?)")
+ params.append(cursor)
+
+ if where_clauses:
+ query += " WHERE " + " AND ".join(where_clauses)
+
+ query += " GROUP BY image.id"
+
+ if "and" in tag_dict:
+ query += " HAVING COUNT(DISTINCT tag_id) = ?"
+ params.append(len(tag_dict["and"]))
+
+ query += " ORDER BY date DESC LIMIT ?"
+ params.append(limit)
+ api_cur = Cursor()
+ with closing(conn.cursor()) as cur:
+ cur.execute(query, params)
+ rows = cur.fetchall()
+ images = []
+ deleted_ids = []
+ for row in rows:
+ img = Image(id=row[0], path=row[1], size=row[2], date=row[3])
+ if os.path.exists(img.path):
+ images.append(img)
+ else:
+ deleted_ids.append(img.id)
+ Image.safe_batch_remove(conn, deleted_ids)
+ api_cur.has_next = len(rows) >= limit
+ if images:
+ api_cur.next = str(images[-1].date)
+ return images, api_cur
+
+ @classmethod
+ def batch_get_tags_by_path(cls, conn: Connection, paths: List[str], type = "custom") -> Dict[str, List[Tag]]:
+ if not paths:
+ return {}
+ tag_dict = {}
+ with closing(conn.cursor()) as cur:
+ placeholders = ",".join("?" * len(paths))
+ query = f"""
+ SELECT image.path, tag.* FROM image_tag
+ INNER JOIN image ON image_tag.image_id = image.id
+ INNER JOIN tag ON image_tag.tag_id = tag.id
+ WHERE tag.type = '{type}' AND image.path IN ({placeholders})
+ """
+ cur.execute(query, paths)
+ rows = cur.fetchall()
+ for row in rows:
+ path = row[0]
+ tag = Tag.from_row(row[1:])
+ if path in tag_dict:
+ tag_dict[path].append(tag)
+ else:
+ tag_dict[path] = [tag]
+ return tag_dict
+
+ @classmethod
+ def remove(
+ cls,
+ conn: Connection,
+ image_id: Optional[int] = None,
+ tag_id: Optional[int] = None,
+ ) -> None:
+ assert image_id or tag_id
+ with closing(conn.cursor()) as cur:
+ if tag_id and image_id:
+ cur.execute(
+ "DELETE FROM image_tag WHERE image_id = ? and tag_id = ?",
+ (image_id, tag_id),
+ )
+ elif tag_id:
+ cur.execute("DELETE FROM image_tag WHERE tag_id = ?", (tag_id,))
+ else:
+ cur.execute("DELETE FROM image_tag WHERE image_id = ?", (image_id,))
+ conn.commit()
+
+
+class Floder:
+ def __init__(self, id: int, path: str, modified_date: str):
+ self.id = id
+ self.path = path
+ self.modified_date = modified_date
+
+ @classmethod
+ def create_table(cls, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ """CREATE TABLE IF NOT EXISTS folders
+ (id INTEGER PRIMARY KEY AUTOINCREMENT,
+ path TEXT,
+ modified_date TEXT)"""
+ )
+ cur.execute("CREATE INDEX IF NOT EXISTS folders_idx_path ON folders(path)")
+
+ @classmethod
+ def check_need_update(cls, conn: Connection, folder_path: str):
+ folder_path = os.path.normpath(folder_path)
+ with closing(conn.cursor()) as cur:
+ if not os.path.exists(folder_path):
+ return False
+ cur.execute("SELECT * FROM folders WHERE path=?", (folder_path,))
+ folder_record = cur.fetchone() # 如果这个文件夹没有记录，或者修改时间与数据库不同，则需要修改
+ return not folder_record or (
+ folder_record[2] != get_modified_date(folder_path)
+ )
+
+ @classmethod
+ def update_modified_date_or_create(cls, conn: Connection, folder_path: str):
+ folder_path = os.path.normpath(folder_path)
+ with closing(conn.cursor()) as cur:
+ cur.execute("SELECT * FROM folders WHERE path = ?", (folder_path,))
+ row = cur.fetchone()
+ if row:
+ cur.execute(
+ "UPDATE folders SET modified_date = ? WHERE path = ?",
+ (get_modified_date(folder_path), folder_path),
+ )
+ else:
+ cur.execute(
+ "INSERT INTO folders (path, modified_date) VALUES (?, ?)",
+ (folder_path, get_modified_date(folder_path)),
+ )
+
+ @classmethod
+ def get_expired_dirs(cls, conn: Connection):
+ dirs: List[str] = []
+ with closing(conn.cursor()) as cur:
+ cur.execute("SELECT * FROM folders")
+ result_set = cur.fetchall()
+ extra_paths = ExtraPath.get_extra_paths(conn)
+ for ep in extra_paths:
+ if not find(result_set, lambda x : x[1] == ep.path):
+ dirs.append(ep.path)
+ for row in result_set:
+ folder_path = row[1]
+ if (
+ os.path.exists(folder_path)
+ and get_modified_date(folder_path) != row[2]
+ ):
+ dirs.append(folder_path)
+ return unique_by(dirs, os.path.normpath)
+
+ @classmethod
+ def remove_folder(cls, conn: Connection, folder_path: str):
+ folder_path = os.path.normpath(folder_path)
+ with closing(conn.cursor()) as cur:
+ cur.execute("DELETE FROM folders WHERE path = ?", (folder_path,))
+
+class ExtraPathType(Enum):
+ scanned = 'scanned'
+ walk = 'walk'
+ cli_only = 'cli_access_only'
+
+class ExtraPath:
+ def __init__(self, path: str, type: Optional[ExtraPathType] = None):
+ assert type
+ self.path = os.path.normpath(path)
+ self.type = type
+
+ def save(self, conn):
+ assert self.type in [ExtraPathType.walk, ExtraPathType.scanned]
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ "INSERT INTO extra_path (path, type) VALUES (?, ?) ON CONFLICT (path) DO UPDATE SET type = ?",
+ (self.path, self.type.value, self.type.value),
+ )
+
+ @classmethod
+ def get_extra_paths(cls, conn, type: Optional[ExtraPathType] = None) -> List['ExtraPath']:
+ query = "SELECT * FROM extra_path"
+ params = ()
+ if type:
+ query += " WHERE type = ?"
+ params = (type.value,)
+ with closing(conn.cursor()) as cur:
+ cur.execute(query, params)
+ rows = cur.fetchall()
+ paths: List[ExtraPath] = []
+ for row in rows:
+ path = row[0]
+ if os.path.exists(path):
+ paths.append(ExtraPath(path, ExtraPathType(row[1])))
+ else:
+ cls.remove(conn, path)
+ return paths
+
+ @classmethod
+ def remove(cls, conn, path: str, type: Optional[ExtraPathType] = None, img_search_dirs: Optional[List[str]] = []):
+ with closing(conn.cursor()) as cur:
+ sql = "DELETE FROM extra_path WHERE path = ?"
+ path = os.path.normpath(path)
+ if type:
+ cur.execute(sql, (path,))
+ else:
+ cur.execute(sql + "AND type = ?", (path, type.value))
+ if path not in img_search_dirs:
+ Floder.remove_folder(conn, path)
+ conn.commit()
+
+ @classmethod
+ def create_table(cls, conn):
+ with closing(conn.cursor()) as cur:
+ cur.execute(
+ """CREATE TABLE IF NOT EXISTS extra_path (
+ path TEXT PRIMARY KEY,
+ type TEXT NOT NULL
+ )"""
+ )
\ No newline at end of file
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/db/update_image_data.py b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/db/update_image_data.py
new file mode 100644
index 0000000000000000000000000000000000000000..54ac559665c75c7a157111c40bdeefd1d4465504
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/db/update_image_data.py
@@ -0,0 +1,125 @@
+from typing import Dict, List
+from scripts.iib.db.datamodel import Image as DbImg, Tag, ImageTag, DataBase, Floder
+import os
+from PIL import Image
+from scripts.iib.tool import (
+ read_sd_webui_gen_info_from_image,
+ parse_generation_parameters,
+ is_valid_image_path,
+ get_modified_date,
+ is_dev,
+ get_comfyui_exif_data,
+ comfyui_exif_data_to_str,
+ is_img_created_by_comfyui,
+ is_img_created_by_comfyui_with_webui_gen_info
+)
+
+from scripts.iib.logger import logger
+
+
+# 定义一个函数来获取图片文件的EXIF数据
+def get_exif_data(file_path):
+ info = ''
+ params = None
+ try:
+ with Image.open(file_path) as img:
+ if is_img_created_by_comfyui(img):
+ if is_img_created_by_comfyui_with_webui_gen_info(img):
+ info = read_sd_webui_gen_info_from_image(img, file_path)
+ params = parse_generation_parameters(info)
+ else:
+ params = get_comfyui_exif_data(img)
+ info = comfyui_exif_data_to_str(params)
+ else:
+ info = read_sd_webui_gen_info_from_image(img, file_path)
+ params = parse_generation_parameters(info)
+ except Exception as e:
+ if is_dev:
+ logger.error("get_exif_data %s", e)
+ return params, info
+
+
+def update_image_data(search_dirs: List[str]):
+ conn = DataBase.get_conn()
+ tag_incr_count_rec: Dict[int, int] = {}
+
+ def safe_save_img_tag(img_tag: ImageTag):
+ tag_incr_count_rec[img_tag.tag_id] = (
+ tag_incr_count_rec.get(img_tag.tag_id, 0) + 1
+ )
+ img_tag.save(conn) # 原先用来处理一些意外，但是写的正确完全没问题,去掉了try catch
+
+ # 递归处理每个文件夹
+ def process_folder(folder_path: str):
+ if not Floder.check_need_update(conn, folder_path):
+ return
+ print(f"Processing folder: {folder_path}")
+ for filename in os.listdir(folder_path):
+ file_path = os.path.normpath(os.path.join(folder_path, filename))
+
+ if os.path.isdir(file_path):
+ process_folder(file_path)
+
+ elif is_valid_image_path(file_path):
+ img = DbImg.get(conn, file_path)
+ if img: # 已存在的跳过
+ if img.date == get_modified_date(img.path):
+ continue
+ else:
+ DbImg.safe_batch_remove(conn=conn, image_ids=[img.id])
+ parsed_params, info = get_exif_data(file_path)
+ img = DbImg(
+ file_path,
+ info,
+ os.path.getsize(file_path),
+ get_modified_date(file_path),
+ )
+ img.save(conn)
+
+ if not parsed_params:
+ continue
+ meta = parsed_params.get("meta", {})
+ lora = parsed_params.get("lora", [])
+ lyco = parsed_params.get("lyco", [])
+ pos = parsed_params["pos_prompt"]
+ size_tag = Tag.get_or_create(
+ conn,
+ str(meta.get("Size-1", 0)) + " * " + str(meta.get("Size-2", 0)),
+ type="size",
+ )
+ safe_save_img_tag(ImageTag(img.id, size_tag.id))
+
+ for k in [
+ "Model",
+ "Sampler",
+ "Postprocess upscale by",
+ "Postprocess upscaler",
+ ]:
+ v = meta.get(k)
+ if not v:
+ continue
+ tag = Tag.get_or_create(conn, str(v), k)
+ safe_save_img_tag(ImageTag(img.id, tag.id))
+ for i in lora:
+ tag = Tag.get_or_create(conn, i["name"], "lora")
+ safe_save_img_tag(ImageTag(img.id, tag.id))
+ for i in lyco:
+ tag = Tag.get_or_create(conn, i["name"], "lyco")
+ safe_save_img_tag(ImageTag(img.id, tag.id))
+ for k in pos:
+ tag = Tag.get_or_create(conn, k, "pos")
+ safe_save_img_tag(ImageTag(img.id, tag.id))
+ # neg暂时跳过感觉个没人会搜索这个
+
+ # 提交对数据库的更改
+ Floder.update_modified_date_or_create(conn, folder_path)
+ conn.commit()
+
+ for dir in search_dirs:
+ process_folder(dir)
+ conn.commit()
+ for tag_id in tag_incr_count_rec:
+ tag = Tag.get(conn, tag_id)
+ tag.count += tag_incr_count_rec[tag_id]
+ tag.save(conn)
+ conn.commit()
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/logger.py b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/logger.py
new file mode 100644
index 0000000000000000000000000000000000000000..3d795279695ff805e6076982e02794c5c6912581
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/logger.py
@@ -0,0 +1,20 @@
+
+from scripts.iib.tool import is_dev,cwd
+
+import logging
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+
+console_handler = logging.StreamHandler()
+console_handler.setLevel(logging.INFO)
+
+file_handler = logging.FileHandler(f"{cwd}/log.log")
+file_handler.setLevel(logging.DEBUG)
+
+formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+console_handler.setFormatter(formatter)
+file_handler.setFormatter(formatter)
+
+logger.addHandler(file_handler)
+if is_dev:
+ logger.addHandler(console_handler)
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/tool.py b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/tool.py
new file mode 100644
index 0000000000000000000000000000000000000000..6fada105b371ea6314296495238f1842871b9a92
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/scripts/iib/tool.py
@@ -0,0 +1,523 @@
+from datetime import datetime
+import os
+import platform
+import re
+import tempfile
+import imghdr
+import subprocess
+from typing import Dict, List
+import sys
+import piexif
+import piexif.helper
+import json
+import zipfile
+from PIL import Image
+
+sd_img_dirs = [
+ "outdir_txt2img_samples",
+ "outdir_img2img_samples",
+ "outdir_save",
+ "outdir_extras_samples",
+ "outdir_grids",
+ "outdir_img2img_grids",
+ "outdir_samples",
+ "outdir_txt2img_grids",
+]
+
+
+is_dev = os.getenv("APP_ENV") == "dev"
+is_nuitka = "__compiled__" in globals()
+cwd = os.getcwd() if is_nuitka else os.path.normpath(os.path.join(__file__, "../../../"))
+is_win = platform.system().lower().find("windows") != -1
+
+
+
+
+try:
+ from dotenv import load_dotenv
+
+ load_dotenv(os.path.join(cwd, ".env"))
+except Exception as e:
+ print(e)
+
+
+def get_sd_webui_conf(**kwargs):
+ try:
+ from modules.shared import opts
+
+ return opts.data
+ except:
+ pass
+ try:
+ sd_conf_path = kwargs.get("sd_webui_config")
+ with open(sd_conf_path, "r") as f:
+ obj = json.loads(f.read())
+ if kwargs.get("sd_webui_path_relative_to_config"):
+ for dir in sd_img_dirs:
+ if obj[dir] and not os.path.isabs(obj[dir]):
+ obj[dir] = os.path.normpath(
+ os.path.join(sd_conf_path, "../", obj[dir])
+ )
+ return obj
+ except:
+ pass
+ return {}
+
+def normalize_paths(paths: List[str], base = cwd):
+ """
+ Normalize a list of paths, ensuring that each path is an absolute path with no redundant components.
+
+ Args:
+ paths (List[str]): A list of paths to be normalized.
+
+ Returns:
+ List[str]: A list of normalized paths.
+ """
+ res: List[str] = []
+ for path in paths:
+ # Skip empty or blank paths
+ if not path or len(path.strip()) == 0:
+ continue
+ # If the path is already an absolute path, use it as is
+ if os.path.isabs(path):
+ abs_path = path
+ # Otherwise, make the path absolute by joining it with the current working directory
+ else:
+ abs_path = os.path.join(base, path)
+ # If the absolute path exists, add it to the result after normalizing it
+ if os.path.exists(abs_path):
+ res.append(os.path.normpath(abs_path))
+ return res
+
+def to_abs_path(path):
+ if not os.path.isabs(path):
+ path = os.path.join(os.getcwd(), path)
+ return os.path.normpath(path)
+
+
+def get_valid_img_dirs(
+ conf,
+ keys=sd_img_dirs,
+):
+ # 获取配置项
+ paths = [conf.get(key) for key in keys]
+
+ # 判断路径是否有效并转为绝对路径
+ abs_paths = []
+ for path in paths:
+ if not path or len(path.strip()) == 0:
+ continue
+ if os.path.isabs(path): # 已经是绝对路径
+ abs_path = path
+ else: # 转为绝对路径
+ abs_path = os.path.join(os.getcwd(), path)
+ if os.path.exists(abs_path): # 判断路径是否存在
+ abs_paths.append(os.path.normpath(abs_path))
+
+ return abs_paths
+
+
+def human_readable_size(size_bytes):
+ """
+ Converts bytes to a human-readable format.
+ """
+ # define the size units
+ units = ("B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")
+ # calculate the logarithm of the input value with base 1024
+ size = int(size_bytes)
+ if size == 0:
+ return "0B"
+ i = 0
+ while size >= 1024 and i < len(units) - 1:
+ size /= 1024
+ i += 1
+ # round the result to two decimal points and return as a string
+ return "{:.2f} {}".format(size, units[i])
+
+
+def get_windows_drives():
+ drives = []
+ for drive in range(ord("A"), ord("Z") + 1):
+ drive_name = chr(drive) + ":/"
+ if os.path.exists(drive_name):
+ drives.append(drive_name)
+ return drives
+
+
+pattern = re.compile(r"(\d+\.?\d*)([KMGT]?B)", re.IGNORECASE)
+
+
+def convert_to_bytes(file_size_str):
+ match = re.match(pattern, file_size_str)
+ if match:
+ size_str, unit_str = match.groups()
+ size = float(size_str)
+ unit = unit_str.upper()
+ if unit == "KB":
+ size *= 1024
+ elif unit == "MB":
+ size *= 1024**2
+ elif unit == "GB":
+ size *= 1024**3
+ elif unit == "TB":
+ size *= 1024**4
+ return int(size)
+ else:
+ raise ValueError(f"Invalid file size string '{file_size_str}'")
+
+
+def is_valid_image_path(path):
+ """
+ 判断给定的路径是否是图像文件
+ """
+ abs_path = os.path.abspath(path) # 转为绝对路径
+ if not os.path.exists(abs_path): # 判断路径是否存在
+ return False
+ if not os.path.isfile(abs_path): # 判断是否是文件
+ return False
+ if not imghdr.what(abs_path): # 判断是否是图像文件
+ return False
+ return True
+
+
+
+def create_zip_file(file_paths: List[str], zip_file_name: str):
+ """
+ 将文件打包成一个压缩包
+
+ Args:
+ file_paths: 文件路径的列表
+ zip_file_name: 压缩包的文件名
+
+ Returns:
+ 无返回值
+ """
+ with zipfile.ZipFile(zip_file_name, 'w', zipfile.ZIP_DEFLATED) as zip_file:
+ for file_path in file_paths:
+ if os.path.isfile(file_path):
+ zip_file.write(file_path, os.path.basename(file_path))
+ elif os.path.isdir(file_path):
+ for root, _, files in os.walk(file_path):
+ for file in files:
+ full_path = os.path.join(root, file)
+ zip_file.write(full_path, os.path.relpath(full_path, file_path))
+
+def get_temp_path():
+ """获取跨平台的临时文件目录路径"""
+ temp_path = None
+ try:
+ # 尝试获取系统环境变量中的临时文件目录路径
+ temp_path = (
+ os.environ.get("TMPDIR") or os.environ.get("TMP") or os.environ.get("TEMP")
+ )
+ except Exception as e:
+ print("获取系统环境变量临时文件目录路径失败，错误信息：", e)
+
+ # 如果系统环境变量中没有设置临时文件目录路径，则使用 Python 的 tempfile 模块创建临时文件目录
+ if not temp_path:
+ try:
+ temp_path = tempfile.gettempdir()
+ except Exception as e:
+ print("使用 Python 的 tempfile 模块创建临时文件目录失败，错误信息：", e)
+
+ # 确保临时文件目录存在
+ if not os.path.exists(temp_path):
+ try:
+ os.makedirs(temp_path)
+ except Exception as e:
+ print("创建临时文件目录失败，错误信息：", e)
+
+ return temp_path
+
+
+temp_path = get_temp_path()
+
+def get_secret_key_required():
+ try:
+ from modules.shared import cmd_opts
+ return bool(cmd_opts.gradio_auth)
+ except:
+ return False
+
+is_secret_key_required = get_secret_key_required()
+
+def get_enable_access_control():
+ ctrl = os.getenv("IIB_ACCESS_CONTROL")
+ if ctrl == "enable":
+ return True
+ if ctrl == "disable":
+ return False
+ try:
+ from modules.shared import cmd_opts
+
+ return (
+ cmd_opts.share or cmd_opts.ngrok or cmd_opts.listen or cmd_opts.server_name
+ )
+ except:
+ pass
+ return False
+
+
+enable_access_control = get_enable_access_control()
+
+
+def get_locale():
+ import locale
+
+ env_lang = os.getenv("IIB_SERVER_LANG")
+ if env_lang in ["zh", "en"]:
+ return env_lang
+ lang, _ = locale.getdefaultlocale()
+ return "zh" if lang and lang.startswith("zh") else "en"
+
+
+locale = get_locale()
+
+
+def get_formatted_date(timestamp: float) -> str:
+ return datetime.fromtimestamp(timestamp).strftime("%Y-%m-%d %H:%M:%S")
+
+
+def get_modified_date(folder_path: str):
+ return get_formatted_date(os.path.getmtime(folder_path))
+
+
+def get_created_date(folder_path: str):
+ return get_formatted_date(os.path.getctime(folder_path))
+
+
+def unique_by(seq, key_func=lambda x: x):
+ seen = set()
+ return [x for x in seq if not (key := key_func(x)) in seen and not seen.add(key)]
+
+
+def find(lst, comparator):
+ return next((item for item in lst if comparator(item)), None)
+
+
+def findIndex(lst, comparator):
+ return next((i for i, item in enumerate(lst) if comparator(item)), -1)
+
+
+def get_img_geninfo_txt_path(path: str):
+ txt_path = re.sub(r"\.\w+$", ".txt", path)
+ if os.path.exists(txt_path):
+ return txt_path
+
+def is_img_created_by_comfyui(img: Image):
+ return img.info.get('prompt') and img.info.get('workflow')
+
+def is_img_created_by_comfyui_with_webui_gen_info(img: Image):
+ return img.info.get('parameters')
+
+def get_comfyui_exif_data(img: Image):
+ prompt = img.info.get('prompt')
+ if not prompt:
+ return {}
+ meta_key = '3'
+ data: Dict[str, any] = json.loads(prompt)
+ for i in range(3, 32):
+ try:
+ i = str(i)
+ if data[i]["class_type"].startswith("KSampler"):
+ meta_key = i
+ break
+ except:
+ pass
+ meta = {}
+ KSampler_entry = data[meta_key]["inputs"]
+ meta["Sampler"] = KSampler_entry["sampler_name"]
+ meta["Model"] = data[KSampler_entry["model"][0]]["inputs"]["ckpt_name"]
+ def get_text_from_clip(idx: str) :
+ text = data[idx]["inputs"]["text"]
+ if isinstance(text, list): # type:CLIPTextEncode (NSP) mode:Wildcards
+ text = data[text[0]]["inputs"]["text"]
+ return text.strip()
+ pos_prompt = get_text_from_clip(KSampler_entry["positive"][0])
+ neg_prompt = get_text_from_clip(KSampler_entry["negative"][0])
+ pos_prompt_arr = unique_by(parse_prompt(pos_prompt)["pos_prompt"])
+ return {
+ "meta": meta,
+ "pos_prompt": pos_prompt_arr,
+ "pos_prompt_raw": pos_prompt,
+ "neg_prompt_raw" : neg_prompt
+ }
+
+def comfyui_exif_data_to_str(data):
+ res = data["pos_prompt_raw"] + "\nNegative prompt: " + data["neg_prompt_raw"] + "\n"
+ meta_arr = []
+ for k,v in data["meta"].items():
+ meta_arr.append(f'{k}: {v}')
+ return res + ", ".join(meta_arr)
+
+def read_sd_webui_gen_info_from_image(image: Image, path="") -> str:
+ """
+ Reads metadata from an image file.
+
+ Args:
+ image (PIL.Image.Image): The image object to read metadata from.
+ path (str): Optional. The path to the image file. Used to look for a .txt file with additional metadata.
+
+ Returns:
+ str: The metadata as a string.
+ """
+ items = image.info or {}
+ geninfo = items.pop("parameters", None)
+
+ if "exif" in items:
+ exif = piexif.load(items["exif"])
+ exif_comment = (exif or {}).get("Exif", {}).get(piexif.ExifIFD.UserComment, b"")
+
+ try:
+ exif_comment = piexif.helper.UserComment.load(exif_comment)
+ except ValueError:
+ exif_comment = exif_comment.decode("utf8", errors="ignore")
+
+ if exif_comment:
+ items["exif comment"] = exif_comment
+ geninfo = exif_comment
+
+ if not geninfo and path:
+ try:
+ txt_path = get_img_geninfo_txt_path(path)
+ if txt_path:
+ with open(txt_path) as f:
+ geninfo = f.read()
+ except Exception as e:
+ pass
+
+ return geninfo
+
+
+re_param_code = r'\s*([\w ]+):\s*("(?:\\"[^,]|\\"|\\|[^\"])+"|[^,]*)(?:,|$)'
+re_param = re.compile(re_param_code)
+re_imagesize = re.compile(r"^(\d+)x(\d+)$")
+re_lora_prompt = re.compile("If that doesn't work, click on FAQ for help
'};n.configure=function(l){var s,o;for(s in l)o=l[s],o!==void 0&&l.hasOwnProperty(s)&&(i[s]=o);return this},n.status=null,n.set=function(l){var s=n.isStarted();l=a(l,i.minimum,1),n.status=l===1?null:l;var o=n.render(!s),f=o.querySelector(i.barSelector),A=i.speed,g=i.easing;return o.offsetWidth,c(function(u){i.positionUsing===""&&(i.positionUsing=n.getPositioningCSS()),h(f,v(l,A,g)),l===1?(h(o,{transition:"none",opacity:1}),o.offsetWidth,setTimeout(function(){h(o,{transition:"all "+A+"ms linear",opacity:0}),setTimeout(function(){n.remove(),u()},A)},A)):setTimeout(u,A)}),this},n.isStarted=function(){return typeof n.status=="number"},n.start=function(){n.status||n.set(0);var l=function(){setTimeout(function(){n.status&&(n.trickle(),l())},i.trickleSpeed)};return i.trickle&&l(),this},n.done=function(l){return!l&&!n.status?this:n.inc(.3+.5*Math.random()).set(1)},n.inc=function(l){var s=n.status;return s?s>1?void 0:(typeof l!="number"&&(s>=0&&s<.2?l=.1:s>=.2&&s<.5?l=.04:s>=.5&&s<.8?l=.02:s>=.8&&s<.99?l=.005:l=0),s=a(s+l,0,.994),n.set(s)):n.start()},n.trickle=function(){return n.inc()},function(){var l=0,s=0;n.promise=function(o){return!o||o.state()==="resolved"?this:(s===0&&n.start(),l++,s++,o.always(function(){s--,s===0?(l=0,n.done()):n.set((l-s)/l)}),this)}}(),n.getElement=function(){var l=n.getParent();if(l){var s=Array.prototype.slice.call(l.querySelectorAll(".nprogress")).filter(function(o){return o.parentElement===l});if(s.length>0)return s[0]}return null},n.getParent=function(){if(i.parent instanceof HTMLElement)return i.parent;if(typeof i.parent=="string")return document.querySelector(i.parent)},n.render=function(l){if(n.isRendered())return n.getElement();d(document.documentElement,"nprogress-busy");var s=document.createElement("div");s.id="nprogress",s.className="nprogress",s.innerHTML=i.template;var o=s.querySelector(i.barSelector),f=l?"-100":r(n.status||0),A=n.getParent(),g;return h(o,{transition:"all 0 linear",transform:"translate3d("+f+"%,0,0)"}),i.showSpinner||(g=s.querySelector(i.spinnerSelector),g&&p(g)),A!=document.body&&d(A,"nprogress-custom-parent"),A.appendChild(s),s},n.remove=function(){n.status=null,m(document.documentElement,"nprogress-busy"),m(n.getParent(),"nprogress-custom-parent");var l=n.getElement();l&&p(l)},n.isRendered=function(){return!!n.getElement()},n.getPositioningCSS=function(){var l=document.body.style,s="WebkitTransform"in l?"Webkit":"MozTransform"in l?"Moz":"msTransform"in l?"ms":"OTransform"in l?"O":"";return s+"Perspective"in l?"translate3d":s+"Transform"in l?"translate":"margin"};function a(l,s,o){return l
+ It seems to have failed to load. You can try refreshing the page.
If that doesn't work, click on FAQ for help
+
+
+
+
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/dist/vite.svg b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/dist/vite.svg
new file mode 100644
index 0000000000000000000000000000000000000000..e7b8dfb1b2a60bd50538bec9f876511b9cac21e3
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/dist/vite.svg
@@ -0,0 +1 @@
+
\ No newline at end of file
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/env.d.ts b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/env.d.ts
new file mode 100644
index 0000000000000000000000000000000000000000..2964e97b5847c9602be9a3d21835337aa9bddcff
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/env.d.ts
@@ -0,0 +1,8 @@
+
+/// If that doesn't work, click on FAQ for help
+ It seems to have failed to load. You can try refreshing the page.
If that doesn't work, click on FAQ for help
+
+
+
+
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/index.tpl.js b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/index.tpl.js
new file mode 100644
index 0000000000000000000000000000000000000000..873875c8503292b1df523dc818fd2bee86d157ee
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/index.tpl.js
@@ -0,0 +1,156 @@
+Promise.resolve().then(async () => {
+ /**
+ * This is a file generated using `yarn build`.
+ * If you want to make changes, please modify `index.tpl.js` and run the command to generate it again.
+ */
+ const html = `__built_html__`
+ let containerSelector = '#infinite_image_browsing_container_wrapper'
+ let shouldMaximize = true
+
+ try {
+ containerSelector = __iib_root_container__
+ shouldMaximize = __iib_should_maximize__
+ } catch (e) {}
+
+ const delay = (timeout = 0) => new Promise((resolve) => setTimeout(resolve, timeout))
+ const asyncCheck = async (getter, checkSize = 100, timeout = 1000) => {
+ let target = getter()
+ let num = 0
+ while (checkSize * num < timeout && (target === undefined || target === null)) {
+ await delay(checkSize)
+ target = getter()
+ num++
+ }
+ return target
+ }
+ const getTabIdxById = (id) => {
+ const tabList = gradioApp().querySelectorAll('#tabs > .tabitem[id^=tab_]')
+ return Array.from(tabList).findIndex((v) => v.id.includes(id))
+ }
+
+ const switch2targetTab = (idx) => {
+ try {
+ gradioApp().querySelector('#tabs').querySelectorAll('button')[idx].click()
+ } catch (error) {
+ console.error(error)
+ }
+ }
+
+ /**
+ * @type {HTMLDivElement}
+ */
+ const wrap = await asyncCheck(() => gradioApp().querySelector(containerSelector), 500, Infinity)
+ wrap.childNodes.forEach((v) => wrap.removeChild(v))
+ const iframe = document.createElement('iframe')
+ iframe.srcdoc = html
+ iframe.style = `width: 100%;height:100vh`
+ wrap.appendChild(iframe)
+
+ if (shouldMaximize) {
+ onUiTabChange(() => {
+ const el = get_uiCurrentTabContent()
+ if (el?.id.includes('infinite-image-browsing')) {
+ try {
+ const iibTop = gradioApp().querySelector('#iib_top')
+ if (!iibTop) {
+ throw new Error("element '#iib_top' is not found")
+ }
+ const topRect = iibTop.getBoundingClientRect()
+ wrap.style = `
+ top:${Math.max(48, topRect.top) - 10}px;
+ position: fixed;
+ left: 10px;
+ right: 10px;
+ z-index: 100;
+ width: unset;
+ bottom: 10px;`
+ iframe.style = `width: 100%;height:100%`
+ } catch (error) {
+ console.error('Error mounting IIB. Running fallback.', error)
+ wrap.style = ''
+ iframe.style = `width: 100%;height:100vh`
+ }
+ }
+ })
+ }
+
+ const IIB_container_id = [Date.now(), Math.random()].join()
+ window.IIB_container_id = IIB_container_id
+ const imgTransferBus = new BroadcastChannel('iib-image-transfer-bus')
+ imgTransferBus.addEventListener('message', async (ev) => {
+ const data = ev.data
+ if (
+ typeof data !== 'object' ||
+ (typeof data.IIB_container_id === 'string' && data.IIB_container_id !== IIB_container_id)
+ ) {
+ return
+ }
+ console.log(`iib-message:`, data)
+ const appDoc = gradioApp()
+ switch (data.event) {
+ case 'click_hidden_button': {
+ const btn = gradioApp().querySelector(`#${data.btnEleId}`)
+ btn.click()
+ break
+ }
+ case 'send_to_control_net': {
+ data.type === 'img2img' ? window.switch_to_img2img() : window.switch_to_txt2img()
+ await delay(100)
+ const cn = appDoc.querySelector(`#${data.type}_controlnet`)
+ const wrap = cn.querySelector('.label-wrap')
+ if (!wrap.className.includes('open')) {
+ wrap.click()
+ await delay(100)
+ }
+ wrap.scrollIntoView()
+ wrap.dispatchEvent(await createPasteEvent(data.url))
+ break
+ }
+ case 'send_to_outpaint': {
+ switch2targetTab(getTabIdxById('openOutpaint'))
+ await delay(100)
+ const iframe = appDoc.querySelector('#openoutpaint-iframe')
+ openoutpaint_send_image(await imgUrl2DataUrl(data.url))
+ iframe.contentWindow.postMessage({
+ key: appDoc.querySelector('#openoutpaint-key').value,
+ type: 'openoutpaint/set-prompt',
+ prompt: data.prompt,
+ negPrompt: data.negPrompt
+ })
+ break
+ }
+ }
+
+ function imgUrl2DataUrl(imgUrl) {
+ return new Promise((resolve, reject) => {
+ fetch(imgUrl)
+ .then((response) => response.blob())
+ .then((blob) => {
+ const reader = new FileReader()
+ reader.readAsDataURL(blob)
+ reader.onloadend = function () {
+ const dataURL = reader.result
+ resolve(dataURL)
+ }
+ })
+ .catch((error) => reject(error))
+ })
+ }
+
+ async function createPasteEvent(imgUrl) {
+ const response = await fetch(imgUrl)
+ const imageBlob = await response.blob()
+ const imageFile = new File([imageBlob], 'image.jpg', {
+ type: imageBlob.type,
+ lastModified: Date.now()
+ })
+ const dataTransfer = new DataTransfer()
+ dataTransfer.items.add(imageFile)
+ const pasteEvent = new ClipboardEvent('paste', {
+ clipboardData: dataTransfer,
+ bubbles: true
+ })
+ return pasteEvent
+ }
+ })
+})
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/package.json b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/package.json
new file mode 100644
index 0000000000000000000000000000000000000000..3440514572ad017e19c311e6cfa4672a011a6e40
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/package.json
@@ -0,0 +1,53 @@
+{
+ "name": "vue",
+ "private": true,
+ "version": "0.0.0",
+ "type": "module",
+ "scripts": {
+ "dev": "vite",
+ "preview": "vite preview",
+ "type-check": "vue-tsc --noEmit",
+ "build": "tsx build",
+ "tauri": "tauri dev",
+ "tauri-build": "tauri build",
+ "tauri-build-debug": "tauri build --debug"
+ },
+ "dependencies": {
+ "@tauri-apps/api": "^1.4.0",
+ "@vueuse/core": "^10.1.2",
+ "@zanllp/vue-virtual-scroller": "^2.0.0-beta.7",
+ "ant-design-vue": "^3.2.20",
+ "axios": "^1.4.0",
+ "js-cookie": "^3.0.5",
+ "multi-nprogress": "^0.3.5",
+ "pinia": "^2.1.3",
+ "pinia-plugin-persistedstate": "^3.1.0",
+ "sjcl": "^1.0.8",
+ "splitpanes": "^3.1.5",
+ "vue": "^3.3.4",
+ "vue-i18n": "^9.3.0-beta.19",
+ "vue3-ts-util": "^1.0.4"
+ },
+ "devDependencies": {
+ "@tauri-apps/cli": "^1.4.0",
+ "@types/js-cookie": "^3.0.3",
+ "@types/node": "^20.2.5",
+ "@types/nprogress": "^0.2.0",
+ "@types/sjcl": "^1.0.30",
+ "@vitejs/plugin-vue": "^4.2.3",
+ "@vitejs/plugin-vue-jsx": "^3.0.1",
+ "@vue/eslint-config-typescript": "^11.0.3",
+ "@vue/tsconfig": "^0.4.0",
+ "antd-vue-volar": "^1.0.1",
+ "eslint": "^8.41.0",
+ "eslint-plugin-vue": "^9.14.1",
+ "less": "^4.1.3",
+ "sass": "^1.62.1",
+ "tsx": "^3.12.7",
+ "typescript": "^5.0.2",
+ "unplugin-vue-components": "^0.25.0",
+ "vite": "^4.3.9",
+ "vue-router": "^4.2.2",
+ "vue-tsc": "^1.4.2"
+ }
+}
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/public/vite.svg b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/public/vite.svg
new file mode 100644
index 0000000000000000000000000000000000000000..e7b8dfb1b2a60bd50538bec9f876511b9cac21e3
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/public/vite.svg
@@ -0,0 +1 @@
+
\ No newline at end of file
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/.gitignore b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/.gitignore
new file mode 100644
index 0000000000000000000000000000000000000000..af3522111c2a4220dab07d6c308f6125e074c73a
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/.gitignore
@@ -0,0 +1,6 @@
+# Generated by Cargo
+# will have compiled files and executables
+/target/
+app.conf.json
+iib_api_server-x86_64-pc-windows-msvc.exe
+iib_api_server-x86_64-unknown-linux-gnu
\ No newline at end of file
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/.taurignore b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/.taurignore
new file mode 100644
index 0000000000000000000000000000000000000000..b0bf3470f676c804a4d1fd1e8b23337cddfd363d
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/.taurignore
@@ -0,0 +1 @@
+app.conf.json
\ No newline at end of file
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/Cargo.lock b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/Cargo.lock
new file mode 100644
index 0000000000000000000000000000000000000000..41a5ee2369429b77296afe206ab7a58258986b7f
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/Cargo.lock
@@ -0,0 +1,4572 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 3
+
+[[package]]
+name = "addr2line"
+version = "0.20.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f4fa78e18c64fce05e902adecd7a5eed15a5e0a3439f7b0e169f0252214865e3"
+dependencies = [
+ "gimli",
+]
+
+[[package]]
+name = "adler"
+version = "1.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f26201604c87b1e01bd3d98f8d5d9a8fcbb815e8cedb41ffccbeb4bf593a35fe"
+
+[[package]]
+name = "aho-corasick"
+version = "0.7.20"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cc936419f96fa211c1b9166887b38e5e40b19958e5b895be7c1f93adec7071ac"
+dependencies = [
+ "memchr",
+]
+
+[[package]]
+name = "aho-corasick"
+version = "1.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "43f6cb1bf222025340178f382c426f13757b2960e89779dfcb319c32542a5a41"
+dependencies = [
+ "memchr",
+]
+
+[[package]]
+name = "alloc-no-stdlib"
+version = "2.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cc7bb162ec39d46ab1ca8c77bf72e890535becd1751bb45f64c597edb4c8c6b3"
+
+[[package]]
+name = "alloc-stdlib"
+version = "0.2.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "94fb8275041c72129eb51b7d0322c29b8387a0386127718b096429201a5d6ece"
+dependencies = [
+ "alloc-no-stdlib",
+]
+
+[[package]]
+name = "android-tzdata"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e999941b234f3131b00bc13c22d06e8c5ff726d1b6318ac7eb276997bbb4fef0"
+
+[[package]]
+name = "android_system_properties"
+version = "0.1.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "819e7219dbd41043ac279b19830f2efc897156490d7fd6ea916720117ee66311"
+dependencies = [
+ "libc",
+]
+
+[[package]]
+name = "anyhow"
+version = "1.0.71"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9c7d0618f0e0b7e8ff11427422b64564d5fb0be1940354bfe2e0529b18a9d9b8"
+
+[[package]]
+name = "async-broadcast"
+version = "0.5.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7c48ccdbf6ca6b121e0f586cbc0e73ae440e56c67c30fa0873b4e110d9c26d2b"
+dependencies = [
+ "event-listener",
+ "futures-core",
+]
+
+[[package]]
+name = "async-channel"
+version = "1.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cf46fee83e5ccffc220104713af3292ff9bc7c64c7de289f66dae8e38d826833"
+dependencies = [
+ "concurrent-queue",
+ "event-listener",
+ "futures-core",
+]
+
+[[package]]
+name = "async-executor"
+version = "1.5.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6fa3dc5f2a8564f07759c008b9109dc0d39de92a88d5588b8a5036d286383afb"
+dependencies = [
+ "async-lock",
+ "async-task",
+ "concurrent-queue",
+ "fastrand",
+ "futures-lite",
+ "slab",
+]
+
+[[package]]
+name = "async-fs"
+version = "1.6.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "279cf904654eeebfa37ac9bb1598880884924aab82e290aa65c9e77a0e142e06"
+dependencies = [
+ "async-lock",
+ "autocfg",
+ "blocking",
+ "futures-lite",
+]
+
+[[package]]
+name = "async-io"
+version = "1.13.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0fc5b45d93ef0529756f812ca52e44c221b35341892d3dcc34132ac02f3dd2af"
+dependencies = [
+ "async-lock",
+ "autocfg",
+ "cfg-if",
+ "concurrent-queue",
+ "futures-lite",
+ "log",
+ "parking",
+ "polling",
+ "rustix",
+ "slab",
+ "socket2",
+ "waker-fn",
+]
+
+[[package]]
+name = "async-lock"
+version = "2.7.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fa24f727524730b077666307f2734b4a1a1c57acb79193127dcc8914d5242dd7"
+dependencies = [
+ "event-listener",
+]
+
+[[package]]
+name = "async-process"
+version = "1.7.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7a9d28b1d97e08915212e2e45310d47854eafa69600756fc735fb788f75199c9"
+dependencies = [
+ "async-io",
+ "async-lock",
+ "autocfg",
+ "blocking",
+ "cfg-if",
+ "event-listener",
+ "futures-lite",
+ "rustix",
+ "signal-hook",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "async-recursion"
+version = "1.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0e97ce7de6cf12de5d7226c73f5ba9811622f4db3a5b91b55c53e987e5f91cba"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "async-task"
+version = "4.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ecc7ab41815b3c653ccd2978ec3255c81349336702dfdf62ee6f7069b12a3aae"
+
+[[package]]
+name = "async-trait"
+version = "0.1.69"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7b2d0f03b3640e3a630367e40c468cb7f309529c708ed1d88597047b0e7c6ef7"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "atk"
+version = "0.15.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2c3d816ce6f0e2909a96830d6911c2aff044370b1ef92d7f267b43bae5addedd"
+dependencies = [
+ "atk-sys",
+ "bitflags",
+ "glib",
+ "libc",
+]
+
+[[package]]
+name = "atk-sys"
+version = "0.15.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "58aeb089fb698e06db8089971c7ee317ab9644bade33383f63631437b03aafb6"
+dependencies = [
+ "glib-sys",
+ "gobject-sys",
+ "libc",
+ "system-deps 6.1.1",
+]
+
+[[package]]
+name = "atomic-waker"
+version = "1.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1181e1e0d1fce796a03db1ae795d67167da795f9cf4a39c37589e85ef57f26d3"
+
+[[package]]
+name = "autocfg"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d468802bab17cbc0cc575e9b053f41e72aa36bfa6b7f55e3529ffa43161b97fa"
+
+[[package]]
+name = "backtrace"
+version = "0.3.68"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4319208da049c43661739c5fade2ba182f09d1dc2299b32298d3a31692b17e12"
+dependencies = [
+ "addr2line",
+ "cc",
+ "cfg-if",
+ "libc",
+ "miniz_oxide",
+ "object",
+ "rustc-demangle",
+]
+
+[[package]]
+name = "base64"
+version = "0.13.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9e1b586273c5702936fe7b7d6896644d8be71e6314cfe09d3167c95f712589e8"
+
+[[package]]
+name = "base64"
+version = "0.21.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "604178f6c5c21f02dc555784810edfb88d34ac2c73b2eae109655649ee73ce3d"
+
+[[package]]
+name = "bitflags"
+version = "1.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
+
+[[package]]
+name = "block"
+version = "0.1.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0d8c1fef690941d3e7788d328517591fecc684c084084702d6ff1641e993699a"
+
+[[package]]
+name = "block-buffer"
+version = "0.10.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3078c7629b62d3f0439517fa394996acacc5cbc91c5a20d8c658e77abd503a71"
+dependencies = [
+ "generic-array",
+]
+
+[[package]]
+name = "blocking"
+version = "1.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "77231a1c8f801696fc0123ec6150ce92cffb8e164a02afb9c8ddee0e9b65ad65"
+dependencies = [
+ "async-channel",
+ "async-lock",
+ "async-task",
+ "atomic-waker",
+ "fastrand",
+ "futures-lite",
+ "log",
+]
+
+[[package]]
+name = "brotli"
+version = "3.3.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a1a0b1dbcc8ae29329621f8d4f0d835787c1c38bb1401979b49d13b0b305ff68"
+dependencies = [
+ "alloc-no-stdlib",
+ "alloc-stdlib",
+ "brotli-decompressor",
+]
+
+[[package]]
+name = "brotli-decompressor"
+version = "2.3.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4b6561fd3f895a11e8f72af2cb7d22e08366bebc2b6b57f7744c4bda27034744"
+dependencies = [
+ "alloc-no-stdlib",
+ "alloc-stdlib",
+]
+
+[[package]]
+name = "bstr"
+version = "1.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a246e68bb43f6cd9db24bea052a53e40405417c5fb372e3d1a8a7f770a564ef5"
+dependencies = [
+ "memchr",
+ "serde",
+]
+
+[[package]]
+name = "bumpalo"
+version = "3.13.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a3e2c3daef883ecc1b5d58c15adae93470a91d425f3532ba1695849656af3fc1"
+
+[[package]]
+name = "bytemuck"
+version = "1.13.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "17febce684fd15d89027105661fec94afb475cb995fbc59d2865198446ba2eea"
+
+[[package]]
+name = "byteorder"
+version = "1.4.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "14c189c53d098945499cdfa7ecc63567cf3886b3332b312a5b4585d8d3a6a610"
+
+[[package]]
+name = "bytes"
+version = "1.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "89b2fd2a0dcf38d7971e2194b6b6eebab45ae01067456a7fd93d5547a61b70be"
+dependencies = [
+ "serde",
+]
+
+[[package]]
+name = "cairo-rs"
+version = "0.15.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c76ee391b03d35510d9fa917357c7f1855bd9a6659c95a1b392e33f49b3369bc"
+dependencies = [
+ "bitflags",
+ "cairo-sys-rs",
+ "glib",
+ "libc",
+ "thiserror",
+]
+
+[[package]]
+name = "cairo-sys-rs"
+version = "0.15.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3c55d429bef56ac9172d25fecb85dc8068307d17acd74b377866b7a1ef25d3c8"
+dependencies = [
+ "glib-sys",
+ "libc",
+ "system-deps 6.1.1",
+]
+
+[[package]]
+name = "cargo_toml"
+version = "0.15.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "599aa35200ffff8f04c1925aa1acc92fa2e08874379ef42e210a80e527e60838"
+dependencies = [
+ "serde",
+ "toml 0.7.5",
+]
+
+[[package]]
+name = "cc"
+version = "1.0.79"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "50d30906286121d95be3d479533b458f87493b30a4b5f79a607db8f5d11aa91f"
+
+[[package]]
+name = "cesu8"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6d43a04d8753f35258c91f8ec639f792891f748a1edbd759cf1dcea3382ad83c"
+
+[[package]]
+name = "cfb"
+version = "0.7.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d38f2da7a0a2c4ccf0065be06397cc26a81f4e528be095826eee9d4adbb8c60f"
+dependencies = [
+ "byteorder",
+ "fnv",
+ "uuid",
+]
+
+[[package]]
+name = "cfg-expr"
+version = "0.9.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3431df59f28accaf4cb4eed4a9acc66bea3f3c3753aa6cdc2f024174ef232af7"
+dependencies = [
+ "smallvec",
+]
+
+[[package]]
+name = "cfg-expr"
+version = "0.15.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "215c0072ecc28f92eeb0eea38ba63ddfcb65c2828c46311d646f1a3ff5f9841c"
+dependencies = [
+ "smallvec",
+ "target-lexicon",
+]
+
+[[package]]
+name = "cfg-if"
+version = "1.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
+
+[[package]]
+name = "chrono"
+version = "0.4.26"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ec837a71355b28f6556dbd569b37b3f363091c0bd4b2e735674521b4c5fd9bc5"
+dependencies = [
+ "android-tzdata",
+ "iana-time-zone",
+ "num-traits",
+ "serde",
+ "winapi",
+]
+
+[[package]]
+name = "cocoa"
+version = "0.24.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f425db7937052c684daec3bd6375c8abe2d146dca4b8b143d6db777c39138f3a"
+dependencies = [
+ "bitflags",
+ "block",
+ "cocoa-foundation",
+ "core-foundation",
+ "core-graphics",
+ "foreign-types",
+ "libc",
+ "objc",
+]
+
+[[package]]
+name = "cocoa-foundation"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "931d3837c286f56e3c58423ce4eba12d08db2374461a785c86f672b08b5650d6"
+dependencies = [
+ "bitflags",
+ "block",
+ "core-foundation",
+ "core-graphics-types",
+ "foreign-types",
+ "libc",
+ "objc",
+]
+
+[[package]]
+name = "color_quant"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3d7b894f5411737b7867f4827955924d7c254fc9f4d91a6aad6b097804b1018b"
+
+[[package]]
+name = "combine"
+version = "4.6.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "35ed6e9d84f0b51a7f52daf1c7d71dd136fd7a3f41a8462b8cdb8c78d920fad4"
+dependencies = [
+ "bytes",
+ "memchr",
+]
+
+[[package]]
+name = "concurrent-queue"
+version = "2.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "62ec6771ecfa0762d24683ee5a32ad78487a3d3afdc0fb8cae19d2c5deb50b7c"
+dependencies = [
+ "crossbeam-utils",
+]
+
+[[package]]
+name = "convert_case"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6245d59a3e82a7fc217c5828a6692dbc6dfb63a0c8c90495621f7b9d79704a0e"
+
+[[package]]
+name = "core-foundation"
+version = "0.9.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "194a7a9e6de53fa55116934067c844d9d749312f75c6f6d0980e8c252f8c2146"
+dependencies = [
+ "core-foundation-sys",
+ "libc",
+]
+
+[[package]]
+name = "core-foundation-sys"
+version = "0.8.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e496a50fda8aacccc86d7529e2c1e0892dbd0f898a6b5645b5561b89c3210efa"
+
+[[package]]
+name = "core-graphics"
+version = "0.22.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2581bbab3b8ffc6fcbd550bf46c355135d16e9ff2a6ea032ad6b9bf1d7efe4fb"
+dependencies = [
+ "bitflags",
+ "core-foundation",
+ "core-graphics-types",
+ "foreign-types",
+ "libc",
+]
+
+[[package]]
+name = "core-graphics-types"
+version = "0.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2bb142d41022986c1d8ff29103a1411c8a3dfad3552f87a4f8dc50d61d4f4e33"
+dependencies = [
+ "bitflags",
+ "core-foundation",
+ "libc",
+]
+
+[[package]]
+name = "cpufeatures"
+version = "0.2.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "03e69e28e9f7f77debdedbaafa2866e1de9ba56df55a8bd7cfc724c25a09987c"
+dependencies = [
+ "libc",
+]
+
+[[package]]
+name = "crc32fast"
+version = "1.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b540bd8bc810d3885c6ea91e2018302f68baba2129ab3e88f32389ee9370880d"
+dependencies = [
+ "cfg-if",
+]
+
+[[package]]
+name = "crossbeam-channel"
+version = "0.5.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a33c2bf77f2df06183c3aa30d1e96c0695a313d4f9c453cc3762a6db39f99200"
+dependencies = [
+ "cfg-if",
+ "crossbeam-utils",
+]
+
+[[package]]
+name = "crossbeam-utils"
+version = "0.8.16"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5a22b2d63d4d1dc0b7f1b6b2747dd0088008a9be28b6ddf0b1e7d335e3037294"
+dependencies = [
+ "cfg-if",
+]
+
+[[package]]
+name = "crypto-common"
+version = "0.1.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1bfb12502f3fc46cca1bb51ac28df9d618d813cdc3d2f25b9fe775a34af26bb3"
+dependencies = [
+ "generic-array",
+ "typenum",
+]
+
+[[package]]
+name = "cssparser"
+version = "0.27.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "754b69d351cdc2d8ee09ae203db831e005560fc6030da058f86ad60c92a9cb0a"
+dependencies = [
+ "cssparser-macros",
+ "dtoa-short",
+ "itoa 0.4.8",
+ "matches",
+ "phf 0.8.0",
+ "proc-macro2",
+ "quote",
+ "smallvec",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "cssparser-macros"
+version = "0.6.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "13b588ba4ac1a99f7f2964d24b3d896ddc6bf847ee3855dbd4366f058cfcd331"
+dependencies = [
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "ctor"
+version = "0.1.26"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6d2301688392eb071b0bf1a37be05c469d3cc4dbbd95df672fe28ab021e6a096"
+dependencies = [
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "darling"
+version = "0.20.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0558d22a7b463ed0241e993f76f09f30b126687447751a8638587b864e4b3944"
+dependencies = [
+ "darling_core",
+ "darling_macro",
+]
+
+[[package]]
+name = "darling_core"
+version = "0.20.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ab8bfa2e259f8ee1ce5e97824a3c55ec4404a0d772ca7fa96bf19f0752a046eb"
+dependencies = [
+ "fnv",
+ "ident_case",
+ "proc-macro2",
+ "quote",
+ "strsim",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "darling_macro"
+version = "0.20.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "29a358ff9f12ec09c3e61fef9b5a9902623a695a46a917b07f269bff1445611a"
+dependencies = [
+ "darling_core",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "derivative"
+version = "2.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fcc3dd5e9e9c0b295d6e1e4d811fb6f157d5ffd784b8d202fc62eac8035a770b"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "derive_more"
+version = "0.99.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4fb810d30a7c1953f91334de7244731fc3f3c10d7fe163338a35b9f640960321"
+dependencies = [
+ "convert_case",
+ "proc-macro2",
+ "quote",
+ "rustc_version",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "digest"
+version = "0.10.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9ed9a281f7bc9b7576e61468ba615a66a5c8cfdff42420a70aa82701a3b1e292"
+dependencies = [
+ "block-buffer",
+ "crypto-common",
+]
+
+[[package]]
+name = "dirs-next"
+version = "2.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b98cf8ebf19c3d1b223e151f99a4f9f0690dca41414773390fc824184ac833e1"
+dependencies = [
+ "cfg-if",
+ "dirs-sys-next",
+]
+
+[[package]]
+name = "dirs-sys-next"
+version = "0.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4ebda144c4fe02d1f7ea1a7d9641b6fc6b580adcfa024ae48797ecdeb6825b4d"
+dependencies = [
+ "libc",
+ "redox_users",
+ "winapi",
+]
+
+[[package]]
+name = "dispatch"
+version = "0.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bd0c93bb4b0c6d9b77f4435b0ae98c24d17f1c45b2ff844c6151a07256ca923b"
+
+[[package]]
+name = "dtoa"
+version = "1.0.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "65d09067bfacaa79114679b279d7f5885b53295b1e2cfb4e79c8e4bd3d633169"
+
+[[package]]
+name = "dtoa-short"
+version = "0.3.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dbaceec3c6e4211c79e7b1800fb9680527106beb2f9c51904a3210c03a448c74"
+dependencies = [
+ "dtoa",
+]
+
+[[package]]
+name = "dunce"
+version = "1.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "56ce8c6da7551ec6c462cbaf3bfbc75131ebbfa1c944aeaa9dab51ca1c5f0c3b"
+
+[[package]]
+name = "embed-resource"
+version = "2.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f7f1e82a60222fc67bfd50d752a9c89da5cce4c39ed39decc84a443b07bbd69a"
+dependencies = [
+ "cc",
+ "rustc_version",
+ "toml 0.7.5",
+ "vswhom",
+ "winreg 0.11.0",
+]
+
+[[package]]
+name = "embed_plist"
+version = "1.2.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4ef6b89e5b37196644d8796de5268852ff179b44e96276cf4290264843743bb7"
+
+[[package]]
+name = "encoding_rs"
+version = "0.8.32"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "071a31f4ee85403370b58aca746f01041ede6f0da2730960ad001edc2b71b394"
+dependencies = [
+ "cfg-if",
+]
+
+[[package]]
+name = "enumflags2"
+version = "0.7.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c041f5090df68b32bcd905365fd51769c8b9d553fe87fde0b683534f10c01bd2"
+dependencies = [
+ "enumflags2_derive",
+ "serde",
+]
+
+[[package]]
+name = "enumflags2_derive"
+version = "0.7.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5e9a1f9f7d83e59740248a6e14ecf93929ade55027844dfcea78beafccc15745"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "equivalent"
+version = "1.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "88bffebc5d80432c9b140ee17875ff173a8ab62faad5b257da912bd2f6c1c0a1"
+
+[[package]]
+name = "errno"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4bcfec3a70f97c962c307b2d2c56e358cf1d00b558d74262b5f929ee8cc7e73a"
+dependencies = [
+ "errno-dragonfly",
+ "libc",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "errno-dragonfly"
+version = "0.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "aa68f1b12764fab894d2755d2518754e71b4fd80ecfb822714a1206c2aab39bf"
+dependencies = [
+ "cc",
+ "libc",
+]
+
+[[package]]
+name = "event-listener"
+version = "2.5.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0206175f82b8d6bf6652ff7d71a1e27fd2e4efde587fd368662814d6ec1d9ce0"
+
+[[package]]
+name = "fastrand"
+version = "1.9.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e51093e27b0797c359783294ca4f0a911c270184cb10f85783b118614a1501be"
+dependencies = [
+ "instant",
+]
+
+[[package]]
+name = "fdeflate"
+version = "0.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d329bdeac514ee06249dabc27877490f17f5d371ec693360768b838e19f3ae10"
+dependencies = [
+ "simd-adler32",
+]
+
+[[package]]
+name = "field-offset"
+version = "0.3.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "38e2275cc4e4fc009b0669731a1e5ab7ebf11f469eaede2bab9309a5b4d6057f"
+dependencies = [
+ "memoffset 0.9.0",
+ "rustc_version",
+]
+
+[[package]]
+name = "filetime"
+version = "0.2.21"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5cbc844cecaee9d4443931972e1289c8ff485cb4cc2767cb03ca139ed6885153"
+dependencies = [
+ "cfg-if",
+ "libc",
+ "redox_syscall 0.2.16",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "flate2"
+version = "1.0.26"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3b9429470923de8e8cbd4d2dc513535400b4b3fef0319fb5c4e1f520a7bef743"
+dependencies = [
+ "crc32fast",
+ "miniz_oxide",
+]
+
+[[package]]
+name = "fnv"
+version = "1.0.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3f9eec918d3f24069decb9af1554cad7c880e2da24a9afd88aca000531ab82c1"
+
+[[package]]
+name = "foreign-types"
+version = "0.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f6f339eb8adc052cd2ca78910fda869aefa38d22d5cb648e6485e4d3fc06f3b1"
+dependencies = [
+ "foreign-types-shared",
+]
+
+[[package]]
+name = "foreign-types-shared"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "00b0228411908ca8685dba7fc2cdd70ec9990a6e753e89b6ac91a84c40fbaf4b"
+
+[[package]]
+name = "form_urlencoded"
+version = "1.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a62bc1cf6f830c2ec14a513a9fb124d0a213a629668a4186f329db21fe045652"
+dependencies = [
+ "percent-encoding",
+]
+
+[[package]]
+name = "futf"
+version = "0.1.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "df420e2e84819663797d1ec6544b13c5be84629e7bb00dc960d6917db2987843"
+dependencies = [
+ "mac",
+ "new_debug_unreachable",
+]
+
+[[package]]
+name = "futures-channel"
+version = "0.3.28"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "955518d47e09b25bbebc7a18df10b81f0c766eaf4c4f1cccef2fca5f2a4fb5f2"
+dependencies = [
+ "futures-core",
+]
+
+[[package]]
+name = "futures-core"
+version = "0.3.28"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4bca583b7e26f571124fe5b7561d49cb2868d79116cfa0eefce955557c6fee8c"
+
+[[package]]
+name = "futures-executor"
+version = "0.3.28"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ccecee823288125bd88b4d7f565c9e58e41858e47ab72e8ea2d64e93624386e0"
+dependencies = [
+ "futures-core",
+ "futures-task",
+ "futures-util",
+]
+
+[[package]]
+name = "futures-io"
+version = "0.3.28"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4fff74096e71ed47f8e023204cfd0aa1289cd54ae5430a9523be060cdb849964"
+
+[[package]]
+name = "futures-lite"
+version = "1.13.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "49a9d51ce47660b1e808d3c990b4709f2f415d928835a17dfd16991515c46bce"
+dependencies = [
+ "fastrand",
+ "futures-core",
+ "futures-io",
+ "memchr",
+ "parking",
+ "pin-project-lite",
+ "waker-fn",
+]
+
+[[package]]
+name = "futures-macro"
+version = "0.3.28"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "89ca545a94061b6365f2c7355b4b32bd20df3ff95f02da9329b34ccc3bd6ee72"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "futures-sink"
+version = "0.3.28"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f43be4fe21a13b9781a69afa4985b0f6ee0e1afab2c6f454a8cf30e2b2237b6e"
+
+[[package]]
+name = "futures-task"
+version = "0.3.28"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "76d3d132be6c0e6aa1534069c705a74a5997a356c0dc2f86a47765e5617c5b65"
+
+[[package]]
+name = "futures-util"
+version = "0.3.28"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "26b01e40b772d54cf6c6d721c1d1abd0647a0106a12ecaa1c186273392a69533"
+dependencies = [
+ "futures-core",
+ "futures-io",
+ "futures-macro",
+ "futures-sink",
+ "futures-task",
+ "memchr",
+ "pin-project-lite",
+ "pin-utils",
+ "slab",
+]
+
+[[package]]
+name = "fxhash"
+version = "0.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c31b6d751ae2c7f11320402d34e41349dd1016f8d5d45e48c4312bc8625af50c"
+dependencies = [
+ "byteorder",
+]
+
+[[package]]
+name = "gdk"
+version = "0.15.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a6e05c1f572ab0e1f15be94217f0dc29088c248b14f792a5ff0af0d84bcda9e8"
+dependencies = [
+ "bitflags",
+ "cairo-rs",
+ "gdk-pixbuf",
+ "gdk-sys",
+ "gio",
+ "glib",
+ "libc",
+ "pango",
+]
+
+[[package]]
+name = "gdk-pixbuf"
+version = "0.15.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ad38dd9cc8b099cceecdf41375bb6d481b1b5a7cd5cd603e10a69a9383f8619a"
+dependencies = [
+ "bitflags",
+ "gdk-pixbuf-sys",
+ "gio",
+ "glib",
+ "libc",
+]
+
+[[package]]
+name = "gdk-pixbuf-sys"
+version = "0.15.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "140b2f5378256527150350a8346dbdb08fadc13453a7a2d73aecd5fab3c402a7"
+dependencies = [
+ "gio-sys",
+ "glib-sys",
+ "gobject-sys",
+ "libc",
+ "system-deps 6.1.1",
+]
+
+[[package]]
+name = "gdk-sys"
+version = "0.15.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "32e7a08c1e8f06f4177fb7e51a777b8c1689f743a7bc11ea91d44d2226073a88"
+dependencies = [
+ "cairo-sys-rs",
+ "gdk-pixbuf-sys",
+ "gio-sys",
+ "glib-sys",
+ "gobject-sys",
+ "libc",
+ "pango-sys",
+ "pkg-config",
+ "system-deps 6.1.1",
+]
+
+[[package]]
+name = "gdkwayland-sys"
+version = "0.15.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cca49a59ad8cfdf36ef7330fe7bdfbe1d34323220cc16a0de2679ee773aee2c2"
+dependencies = [
+ "gdk-sys",
+ "glib-sys",
+ "gobject-sys",
+ "libc",
+ "pkg-config",
+ "system-deps 6.1.1",
+]
+
+[[package]]
+name = "gdkx11-sys"
+version = "0.15.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b4b7f8c7a84b407aa9b143877e267e848ff34106578b64d1e0a24bf550716178"
+dependencies = [
+ "gdk-sys",
+ "glib-sys",
+ "libc",
+ "system-deps 6.1.1",
+ "x11",
+]
+
+[[package]]
+name = "generator"
+version = "0.7.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5cc16584ff22b460a382b7feec54b23d2908d858152e5739a120b949293bd74e"
+dependencies = [
+ "cc",
+ "libc",
+ "log",
+ "rustversion",
+ "windows 0.48.0",
+]
+
+[[package]]
+name = "generic-array"
+version = "0.14.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "85649ca51fd72272d7821adaf274ad91c288277713d9c18820d8499a7ff69e9a"
+dependencies = [
+ "typenum",
+ "version_check",
+]
+
+[[package]]
+name = "getrandom"
+version = "0.1.16"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8fc3cb4d91f53b50155bdcfd23f6a4c39ae1969c2ae85982b135750cccaf5fce"
+dependencies = [
+ "cfg-if",
+ "libc",
+ "wasi 0.9.0+wasi-snapshot-preview1",
+]
+
+[[package]]
+name = "getrandom"
+version = "0.2.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "be4136b2a15dd319360be1c07d9933517ccf0be8f16bf62a3bee4f0d618df427"
+dependencies = [
+ "cfg-if",
+ "libc",
+ "wasi 0.11.0+wasi-snapshot-preview1",
+]
+
+[[package]]
+name = "gimli"
+version = "0.27.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b6c80984affa11d98d1b88b66ac8853f143217b399d3c74116778ff8fdb4ed2e"
+
+[[package]]
+name = "gio"
+version = "0.15.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "68fdbc90312d462781a395f7a16d96a2b379bb6ef8cd6310a2df272771c4283b"
+dependencies = [
+ "bitflags",
+ "futures-channel",
+ "futures-core",
+ "futures-io",
+ "gio-sys",
+ "glib",
+ "libc",
+ "once_cell",
+ "thiserror",
+]
+
+[[package]]
+name = "gio-sys"
+version = "0.15.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "32157a475271e2c4a023382e9cab31c4584ee30a97da41d3c4e9fdd605abcf8d"
+dependencies = [
+ "glib-sys",
+ "gobject-sys",
+ "libc",
+ "system-deps 6.1.1",
+ "winapi",
+]
+
+[[package]]
+name = "glib"
+version = "0.15.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "edb0306fbad0ab5428b0ca674a23893db909a98582969c9b537be4ced78c505d"
+dependencies = [
+ "bitflags",
+ "futures-channel",
+ "futures-core",
+ "futures-executor",
+ "futures-task",
+ "glib-macros",
+ "glib-sys",
+ "gobject-sys",
+ "libc",
+ "once_cell",
+ "smallvec",
+ "thiserror",
+]
+
+[[package]]
+name = "glib-macros"
+version = "0.15.13"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "10c6ae9f6fa26f4fb2ac16b528d138d971ead56141de489f8111e259b9df3c4a"
+dependencies = [
+ "anyhow",
+ "heck 0.4.1",
+ "proc-macro-crate",
+ "proc-macro-error",
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "glib-sys"
+version = "0.15.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ef4b192f8e65e9cf76cbf4ea71fa8e3be4a0e18ffe3d68b8da6836974cc5bad4"
+dependencies = [
+ "libc",
+ "system-deps 6.1.1",
+]
+
+[[package]]
+name = "glob"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d2fabcfbdc87f4758337ca535fb41a6d701b65693ce38287d856d1674551ec9b"
+
+[[package]]
+name = "globset"
+version = "0.4.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "029d74589adefde59de1a0c4f4732695c32805624aec7b68d91503d4dba79afc"
+dependencies = [
+ "aho-corasick 0.7.20",
+ "bstr",
+ "fnv",
+ "log",
+ "regex",
+]
+
+[[package]]
+name = "gobject-sys"
+version = "0.15.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0d57ce44246becd17153bd035ab4d32cfee096a657fc01f2231c9278378d1e0a"
+dependencies = [
+ "glib-sys",
+ "libc",
+ "system-deps 6.1.1",
+]
+
+[[package]]
+name = "gtk"
+version = "0.15.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "92e3004a2d5d6d8b5057d2b57b3712c9529b62e82c77f25c1fecde1fd5c23bd0"
+dependencies = [
+ "atk",
+ "bitflags",
+ "cairo-rs",
+ "field-offset",
+ "futures-channel",
+ "gdk",
+ "gdk-pixbuf",
+ "gio",
+ "glib",
+ "gtk-sys",
+ "gtk3-macros",
+ "libc",
+ "once_cell",
+ "pango",
+ "pkg-config",
+]
+
+[[package]]
+name = "gtk-sys"
+version = "0.15.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d5bc2f0587cba247f60246a0ca11fe25fb733eabc3de12d1965fc07efab87c84"
+dependencies = [
+ "atk-sys",
+ "cairo-sys-rs",
+ "gdk-pixbuf-sys",
+ "gdk-sys",
+ "gio-sys",
+ "glib-sys",
+ "gobject-sys",
+ "libc",
+ "pango-sys",
+ "system-deps 6.1.1",
+]
+
+[[package]]
+name = "gtk3-macros"
+version = "0.15.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "684c0456c086e8e7e9af73ec5b84e35938df394712054550e81558d21c44ab0d"
+dependencies = [
+ "anyhow",
+ "proc-macro-crate",
+ "proc-macro-error",
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "h2"
+version = "0.3.20"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "97ec8491ebaf99c8eaa73058b045fe58073cd6be7f596ac993ced0b0a0c01049"
+dependencies = [
+ "bytes",
+ "fnv",
+ "futures-core",
+ "futures-sink",
+ "futures-util",
+ "http",
+ "indexmap 1.9.3",
+ "slab",
+ "tokio",
+ "tokio-util",
+ "tracing",
+]
+
+[[package]]
+name = "hashbrown"
+version = "0.12.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8a9ee70c43aaf417c914396645a0fa852624801b24ebb7ae78fe8272889ac888"
+
+[[package]]
+name = "hashbrown"
+version = "0.14.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2c6201b9ff9fd90a5a3bac2e56a830d0caa509576f0e503818ee82c181b3437a"
+
+[[package]]
+name = "heck"
+version = "0.3.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6d621efb26863f0e9924c6ac577e8275e5e6b77455db64ffa6c65c904e9e132c"
+dependencies = [
+ "unicode-segmentation",
+]
+
+[[package]]
+name = "heck"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "95505c38b4572b2d910cecb0281560f54b440a19336cbbcb27bf6ce6adc6f5a8"
+
+[[package]]
+name = "hermit-abi"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fed44880c466736ef9a5c5b5facefb5ed0785676d0c02d612db14e54f0d84286"
+
+[[package]]
+name = "hex"
+version = "0.4.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7f24254aa9a54b5c858eaee2f5bccdb46aaf0e486a595ed5fd8f86ba55232a70"
+
+[[package]]
+name = "html5ever"
+version = "0.25.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e5c13fb08e5d4dfc151ee5e88bae63f7773d61852f3bdc73c9f4b9e1bde03148"
+dependencies = [
+ "log",
+ "mac",
+ "markup5ever",
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "http"
+version = "0.2.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bd6effc99afb63425aff9b05836f029929e345a6148a14b7ecd5ab67af944482"
+dependencies = [
+ "bytes",
+ "fnv",
+ "itoa 1.0.6",
+]
+
+[[package]]
+name = "http-body"
+version = "0.4.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d5f38f16d184e36f2408a55281cd658ecbd3ca05cce6d6510a176eca393e26d1"
+dependencies = [
+ "bytes",
+ "http",
+ "pin-project-lite",
+]
+
+[[package]]
+name = "http-range"
+version = "0.1.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "21dec9db110f5f872ed9699c3ecf50cf16f423502706ba5c72462e28d3157573"
+
+[[package]]
+name = "httparse"
+version = "1.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d897f394bad6a705d5f4104762e116a75639e470d80901eed05a860a95cb1904"
+
+[[package]]
+name = "httpdate"
+version = "1.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c4a1e36c821dbe04574f602848a19f742f4fb3c98d40449f11bcad18d6b17421"
+
+[[package]]
+name = "hyper"
+version = "0.14.27"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ffb1cfd654a8219eaef89881fdb3bb3b1cdc5fa75ded05d6933b2b382e395468"
+dependencies = [
+ "bytes",
+ "futures-channel",
+ "futures-core",
+ "futures-util",
+ "h2",
+ "http",
+ "http-body",
+ "httparse",
+ "httpdate",
+ "itoa 1.0.6",
+ "pin-project-lite",
+ "socket2",
+ "tokio",
+ "tower-service",
+ "tracing",
+ "want",
+]
+
+[[package]]
+name = "hyper-tls"
+version = "0.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d6183ddfa99b85da61a140bea0efc93fdf56ceaa041b37d553518030827f9905"
+dependencies = [
+ "bytes",
+ "hyper",
+ "native-tls",
+ "tokio",
+ "tokio-native-tls",
+]
+
+[[package]]
+name = "iana-time-zone"
+version = "0.1.57"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2fad5b825842d2b38bd206f3e81d6957625fd7f0a361e345c30e01a0ae2dd613"
+dependencies = [
+ "android_system_properties",
+ "core-foundation-sys",
+ "iana-time-zone-haiku",
+ "js-sys",
+ "wasm-bindgen",
+ "windows 0.48.0",
+]
+
+[[package]]
+name = "iana-time-zone-haiku"
+version = "0.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f31827a206f56af32e590ba56d5d2d085f558508192593743f16b2306495269f"
+dependencies = [
+ "cc",
+]
+
+[[package]]
+name = "ico"
+version = "0.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e3804960be0bb5e4edb1e1ad67afd321a9ecfd875c3e65c099468fd2717d7cae"
+dependencies = [
+ "byteorder",
+ "png",
+]
+
+[[package]]
+name = "ident_case"
+version = "1.0.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b9e0384b61958566e926dc50660321d12159025e767c18e043daf26b70104c39"
+
+[[package]]
+name = "idna"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7d20d6b07bfbc108882d88ed8e37d39636dcc260e15e30c45e6ba089610b917c"
+dependencies = [
+ "unicode-bidi",
+ "unicode-normalization",
+]
+
+[[package]]
+name = "ignore"
+version = "0.4.20"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dbe7873dab538a9a44ad79ede1faf5f30d49f9a5c883ddbab48bce81b64b7492"
+dependencies = [
+ "globset",
+ "lazy_static",
+ "log",
+ "memchr",
+ "regex",
+ "same-file",
+ "thread_local",
+ "walkdir",
+ "winapi-util",
+]
+
+[[package]]
+name = "image"
+version = "0.24.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "527909aa81e20ac3a44803521443a765550f09b5130c2c2fa1ea59c2f8f50a3a"
+dependencies = [
+ "bytemuck",
+ "byteorder",
+ "color_quant",
+ "num-rational",
+ "num-traits",
+]
+
+[[package]]
+name = "indexmap"
+version = "1.9.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bd070e393353796e801d209ad339e89596eb4c8d430d18ede6a1cced8fafbd99"
+dependencies = [
+ "autocfg",
+ "hashbrown 0.12.3",
+ "serde",
+]
+
+[[package]]
+name = "indexmap"
+version = "2.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d5477fe2230a79769d8dc68e0eabf5437907c0457a5614a9e8dddb67f65eb65d"
+dependencies = [
+ "equivalent",
+ "hashbrown 0.14.0",
+]
+
+[[package]]
+name = "infer"
+version = "0.12.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a898e4b7951673fce96614ce5751d13c40fc5674bc2d759288e46c3ab62598b3"
+dependencies = [
+ "cfb",
+]
+
+[[package]]
+name = "infinite_image_browsing"
+version = "0.0.1"
+dependencies = [
+ "reqwest",
+ "serde",
+ "serde_json",
+ "tauri",
+ "tauri-build",
+]
+
+[[package]]
+name = "instant"
+version = "0.1.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7a5bbe824c507c5da5956355e86a746d82e0e1464f65d862cc5e71da70e94b2c"
+dependencies = [
+ "cfg-if",
+]
+
+[[package]]
+name = "io-lifetimes"
+version = "1.0.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "eae7b9aee968036d54dce06cebaefd919e4472e753296daccd6d344e3e2df0c2"
+dependencies = [
+ "hermit-abi",
+ "libc",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "ipnet"
+version = "2.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "28b29a3cd74f0f4598934efe3aeba42bae0eb4680554128851ebbecb02af14e6"
+
+[[package]]
+name = "itoa"
+version = "0.4.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b71991ff56294aa922b450139ee08b3bfc70982c6b2c7562771375cf73542dd4"
+
+[[package]]
+name = "itoa"
+version = "1.0.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "453ad9f582a441959e5f0d088b02ce04cfe8d51a8eaf077f12ac6d3e94164ca6"
+
+[[package]]
+name = "javascriptcore-rs"
+version = "0.16.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bf053e7843f2812ff03ef5afe34bb9c06ffee120385caad4f6b9967fcd37d41c"
+dependencies = [
+ "bitflags",
+ "glib",
+ "javascriptcore-rs-sys",
+]
+
+[[package]]
+name = "javascriptcore-rs-sys"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "905fbb87419c5cde6e3269537e4ea7d46431f3008c5d057e915ef3f115e7793c"
+dependencies = [
+ "glib-sys",
+ "gobject-sys",
+ "libc",
+ "system-deps 5.0.0",
+]
+
+[[package]]
+name = "jni"
+version = "0.20.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "039022cdf4d7b1cf548d31f60ae783138e5fd42013f6271049d7df7afadef96c"
+dependencies = [
+ "cesu8",
+ "combine",
+ "jni-sys",
+ "log",
+ "thiserror",
+ "walkdir",
+]
+
+[[package]]
+name = "jni-sys"
+version = "0.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8eaf4bc02d17cbdd7ff4c7438cafcdf7fb9a4613313ad11b4f8fefe7d3fa0130"
+
+[[package]]
+name = "js-sys"
+version = "0.3.64"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c5f195fe497f702db0f318b07fdd68edb16955aed830df8363d837542f8f935a"
+dependencies = [
+ "wasm-bindgen",
+]
+
+[[package]]
+name = "json-patch"
+version = "1.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1f54898088ccb91df1b492cc80029a6fdf1c48ca0db7c6822a8babad69c94658"
+dependencies = [
+ "serde",
+ "serde_json",
+ "thiserror",
+ "treediff",
+]
+
+[[package]]
+name = "kuchiki"
+version = "0.8.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1ea8e9c6e031377cff82ee3001dc8026cdf431ed4e2e6b51f98ab8c73484a358"
+dependencies = [
+ "cssparser",
+ "html5ever",
+ "matches",
+ "selectors",
+]
+
+[[package]]
+name = "lazy_static"
+version = "1.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e2abad23fbc42b3700f2f279844dc832adb2b2eb069b2df918f455c4e18cc646"
+
+[[package]]
+name = "libc"
+version = "0.2.147"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b4668fb0ea861c1df094127ac5f1da3409a82116a4ba74fca2e58ef927159bb3"
+
+[[package]]
+name = "line-wrap"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f30344350a2a51da54c1d53be93fade8a237e545dbcc4bdbe635413f2117cab9"
+dependencies = [
+ "safemem",
+]
+
+[[package]]
+name = "linux-raw-sys"
+version = "0.3.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ef53942eb7bf7ff43a617b3e2c1c4a5ecf5944a7c1bc12d7ee39bbb15e5c1519"
+
+[[package]]
+name = "lock_api"
+version = "0.4.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c1cc9717a20b1bb222f333e6a92fd32f7d8a18ddc5a3191a11af45dcbf4dcd16"
+dependencies = [
+ "autocfg",
+ "scopeguard",
+]
+
+[[package]]
+name = "log"
+version = "0.4.19"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b06a4cde4c0f271a446782e3eff8de789548ce57dbc8eca9292c27f4a42004b4"
+
+[[package]]
+name = "loom"
+version = "0.5.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ff50ecb28bb86013e935fb6683ab1f6d3a20016f123c76fd4c27470076ac30f5"
+dependencies = [
+ "cfg-if",
+ "generator",
+ "scoped-tls",
+ "serde",
+ "serde_json",
+ "tracing",
+ "tracing-subscriber",
+]
+
+[[package]]
+name = "mac"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c41e0c4fef86961ac6d6f8a82609f55f31b05e4fce149ac5710e439df7619ba4"
+
+[[package]]
+name = "mac-notification-sys"
+version = "0.5.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3e72d50edb17756489e79d52eb146927bec8eba9dd48faadf9ef08bca3791ad5"
+dependencies = [
+ "cc",
+ "dirs-next",
+ "objc-foundation",
+ "objc_id",
+ "time",
+]
+
+[[package]]
+name = "malloc_buf"
+version = "0.0.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "62bb907fe88d54d8d9ce32a3cceab4218ed2f6b7d35617cafe9adf84e43919cb"
+dependencies = [
+ "libc",
+]
+
+[[package]]
+name = "markup5ever"
+version = "0.10.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a24f40fb03852d1cdd84330cddcaf98e9ec08a7b7768e952fad3b4cf048ec8fd"
+dependencies = [
+ "log",
+ "phf 0.8.0",
+ "phf_codegen",
+ "string_cache",
+ "string_cache_codegen",
+ "tendril",
+]
+
+[[package]]
+name = "matchers"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8263075bb86c5a1b1427b5ae862e8889656f126e9f77c484496e8b47cf5c5558"
+dependencies = [
+ "regex-automata",
+]
+
+[[package]]
+name = "matches"
+version = "0.1.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2532096657941c2fea9c289d370a250971c689d4f143798ff67113ec042024a5"
+
+[[package]]
+name = "memchr"
+version = "2.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2dffe52ecf27772e601905b7522cb4ef790d2cc203488bbd0e2fe85fcb74566d"
+
+[[package]]
+name = "memoffset"
+version = "0.7.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5de893c32cde5f383baa4c04c5d6dbdd735cfd4a794b0debdb2bb1b421da5ff4"
+dependencies = [
+ "autocfg",
+]
+
+[[package]]
+name = "memoffset"
+version = "0.9.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5a634b1c61a95585bd15607c6ab0c4e5b226e695ff2800ba0cdccddf208c406c"
+dependencies = [
+ "autocfg",
+]
+
+[[package]]
+name = "mime"
+version = "0.3.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6877bb514081ee2a7ff5ef9de3281f14a4dd4bceac4c09388074a6b5df8a139a"
+
+[[package]]
+name = "miniz_oxide"
+version = "0.7.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e7810e0be55b428ada41041c41f32c9f1a42817901b4ccf45fa3d4b6561e74c7"
+dependencies = [
+ "adler",
+ "simd-adler32",
+]
+
+[[package]]
+name = "mio"
+version = "0.8.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "927a765cd3fc26206e66b296465fa9d3e5ab003e651c1b3c060e7956d96b19d2"
+dependencies = [
+ "libc",
+ "wasi 0.11.0+wasi-snapshot-preview1",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "native-tls"
+version = "0.2.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "07226173c32f2926027b63cce4bcd8076c3552846cbe7925f3aaffeac0a3b92e"
+dependencies = [
+ "lazy_static",
+ "libc",
+ "log",
+ "openssl",
+ "openssl-probe",
+ "openssl-sys",
+ "schannel",
+ "security-framework",
+ "security-framework-sys",
+ "tempfile",
+]
+
+[[package]]
+name = "ndk"
+version = "0.6.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2032c77e030ddee34a6787a64166008da93f6a352b629261d0fee232b8742dd4"
+dependencies = [
+ "bitflags",
+ "jni-sys",
+ "ndk-sys",
+ "num_enum",
+ "thiserror",
+]
+
+[[package]]
+name = "ndk-context"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "27b02d87554356db9e9a873add8782d4ea6e3e58ea071a9adb9a2e8ddb884a8b"
+
+[[package]]
+name = "ndk-sys"
+version = "0.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6e5a6ae77c8ee183dcbbba6150e2e6b9f3f4196a7666c02a715a95692ec1fa97"
+dependencies = [
+ "jni-sys",
+]
+
+[[package]]
+name = "new_debug_unreachable"
+version = "1.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e4a24736216ec316047a1fc4252e27dabb04218aa4a3f37c6e7ddbf1f9782b54"
+
+[[package]]
+name = "nix"
+version = "0.26.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bfdda3d196821d6af13126e40375cdf7da646a96114af134d5f417a9a1dc8e1a"
+dependencies = [
+ "bitflags",
+ "cfg-if",
+ "libc",
+ "memoffset 0.7.1",
+ "static_assertions",
+]
+
+[[package]]
+name = "nodrop"
+version = "0.1.14"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "72ef4a56884ca558e5ddb05a1d1e7e1bfd9a68d9ed024c21704cc98872dae1bb"
+
+[[package]]
+name = "notify-rust"
+version = "4.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2bfa211d18e360f08e36c364308f394b5eb23a6629150690e109a916dc6f610e"
+dependencies = [
+ "log",
+ "mac-notification-sys",
+ "serde",
+ "tauri-winrt-notification",
+ "zbus",
+]
+
+[[package]]
+name = "nu-ansi-term"
+version = "0.46.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "77a8165726e8236064dbb45459242600304b42a5ea24ee2948e18e023bf7ba84"
+dependencies = [
+ "overload",
+ "winapi",
+]
+
+[[package]]
+name = "num-integer"
+version = "0.1.45"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "225d3389fb3509a24c93f5c29eb6bde2586b98d9f016636dff58d7c6f7569cd9"
+dependencies = [
+ "autocfg",
+ "num-traits",
+]
+
+[[package]]
+name = "num-rational"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0638a1c9d0a3c0914158145bc76cff373a75a627e6ecbfb71cbe6f453a5a19b0"
+dependencies = [
+ "autocfg",
+ "num-integer",
+ "num-traits",
+]
+
+[[package]]
+name = "num-traits"
+version = "0.2.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "578ede34cf02f8924ab9447f50c28075b4d3e5b269972345e7e0372b38c6cdcd"
+dependencies = [
+ "autocfg",
+]
+
+[[package]]
+name = "num_cpus"
+version = "1.16.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4161fcb6d602d4d2081af7c3a45852d875a03dd337a6bfdd6e06407b61342a43"
+dependencies = [
+ "hermit-abi",
+ "libc",
+]
+
+[[package]]
+name = "num_enum"
+version = "0.5.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1f646caf906c20226733ed5b1374287eb97e3c2a5c227ce668c1f2ce20ae57c9"
+dependencies = [
+ "num_enum_derive",
+]
+
+[[package]]
+name = "num_enum_derive"
+version = "0.5.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dcbff9bc912032c62bf65ef1d5aea88983b420f4f839db1e9b0c281a25c9c799"
+dependencies = [
+ "proc-macro-crate",
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "objc"
+version = "0.2.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "915b1b472bc21c53464d6c8461c9d3af805ba1ef837e1cac254428f4a77177b1"
+dependencies = [
+ "malloc_buf",
+ "objc_exception",
+]
+
+[[package]]
+name = "objc-foundation"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1add1b659e36c9607c7aab864a76c7a4c2760cd0cd2e120f3fb8b952c7e22bf9"
+dependencies = [
+ "block",
+ "objc",
+ "objc_id",
+]
+
+[[package]]
+name = "objc_exception"
+version = "0.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ad970fb455818ad6cba4c122ad012fae53ae8b4795f86378bce65e4f6bab2ca4"
+dependencies = [
+ "cc",
+]
+
+[[package]]
+name = "objc_id"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c92d4ddb4bd7b50d730c215ff871754d0da6b2178849f8a2a2ab69712d0c073b"
+dependencies = [
+ "objc",
+]
+
+[[package]]
+name = "object"
+version = "0.31.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8bda667d9f2b5051b8833f59f3bf748b28ef54f850f4fcb389a252aa383866d1"
+dependencies = [
+ "memchr",
+]
+
+[[package]]
+name = "once_cell"
+version = "1.18.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dd8b5dd2ae5ed71462c540258bedcb51965123ad7e7ccf4b9a8cafaa4a63576d"
+
+[[package]]
+name = "open"
+version = "3.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2078c0039e6a54a0c42c28faa984e115fb4c2d5bf2208f77d1961002df8576f8"
+dependencies = [
+ "pathdiff",
+ "windows-sys 0.42.0",
+]
+
+[[package]]
+name = "openssl"
+version = "0.10.55"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "345df152bc43501c5eb9e4654ff05f794effb78d4efe3d53abc158baddc0703d"
+dependencies = [
+ "bitflags",
+ "cfg-if",
+ "foreign-types",
+ "libc",
+ "once_cell",
+ "openssl-macros",
+ "openssl-sys",
+]
+
+[[package]]
+name = "openssl-macros"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a948666b637a0f465e8564c73e89d4dde00d72d4d473cc972f390fc3dcee7d9c"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "openssl-probe"
+version = "0.1.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ff011a302c396a5197692431fc1948019154afc178baf7d8e37367442a4601cf"
+
+[[package]]
+name = "openssl-sys"
+version = "0.9.90"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "374533b0e45f3a7ced10fcaeccca020e66656bc03dac384f852e4e5a7a8104a6"
+dependencies = [
+ "cc",
+ "libc",
+ "pkg-config",
+ "vcpkg",
+]
+
+[[package]]
+name = "ordered-stream"
+version = "0.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9aa2b01e1d916879f73a53d01d1d6cee68adbb31d6d9177a8cfce093cced1d50"
+dependencies = [
+ "futures-core",
+ "pin-project-lite",
+]
+
+[[package]]
+name = "os_info"
+version = "3.7.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "006e42d5b888366f1880eda20371fedde764ed2213dc8496f49622fa0c99cd5e"
+dependencies = [
+ "log",
+ "serde",
+ "winapi",
+]
+
+[[package]]
+name = "os_pipe"
+version = "1.1.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0ae859aa07428ca9a929b936690f8b12dc5f11dd8c6992a18ca93919f28bc177"
+dependencies = [
+ "libc",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "overload"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b15813163c1d831bf4a13c3610c05c0d03b39feb07f7e09fa234dac9b15aaf39"
+
+[[package]]
+name = "pango"
+version = "0.15.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "22e4045548659aee5313bde6c582b0d83a627b7904dd20dc2d9ef0895d414e4f"
+dependencies = [
+ "bitflags",
+ "glib",
+ "libc",
+ "once_cell",
+ "pango-sys",
+]
+
+[[package]]
+name = "pango-sys"
+version = "0.15.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d2a00081cde4661982ed91d80ef437c20eacaf6aa1a5962c0279ae194662c3aa"
+dependencies = [
+ "glib-sys",
+ "gobject-sys",
+ "libc",
+ "system-deps 6.1.1",
+]
+
+[[package]]
+name = "parking"
+version = "2.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "14f2252c834a40ed9bb5422029649578e63aa341ac401f74e719dd1afda8394e"
+
+[[package]]
+name = "parking_lot"
+version = "0.12.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3742b2c103b9f06bc9fff0a37ff4912935851bee6d36f3c02bcc755bcfec228f"
+dependencies = [
+ "lock_api",
+ "parking_lot_core",
+]
+
+[[package]]
+name = "parking_lot_core"
+version = "0.9.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "93f00c865fe7cabf650081affecd3871070f26767e7b2070a3ffae14c654b447"
+dependencies = [
+ "cfg-if",
+ "libc",
+ "redox_syscall 0.3.5",
+ "smallvec",
+ "windows-targets 0.48.1",
+]
+
+[[package]]
+name = "pathdiff"
+version = "0.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8835116a5c179084a830efb3adc117ab007512b535bc1a21c991d3b32a6b44dd"
+
+[[package]]
+name = "percent-encoding"
+version = "2.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9b2a4787296e9989611394c33f193f676704af1686e70b8f8033ab5ba9a35a94"
+
+[[package]]
+name = "phf"
+version = "0.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3dfb61232e34fcb633f43d12c58f83c1df82962dcdfa565a4e866ffc17dafe12"
+dependencies = [
+ "phf_macros 0.8.0",
+ "phf_shared 0.8.0",
+ "proc-macro-hack",
+]
+
+[[package]]
+name = "phf"
+version = "0.10.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fabbf1ead8a5bcbc20f5f8b939ee3f5b0f6f281b6ad3468b84656b658b455259"
+dependencies = [
+ "phf_macros 0.10.0",
+ "phf_shared 0.10.0",
+ "proc-macro-hack",
+]
+
+[[package]]
+name = "phf_codegen"
+version = "0.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cbffee61585b0411840d3ece935cce9cb6321f01c45477d30066498cd5e1a815"
+dependencies = [
+ "phf_generator 0.8.0",
+ "phf_shared 0.8.0",
+]
+
+[[package]]
+name = "phf_generator"
+version = "0.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "17367f0cc86f2d25802b2c26ee58a7b23faeccf78a396094c13dced0d0182526"
+dependencies = [
+ "phf_shared 0.8.0",
+ "rand 0.7.3",
+]
+
+[[package]]
+name = "phf_generator"
+version = "0.10.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5d5285893bb5eb82e6aaf5d59ee909a06a16737a8970984dd7746ba9283498d6"
+dependencies = [
+ "phf_shared 0.10.0",
+ "rand 0.8.5",
+]
+
+[[package]]
+name = "phf_macros"
+version = "0.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7f6fde18ff429ffc8fe78e2bf7f8b7a5a5a6e2a8b58bc5a9ac69198bbda9189c"
+dependencies = [
+ "phf_generator 0.8.0",
+ "phf_shared 0.8.0",
+ "proc-macro-hack",
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "phf_macros"
+version = "0.10.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "58fdf3184dd560f160dd73922bea2d5cd6e8f064bf4b13110abd81b03697b4e0"
+dependencies = [
+ "phf_generator 0.10.0",
+ "phf_shared 0.10.0",
+ "proc-macro-hack",
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "phf_shared"
+version = "0.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c00cf8b9eafe68dde5e9eaa2cef8ee84a9336a47d566ec55ca16589633b65af7"
+dependencies = [
+ "siphasher",
+]
+
+[[package]]
+name = "phf_shared"
+version = "0.10.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b6796ad771acdc0123d2a88dc428b5e38ef24456743ddb1744ed628f9815c096"
+dependencies = [
+ "siphasher",
+]
+
+[[package]]
+name = "pin-project-lite"
+version = "0.2.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e0a7ae3ac2f1173085d398531c705756c94a4c56843785df85a60c1a0afac116"
+
+[[package]]
+name = "pin-utils"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8b870d8c151b6f2fb93e84a13146138f05d02ed11c7e7c54f8826aaaf7c9f184"
+
+[[package]]
+name = "pkg-config"
+version = "0.3.27"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "26072860ba924cbfa98ea39c8c19b4dd6a4a25423dbdf219c1eca91aa0cf6964"
+
+[[package]]
+name = "plist"
+version = "1.4.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9bd9647b268a3d3e14ff09c23201133a62589c658db02bb7388c7246aafe0590"
+dependencies = [
+ "base64 0.21.2",
+ "indexmap 1.9.3",
+ "line-wrap",
+ "quick-xml 0.28.2",
+ "serde",
+ "time",
+]
+
+[[package]]
+name = "png"
+version = "0.17.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "59871cc5b6cce7eaccca5a802b4173377a1c2ba90654246789a8fa2334426d11"
+dependencies = [
+ "bitflags",
+ "crc32fast",
+ "fdeflate",
+ "flate2",
+ "miniz_oxide",
+]
+
+[[package]]
+name = "polling"
+version = "2.8.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4b2d323e8ca7996b3e23126511a523f7e62924d93ecd5ae73b333815b0eb3dce"
+dependencies = [
+ "autocfg",
+ "bitflags",
+ "cfg-if",
+ "concurrent-queue",
+ "libc",
+ "log",
+ "pin-project-lite",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "ppv-lite86"
+version = "0.2.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5b40af805b3121feab8a3c29f04d8ad262fa8e0561883e7653e024ae4479e6de"
+
+[[package]]
+name = "precomputed-hash"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "925383efa346730478fb4838dbe9137d2a47675ad789c546d150a6e1dd4ab31c"
+
+[[package]]
+name = "proc-macro-crate"
+version = "1.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7f4c021e1093a56626774e81216a4ce732a735e5bad4868a03f3ed65ca0c3919"
+dependencies = [
+ "once_cell",
+ "toml_edit",
+]
+
+[[package]]
+name = "proc-macro-error"
+version = "1.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "da25490ff9892aab3fcf7c36f08cfb902dd3e71ca0f9f9517bea02a73a5ce38c"
+dependencies = [
+ "proc-macro-error-attr",
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+ "version_check",
+]
+
+[[package]]
+name = "proc-macro-error-attr"
+version = "1.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a1be40180e52ecc98ad80b184934baf3d0d29f979574e439af5a55274b35f869"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "version_check",
+]
+
+[[package]]
+name = "proc-macro-hack"
+version = "0.5.20+deprecated"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dc375e1527247fe1a97d8b7156678dfe7c1af2fc075c9a4db3690ecd2a148068"
+
+[[package]]
+name = "proc-macro2"
+version = "1.0.63"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7b368fba921b0dce7e60f5e04ec15e565b3303972b42bcfde1d0713b881959eb"
+dependencies = [
+ "unicode-ident",
+]
+
+[[package]]
+name = "quick-xml"
+version = "0.23.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "11bafc859c6815fbaffbbbf4229ecb767ac913fecb27f9ad4343662e9ef099ea"
+dependencies = [
+ "memchr",
+]
+
+[[package]]
+name = "quick-xml"
+version = "0.28.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0ce5e73202a820a31f8a0ee32ada5e21029c81fd9e3ebf668a40832e4219d9d1"
+dependencies = [
+ "memchr",
+]
+
+[[package]]
+name = "quote"
+version = "1.0.29"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "573015e8ab27661678357f27dc26460738fd2b6c86e46f386fde94cb5d913105"
+dependencies = [
+ "proc-macro2",
+]
+
+[[package]]
+name = "rand"
+version = "0.7.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6a6b1679d49b24bbfe0c803429aa1874472f50d9b363131f0e89fc356b544d03"
+dependencies = [
+ "getrandom 0.1.16",
+ "libc",
+ "rand_chacha 0.2.2",
+ "rand_core 0.5.1",
+ "rand_hc",
+ "rand_pcg",
+]
+
+[[package]]
+name = "rand"
+version = "0.8.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "34af8d1a0e25924bc5b7c43c079c942339d8f0a8b57c39049bef581b46327404"
+dependencies = [
+ "libc",
+ "rand_chacha 0.3.1",
+ "rand_core 0.6.4",
+]
+
+[[package]]
+name = "rand_chacha"
+version = "0.2.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f4c8ed856279c9737206bf725bf36935d8666ead7aa69b52be55af369d193402"
+dependencies = [
+ "ppv-lite86",
+ "rand_core 0.5.1",
+]
+
+[[package]]
+name = "rand_chacha"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e6c10a63a0fa32252be49d21e7709d4d4baf8d231c2dbce1eaa8141b9b127d88"
+dependencies = [
+ "ppv-lite86",
+ "rand_core 0.6.4",
+]
+
+[[package]]
+name = "rand_core"
+version = "0.5.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "90bde5296fc891b0cef12a6d03ddccc162ce7b2aff54160af9338f8d40df6d19"
+dependencies = [
+ "getrandom 0.1.16",
+]
+
+[[package]]
+name = "rand_core"
+version = "0.6.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ec0be4795e2f6a28069bec0b5ff3e2ac9bafc99e6a9a7dc3547996c5c816922c"
+dependencies = [
+ "getrandom 0.2.10",
+]
+
+[[package]]
+name = "rand_hc"
+version = "0.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ca3129af7b92a17112d59ad498c6f81eaf463253766b90396d39ea7a39d6613c"
+dependencies = [
+ "rand_core 0.5.1",
+]
+
+[[package]]
+name = "rand_pcg"
+version = "0.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "16abd0c1b639e9eb4d7c50c0b8100b0d0f849be2349829c740fe8e6eb4816429"
+dependencies = [
+ "rand_core 0.5.1",
+]
+
+[[package]]
+name = "raw-window-handle"
+version = "0.5.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f2ff9a1f06a88b01621b7ae906ef0211290d1c8a168a15542486a8f61c0833b9"
+
+[[package]]
+name = "redox_syscall"
+version = "0.2.16"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fb5a58c1855b4b6819d59012155603f0b22ad30cad752600aadfcb695265519a"
+dependencies = [
+ "bitflags",
+]
+
+[[package]]
+name = "redox_syscall"
+version = "0.3.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "567664f262709473930a4bf9e51bf2ebf3348f2e748ccc50dea20646858f8f29"
+dependencies = [
+ "bitflags",
+]
+
+[[package]]
+name = "redox_users"
+version = "0.4.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b033d837a7cf162d7993aded9304e30a83213c648b6e389db233191f891e5c2b"
+dependencies = [
+ "getrandom 0.2.10",
+ "redox_syscall 0.2.16",
+ "thiserror",
+]
+
+[[package]]
+name = "regex"
+version = "1.8.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d0ab3ca65655bb1e41f2a8c8cd662eb4fb035e67c3f78da1d61dffe89d07300f"
+dependencies = [
+ "aho-corasick 1.0.2",
+ "memchr",
+ "regex-syntax 0.7.2",
+]
+
+[[package]]
+name = "regex-automata"
+version = "0.1.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6c230d73fb8d8c1b9c0b3135c5142a8acee3a0558fb8db5cf1cb65f8d7862132"
+dependencies = [
+ "regex-syntax 0.6.29",
+]
+
+[[package]]
+name = "regex-syntax"
+version = "0.6.29"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f162c6dd7b008981e4d40210aca20b4bd0f9b60ca9271061b07f78537722f2e1"
+
+[[package]]
+name = "regex-syntax"
+version = "0.7.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "436b050e76ed2903236f032a59761c1eb99e1b0aead2c257922771dab1fc8c78"
+
+[[package]]
+name = "reqwest"
+version = "0.11.18"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cde824a14b7c14f85caff81225f411faacc04a2013f41670f41443742b1c1c55"
+dependencies = [
+ "base64 0.21.2",
+ "bytes",
+ "encoding_rs",
+ "futures-core",
+ "futures-util",
+ "h2",
+ "http",
+ "http-body",
+ "hyper",
+ "hyper-tls",
+ "ipnet",
+ "js-sys",
+ "log",
+ "mime",
+ "native-tls",
+ "once_cell",
+ "percent-encoding",
+ "pin-project-lite",
+ "serde",
+ "serde_json",
+ "serde_urlencoded",
+ "tokio",
+ "tokio-native-tls",
+ "tokio-util",
+ "tower-service",
+ "url",
+ "wasm-bindgen",
+ "wasm-bindgen-futures",
+ "wasm-streams",
+ "web-sys",
+ "winreg 0.10.1",
+]
+
+[[package]]
+name = "rfd"
+version = "0.10.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0149778bd99b6959285b0933288206090c50e2327f47a9c463bfdbf45c8823ea"
+dependencies = [
+ "block",
+ "dispatch",
+ "glib-sys",
+ "gobject-sys",
+ "gtk-sys",
+ "js-sys",
+ "lazy_static",
+ "log",
+ "objc",
+ "objc-foundation",
+ "objc_id",
+ "raw-window-handle",
+ "wasm-bindgen",
+ "wasm-bindgen-futures",
+ "web-sys",
+ "windows 0.37.0",
+]
+
+[[package]]
+name = "rustc-demangle"
+version = "0.1.23"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d626bb9dae77e28219937af045c257c28bfd3f69333c512553507f5f9798cb76"
+
+[[package]]
+name = "rustc_version"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bfa0f585226d2e68097d4f95d113b15b83a82e819ab25717ec0590d9584ef366"
+dependencies = [
+ "semver",
+]
+
+[[package]]
+name = "rustix"
+version = "0.37.21"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "62f25693a73057a1b4cb56179dd3c7ea21a7c6c5ee7d85781f5749b46f34b79c"
+dependencies = [
+ "bitflags",
+ "errno",
+ "io-lifetimes",
+ "libc",
+ "linux-raw-sys",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "rustversion"
+version = "1.0.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4f3208ce4d8448b3f3e7d168a73f5e0c43a61e32930de3bceeccedb388b6bf06"
+
+[[package]]
+name = "ryu"
+version = "1.0.13"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f91339c0467de62360649f8d3e185ca8de4224ff281f66000de5eb2a77a79041"
+
+[[package]]
+name = "safemem"
+version = "0.3.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ef703b7cb59335eae2eb93ceb664c0eb7ea6bf567079d843e09420219668e072"
+
+[[package]]
+name = "same-file"
+version = "1.0.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "93fc1dc3aaa9bfed95e02e6eadabb4baf7e3078b0bd1b4d7b6b0b68378900502"
+dependencies = [
+ "winapi-util",
+]
+
+[[package]]
+name = "schannel"
+version = "0.1.22"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0c3733bf4cf7ea0880754e19cb5a462007c4a8c1914bff372ccc95b464f1df88"
+dependencies = [
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "scoped-tls"
+version = "1.0.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e1cf6437eb19a8f4a6cc0f7dca544973b0b78843adbfeb3683d1a94a0024a294"
+
+[[package]]
+name = "scopeguard"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d29ab0c6d3fc0ee92fe66e2d99f700eab17a8d57d1c1d3b748380fb20baa78cd"
+
+[[package]]
+name = "security-framework"
+version = "2.9.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1fc758eb7bffce5b308734e9b0c1468893cae9ff70ebf13e7090be8dcbcc83a8"
+dependencies = [
+ "bitflags",
+ "core-foundation",
+ "core-foundation-sys",
+ "libc",
+ "security-framework-sys",
+]
+
+[[package]]
+name = "security-framework-sys"
+version = "2.9.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f51d0c0d83bec45f16480d0ce0058397a69e48fcdc52d1dc8855fb68acbd31a7"
+dependencies = [
+ "core-foundation-sys",
+ "libc",
+]
+
+[[package]]
+name = "selectors"
+version = "0.22.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "df320f1889ac4ba6bc0cdc9c9af7af4bd64bb927bccdf32d81140dc1f9be12fe"
+dependencies = [
+ "bitflags",
+ "cssparser",
+ "derive_more",
+ "fxhash",
+ "log",
+ "matches",
+ "phf 0.8.0",
+ "phf_codegen",
+ "precomputed-hash",
+ "servo_arc",
+ "smallvec",
+ "thin-slice",
+]
+
+[[package]]
+name = "semver"
+version = "1.0.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bebd363326d05ec3e2f532ab7660680f3b02130d780c299bca73469d521bc0ed"
+dependencies = [
+ "serde",
+]
+
+[[package]]
+name = "serde"
+version = "1.0.164"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9e8c8cf938e98f769bc164923b06dce91cea1751522f46f8466461af04c9027d"
+dependencies = [
+ "serde_derive",
+]
+
+[[package]]
+name = "serde_derive"
+version = "1.0.164"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d9735b638ccc51c28bf6914d90a2e9725b377144fc612c49a611fddd1b631d68"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "serde_json"
+version = "1.0.99"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "46266871c240a00b8f503b877622fe33430b3c7d963bdc0f2adc511e54a1eae3"
+dependencies = [
+ "itoa 1.0.6",
+ "ryu",
+ "serde",
+]
+
+[[package]]
+name = "serde_repr"
+version = "0.1.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bcec881020c684085e55a25f7fd888954d56609ef363479dc5a1305eb0d40cab"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "serde_spanned"
+version = "0.6.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "96426c9936fd7a0124915f9185ea1d20aa9445cc9821142f0a73bc9207a2e186"
+dependencies = [
+ "serde",
+]
+
+[[package]]
+name = "serde_urlencoded"
+version = "0.7.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d3491c14715ca2294c4d6a88f15e84739788c1d030eed8c110436aafdaa2f3fd"
+dependencies = [
+ "form_urlencoded",
+ "itoa 1.0.6",
+ "ryu",
+ "serde",
+]
+
+[[package]]
+name = "serde_with"
+version = "3.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9f02d8aa6e3c385bf084924f660ce2a3a6bd333ba55b35e8590b321f35d88513"
+dependencies = [
+ "base64 0.21.2",
+ "chrono",
+ "hex",
+ "indexmap 1.9.3",
+ "serde",
+ "serde_json",
+ "serde_with_macros",
+ "time",
+]
+
+[[package]]
+name = "serde_with_macros"
+version = "3.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "edc7d5d3932fb12ce722ee5e64dd38c504efba37567f0c402f6ca728c3b8b070"
+dependencies = [
+ "darling",
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "serialize-to-javascript"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c9823f2d3b6a81d98228151fdeaf848206a7855a7a042bbf9bf870449a66cafb"
+dependencies = [
+ "serde",
+ "serde_json",
+ "serialize-to-javascript-impl",
+]
+
+[[package]]
+name = "serialize-to-javascript-impl"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "74064874e9f6a15f04c1f3cb627902d0e6b410abbf36668afa873c61889f1763"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "servo_arc"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d98238b800e0d1576d8b6e3de32827c2d74bee68bb97748dcf5071fb53965432"
+dependencies = [
+ "nodrop",
+ "stable_deref_trait",
+]
+
+[[package]]
+name = "sha1"
+version = "0.10.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f04293dc80c3993519f2d7f6f511707ee7094fe0c6d3406feb330cdb3540eba3"
+dependencies = [
+ "cfg-if",
+ "cpufeatures",
+ "digest",
+]
+
+[[package]]
+name = "sha2"
+version = "0.10.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "479fb9d862239e610720565ca91403019f2f00410f1864c5aa7479b950a76ed8"
+dependencies = [
+ "cfg-if",
+ "cpufeatures",
+ "digest",
+]
+
+[[package]]
+name = "sharded-slab"
+version = "0.1.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "900fba806f70c630b0a382d0d825e17a0f19fcd059a2ade1ff237bcddf446b31"
+dependencies = [
+ "lazy_static",
+]
+
+[[package]]
+name = "shared_child"
+version = "1.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b0d94659ad3c2137fef23ae75b03d5241d633f8acded53d672decfa0e6e0caef"
+dependencies = [
+ "libc",
+ "winapi",
+]
+
+[[package]]
+name = "signal-hook"
+version = "0.3.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "732768f1176d21d09e076c23a93123d40bba92d50c4058da34d45c8de8e682b9"
+dependencies = [
+ "libc",
+ "signal-hook-registry",
+]
+
+[[package]]
+name = "signal-hook-registry"
+version = "1.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d8229b473baa5980ac72ef434c4415e70c4b5e71b423043adb4ba059f89c99a1"
+dependencies = [
+ "libc",
+]
+
+[[package]]
+name = "simd-adler32"
+version = "0.3.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "238abfbb77c1915110ad968465608b68e869e0772622c9656714e73e5a1a522f"
+
+[[package]]
+name = "siphasher"
+version = "0.3.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7bd3e3206899af3f8b12af284fafc038cc1dc2b41d1b89dd17297221c5d225de"
+
+[[package]]
+name = "slab"
+version = "0.4.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6528351c9bc8ab22353f9d776db39a20288e8d6c37ef8cfe3317cf875eecfc2d"
+dependencies = [
+ "autocfg",
+]
+
+[[package]]
+name = "smallvec"
+version = "1.10.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a507befe795404456341dfab10cef66ead4c041f62b8b11bbb92bffe5d0953e0"
+
+[[package]]
+name = "socket2"
+version = "0.4.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "64a4a911eed85daf18834cfaa86a79b7d266ff93ff5ba14005426219480ed662"
+dependencies = [
+ "libc",
+ "winapi",
+]
+
+[[package]]
+name = "soup2"
+version = "0.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b2b4d76501d8ba387cf0fefbe055c3e0a59891d09f0f995ae4e4b16f6b60f3c0"
+dependencies = [
+ "bitflags",
+ "gio",
+ "glib",
+ "libc",
+ "once_cell",
+ "soup2-sys",
+]
+
+[[package]]
+name = "soup2-sys"
+version = "0.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "009ef427103fcb17f802871647a7fa6c60cbb654b4c4e4c0ac60a31c5f6dc9cf"
+dependencies = [
+ "bitflags",
+ "gio-sys",
+ "glib-sys",
+ "gobject-sys",
+ "libc",
+ "system-deps 5.0.0",
+]
+
+[[package]]
+name = "stable_deref_trait"
+version = "1.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a8f112729512f8e442d81f95a8a7ddf2b7c6b8a1a6f509a95864142b30cab2d3"
+
+[[package]]
+name = "state"
+version = "0.5.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dbe866e1e51e8260c9eed836a042a5e7f6726bb2b411dffeaa712e19c388f23b"
+dependencies = [
+ "loom",
+]
+
+[[package]]
+name = "static_assertions"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a2eb9349b6444b326872e140eb1cf5e7c522154d69e7a0ffb0fb81c06b37543f"
+
+[[package]]
+name = "string_cache"
+version = "0.8.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f91138e76242f575eb1d3b38b4f1362f10d3a43f47d182a5b359af488a02293b"
+dependencies = [
+ "new_debug_unreachable",
+ "once_cell",
+ "parking_lot",
+ "phf_shared 0.10.0",
+ "precomputed-hash",
+ "serde",
+]
+
+[[package]]
+name = "string_cache_codegen"
+version = "0.5.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6bb30289b722be4ff74a408c3cc27edeaad656e06cb1fe8fa9231fa59c728988"
+dependencies = [
+ "phf_generator 0.10.0",
+ "phf_shared 0.10.0",
+ "proc-macro2",
+ "quote",
+]
+
+[[package]]
+name = "strsim"
+version = "0.10.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "73473c0e59e6d5812c5dfe2a064a6444949f089e20eec9a2e5506596494e4623"
+
+[[package]]
+name = "syn"
+version = "1.0.109"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "72b64191b275b66ffe2469e8af2c1cfe3bafa67b529ead792a6d0160888b4237"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "unicode-ident",
+]
+
+[[package]]
+name = "syn"
+version = "2.0.22"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2efbeae7acf4eabd6bcdcbd11c92f45231ddda7539edc7806bd1a04a03b24616"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "unicode-ident",
+]
+
+[[package]]
+name = "sys-locale"
+version = "0.2.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f8a11bd9c338fdba09f7881ab41551932ad42e405f61d01e8406baea71c07aee"
+dependencies = [
+ "js-sys",
+ "libc",
+ "wasm-bindgen",
+ "web-sys",
+ "windows-sys 0.45.0",
+]
+
+[[package]]
+name = "system-deps"
+version = "5.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "18db855554db7bd0e73e06cf7ba3df39f97812cb11d3f75e71c39bf45171797e"
+dependencies = [
+ "cfg-expr 0.9.1",
+ "heck 0.3.3",
+ "pkg-config",
+ "toml 0.5.11",
+ "version-compare 0.0.11",
+]
+
+[[package]]
+name = "system-deps"
+version = "6.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "30c2de8a4d8f4b823d634affc9cd2a74ec98c53a756f317e529a48046cbf71f3"
+dependencies = [
+ "cfg-expr 0.15.3",
+ "heck 0.4.1",
+ "pkg-config",
+ "toml 0.7.5",
+ "version-compare 0.1.1",
+]
+
+[[package]]
+name = "tao"
+version = "0.16.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6a6d198e01085564cea63e976ad1566c1ba2c2e4cc79578e35d9f05521505e31"
+dependencies = [
+ "bitflags",
+ "cairo-rs",
+ "cc",
+ "cocoa",
+ "core-foundation",
+ "core-graphics",
+ "crossbeam-channel",
+ "dispatch",
+ "gdk",
+ "gdk-pixbuf",
+ "gdk-sys",
+ "gdkwayland-sys",
+ "gdkx11-sys",
+ "gio",
+ "glib",
+ "glib-sys",
+ "gtk",
+ "image",
+ "instant",
+ "jni",
+ "lazy_static",
+ "libc",
+ "log",
+ "ndk",
+ "ndk-context",
+ "ndk-sys",
+ "objc",
+ "once_cell",
+ "parking_lot",
+ "png",
+ "raw-window-handle",
+ "scopeguard",
+ "serde",
+ "tao-macros",
+ "unicode-segmentation",
+ "uuid",
+ "windows 0.39.0",
+ "windows-implement",
+ "x11-dl",
+]
+
+[[package]]
+name = "tao-macros"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3b27a4bcc5eb524658234589bdffc7e7bfb996dbae6ce9393bfd39cb4159b445"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "tar"
+version = "0.4.38"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4b55807c0344e1e6c04d7c965f5289c39a8d94ae23ed5c0b57aabac549f871c6"
+dependencies = [
+ "filetime",
+ "libc",
+ "xattr",
+]
+
+[[package]]
+name = "target-lexicon"
+version = "0.12.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1b1c7f239eb94671427157bd93b3694320f3668d4e1eff08c7285366fd777fac"
+
+[[package]]
+name = "tauri"
+version = "1.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7fbe522898e35407a8e60dc3870f7579fea2fc262a6a6072eccdd37ae1e1d91e"
+dependencies = [
+ "anyhow",
+ "bytes",
+ "cocoa",
+ "dirs-next",
+ "embed_plist",
+ "encoding_rs",
+ "flate2",
+ "futures-util",
+ "glib",
+ "glob",
+ "gtk",
+ "heck 0.4.1",
+ "http",
+ "ignore",
+ "notify-rust",
+ "objc",
+ "once_cell",
+ "open",
+ "os_info",
+ "os_pipe",
+ "percent-encoding",
+ "rand 0.8.5",
+ "raw-window-handle",
+ "regex",
+ "reqwest",
+ "rfd",
+ "semver",
+ "serde",
+ "serde_json",
+ "serde_repr",
+ "serialize-to-javascript",
+ "shared_child",
+ "state",
+ "sys-locale",
+ "tar",
+ "tauri-macros",
+ "tauri-runtime",
+ "tauri-runtime-wry",
+ "tauri-utils",
+ "tempfile",
+ "thiserror",
+ "tokio",
+ "url",
+ "uuid",
+ "webkit2gtk",
+ "webview2-com",
+ "windows 0.39.0",
+]
+
+[[package]]
+name = "tauri-build"
+version = "1.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7d2edd6a259b5591c8efdeb9d5702cb53515b82a6affebd55c7fd6d3a27b7d1b"
+dependencies = [
+ "anyhow",
+ "cargo_toml",
+ "heck 0.4.1",
+ "json-patch",
+ "semver",
+ "serde",
+ "serde_json",
+ "tauri-utils",
+ "tauri-winres",
+]
+
+[[package]]
+name = "tauri-codegen"
+version = "1.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "54ad2d49fdeab4a08717f5b49a163bdc72efc3b1950b6758245fcde79b645e1a"
+dependencies = [
+ "base64 0.21.2",
+ "brotli",
+ "ico",
+ "json-patch",
+ "plist",
+ "png",
+ "proc-macro2",
+ "quote",
+ "regex",
+ "semver",
+ "serde",
+ "serde_json",
+ "sha2",
+ "tauri-utils",
+ "thiserror",
+ "time",
+ "uuid",
+ "walkdir",
+]
+
+[[package]]
+name = "tauri-macros"
+version = "1.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8eb12a2454e747896929338d93b0642144bb51e0dddbb36e579035731f0d76b7"
+dependencies = [
+ "heck 0.4.1",
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+ "tauri-codegen",
+ "tauri-utils",
+]
+
+[[package]]
+name = "tauri-runtime"
+version = "0.14.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "108683199cb18f96d2d4134187bb789964143c845d2d154848dda209191fd769"
+dependencies = [
+ "gtk",
+ "http",
+ "http-range",
+ "rand 0.8.5",
+ "raw-window-handle",
+ "serde",
+ "serde_json",
+ "tauri-utils",
+ "thiserror",
+ "url",
+ "uuid",
+ "webview2-com",
+ "windows 0.39.0",
+]
+
+[[package]]
+name = "tauri-runtime-wry"
+version = "0.14.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0b7aa256a1407a3a091b5d843eccc1a5042289baf0a43d1179d9f0fcfea37c1b"
+dependencies = [
+ "cocoa",
+ "gtk",
+ "percent-encoding",
+ "rand 0.8.5",
+ "raw-window-handle",
+ "tauri-runtime",
+ "tauri-utils",
+ "uuid",
+ "webkit2gtk",
+ "webview2-com",
+ "windows 0.39.0",
+ "wry",
+]
+
+[[package]]
+name = "tauri-utils"
+version = "1.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "03fc02bb6072bb397e1d473c6f76c953cda48b4a2d0cce605df284aa74a12e84"
+dependencies = [
+ "brotli",
+ "ctor",
+ "dunce",
+ "glob",
+ "heck 0.4.1",
+ "html5ever",
+ "infer",
+ "json-patch",
+ "kuchiki",
+ "memchr",
+ "phf 0.10.1",
+ "proc-macro2",
+ "quote",
+ "semver",
+ "serde",
+ "serde_json",
+ "serde_with",
+ "thiserror",
+ "url",
+ "walkdir",
+ "windows 0.39.0",
+]
+
+[[package]]
+name = "tauri-winres"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5993dc129e544393574288923d1ec447c857f3f644187f4fbf7d9a875fbfc4fb"
+dependencies = [
+ "embed-resource",
+ "toml 0.7.5",
+]
+
+[[package]]
+name = "tauri-winrt-notification"
+version = "0.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4f5bff1d532fead7c43324a0fa33643b8621a47ce2944a633be4cb6c0240898f"
+dependencies = [
+ "quick-xml 0.23.1",
+ "windows 0.39.0",
+]
+
+[[package]]
+name = "tempfile"
+version = "3.6.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "31c0432476357e58790aaa47a8efb0c5138f137343f3b5f23bd36a27e3b0a6d6"
+dependencies = [
+ "autocfg",
+ "cfg-if",
+ "fastrand",
+ "redox_syscall 0.3.5",
+ "rustix",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "tendril"
+version = "0.4.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d24a120c5fc464a3458240ee02c299ebcb9d67b5249c8848b09d639dca8d7bb0"
+dependencies = [
+ "futf",
+ "mac",
+ "utf-8",
+]
+
+[[package]]
+name = "thin-slice"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8eaa81235c7058867fa8c0e7314f33dcce9c215f535d1913822a2b3f5e289f3c"
+
+[[package]]
+name = "thiserror"
+version = "1.0.40"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "978c9a314bd8dc99be594bc3c175faaa9794be04a5a5e153caba6915336cebac"
+dependencies = [
+ "thiserror-impl",
+]
+
+[[package]]
+name = "thiserror-impl"
+version = "1.0.40"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f9456a42c5b0d803c8cd86e73dd7cc9edd429499f37a3550d286d5e86720569f"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "thread_local"
+version = "1.1.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3fdd6f064ccff2d6567adcb3873ca630700f00b5ad3f060c25b5dcfd9a4ce152"
+dependencies = [
+ "cfg-if",
+ "once_cell",
+]
+
+[[package]]
+name = "time"
+version = "0.3.22"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ea9e1b3cf1243ae005d9e74085d4d542f3125458f3a81af210d901dcd7411efd"
+dependencies = [
+ "itoa 1.0.6",
+ "serde",
+ "time-core",
+ "time-macros",
+]
+
+[[package]]
+name = "time-core"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7300fbefb4dadc1af235a9cef3737cea692a9d97e1b9cbcd4ebdae6f8868e6fb"
+
+[[package]]
+name = "time-macros"
+version = "0.2.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "372950940a5f07bf38dbe211d7283c9e6d7327df53794992d293e534c733d09b"
+dependencies = [
+ "time-core",
+]
+
+[[package]]
+name = "tinyvec"
+version = "1.6.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "87cc5ceb3875bb20c2890005a4e226a4651264a5c75edb2421b52861a0a0cb50"
+dependencies = [
+ "tinyvec_macros",
+]
+
+[[package]]
+name = "tinyvec_macros"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1f3ccbac311fea05f86f61904b462b55fb3df8837a366dfc601a0161d0532f20"
+
+[[package]]
+name = "tokio"
+version = "1.29.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "532826ff75199d5833b9d2c5fe410f29235e25704ee5f0ef599fb51c21f4a4da"
+dependencies = [
+ "autocfg",
+ "backtrace",
+ "bytes",
+ "libc",
+ "mio",
+ "num_cpus",
+ "pin-project-lite",
+ "socket2",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "tokio-native-tls"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bbae76ab933c85776efabc971569dd6119c580d8f5d448769dec1764bf796ef2"
+dependencies = [
+ "native-tls",
+ "tokio",
+]
+
+[[package]]
+name = "tokio-util"
+version = "0.7.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "806fe8c2c87eccc8b3267cbae29ed3ab2d0bd37fca70ab622e46aaa9375ddb7d"
+dependencies = [
+ "bytes",
+ "futures-core",
+ "futures-sink",
+ "pin-project-lite",
+ "tokio",
+ "tracing",
+]
+
+[[package]]
+name = "toml"
+version = "0.5.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f4f7f0dd8d50a853a531c426359045b1998f04219d88799810762cd4ad314234"
+dependencies = [
+ "serde",
+]
+
+[[package]]
+name = "toml"
+version = "0.7.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1ebafdf5ad1220cb59e7d17cf4d2c72015297b75b19a10472f99b89225089240"
+dependencies = [
+ "serde",
+ "serde_spanned",
+ "toml_datetime",
+ "toml_edit",
+]
+
+[[package]]
+name = "toml_datetime"
+version = "0.6.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7cda73e2f1397b1262d6dfdcef8aafae14d1de7748d66822d3bfeeb6d03e5e4b"
+dependencies = [
+ "serde",
+]
+
+[[package]]
+name = "toml_edit"
+version = "0.19.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "266f016b7f039eec8a1a80dfe6156b633d208b9fccca5e4db1d6775b0c4e34a7"
+dependencies = [
+ "indexmap 2.0.0",
+ "serde",
+ "serde_spanned",
+ "toml_datetime",
+ "winnow",
+]
+
+[[package]]
+name = "tower-service"
+version = "0.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b6bc1c9ce2b5135ac7f93c72918fc37feb872bdc6a5533a8b85eb4b86bfdae52"
+
+[[package]]
+name = "tracing"
+version = "0.1.37"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8ce8c33a8d48bd45d624a6e523445fd21ec13d3653cd51f681abf67418f54eb8"
+dependencies = [
+ "cfg-if",
+ "pin-project-lite",
+ "tracing-attributes",
+ "tracing-core",
+]
+
+[[package]]
+name = "tracing-attributes"
+version = "0.1.26"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5f4f31f56159e98206da9efd823404b79b6ef3143b4a7ab76e67b1751b25a4ab"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+]
+
+[[package]]
+name = "tracing-core"
+version = "0.1.31"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0955b8137a1df6f1a2e9a37d8a6656291ff0297c1a97c24e0d8425fe2312f79a"
+dependencies = [
+ "once_cell",
+ "valuable",
+]
+
+[[package]]
+name = "tracing-log"
+version = "0.1.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "78ddad33d2d10b1ed7eb9d1f518a5674713876e97e5bb9b7345a7984fbb4f922"
+dependencies = [
+ "lazy_static",
+ "log",
+ "tracing-core",
+]
+
+[[package]]
+name = "tracing-subscriber"
+version = "0.3.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "30a651bc37f915e81f087d86e62a18eec5f79550c7faff886f7090b4ea757c77"
+dependencies = [
+ "matchers",
+ "nu-ansi-term",
+ "once_cell",
+ "regex",
+ "sharded-slab",
+ "smallvec",
+ "thread_local",
+ "tracing",
+ "tracing-core",
+ "tracing-log",
+]
+
+[[package]]
+name = "treediff"
+version = "4.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "52984d277bdf2a751072b5df30ec0377febdb02f7696d64c2d7d54630bac4303"
+dependencies = [
+ "serde_json",
+]
+
+[[package]]
+name = "try-lock"
+version = "0.2.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3528ecfd12c466c6f163363caf2d02a71161dd5e1cc6ae7b34207ea2d42d81ed"
+
+[[package]]
+name = "typenum"
+version = "1.16.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "497961ef93d974e23eb6f433eb5fe1b7930b659f06d12dec6fc44a8f554c0bba"
+
+[[package]]
+name = "uds_windows"
+version = "1.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ce65604324d3cce9b966701489fbd0cf318cb1f7bd9dd07ac9a4ee6fb791930d"
+dependencies = [
+ "tempfile",
+ "winapi",
+]
+
+[[package]]
+name = "unicode-bidi"
+version = "0.3.13"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "92888ba5573ff080736b3648696b70cafad7d250551175acbaa4e0385b3e1460"
+
+[[package]]
+name = "unicode-ident"
+version = "1.0.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b15811caf2415fb889178633e7724bad2509101cde276048e013b9def5e51fa0"
+
+[[package]]
+name = "unicode-normalization"
+version = "0.1.22"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5c5713f0fc4b5db668a2ac63cdb7bb4469d8c9fed047b1d0292cc7b0ce2ba921"
+dependencies = [
+ "tinyvec",
+]
+
+[[package]]
+name = "unicode-segmentation"
+version = "1.10.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1dd624098567895118886609431a7c3b8f516e41d30e0643f03d94592a147e36"
+
+[[package]]
+name = "url"
+version = "2.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "50bff7831e19200a85b17131d085c25d7811bc4e186efdaf54bbd132994a88cb"
+dependencies = [
+ "form_urlencoded",
+ "idna",
+ "percent-encoding",
+ "serde",
+]
+
+[[package]]
+name = "utf-8"
+version = "0.7.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "09cc8ee72d2a9becf2f2febe0205bbed8fc6615b7cb429ad062dc7b7ddd036a9"
+
+[[package]]
+name = "uuid"
+version = "1.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d023da39d1fde5a8a3fe1f3e01ca9632ada0a63e9797de55a879d6e2236277be"
+dependencies = [
+ "getrandom 0.2.10",
+]
+
+[[package]]
+name = "valuable"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "830b7e5d4d90034032940e4ace0d9a9a057e7a45cd94e6c007832e39edb82f6d"
+
+[[package]]
+name = "vcpkg"
+version = "0.2.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "accd4ea62f7bb7a82fe23066fb0957d48ef677f6eeb8215f372f52e48bb32426"
+
+[[package]]
+name = "version-compare"
+version = "0.0.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1c18c859eead79d8b95d09e4678566e8d70105c4e7b251f707a03df32442661b"
+
+[[package]]
+name = "version-compare"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "579a42fc0b8e0c63b76519a339be31bed574929511fa53c1a3acae26eb258f29"
+
+[[package]]
+name = "version_check"
+version = "0.9.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "49874b5167b65d7193b8aba1567f5c7d93d001cafc34600cee003eda787e483f"
+
+[[package]]
+name = "vswhom"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "be979b7f07507105799e854203b470ff7c78a1639e330a58f183b5fea574608b"
+dependencies = [
+ "libc",
+ "vswhom-sys",
+]
+
+[[package]]
+name = "vswhom-sys"
+version = "0.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d3b17ae1f6c8a2b28506cd96d412eebf83b4a0ff2cbefeeb952f2f9dfa44ba18"
+dependencies = [
+ "cc",
+ "libc",
+]
+
+[[package]]
+name = "waker-fn"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9d5b2c62b4012a3e1eca5a7e077d13b3bf498c4073e33ccd58626607748ceeca"
+
+[[package]]
+name = "walkdir"
+version = "2.3.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "36df944cda56c7d8d8b7496af378e6b16de9284591917d307c9b4d313c44e698"
+dependencies = [
+ "same-file",
+ "winapi-util",
+]
+
+[[package]]
+name = "want"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bfa7760aed19e106de2c7c0b581b509f2f25d3dacaf737cb82ac61bc6d760b0e"
+dependencies = [
+ "try-lock",
+]
+
+[[package]]
+name = "wasi"
+version = "0.9.0+wasi-snapshot-preview1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cccddf32554fecc6acb585f82a32a72e28b48f8c4c1883ddfeeeaa96f7d8e519"
+
+[[package]]
+name = "wasi"
+version = "0.11.0+wasi-snapshot-preview1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9c8d87e72b64a3b4db28d11ce29237c246188f4f51057d65a7eab63b7987e423"
+
+[[package]]
+name = "wasm-bindgen"
+version = "0.2.87"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7706a72ab36d8cb1f80ffbf0e071533974a60d0a308d01a5d0375bf60499a342"
+dependencies = [
+ "cfg-if",
+ "wasm-bindgen-macro",
+]
+
+[[package]]
+name = "wasm-bindgen-backend"
+version = "0.2.87"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5ef2b6d3c510e9625e5fe6f509ab07d66a760f0885d858736483c32ed7809abd"
+dependencies = [
+ "bumpalo",
+ "log",
+ "once_cell",
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+ "wasm-bindgen-shared",
+]
+
+[[package]]
+name = "wasm-bindgen-futures"
+version = "0.4.37"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c02dbc21516f9f1f04f187958890d7e6026df8d16540b7ad9492bc34a67cea03"
+dependencies = [
+ "cfg-if",
+ "js-sys",
+ "wasm-bindgen",
+ "web-sys",
+]
+
+[[package]]
+name = "wasm-bindgen-macro"
+version = "0.2.87"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dee495e55982a3bd48105a7b947fd2a9b4a8ae3010041b9e0faab3f9cd028f1d"
+dependencies = [
+ "quote",
+ "wasm-bindgen-macro-support",
+]
+
+[[package]]
+name = "wasm-bindgen-macro-support"
+version = "0.2.87"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "54681b18a46765f095758388f2d0cf16eb8d4169b639ab575a8f5693af210c7b"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.22",
+ "wasm-bindgen-backend",
+ "wasm-bindgen-shared",
+]
+
+[[package]]
+name = "wasm-bindgen-shared"
+version = "0.2.87"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ca6ad05a4870b2bf5fe995117d3728437bd27d7cd5f06f13c17443ef369775a1"
+
+[[package]]
+name = "wasm-streams"
+version = "0.2.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6bbae3363c08332cadccd13b67db371814cd214c2524020932f0804b8cf7c078"
+dependencies = [
+ "futures-util",
+ "js-sys",
+ "wasm-bindgen",
+ "wasm-bindgen-futures",
+ "web-sys",
+]
+
+[[package]]
+name = "web-sys"
+version = "0.3.64"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9b85cbef8c220a6abc02aefd892dfc0fc23afb1c6a426316ec33253a3877249b"
+dependencies = [
+ "js-sys",
+ "wasm-bindgen",
+]
+
+[[package]]
+name = "webkit2gtk"
+version = "0.18.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b8f859735e4a452aeb28c6c56a852967a8a76c8eb1cc32dbf931ad28a13d6370"
+dependencies = [
+ "bitflags",
+ "cairo-rs",
+ "gdk",
+ "gdk-sys",
+ "gio",
+ "gio-sys",
+ "glib",
+ "glib-sys",
+ "gobject-sys",
+ "gtk",
+ "gtk-sys",
+ "javascriptcore-rs",
+ "libc",
+ "once_cell",
+ "soup2",
+ "webkit2gtk-sys",
+]
+
+[[package]]
+name = "webkit2gtk-sys"
+version = "0.18.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4d76ca6ecc47aeba01ec61e480139dda143796abcae6f83bcddf50d6b5b1dcf3"
+dependencies = [
+ "atk-sys",
+ "bitflags",
+ "cairo-sys-rs",
+ "gdk-pixbuf-sys",
+ "gdk-sys",
+ "gio-sys",
+ "glib-sys",
+ "gobject-sys",
+ "gtk-sys",
+ "javascriptcore-rs-sys",
+ "libc",
+ "pango-sys",
+ "pkg-config",
+ "soup2-sys",
+ "system-deps 6.1.1",
+]
+
+[[package]]
+name = "webview2-com"
+version = "0.19.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b4a769c9f1a64a8734bde70caafac2b96cada12cd4aefa49196b3a386b8b4178"
+dependencies = [
+ "webview2-com-macros",
+ "webview2-com-sys",
+ "windows 0.39.0",
+ "windows-implement",
+]
+
+[[package]]
+name = "webview2-com-macros"
+version = "0.6.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "eaebe196c01691db62e9e4ca52c5ef1e4fd837dcae27dae3ada599b5a8fd05ac"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "webview2-com-sys"
+version = "0.19.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "aac48ef20ddf657755fdcda8dfed2a7b4fc7e4581acce6fe9b88c3d64f29dee7"
+dependencies = [
+ "regex",
+ "serde",
+ "serde_json",
+ "thiserror",
+ "windows 0.39.0",
+ "windows-bindgen",
+ "windows-metadata",
+]
+
+[[package]]
+name = "winapi"
+version = "0.3.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5c839a674fcd7a98952e593242ea400abe93992746761e38641405d28b00f419"
+dependencies = [
+ "winapi-i686-pc-windows-gnu",
+ "winapi-x86_64-pc-windows-gnu",
+]
+
+[[package]]
+name = "winapi-i686-pc-windows-gnu"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"
+
+[[package]]
+name = "winapi-util"
+version = "0.1.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "70ec6ce85bb158151cae5e5c87f95a8e97d2c0c4b001223f33a334e3ce5de178"
+dependencies = [
+ "winapi",
+]
+
+[[package]]
+name = "winapi-x86_64-pc-windows-gnu"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f"
+
+[[package]]
+name = "windows"
+version = "0.37.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "57b543186b344cc61c85b5aab0d2e3adf4e0f99bc076eff9aa5927bcc0b8a647"
+dependencies = [
+ "windows_aarch64_msvc 0.37.0",
+ "windows_i686_gnu 0.37.0",
+ "windows_i686_msvc 0.37.0",
+ "windows_x86_64_gnu 0.37.0",
+ "windows_x86_64_msvc 0.37.0",
+]
+
+[[package]]
+name = "windows"
+version = "0.39.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f1c4bd0a50ac6020f65184721f758dba47bb9fbc2133df715ec74a237b26794a"
+dependencies = [
+ "windows-implement",
+ "windows_aarch64_msvc 0.39.0",
+ "windows_i686_gnu 0.39.0",
+ "windows_i686_msvc 0.39.0",
+ "windows_x86_64_gnu 0.39.0",
+ "windows_x86_64_msvc 0.39.0",
+]
+
+[[package]]
+name = "windows"
+version = "0.48.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e686886bc078bc1b0b600cac0147aadb815089b6e4da64016cbd754b6342700f"
+dependencies = [
+ "windows-targets 0.48.1",
+]
+
+[[package]]
+name = "windows-bindgen"
+version = "0.39.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "68003dbd0e38abc0fb85b939240f4bce37c43a5981d3df37ccbaaa981b47cb41"
+dependencies = [
+ "windows-metadata",
+ "windows-tokens",
+]
+
+[[package]]
+name = "windows-implement"
+version = "0.39.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ba01f98f509cb5dc05f4e5fc95e535f78260f15fea8fe1a8abdd08f774f1cee7"
+dependencies = [
+ "syn 1.0.109",
+ "windows-tokens",
+]
+
+[[package]]
+name = "windows-metadata"
+version = "0.39.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9ee5e275231f07c6e240d14f34e1b635bf1faa1c76c57cfd59a5cdb9848e4278"
+
+[[package]]
+name = "windows-sys"
+version = "0.42.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5a3e1820f08b8513f676f7ab6c1f99ff312fb97b553d30ff4dd86f9f15728aa7"
+dependencies = [
+ "windows_aarch64_gnullvm 0.42.2",
+ "windows_aarch64_msvc 0.42.2",
+ "windows_i686_gnu 0.42.2",
+ "windows_i686_msvc 0.42.2",
+ "windows_x86_64_gnu 0.42.2",
+ "windows_x86_64_gnullvm 0.42.2",
+ "windows_x86_64_msvc 0.42.2",
+]
+
+[[package]]
+name = "windows-sys"
+version = "0.45.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "75283be5efb2831d37ea142365f009c02ec203cd29a3ebecbc093d52315b66d0"
+dependencies = [
+ "windows-targets 0.42.2",
+]
+
+[[package]]
+name = "windows-sys"
+version = "0.48.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "677d2418bec65e3338edb076e806bc1ec15693c5d0104683f2efe857f61056a9"
+dependencies = [
+ "windows-targets 0.48.1",
+]
+
+[[package]]
+name = "windows-targets"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8e5180c00cd44c9b1c88adb3693291f1cd93605ded80c250a75d472756b4d071"
+dependencies = [
+ "windows_aarch64_gnullvm 0.42.2",
+ "windows_aarch64_msvc 0.42.2",
+ "windows_i686_gnu 0.42.2",
+ "windows_i686_msvc 0.42.2",
+ "windows_x86_64_gnu 0.42.2",
+ "windows_x86_64_gnullvm 0.42.2",
+ "windows_x86_64_msvc 0.42.2",
+]
+
+[[package]]
+name = "windows-targets"
+version = "0.48.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "05d4b17490f70499f20b9e791dcf6a299785ce8af4d709018206dc5b4953e95f"
+dependencies = [
+ "windows_aarch64_gnullvm 0.48.0",
+ "windows_aarch64_msvc 0.48.0",
+ "windows_i686_gnu 0.48.0",
+ "windows_i686_msvc 0.48.0",
+ "windows_x86_64_gnu 0.48.0",
+ "windows_x86_64_gnullvm 0.48.0",
+ "windows_x86_64_msvc 0.48.0",
+]
+
+[[package]]
+name = "windows-tokens"
+version = "0.39.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f838de2fe15fe6bac988e74b798f26499a8b21a9d97edec321e79b28d1d7f597"
+
+[[package]]
+name = "windows_aarch64_gnullvm"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "597a5118570b68bc08d8d59125332c54f1ba9d9adeedeef5b99b02ba2b0698f8"
+
+[[package]]
+name = "windows_aarch64_gnullvm"
+version = "0.48.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "91ae572e1b79dba883e0d315474df7305d12f569b400fcf90581b06062f7e1bc"
+
+[[package]]
+name = "windows_aarch64_msvc"
+version = "0.37.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2623277cb2d1c216ba3b578c0f3cf9cdebeddb6e66b1b218bb33596ea7769c3a"
+
+[[package]]
+name = "windows_aarch64_msvc"
+version = "0.39.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ec7711666096bd4096ffa835238905bb33fb87267910e154b18b44eaabb340f2"
+
+[[package]]
+name = "windows_aarch64_msvc"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e08e8864a60f06ef0d0ff4ba04124db8b0fb3be5776a5cd47641e942e58c4d43"
+
+[[package]]
+name = "windows_aarch64_msvc"
+version = "0.48.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b2ef27e0d7bdfcfc7b868b317c1d32c641a6fe4629c171b8928c7b08d98d7cf3"
+
+[[package]]
+name = "windows_i686_gnu"
+version = "0.37.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d3925fd0b0b804730d44d4b6278c50f9699703ec49bcd628020f46f4ba07d9e1"
+
+[[package]]
+name = "windows_i686_gnu"
+version = "0.39.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "763fc57100a5f7042e3057e7e8d9bdd7860d330070251a73d003563a3bb49e1b"
+
+[[package]]
+name = "windows_i686_gnu"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c61d927d8da41da96a81f029489353e68739737d3beca43145c8afec9a31a84f"
+
+[[package]]
+name = "windows_i686_gnu"
+version = "0.48.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "622a1962a7db830d6fd0a69683c80a18fda201879f0f447f065a3b7467daa241"
+
+[[package]]
+name = "windows_i686_msvc"
+version = "0.37.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ce907ac74fe331b524c1298683efbf598bb031bc84d5e274db2083696d07c57c"
+
+[[package]]
+name = "windows_i686_msvc"
+version = "0.39.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7bc7cbfe58828921e10a9f446fcaaf649204dcfe6c1ddd712c5eebae6bda1106"
+
+[[package]]
+name = "windows_i686_msvc"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "44d840b6ec649f480a41c8d80f9c65108b92d89345dd94027bfe06ac444d1060"
+
+[[package]]
+name = "windows_i686_msvc"
+version = "0.48.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4542c6e364ce21bf45d69fdd2a8e455fa38d316158cfd43b3ac1c5b1b19f8e00"
+
+[[package]]
+name = "windows_x86_64_gnu"
+version = "0.37.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2babfba0828f2e6b32457d5341427dcbb577ceef556273229959ac23a10af33d"
+
+[[package]]
+name = "windows_x86_64_gnu"
+version = "0.39.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6868c165637d653ae1e8dc4d82c25d4f97dd6605eaa8d784b5c6e0ab2a252b65"
+
+[[package]]
+name = "windows_x86_64_gnu"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8de912b8b8feb55c064867cf047dda097f92d51efad5b491dfb98f6bbb70cb36"
+
+[[package]]
+name = "windows_x86_64_gnu"
+version = "0.48.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ca2b8a661f7628cbd23440e50b05d705db3686f894fc9580820623656af974b1"
+
+[[package]]
+name = "windows_x86_64_gnullvm"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "26d41b46a36d453748aedef1486d5c7a85db22e56aff34643984ea85514e94a3"
+
+[[package]]
+name = "windows_x86_64_gnullvm"
+version = "0.48.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7896dbc1f41e08872e9d5e8f8baa8fdd2677f29468c4e156210174edc7f7b953"
+
+[[package]]
+name = "windows_x86_64_msvc"
+version = "0.37.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f4dd6dc7df2d84cf7b33822ed5b86318fb1781948e9663bacd047fc9dd52259d"
+
+[[package]]
+name = "windows_x86_64_msvc"
+version = "0.39.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5e4d40883ae9cae962787ca76ba76390ffa29214667a111db9e0a1ad8377e809"
+
+[[package]]
+name = "windows_x86_64_msvc"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9aec5da331524158c6d1a4ac0ab1541149c0b9505fde06423b02f5ef0106b9f0"
+
+[[package]]
+name = "windows_x86_64_msvc"
+version = "0.48.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1a515f5799fe4961cb532f983ce2b23082366b898e52ffbce459c86f67c8378a"
+
+[[package]]
+name = "winnow"
+version = "0.4.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ca0ace3845f0d96209f0375e6d367e3eb87eb65d27d445bdc9f1843a26f39448"
+dependencies = [
+ "memchr",
+]
+
+[[package]]
+name = "winreg"
+version = "0.10.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "80d0f4e272c85def139476380b12f9ac60926689dd2e01d4923222f40580869d"
+dependencies = [
+ "winapi",
+]
+
+[[package]]
+name = "winreg"
+version = "0.11.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "76a1a57ff50e9b408431e8f97d5456f2807f8eb2a2cd79b06068fc87f8ecf189"
+dependencies = [
+ "cfg-if",
+ "winapi",
+]
+
+[[package]]
+name = "wry"
+version = "0.24.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "33748f35413c8a98d45f7a08832d848c0c5915501803d1faade5a4ebcd258cea"
+dependencies = [
+ "base64 0.13.1",
+ "block",
+ "cocoa",
+ "core-graphics",
+ "crossbeam-channel",
+ "dunce",
+ "gdk",
+ "gio",
+ "glib",
+ "gtk",
+ "html5ever",
+ "http",
+ "kuchiki",
+ "libc",
+ "log",
+ "objc",
+ "objc_id",
+ "once_cell",
+ "serde",
+ "serde_json",
+ "sha2",
+ "soup2",
+ "tao",
+ "thiserror",
+ "url",
+ "webkit2gtk",
+ "webkit2gtk-sys",
+ "webview2-com",
+ "windows 0.39.0",
+ "windows-implement",
+]
+
+[[package]]
+name = "x11"
+version = "2.21.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "502da5464ccd04011667b11c435cb992822c2c0dbde1770c988480d312a0db2e"
+dependencies = [
+ "libc",
+ "pkg-config",
+]
+
+[[package]]
+name = "x11-dl"
+version = "2.21.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "38735924fedd5314a6e548792904ed8c6de6636285cb9fec04d5b1db85c1516f"
+dependencies = [
+ "libc",
+ "once_cell",
+ "pkg-config",
+]
+
+[[package]]
+name = "xattr"
+version = "0.2.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6d1526bbe5aaeb5eb06885f4d987bcdfa5e23187055de9b83fe00156a821fabc"
+dependencies = [
+ "libc",
+]
+
+[[package]]
+name = "xdg-home"
+version = "1.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2769203cd13a0c6015d515be729c526d041e9cf2c0cc478d57faee85f40c6dcd"
+dependencies = [
+ "nix",
+ "winapi",
+]
+
+[[package]]
+name = "zbus"
+version = "3.14.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "31de390a2d872e4cd04edd71b425e29853f786dc99317ed72d73d6fcf5ebb948"
+dependencies = [
+ "async-broadcast",
+ "async-executor",
+ "async-fs",
+ "async-io",
+ "async-lock",
+ "async-process",
+ "async-recursion",
+ "async-task",
+ "async-trait",
+ "blocking",
+ "byteorder",
+ "derivative",
+ "enumflags2",
+ "event-listener",
+ "futures-core",
+ "futures-sink",
+ "futures-util",
+ "hex",
+ "nix",
+ "once_cell",
+ "ordered-stream",
+ "rand 0.8.5",
+ "serde",
+ "serde_repr",
+ "sha1",
+ "static_assertions",
+ "tracing",
+ "uds_windows",
+ "winapi",
+ "xdg-home",
+ "zbus_macros",
+ "zbus_names",
+ "zvariant",
+]
+
+[[package]]
+name = "zbus_macros"
+version = "3.14.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "41d1794a946878c0e807f55a397187c11fc7a038ba5d868e7db4f3bd7760bc9d"
+dependencies = [
+ "proc-macro-crate",
+ "proc-macro2",
+ "quote",
+ "regex",
+ "syn 1.0.109",
+ "zvariant_utils",
+]
+
+[[package]]
+name = "zbus_names"
+version = "2.6.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fb80bb776dbda6e23d705cf0123c3b95df99c4ebeaec6c2599d4a5419902b4a9"
+dependencies = [
+ "serde",
+ "static_assertions",
+ "zvariant",
+]
+
+[[package]]
+name = "zvariant"
+version = "3.15.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "44b291bee0d960c53170780af148dca5fa260a63cdd24f1962fa82e03e53338c"
+dependencies = [
+ "byteorder",
+ "enumflags2",
+ "libc",
+ "serde",
+ "static_assertions",
+ "zvariant_derive",
+]
+
+[[package]]
+name = "zvariant_derive"
+version = "3.15.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "934d7a7dfc310d6ee06c87ffe88ef4eca7d3e37bb251dece2ef93da8f17d8ecd"
+dependencies = [
+ "proc-macro-crate",
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+ "zvariant_utils",
+]
+
+[[package]]
+name = "zvariant_utils"
+version = "1.0.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7234f0d811589db492d16893e3f21e8e2fd282e6d01b0cddee310322062cc200"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/Cargo.toml b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/Cargo.toml
new file mode 100644
index 0000000000000000000000000000000000000000..e30fc2ffe9afb50c9c97fed5fd67e37a0777c41b
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/Cargo.toml
@@ -0,0 +1,25 @@
+[package]
+name = "infinite_image_browsing"
+version = "0.0.1"
+description = "Infinite Image Browsing"
+authors = ["zanllp"]
+license = "mit"
+repository = ""
+edition = "2021"
+
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[build-dependencies]
+tauri-build = { version = "1.4", features = [] }
+
+[dependencies]
+tauri = { version = "1.4", features = [ "api-all", "devtools", "process-command-api"] }
+serde = { version = "1.0", features = ["derive"] }
+serde_json = "1.0"
+reqwest = { version = "0.11", features = ["blocking", "json"] }
+
+[features]
+# this feature is used for production builds or when `devPath` points to the filesystem
+# DO NOT REMOVE!!
+custom-protocol = ["tauri/custom-protocol"]
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/build.rs b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/build.rs
new file mode 100644
index 0000000000000000000000000000000000000000..795b9b7c83dbfc6bf454958eef78f93e211703d2
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/build.rs
@@ -0,0 +1,3 @@
+fn main() {
+ tauri_build::build()
+}
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/128x128.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/128x128.png
new file mode 100644
index 0000000000000000000000000000000000000000..4398e925dacc3de445d58f32b998b0975c41196d
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/128x128.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/128x128@2x.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/128x128@2x.png
new file mode 100644
index 0000000000000000000000000000000000000000..dfe8ac4deb698e20adc85fac8998aaf623edd90b
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/128x128@2x.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/32x32.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/32x32.png
new file mode 100644
index 0000000000000000000000000000000000000000..0111967051321edbe23ebfd34578b9188e59998c
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/32x32.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square107x107Logo.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square107x107Logo.png
new file mode 100644
index 0000000000000000000000000000000000000000..77999ba944050316ecf5a1f13fc9aa0a4c7a8c9e
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square107x107Logo.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square142x142Logo.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square142x142Logo.png
new file mode 100644
index 0000000000000000000000000000000000000000..2ca2ea7ef24e853f32018c9b02dd026861734f5a
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square142x142Logo.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square150x150Logo.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square150x150Logo.png
new file mode 100644
index 0000000000000000000000000000000000000000..c5dd45eac6bcf7dfe42127c8757ee325f071da30
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square150x150Logo.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square284x284Logo.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square284x284Logo.png
new file mode 100644
index 0000000000000000000000000000000000000000..8052711ca8845696ea3668760a63991aade5390a
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square284x284Logo.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square30x30Logo.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square30x30Logo.png
new file mode 100644
index 0000000000000000000000000000000000000000..b2ea9712be0a92ad10997b8f6fd535e8e0535445
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square30x30Logo.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square310x310Logo.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square310x310Logo.png
new file mode 100644
index 0000000000000000000000000000000000000000..8d9025588da30e7ecea896a0a48511688c31709f
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square310x310Logo.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square44x44Logo.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square44x44Logo.png
new file mode 100644
index 0000000000000000000000000000000000000000..d76ef105578f9879d5e84dc987c6bbbc57b28949
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square44x44Logo.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square71x71Logo.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square71x71Logo.png
new file mode 100644
index 0000000000000000000000000000000000000000..84feca193d41983b04794f33b61d34b27c59bd23
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square71x71Logo.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square89x89Logo.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square89x89Logo.png
new file mode 100644
index 0000000000000000000000000000000000000000..5684cb88afa0b9ae803d51f902307de3a2233756
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/Square89x89Logo.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/StoreLogo.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/StoreLogo.png
new file mode 100644
index 0000000000000000000000000000000000000000..477ec1ab15392b6c5f852a869d1c35b4e085e7e9
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/StoreLogo.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/icon.icns b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/icon.icns
new file mode 100644
index 0000000000000000000000000000000000000000..0f79fdef3a73fcc512e7cba0e30988e3d3001bec
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/icon.icns differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/icon.ico b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/icon.ico
new file mode 100644
index 0000000000000000000000000000000000000000..eaefe0037ea42a1228b094c72110751222aef1f0
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/icon.ico differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/icon.png b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/icon.png
new file mode 100644
index 0000000000000000000000000000000000000000..16a616556b16d5f87d3c09eb8b75da6153f1fcfd
Binary files /dev/null and b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/icons/icon.png differ
diff --git a/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/src/main.rs b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/src/main.rs
new file mode 100644
index 0000000000000000000000000000000000000000..d157c62126222aa6ff740898f6db88ee4e65a29f
--- /dev/null
+++ b/TestingforfaivsixEx/content/eichellis/extensions/images-browser/vue/src-tauri/src/main.rs
@@ -0,0 +1,94 @@
+// Prevents additional console window on Windows in release, DO NOT REMOVE!!
+#![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]
+
+use serde::Deserialize;
+use std::fs::File;
+use std::io::prelude::*;
+use std::io::Error;
+use tauri::api::process::Command;
+use tauri::api::process::CommandEvent;
+use tauri::WindowEvent;
+// Learn more about Tauri commands at https://tauri.app/v1/guides/features/command
+#[tauri::command]
+fn greet(name: &str) -> String {
+ format!("Hello, {}! You've been greeted from Rust!", name)
+}
+struct AppState {
+ port: u16,
+}
+#[derive(serde::Serialize)]
+struct AppConf {
+ port: u16,
+}
+#[tauri::command]
+fn get_tauri_conf(state: tauri::State<'_, AppState>) -> AppConf {
+ AppConf { port: state.port }
+}
+
+#[derive(Deserialize, Default, Debug)]
+struct Config {
+ sdwebui_dir: String,
+}
+
+fn read_config_file(path: &str) -> ResultIf that doesn't work, click on FAQ for help