chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: state

Browse Source
This commit is contained in:
bennykok 2024-09-23 19:01:47 -07:00
parent 3a6c3b1ae9
commit d1c54b2b6d
1 changed files with 23 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
web-plugin/index.js
View File

@ -83,7 +83,9 @@ function dispatchAPIEventData(data) {
} }
} }
let selectedWorkflowInfo const context = {
selectedWorkflowInfo: null,
};
// let selectedWorkflowInfo = { // let selectedWorkflowInfo = {
// workflow_id: "05da8f2b-63af-4c0c-86dd-08d01ec512b7", // workflow_id: "05da8f2b-63af-4c0c-86dd-08d01ec512b7",
// machine_id: "45ac5f85-b7b6-436f-8d97-2383b25485f3", // machine_id: "45ac5f85-b7b6-436f-8d97-2383b25485f3",
@ -91,11 +93,11 @@ let selectedWorkflowInfo
// }; // };
function getSelectedWorkflowInfo() { function getSelectedWorkflowInfo() {
return selectedWorkflowInfo; return context.selectedWorkflowInfo;
} }
function setSelectedWorkflowInfo(info) { function setSelectedWorkflowInfo(info) {
selectedWorkflowInfo = info; context.selectedWorkflowInfo = info;
} }
/** @typedef {import('../../../web/types/comfy.js').ComfyExtension} ComfyExtension*/ /** @typedef {import('../../../web/types/comfy.js').ComfyExtension} ComfyExtension*/
@ -118,10 +120,10 @@ const ext = {
sendEventToCD("cd_plugin_onInit"); sendEventToCD("cd_plugin_onInit");
app.queuePrompt = ((originalFunction) => async () => { // app.queuePrompt = ((originalFunction) => async () => {
// const prompt = await app.graphToPrompt(); // // const prompt = await app.graphToPrompt();
sendEventToCD("cd_plugin_onQueuePromptTrigger"); // sendEventToCD("cd_plugin_onQueuePromptTrigger");
})(app.queuePrompt); // })(app.queuePrompt);
// // Intercept the onkeydown event // // Intercept the onkeydown event
// window.addEventListener( // window.addEventListener(
@ -1525,18 +1527,25 @@ api.fetchApi = async (route, options) => {
const body = JSON.parse(options.body); const body = JSON.parse(options.body);
// getData().apiKey
const data = {
client_id: body.client_id,
workflow_api_json: body.prompt,
is_native_run: true,
machine_id: info.machine_id,
workflow_id: info.workflow_id,
native_run_api_endpoint: info.native_run_api_endpoint,
};
return await fetch("/comfyui-deploy/run", { return await fetch("/comfyui-deploy/run", {
method: "POST", method: "POST",
headers: { headers: {
Authorization: `Bearer ${getData().apiKey}`, Authorization: `Bearer ${info.cd_token}`,
"Content-Type": "application/json", "Content-Type": "application/json",
}, },
body: JSON.stringify({ body: JSON.stringify(data),
client_id: body.client_id,
workflow_api_json: body.prompt,
is_native_run: true,
...info,
}),
}); });
} }