Skip to content

GitLab

Commit 72cdd83c authored by pythongosssss's avatar pythongosssss
Browse files

Readded loading file

parent a5c5c97d
Show whitespace changes
Inline Side-by-side
Showing with 26 additions and 48 deletions
web/index.html
View file @ 72cdd83c
......@@ -12,28 +12,5 @@
window.graph = app.graph;
</script>
</head>
<body>
<script>
return;
let runningNodeId = null;
let progress = null;
let clientId = null;
var input = document.createElement("input");
input.setAttribute("type", "file");
input.setAttribute("accept", ".json,image/png");
input.style.display = "none";
document.body.appendChild(input);
input.addEventListener("change", function () {
var file = input.files[0];
prompt_file_load(file);
});
function loadGraph() {
input.click();
}
</script>
</body>
<body></body>
</html>
web/scripts/app.js
View file @ 72cdd83c
......@@ -300,19 +300,7 @@ class ComfyApp {
event.preventDefault();
event.stopPropagation();
const file = event.dataTransfer.files[0];
if (file.type === "image/png") {
const pngInfo = await getPngMetadata(file);
if (pngInfo && pngInfo.workflow) {
this.loadGraphData(JSON.parse(pngInfo.workflow));
}
} else if (file.type === "application/json" || file.name.endsWith(".json")) {
const reader = new FileReader();
reader.onload = () => {
this.loadGraphData(JSON.parse(reader.result));
};
reader.readAsText(file);
}
await this.handleFile(file);
});
}
......@@ -630,6 +618,21 @@ class ComfyApp {
this.canvas.draw(true, true);
await this.ui.queue.update();
}
async handleFile(file) {
if (file.type === "image/png") {
const pngInfo = await getPngMetadata(file);
if (pngInfo && pngInfo.workflow) {
this.loadGraphData(JSON.parse(pngInfo.workflow));
}
} else if (file.type === "application/json" || file.name.endsWith(".json")) {
const reader = new FileReader();
reader.onload = () => {
this.loadGraphData(JSON.parse(reader.result));
};
reader.readAsText(file);
}
}
}
export const app = new ComfyApp();
web/scripts/ui.js
View file @ 72cdd83c
......@@ -159,18 +159,16 @@ export class ComfyUI {
this.history.update();
});
var input = document.createElement("input");
input.setAttribute("type", "file");
input.setAttribute("accept", ".json,image/png");
input.style.display = "none";
document.body.appendChild(input);
input.addEventListener("change", function () {
var file = input.files[0];
prompt_file_load(file);
const fileInput = $el("input", {
type: "file",
accept: ".json,image/png",
style: "display: none",
parent: document.body,
onchange: () => {
app.handleFile(fileInput.files[0]);
},
});
this.menuContainer = $el("div.comfy-menu", { parent: document.body }, [
$el("span", { $: (q) => (this.queueSize = q) }),
$el("button.comfy-queue-btn", { textContent: "Queue Prompt", onclick: () => app.queuePrompt(0) }),
......@@ -214,7 +212,7 @@ export class ComfyUI {
}, 0);
},
}),
$el("button", { textContent: "Load", onclick: () => {} }),
$el("button", { textContent: "Load", onclick: () => fileInput.click() }),
$el("button", { textContent: "Clear", onclick: () => app.graph.clear() }),
$el("button", { textContent: "Load Default", onclick: () => app.loadGraphData() }),
]);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment