zjt/comfy_backend
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

get image

Browse Source
This commit is contained in:
zjt 2024-05-25 22:31:49 +08:00
parent 6f43190f9d
commit a32007c8fd
3 changed files with 62 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/biz/txt23d.ts
View File

@ -20,6 +20,7 @@ const Txt23DHandler: RequestHandler<any, any> = async (ctx) => {
const { prompt, url } = requestBody;
const inputNode = selectNodeFromApiJSONbyID(txt23dApiFormatJSON, "33");
inputNode.inputs.string = prompt;
try {
ctx.body = {
url: ctx.body = baseUrl + '/view?filename=' + await new Promise((resolve, reject) => {
const taskID = Math.random().toFixed(10);
@ -52,5 +53,9 @@ const Txt23DHandler: RequestHandler<any, any> = async (ctx) => {
}
})
}
} catch (error) {
console.log(error);
}
}
export default Txt23DHandler;

5
src/biz/txt2img.ts
View File

@ -24,6 +24,7 @@ const Txt2ImgHandler: RequestHandler<Txt2ImgRequest, any> = async (ctx, next) =>
let temp = "";
let preViewBlobs: any = [];
inputNode.inputs.string = prompt;
try {
ctx.body = {
data: await new Promise((resolve, reject) => {
const taskID = Math.random().toFixed(10);
@ -75,5 +76,9 @@ const Txt2ImgHandler: RequestHandler<Txt2ImgRequest, any> = async (ctx, next) =>
}
})
}
} catch (error) {
console.log(error);
}
}
export default Txt2ImgHandler;

5
src/biz/upScale.ts
View File

@ -21,6 +21,7 @@ const UpscaleHandler: RequestHandler<any ,any> = async (ctx) => {
const { prompt, url } = requestBody;
const inputNode = selectNodeFromApiJSONbyID(upScaleAPIFormatJSON, "12");
inputNode.inputs.image = url;
try {
ctx.body = {
url: baseUrl + '/view?filename=' + await new Promise((resolve, reject) => {
const taskID = Math.random().toFixed(10);
@ -52,5 +53,9 @@ const UpscaleHandler: RequestHandler<any ,any> = async (ctx) => {
}
})
}
}catch (error) {
console.log(error);
}
}
export default UpscaleHandler;