From cfb90a938eff6d5d4cfa39f58ebc0ab32ffedfb3 Mon Sep 17 00:00:00 2001
From: w-e-w <40751091+w-e-w@users.noreply.github.com>
Date: Sat, 13 Jan 2024 02:22:32 +0900
Subject: [PATCH] allowe hr pass to return multiple images

---
 modules/txt2img.py | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

diff --git a/modules/txt2img.py b/modules/txt2img.py
index 115587859..daf8f51a9 100644
--- a/modules/txt2img.py
+++ b/modules/txt2img.py
@@ -88,18 +88,13 @@ def txt2img_upscale(id_task: str, request: gr.Request, gallery, gallery_index, g
 
     new_gallery = []
     for i, image in enumerate(gallery):
-        fake_image = Image.new(mode="RGB", size=(1, 1))
-
         if i == gallery_index:
-            already_saved_as = getattr(processed.images[0], 'already_saved_as', None)
-            if already_saved_as is not None:
-                fake_image.already_saved_as = already_saved_as
-            else:
-                fake_image = processed.images[0]
+            geninfo["infotexts"][gallery_index: gallery_index+1] = processed.infotexts
+            new_gallery.extend(processed.images)
         else:
+            fake_image = Image.new(mode="RGB", size=(1, 1))
             fake_image.already_saved_as = image["name"]
-
-        new_gallery.append(fake_image)
+            new_gallery.append(fake_image)
 
     geninfo["infotexts"][gallery_index] = processed.info