Remove hardcoded preview url modifiers, use ResultType instead

Fixes #284
This commit is contained in:
DominikDoom
2024-04-15 18:48:44 +02:00
parent 4cd6174a22
commit a27633da55
2 changed files with 10 additions and 25 deletions

View File

@@ -860,34 +860,19 @@ async function updateSelectionStyle(textArea, newIndex, oldIndex) {
// Show preview if enabled and the selected type supports it
if (newIndex !== null) {
let selected = items[newIndex];
let previewTypes = ["v1 Embedding", "v2 Embedding", "Hypernetwork", "Lora", "Lyco"];
let selectedType = selected.querySelector(".acMetaText").innerText;
let selectedFilename = selected.querySelector(".acListItem").innerText;
let selectedResult = results[newIndex];
let selectedType = selectedResult.type;
// These types support previews (others could technically too, but are not native to the webui gallery)
let previewTypes = [ResultType.embedding, ResultType.hypernetwork, ResultType.lora, ResultType.lyco];
let previewDiv = gradioApp().querySelector(`.autocompleteParent${textAreaId} .sideInfo`);
if (TAC_CFG.showExtraNetworkPreviews && previewTypes.includes(selectedType)) {
let shorthandType = "";
switch (selectedType) {
case "v1 Embedding":
case "v2 Embedding":
shorthandType = "embed";
break;
case "Hypernetwork":
shorthandType = "hyper";
break;
case "Lora":
shorthandType = "lora";
break;
case "Lyco":
shorthandType = "lyco";
break;
}
let img = previewDiv.querySelector("img");
let url = await getTacExtraNetworkPreviewURL(selectedFilename, shorthandType);
// String representation of our type enum
const typeString = Object.keys(ResultType)[selectedType - 1].toLowerCase();
// Get image from API
let url = await getTacExtraNetworkPreviewURL(selectedResult.text, typeString);
if (url) {
img.src = url;
previewDiv.style.display = "block";

View File

@@ -727,9 +727,9 @@ def api_tac(_: gr.Blocks, app: FastAPI):
return LORA_PATH
elif type == "lyco":
return LYCO_PATH
elif type == "hyper":
elif type == "hypernetwork":
return HYP_PATH
elif type == "embed":
elif type == "embedding":
return EMB_PATH
else:
return None