minor patches re live preview gens

Author: mcmonkey4eva

Diff for: docs/Features/Prompt Syntax.md

@@ -126,7 +126,7 @@
             - Without this, if there are multiple people, it will do a bulk segmented refine on all faces combined
             - Note the index order is sorted from leftmost detection to right
         - To control the creativity/threshold with a yolo model just append `,<creativity>,<threshold>`, for example `<segment:yolo-face_yolov8m-seg_60.pt-1,0.8,0.25>` sets a `0.8` creativity and `0.25` threshold.
-            - Note the default "confidence threshold" for Yolo models is `0.25`, which is different than is often used with ClipSeg, and does not have a max value.
+            - Note the default "confidence threshold" for Yolo models is `0.25`, which is different than is often used with ClipSeg, and does not have a "max threshold" like ClipSeg does.
         - If you have a yolo model with multiple supported classes, you can filter specific classes by appending `:<classes>:` to the model name where `<classes>` is a comma-separated list of class IDs or names, e.g., `<segment:yolo-modelnamehere:0,apple,2:,0.8,0.25>`
     - There's an advanced parameter under `Regional Prompting` named `Segment Model` to customize the base model used for segment processing
     - There's also a parameter named `Save Segment Mask` to save a preview copy of the generated mask

Diff for: src/wwwroot/js/genpage/generatehandler.js

@@ -6,7 +6,7 @@ class GenerateHandler {
         this.totalGenRunTime = 0;
         this.validateModel = true;
         this.interrupted = -1;
-        this.socket = null;
+        this.sockets = {};
         this.imageContainerDivId = 'current_image';
         this.imageId = 'current_image_img';
         this.progressBarHtml = `<div class="image-preview-progress-inner"><div class="image-preview-progress-overall"></div><div class="image-preview-progress-current"></div></div>`;
@@ -116,9 +116,11 @@ class GenerateHandler {
             }
             return;
         }
+        let socketId = 'normal';
         let isPreview = '_preview' in input_overrides;
         if (isPreview) {
             delete input_overrides['_preview'];
+            socketId = 'preview';
         }
         this.beforeGenRun();
         let run = () => {
@@ -131,8 +133,8 @@ class GenerateHandler {
             let socket = null;
             let handleData = data => {
                 if ('socket_intention' in data && data.socket_intention == 'close') {
-                    if (this.socket == socket) {
-                        this.socket = null;
+                    if (this.sockets[socketId] == socket) {
+                        this.sockets[socketId] = null;
                     }
                     if (Object.keys(discardable).length > 0) {
                         // clear any lingering previews
@@ -267,12 +269,12 @@ class GenerateHandler {
                 }
                 this.hadError(e);
             };
-            if (this.socket && this.socket.readyState == WebSocket.OPEN) {
-                this.socket.send(JSON.stringify(actualInput));
+            if (this.sockets[socketId] && this.sockets[socketId].readyState == WebSocket.OPEN) {
+                this.sockets[socketId].send(JSON.stringify(actualInput));
             }
             else {
                 socket = makeWSRequestT2I('GenerateText2ImageWS', actualInput, handleData, handleError);
-                this.socket = socket;
+                this.sockets[socketId] = socket;
             }
         };
         if (this.validateModel) {

Diff for: src/wwwroot/js/genpage/main.js

@@ -1064,7 +1064,7 @@ function updateCurrentStatusDirect(data) {
         let estTime = avgGenTime * total;
         timeEstimate = ` (est. ${durationStringify(estTime)})`;
     }
-    elem.innerHTML = total == 0 ? (isGeneratingPreviews ? translatableText.get() : '') : `${autoBlock(num_current_gens, 'current generation%')}${autoBlock(num_live_gens, 'running')}${autoBlock(num_backends_waiting, 'queued')}${autoBlock(num_models_loading, waitingOnModelLoadText.get())} ${timeEstimate}...`;
+    elem.innerHTML = total == 0 ? (isGeneratingPreviews ? generatingPreviewsText.get() : '') : `${autoBlock(num_current_gens, 'current generation%')}${autoBlock(num_live_gens, 'running')}${autoBlock(num_backends_waiting, 'queued')}${autoBlock(num_models_loading, waitingOnModelLoadText.get())} ${timeEstimate}...`;
     let max = Math.max(num_current_gens, num_models_loading, num_live_gens, num_backends_waiting);
     setPageTitle(total == 0 ? curAutoTitle : `(${max} ${generatingText.get()}) ${curAutoTitle}`);
 }