Upload app.py

app.py

@@ -1223,6 +1223,90 @@ def enhance_prompt(prompt, use_model, camera, framing, lighting, pace, add_quali
     )
 
 
+# ---------------------------------------------------------------------------
+# Naming CivitAI links
+# ---------------------------------------------------------------------------
+# A CivitAI download URL is a bare number, which is useless once three slots are
+# filled. The number is the *model version* id, and CivitAI's public API turns it
+# into the model's title, the version, the file behind `fileId`, and the words the
+# adapter was trained on. Public, cached, and never fatal.
+
+from functools import lru_cache
+
+CIVITAI_DOWNLOAD_RE = re.compile(r"/api/download/models/(\d+)")
+
+
+@lru_cache(maxsize=128)
+def civitai_details(version_id: str, file_id: str = ""):
+    import requests
+
+    headers = {"User-Agent": "Mozilla/5.0"}
+    token = os.environ.get("CIVITAI_TOKEN", "").strip()
+    if token:
+        headers["Authorization"] = f"Bearer {token}"
+
+    try:
+        response = requests.get(
+            f"https://civitai.com/api/v1/model-versions/{version_id}", headers=headers, timeout=20
+        )
+        response.raise_for_status()
+        data = response.json()
+    except Exception as error:
+        return f"CivitAI {version_id} (name unavailable: {type(error).__name__})", []
+
+    model_name = (data.get("model") or {}).get("name") or f"model {data.get('modelId', '?')}"
+    version_name = data.get("name") or ""
+    label = f"{model_name} · {version_name}".strip(" ·")
+
+    if file_id:
+        for entry in data.get("files") or []:
+            if str(entry.get("id")) == str(file_id):
+                label += f" · {entry.get('name', '')}"
+                break
+
+    return label, [word for word in (data.get("trainedWords") or []) if word]
+
+
+def describe_lora(reference: str):
+    """A readable line for whatever is in a box: a CivitAI link becomes its real title."""
+    reference = (reference or "").strip()
+    if not reference or reference == "(None)":
+        return "", []
+
+    match = CIVITAI_DOWNLOAD_RE.search(reference)
+    if match:
+        file_id = ""
+        if "fileId=" in reference:
+            file_id = reference.split("fileId=", 1)[1].split("&")[0]
+        return civitai_details(match.group(1), file_id)
+
+    if reference.startswith(("http://", "https://")):
+        return os.path.basename(reference.split("?")[0]) or reference, []
+    return reference, []
+
+
+def identify_loras(*fields):
+    """Name every filled box across the five slots. `fields` arrives as
+    `repo, high, low` per slot, so each slot is reported on one line."""
+    lines = []
+    for slot in range(len(fields) // 3):
+        repo, high, low = fields[slot * 3: slot * 3 + 3]
+        parts = []
+        for role, value in (("repo", repo), ("high", high), ("low", low)):
+            label, words = describe_lora(value)
+            if not label:
+                continue
+            entry = f"{role}: {label}"
+            if words:
+                entry += f" — triggers: {', '.join(words[:6])}"
+            parts.append(entry)
+        if parts:
+            lines.append(f"<b>Custom lora {slot + 1}</b> — " + " · ".join(parts))
+    if not lines:
+        return "<div class='status-idle'>Nothing in the slots yet.</div>"
+    return "<div class='status-ok'>" + "<br>".join(lines) + "</div>"
+
+
 def _target_size(image):
     """Same math as resize_image(), but returns only the final (w, h)."""
     if image is None:
@@ -1378,7 +1462,7 @@ CSS = """
     box-shadow: 0 8px 22px rgba(124, 58, 237, .28);
 }
 #estimate-btn, #enhance-btn { min-height: 42px !important; border-radius: 12px !important; }
-#profile-load, #profile-save, #profile-delete, #profile-refresh {
+#lora-identify, #profile-load, #profile-save, #profile-delete, #profile-refresh {
     min-height: 42px !important; border-radius: 12px !important; font-weight: 600 !important;
 }
 #stop-btn, #clear-btn { min-height: 46px !important; border-radius: 12px !important; }
@@ -1630,8 +1714,15 @@ with gr.Blocks(delete_cache=(3600, 86400)) as demo:
                             "two separate files, so paste the low-noise link straight into the Low box "
                             "(both boxes accept a pasted link). The Space downloads it itself, so being "
                             "signed in to CivitAI in your browser does not help a gated model: add "
-                            "`CIVITAI_TOKEN` under Settings → Variables and secrets. Downloads are cached."
+                            "`CIVITAI_TOKEN` under Settings → Variables and secrets. Downloads are cached.\n\n"
+                            "A CivitAI link is just a number, so press **🔎 name the CivitAI links** and "
+                            "each box is replaced by its real title, version, file name and trigger words, "
+                            "read from CivitAI's public API."
                         )
+                        with gr.Row():
+                            lora_identify_btn = gr.Button("🔎 name the CivitAI links",
+                                                          variant="secondary", elem_id="lora-identify")
+                        lora_names = gr.HTML("<div class='status-idle'>Nothing in the slots yet.</div>")
                         _my_rows = []
                         for _i in range(1, 6):
                             with gr.Accordion(f"Custom lora {_i}", open=(_i == 1)):
@@ -1871,6 +1962,16 @@ with gr.Blocks(delete_cache=(3600, 86400)) as demo:
         outputs=[merged_video, merged_file, merge_queue, merge_status],
     )
 
+    # Naming the links. Registered here with the rest of the additions, and reading every box of every slot.
+    _IDENTIFY_FIELDS = [
+        my1_repo, my1_high, my1_low,
+        my2_repo, my2_high, my2_low,
+        my3_repo, my3_high, my3_low,
+        my4_repo, my4_high, my4_low,
+        my5_repo, my5_high, my5_low,
+    ]
+    lora_identify_btn.click(identify_loras, _IDENTIFY_FIELDS, lora_names)
+
     _SETTINGS_FIELDS = [
         prompt_input, negative_prompt_input, duration_seconds_input, steps_slider, quality_slider,
         seed_input, randomize_seed_checkbox, scheduler_dropdown, flow_shift_slider, frame_multi,