From 0a161088dd2c29c0c166bef380cc15dac34c3e25 Mon Sep 17 00:00:00 2001
From: taabata <57796911+taabata@users.noreply.github.com>
Date: Mon, 18 Nov 2024 03:40:52 +0300
Subject: [PATCH] Update latent_preview.py

---
 latent_preview.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/latent_preview.py b/latent_preview.py
index 0cd9fbb..cc4e8c6 100644
--- a/latent_preview.py
+++ b/latent_preview.py
@@ -10,7 +10,7 @@ import comfy.utils
 import logging
 from pathlib import Path
 from urllib import request
-import base64, io, json
+import base64, io, json, os
 
 MAX_PREVIEW_RESOLUTION = args.preview_size
 
@@ -33,7 +33,7 @@ def preview_to_image(latent_image,newsample=False):
                         }
             }
             data = json.dumps(p).encode('utf-8')
-            req =  request.Request("http://localhost:5000/settaesd", data=data)
+            req =  request.Request(f"http://{args.listen}:{os.environ['CANVASPORT']}/settaesd", data=data)
             req.add_header("Content-Type", "application/json")
             request.urlopen(req)
         except Exception as e: 
@@ -125,4 +125,3 @@ def prepare_callback(model, steps, x0_output_dict=None):
             preview_bytes = previewer.decode_latent_to_preview_image(preview_format, x0)
         pbar.update_absolute(step + 1, total_steps, preview_bytes)
     return callback
-