jeremie louvaert
04e31f5b77
New feature: modal gallery UI over the vault, accessible from the top menu
bar under Extensions → AKURATE: Hash Vault Browser. Reads all sidecar
metadata + thumbnails shipped in v1.3.0 and presents them as a filterable
grid. No graph node required.
Backend: new GET /akurate/hash_vault/list route walks output/hash_vault/,
reads every {hash}.json sidecar, returns JSON sorted by last_accessed_at
desc with pt_size enrichment. Thumbnails reuse ComfyUI's built-in /view
route (type=output&subfolder=hash_vault) — zero new thumbnail endpoint.
Frontend: single web/hash_vault_browser.js (vanilla JS, no framework).
Modal with CSS grid, live substring filter on label+hash, click-to-copy
hash with toast, lazy-loaded thumbnails, relative-age display, ESC and
backdrop-click dismiss.
__init__.py now declares WEB_DIRECTORY and imports the route module.
Deferred to v1.5+: load-into-workflow action, delete, bulk ops, sort
controls, pagination. Current vault (57 entries, 2.33 GB .pt, 3.7 MB
thumbs) loads instantly so none of those are urgent.
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
"""Hash Vault Browser backend route.
|
|
|
|
Registers GET /akurate/hash_vault/list which walks output/hash_vault/, reads
|
|
every {hash}.json sidecar, and returns a JSON array sorted by last_accessed_at
|
|
desc. Thumbnails are served by ComfyUI's built-in /view route using
|
|
type=output&subfolder=hash_vault&filename={hash}.thumb.png — no extra route
|
|
needed on our side.
|
|
|
|
The .pt files themselves are never read here; they are opaque + expensive to
|
|
load, and the sidecar already carries everything the browser UI needs.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
from typing import Any
|
|
|
|
try:
|
|
import folder_paths
|
|
except ImportError:
|
|
folder_paths = None
|
|
|
|
try:
|
|
from server import PromptServer
|
|
from aiohttp import web
|
|
except ImportError:
|
|
PromptServer = None
|
|
web = None
|
|
|
|
|
|
def _vault_dir() -> str | None:
|
|
if folder_paths is None:
|
|
return None
|
|
return os.path.join(folder_paths.get_output_directory(), "hash_vault")
|
|
|
|
|
|
def _list_entries() -> list[dict[str, Any]]:
|
|
"""Walk hash_vault/, read sidecars, return enriched entries sorted by last_accessed_at desc."""
|
|
vault = _vault_dir()
|
|
if not vault or not os.path.isdir(vault):
|
|
return []
|
|
|
|
entries: list[dict[str, Any]] = []
|
|
for name in os.listdir(vault):
|
|
if not name.endswith(".json"):
|
|
continue
|
|
sidecar_path = os.path.join(vault, name)
|
|
hash_key = name[:-5] # strip .json
|
|
|
|
try:
|
|
with open(sidecar_path, "r", encoding="utf-8") as f:
|
|
sidecar = json.load(f)
|
|
except (OSError, json.JSONDecodeError):
|
|
continue
|
|
|
|
pt_path = os.path.join(vault, f"{hash_key}.pt")
|
|
pt_size = os.path.getsize(pt_path) if os.path.exists(pt_path) else 0
|
|
|
|
sidecar["pt_size"] = pt_size
|
|
sidecar["pt_exists"] = pt_size > 0
|
|
entries.append(sidecar)
|
|
|
|
entries.sort(key=lambda e: e.get("last_accessed_at", ""), reverse=True)
|
|
return entries
|
|
|
|
|
|
if PromptServer is not None and web is not None:
|
|
routes = PromptServer.instance.routes
|
|
|
|
@routes.get("/akurate/hash_vault/list")
|
|
async def hash_vault_list(request):
|
|
try:
|
|
data = _list_entries()
|
|
return web.json_response({"entries": data, "count": len(data)})
|
|
except Exception as e:
|
|
return web.json_response({"error": str(e)}, status=500)
|