From 019a6c433d5d80d5a6336b92f9b2505448a7be5e Mon Sep 17 00:00:00 2001
From: zjt <937178071@qq.com>
Date: Tue, 28 May 2024 15:28:45 +0800
Subject: [PATCH] seed

---
 src/biz/getimg.ts  | 11 ++++++++---
 src/biz/txt23d.ts  |  5 +++++
 src/biz/txt2img.ts |  4 ++++
 src/seed.ts        |  3 +++
 4 files changed, 20 insertions(+), 3 deletions(-)
 create mode 100644 src/seed.ts

diff --git a/src/biz/getimg.ts b/src/biz/getimg.ts
index 13e109e..942938a 100644
--- a/src/biz/getimg.ts
+++ b/src/biz/getimg.ts
@@ -18,9 +18,14 @@ const GetImageHandler: RequestHandler<Txt2ImgRequest, any> = async (ctx, next) =
         ctx.body = 200;
         return;
     }
-    const data = await readJSONFile("./data.json");
-    ctx.body = {
-        data: data
+    try {
+        const data = await readJSONFile("./data.json");
+        ctx.body = {
+            data: data
+        }
+    } catch (err) {
+
     }
+
 }
 export default GetImageHandler;
\ No newline at end of file
diff --git a/src/biz/txt23d.ts b/src/biz/txt23d.ts
index 2cac4e0..2a072ba 100644
--- a/src/biz/txt23d.ts
+++ b/src/biz/txt23d.ts
@@ -3,6 +3,8 @@ import { txt23dApiFormatJSON } from "../comfyJson/txt23d";
 import axios from "axios";
 import { selectNodeFromApiJSONbyID } from "../utils/editComfyJson";
 import WebSocket from "ws";
+import { seed } from "../seed";
+import { txt2imgAPIformatJSON } from "../comfyJson/txt2img";
 const baseUrl = "http://47.108.92.176:20000";
 const baseWsUrl = "ws://47.108.92.176:20000";
 // const baseUrl = "http://localhost:8188";
@@ -19,6 +21,9 @@ const Txt23DHandler: RequestHandler<any, any> = async (ctx) => {
     const requestBody = ctx.request.body;
     const { prompt, url } = requestBody;
     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;
     try {
            ctx.body = {
diff --git a/src/biz/txt2img.ts b/src/biz/txt2img.ts
index 2c2917d..b8cc493 100644
--- a/src/biz/txt2img.ts
+++ b/src/biz/txt2img.ts
@@ -4,6 +4,7 @@ import axios from "axios";
 import { selectNodeFromApiJSONbyID } from "../utils/editComfyJson";
 import WebSocket from "ws";
 import { readJSONFile, saveJSONFile } from "../utils/jsonReader";
+import { seed } from "../seed";
 const baseUrl = "http://47.108.92.176:20000";
 const baseWsUrl = "ws://47.108.92.176:20000";
 // const baseUrl = "http://localhost:8188";
@@ -23,6 +24,9 @@ const Txt2ImgHandler: RequestHandler<Txt2ImgRequest, any> = async (ctx, next) =>
     const inputNode = selectNodeFromApiJSONbyID(txt2imgAPIformatJSON, "33");
     let temp = "";
     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;
     try {
         ctx.body = {
diff --git a/src/seed.ts b/src/seed.ts
new file mode 100644
index 0000000..2ceeb65
--- /dev/null
+++ b/src/seed.ts
@@ -0,0 +1,3 @@
+let seed = 0;
+let strength = 1;
+export { seed, strength };
\ No newline at end of file