Hot Patch For loading workflows with custom model, Increased search results to 20, Adjusted strength to be single Precision point, Added Image Captioning workflow

clientlibs/main.js

@@ -3,6 +3,7 @@ import { api } from "../../scripts/api.js";
 import { promptEnhanceHandler, syncDimensionsNodeHandler, searchNodeHandler, APIKeyHandler, captionNodeHandler, handleCustomErrors } from "./utils.js";
 import { RUNWARE_NODE_TYPES, RUNWARE_NODE_PROPS, SEARCH_TERMS } from "./types.js";
 
+const nodeInitList = [];
 app.registerExtension({
 	name: "runware.ai",
     async setup() {
@@ -47,9 +48,14 @@ app.registerExtension({
 
             if(crNodeProps.liveSearch === true) {
                 if(widgetType === "text" && SEARCH_TERMS.includes(widgetName)) {
+                    node.callback = function(){};
                     searchNodeHandler(node, nodeWidget);
+                    nodeInitList.push(node);
                 }
             }
         }
+    },
+    loadedGraphNode(node) {
+        if(nodeInitList.includes(node)) node.callback();
     }
 })

clientlibs/utils.js

@@ -333,8 +333,13 @@ async function searchNodeHandler(searchNode, searchInputWidget) {
         }
     }
 
-    async function searchModels() {
-        let searchQuery = searchInputWidget.value.trim();
+    async function searchModels(exactQuery = null, returnOutput = false) {
+        let searchQuery;
+        if(exactQuery) {
+            searchQuery = exactQuery;
+        } else {
+            searchQuery = searchInputWidget.value.trim();
+        }
         searchQuery = shortenAirCode(searchQuery);
         const modelArchValue = DEFAULT_MODELS_ARCH_LIST[modelArchWidget.value];
         if(isControlNet) {
@@ -365,9 +370,25 @@ async function searchNodeHandler(searchNode, searchInputWidget) {
                 if(isEmbedding) embeddingTriggerWordsList[modelObj.air] = modelObj.positiveTriggerWords.trim();
                 modelListArray.push(`${modelObj.air} (${modelObj.name} ${modelObj.version})`);
             });
+
+            if(exactQuery) {
+                const extraSearch = await searchModels(null, true);
+                if(extraSearch) {
+                    const newModelList = [...new Set([...modelListArray, ...extraSearch])];
+                    modelListWidget.options.values = newModelList;
+                    return;
+                }
+            }
+
+            if(returnOutput) return modelListArray;
             modelListWidget.options.values = modelListArray;
             modelListWidget.value = modelListArray[0];
         } else {
+            if(returnOutput) return false;
+            if(exactQuery) {
+                notifyUser("Failed To Load Workflow's Model List Value!", "error", "Runware Search");
+                return;
+            }
             notifyUser(modelSearchResults.error, "error", "Runware Search");
         }
     }
@@ -406,6 +427,7 @@ async function searchNodeHandler(searchNode, searchInputWidget) {
 
         runwareButton.addEventListener("click", async () => {
             const chosenLora = modelListWidget.value.split(" ")[0];
+            if(!triggerWordsList.hasOwnProperty(chosenLora)) return;
             const triggerWords = triggerWordsList[chosenLora];
             if(triggerWords.length > 0) {
                 const positivePromptWidget = findTargetWidget(searchNode, "positivePrompt");
@@ -434,6 +456,7 @@ async function searchNodeHandler(searchNode, searchInputWidget) {
 
         runwareButton.addEventListener("click", async () => {
             const chosenEmbedding = modelListWidget.value.split(" ")[0];
+            if (!embeddingTriggerWordsList.hasOwnProperty(chosenEmbedding)) return;
             const triggerWords = embeddingTriggerWordsList[chosenEmbedding];
             if(triggerWords.length > 0) {
                 const negativePromptWidget = findTargetWidget(searchNode, "negativePrompt");
@@ -480,6 +503,18 @@ async function searchNodeHandler(searchNode, searchInputWidget) {
             await searchModels();
         });
     }
+
+    appendWidgetCB(searchNode, async function(...args) {
+        try {
+            let modelListValues = modelListWidget.options.values;
+            if(modelListValues.length <= 0) return;
+            modelListValues = modelListValues.map(model => model.split(" ")[0]);
+            const modelListCRValue = modelListWidget.value.split(" ")[0];
+            if(!modelListValues.includes(modelListCRValue)) {
+                searchModels(modelListCRValue);
+            }
+        } catch(e) {}
+    });
 }
 
 export {