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

seed

Browse Source
This commit is contained in:
zjt 2024-05-28 15:28:45 +08:00
parent ed6426bc35
commit 019a6c433d
4 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/biz/getimg.ts
View File

@ -18,9 +18,14 @@ const GetImageHandler: RequestHandler<Txt2ImgRequest, any> = async (ctx, next) =
ctx.body = 200; ctx.body = 200;
return; return;
} }
const data = await readJSONFile("./data.json"); try {
ctx.body = { const data = await readJSONFile("./data.json");
data: data ctx.body = {
data: data
}
} catch (err) {
} }
} }
export default GetImageHandler; export default GetImageHandler;

5
src/biz/txt23d.ts
View File

@ -3,6 +3,8 @@ import { txt23dApiFormatJSON } from "../comfyJson/txt23d";
import axios from "axios"; import axios from "axios";
import { selectNodeFromApiJSONbyID } from "../utils/editComfyJson"; import { selectNodeFromApiJSONbyID } from "../utils/editComfyJson";
import WebSocket from "ws"; import WebSocket from "ws";
import { seed } from "../seed";
import { txt2imgAPIformatJSON } from "../comfyJson/txt2img";
const baseUrl = "http://47.108.92.176:20000"; const baseUrl = "http://47.108.92.176:20000";
const baseWsUrl = "ws://47.108.92.176:20000"; const baseWsUrl = "ws://47.108.92.176:20000";
// const baseUrl = "http://localhost:8188"; // const baseUrl = "http://localhost:8188";
@ -19,6 +21,9 @@ const Txt23DHandler: RequestHandler<any, any> = async (ctx) => {
const requestBody = ctx.request.body; const requestBody = ctx.request.body;
const { prompt, url } = requestBody; const { prompt, url } = requestBody;
const inputNode = selectNodeFromApiJSONbyID(txt23dApiFormatJSON, "33"); const inputNode = selectNodeFromApiJSONbyID(txt23dApiFormatJSON, "33");
const seedNode = selectNodeFromApiJSONbyID(txt23dApiFormatJSON, "10");
const formSeedNode = selectNodeFromApiJSONbyID(txt2imgAPIformatJSON, "10");
seedNode.inputs.seed = formSeedNode.inputs.seed;
inputNode.inputs.string = prompt; inputNode.inputs.string = prompt;
try { try {
ctx.body = { ctx.body = {

4
src/biz/txt2img.ts
View File

@ -4,6 +4,7 @@ import axios from "axios";
import { selectNodeFromApiJSONbyID } from "../utils/editComfyJson"; import { selectNodeFromApiJSONbyID } from "../utils/editComfyJson";
import WebSocket from "ws"; import WebSocket from "ws";
import { readJSONFile, saveJSONFile } from "../utils/jsonReader"; import { readJSONFile, saveJSONFile } from "../utils/jsonReader";
import { seed } from "../seed";
const baseUrl = "http://47.108.92.176:20000"; const baseUrl = "http://47.108.92.176:20000";
const baseWsUrl = "ws://47.108.92.176:20000"; const baseWsUrl = "ws://47.108.92.176:20000";
// const baseUrl = "http://localhost:8188"; // const baseUrl = "http://localhost:8188";
@ -23,6 +24,9 @@ const Txt2ImgHandler: RequestHandler<Txt2ImgRequest, any> = async (ctx, next) =>
const inputNode = selectNodeFromApiJSONbyID(txt2imgAPIformatJSON, "33"); const inputNode = selectNodeFromApiJSONbyID(txt2imgAPIformatJSON, "33");
let temp = ""; let temp = "";
let preViewBlobs: any = []; let preViewBlobs: any = [];
const seedNode = selectNodeFromApiJSONbyID(txt2imgAPIformatJSON, "10");
seedNode.inputs.seed = Math.floor(Math.random() * Number.MAX_SAFE_INTEGER);
seed = seedNode.inputs.seed;
inputNode.inputs.string = prompt; inputNode.inputs.string = prompt;
try { try {
ctx.body = { ctx.body = {

3
src/seed.ts Normal file
View File

@ -0,0 +1,3 @@
let seed = 0;
let strength = 1;
export { seed, strength };