.env add ws

.env.example

@@ -3,4 +3,5 @@ SALAI_TOKEN="Token of the Account from which you paid MidJourney"
 SERVER_ID="Server id here"
 CHANNEL_ID="Channel in which commands are sent"
 NEXT_PUBLIC_IMAGE_PREFIX="/"
-HUGGINGFACE_TOKEN="huggingface token here https://huggingface.co/docs/hub/security-tokens" 
+HUGGINGFACE_TOKEN="huggingface token here https://huggingface.co/docs/hub/security-tokens" 
+WS=true

pages/api/imagine.ts

@@ -15,7 +15,7 @@ const handler = async (req: Request) => {
     SalaiToken: <string>process.env.SALAI_TOKEN,
     HuggingFaceToken: <string>process.env.HUGGINGFACE_TOKEN,
     Debug: true,
-    Ws: true,
+    Ws: process.env.WS === "true",
   });
   await client.init();
   const encoder = new TextEncoder();
@@ -30,10 +30,12 @@ const handler = async (req: Request) => {
         .then((msg) => {
           console.log("imagine.done", msg);
           controller.enqueue(encoder.encode(JSON.stringify(msg)));
+          client.Close();
           controller.close();
         })
         .catch((err: ResponseError) => {
           console.log("imagine.error", err);
+          client.Close();
           controller.close();
         });
     },

pages/api/upscale.ts

@@ -14,7 +14,7 @@ export default async function handler(req: Request) {
     SalaiToken: <string>process.env.SALAI_TOKEN,
     HuggingFaceToken: <string>process.env.HUGGINGFACE_TOKEN,
     Debug: true,
-    Ws: true,
+    Ws: process.env.WS === "true",
   });
   await client.init();
   const encoder = new TextEncoder();
@@ -37,10 +37,12 @@ export default async function handler(req: Request) {
         .then((msg) => {
           console.log("upscale.done", msg);
           controller.enqueue(encoder.encode(JSON.stringify(msg)));
+          client.Close();
           controller.close();
         })
         .catch((err: ResponseError) => {
           console.log("upscale.error", err);
+          client.Close();
           controller.close();
         });
     },

pages/api/variation.ts

@@ -13,7 +13,7 @@ export default async function handler(req: Request) {
     SalaiToken: <string>process.env.SALAI_TOKEN,
     HuggingFaceToken: <string>process.env.HUGGINGFACE_TOKEN,
     Debug: true,
-    Ws: true,
+    Ws: process.env.WS === "true",
   });
   await client.init();
   const encoder = new TextEncoder();
@@ -36,10 +36,12 @@ export default async function handler(req: Request) {
         .then((msg) => {
           console.log("variation.done", msg);
           controller.enqueue(encoder.encode(JSON.stringify(msg)));
+          client.Close();
           controller.close();
         })
         .catch((err: ResponseError) => {
           console.log("variation.error", err);
+          client.Close();
           controller.close();
         });
     },