DemonGatanjieuandClaude Opus 5.5
5667fa2d9e
Types go by their usual names, base model chips narrow the list, the version stands apart from the name, and a model's versions are one card whose version opens the list to show another; a pick goes with it. The grouping, the families and the type names are the Models page's own. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
55 lines
2.1 KiB
Python
55 lines
2.1 KiB
Python
"""Every model file in a few words, for the canvas's quick search (GET /anomalous/quick_index,
|
|
web/modules/quick_search_data.js): its folder type, place and file, the name it shows, its
|
|
base model, a few trigger words and its cover's URL. Read once when the search bar opens;
|
|
the bar keeps it a little while. Files a folder type shares with another (unet and
|
|
diffusion_models) are listed once; node packs' own files are not models."""
|
|
|
|
import asyncio
|
|
|
|
from aiohttp import web
|
|
|
|
from .model_catalog import _collect_all_scan_models
|
|
|
|
MAX_WORDS = 6
|
|
|
|
|
|
def _words(meta):
|
|
words = meta.get("trainedWords") or meta.get("trigger_words") or []
|
|
if isinstance(words, str):
|
|
words = [words]
|
|
return [str(word) for word in words if str(word).strip()][:MAX_WORDS]
|
|
|
|
|
|
def compact_models(models):
|
|
"""`models` (the shape GET /anomalous/all_scan_models gives) in a few words each."""
|
|
seen, found = set(), []
|
|
for model in models:
|
|
if model.get("type") == "custom_nodes":
|
|
continue
|
|
key = (model.get("subfolder") or "", model.get("filename"), model.get("size_bytes"))
|
|
if key in seen:
|
|
continue
|
|
seen.add(key)
|
|
meta = model.get("metadata") or {}
|
|
found.append({
|
|
"type": model.get("type"),
|
|
"path_idx": model.get("path_idx", 0),
|
|
"subfolder": model.get("subfolder") or "",
|
|
"filename": model.get("filename"),
|
|
"name": str(meta.get("custom_name") or "").strip(),
|
|
"civitai_name": str(meta.get("name") or "").strip(),
|
|
"base": str(meta.get("baseModel") or "").strip(),
|
|
"base_guess": str(meta.get("base_guess") or "").strip(),
|
|
"page": meta.get("model_id") or 0,
|
|
"version": meta.get("version_id") or 0,
|
|
"version_name": str(meta.get("version_name") or "").strip(),
|
|
"words": _words(meta),
|
|
"preview_url": model.get("preview_url") or "",
|
|
})
|
|
return found
|
|
|
|
|
|
async def api_quick_index(request):
|
|
payload = await asyncio.to_thread(_collect_all_scan_models, 1, 0)
|
|
return web.json_response({"models": compact_models(payload.get("models") or [])})
|