From bc4b420071d00d3f7ea031dce47415cc173e65fb Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Fri, 11 Aug 2023 05:21:40 -0700
Subject: [PATCH] i

---
 modules/core.py     | 4 ++--
 modules/cv2win32.py | 9 ++++-----
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/modules/core.py b/modules/core.py
index 1e5e2b8..d731ae7 100644
--- a/modules/core.py
+++ b/modules/core.py
@@ -82,9 +82,9 @@ def get_previewer(device, latent_format):
             x_sample = x_sample.cpu().numpy().clip(0, 255).astype(np.uint8)
             for i, s in enumerate(x_sample):
                 if i > 0:
-                    show_preview(s, title=f'Preview Image {i}, step = [{step}/{total_steps}')
+                    show_preview(f'cv2_preview_{i}', s, title=f'Preview Image {i}, step = [{step}/{total_steps}')
                 else:
-                    show_preview(s, title=f'Preview Image, step =  {step}/{total_steps}')
+                    show_preview(f'cv2_preview_{i}', s, title=f'Preview Image, step =  {step}/{total_steps}')
 
     taesd.preview = preview_function
 
diff --git a/modules/cv2win32.py b/modules/cv2win32.py
index 66eabdd..3e9b340 100644
--- a/modules/cv2win32.py
+++ b/modules/cv2win32.py
@@ -2,12 +2,11 @@ import threading
 import cv2
 
 
-flag = 'fooocus_cv2win32'
 buffer = []
 
 
 def worker():
-    global buffer, flag
+    global buffer
     while True:
         cv2.waitKey(50)
         try:
@@ -16,7 +15,7 @@ def worker():
                 if task is None:
                     cv2.destroyAllWindows()
                 else:
-                    img, title = task
+                    flag, img, title = task
                     cv2.imshow(flag, img)
                     cv2.setWindowTitle(flag, title)
                     cv2.setWindowProperty(flag, cv2.WND_PROP_TOPMOST, 1)
@@ -25,8 +24,8 @@ def worker():
     pass
 
 
-def show_preview(img, title='preview'):
-    buffer.append((img, title))
+def show_preview(flag, img, title='preview'):
+    buffer.append((flag, img.copy(), title))
 
 
 def close_all_preview():