nodes/Azornes-Comfyui-Model-Resolver
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add CivitAI URN download UI and metadata

Browse Source 
Move direct URN->CivitAI download handling into the frontend and enrich backend metadata. Removed server-side immediate URN download exposure and adjusted comment numbering. Backend now selects a primary file from resolved CivitAI model info, includes model_id, version_id, size, and a modelVersionId query on model URLs.

Frontend: add renderDownloadSourceSection to render download buttons and CivitAI model-card links, display file sizes, and format names. Store missingModels for later updates; render a placeholder URN download section while resolving; when URN resolution returns, inject civitai info into the missing entry, build a download_source object, replace the placeholder with the rendered download UI, and attach the download button handler. Improved error handling and model URL construction to include modelVersionId when available.
This commit is contained in:
Dariusz L
2026-04-24 20:38:11 +02:00
parent 53f52518ff
commit b64597fd3f
2 changed files with 120 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+19 -50
View File
@@ -210,54 +210,7 @@ class ModelLinkerExtension:
}
continue
# 1. For URNs, we already have an exact CivitAI identity.
# Expose it immediately as a direct download source.
if missing.get("is_urn"):
urn_model_id = missing.get("urn_model_id") or (
missing.get("urn", {}) or {}
).get("model_id")
urn_version_id = missing.get(
"urn_version_id"
) or (missing.get("urn", {}) or {}).get(
"version_id"
)
if urn_model_id and urn_version_id:
civitai_info = resolve_urn(
urn_model_id, urn_version_id
)
if civitai_info:
missing["civitai_info"] = civitai_info
expected_filename = (
civitai_info.get("expected_filename")
or filename
)
primary_file = (
civitai_info.get("files") or [{}]
)[0]
missing["download_source"] = {
"source": "civitai",
"url": get_civitai_download_url(
urn_version_id
),
"filename": expected_filename,
"name": civitai_info.get(
"model_name"
),
"type": missing.get("category"),
"directory": missing.get(
"category", "checkpoints"
),
"match_type": "exact",
"size": primary_file.get("size"),
"model_id": urn_model_id,
"version_id": urn_version_id,
"model_url": f"https://civitai.com/models/{urn_model_id}?modelVersionId={urn_version_id}",
}
continue
# 2. Check popular models (always exact match)
# 1. Check popular models (always exact match)
popular_info = get_popular_model_url(filename)
if popular_info:
missing["download_source"] = {
@@ -270,7 +223,7 @@ class ModelLinkerExtension:
}
continue
# 3. Check model list (ComfyUI Manager database)
# 2. Check model list (ComfyUI Manager database)
# Use exact_only=True to avoid confusing fuzzy matches for downloads
model_list_result = search_model_list(
filename, exact_only=True
@@ -772,6 +725,19 @@ class ModelLinkerExtension:
# Use resolve_urn to get model info (cached)
model_info = resolve_urn(model_id, version_id)
if model_info:
primary_file = None
for file_info in model_info.get("files", []):
if (
file_info.get("name")
== model_info.get("expected_filename")
):
primary_file = file_info
break
if primary_file is None:
primary_file = (
model_info.get("files") or [{}]
)[0]
download_url = get_civitai_download_url(
version_id
)
@@ -783,7 +749,10 @@ class ModelLinkerExtension:
),
"type": category,
"download_url": download_url,
"url": f"https://civitai.com/models/{model_id}",
"url": f"https://civitai.com/models/{model_id}?modelVersionId={version_id}",
"model_id": model_id,
"version_id": version_id,
"size": primary_file.get("size"),
}
results["found"] = True
elif category: