Load workflow with image name.

- Fixed bug using the reserved word `type`.
This commit is contained in:
Christian Bastian
2024-07-24 02:33:35 -04:00
parent 752aac74f7
commit 7e60dbe827

View File

@@ -20,11 +20,14 @@ function request(url, options = undefined) {
/** /**
* @param {string} url * @param {string} url
*/ */
async function load_workflow(url) { async function loadWorkflow(url) {
const uri = (new URL(url)).searchParams.get("uri");
const fileNameIndex = Math.max(uri.lastIndexOf("/"), uri.lastIndexOf("\\")) + 1;
const fileName = uri.substring(fileNameIndex);
const response = await fetch(url); const response = await fetch(url);
const data = await response.blob(); const data = await response.blob();
const type = data.type; const datatype = data.type;
const file = new File([data], "model-manager-workflow-file", { type }); const file = new File([data], fileName, { datatype });
app.handleFile(file); app.handleFile(file);
} }
@@ -1833,7 +1836,7 @@ class ModelGrid {
const uri = urlSearchParams.get("uri"); const uri = urlSearchParams.get("uri");
const v = urlSearchParams.get("v"); const v = urlSearchParams.get("v");
const urlFull = urlString.substring(0, urlString.indexOf("?")) + "?uri=" + uri + "&v=" + v; const urlFull = urlString.substring(0, urlString.indexOf("?")) + "?uri=" + uri + "&v=" + v;
load_workflow(urlFull); await loadWorkflow(urlFull);
}, },
}).element, }).element,
); );
@@ -2376,7 +2379,10 @@ class ModelInfo {
new ComfyButton({ new ComfyButton({
content: "Load Workflow", content: "Load Workflow",
tooltip: "Attempt to load preview image workflow", tooltip: "Attempt to load preview image workflow",
action: () => load_workflow(previewSelect.elements.defaultPreviews.children[0].src), action: async () => {
const urlString = previewSelect.elements.defaultPreviews.children[0].src;
await loadWorkflow(urlString);
},
}).element, }).element,
]), ]),
$el("div.row.tab-header-flex-block", [ $el("div.row.tab-header-flex-block", [