From eed438fed17b592daf1e139158ece1e3036472f6 Mon Sep 17 00:00:00 2001
From: Jopezzia <chehonte60@gmail.com>
Date: Fri, 6 Jan 2023 16:59:53 +0300
Subject: [PATCH] fix size in image info

---
 scripts/ultimate-upscale.py | 23 +++++++++++++++++------
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/scripts/ultimate-upscale.py b/scripts/ultimate-upscale.py
index 616a93e..5f6d154 100644
--- a/scripts/ultimate-upscale.py
+++ b/scripts/ultimate-upscale.py
@@ -163,9 +163,13 @@ class USDURedraw():
                 p.image_mask = mask
                 processed = processing.process_images(p)
                 draw.rectangle(self.calc_rectangle(xi, yi), fill="black")
-                self.initial_info = processed.info
                 if (len(processed.images) > 0):
                     image = processed.images[0]
+
+        p.width = image.width
+        p.height = image.height
+        self.initial_info = processed.infotext(p, 0)
+        
         return image
 
     def chess_process(self, p, image, rows, cols):
@@ -196,7 +200,6 @@ class USDURedraw():
                 p.image_mask = mask
                 processed = processing.process_images(p)
                 draw.rectangle(self.calc_rectangle(xi, yi), fill="black")
-                self.initial_info = processed.info
                 if (len(processed.images) > 0):
                     image = processed.images[0]
 
@@ -211,10 +214,13 @@ class USDURedraw():
                 p.image_mask = mask
                 processed = processing.process_images(p)
                 draw.rectangle(self.calc_rectangle(xi, yi), fill="black")
-                self.initial_info = processed.info
                 if (len(processed.images) > 0):
                     image = processed.images[0]
 
+        p.width = image.width
+        p.height = image.height
+        self.initial_info = processed.infotext(p, 0)
+
         return image
 
     def start(self, p, image, rows, cols):
@@ -258,7 +264,6 @@ class USDUSeamsFix():
                 p.init_images = [image]
                 p.image_mask = mask
                 processed = processing.process_images(p)
-                self.initial_info = processed.info
                 if (len(processed.images) > 0):
                     image = processed.images[0]
 
@@ -276,10 +281,13 @@ class USDUSeamsFix():
                 p.init_images = [image]
                 p.image_mask = mask
                 processed = processing.process_images(p)
-                self.initial_info = processed.info
                 if (len(processed.images) > 0):
                     image = processed.images[0]
 
+        p.width = image.width
+        p.height = image.height
+        self.initial_info = processed.infotext(p, 0)
+
         return image
 
     def band_pass_process(self, p, image, cols, rows):
@@ -322,10 +330,13 @@ class USDUSeamsFix():
             p.init_images = [image]
             p.image_mask = mask
             processed = processing.process_images(p)
-            self.initial_info = processed.info
             if (len(processed.images) > 0):
                 image = processed.images[0]
 
+        p.width = image.width
+        p.height = image.height
+        self.initial_info = processed.infotext(p, 0)
+
         return image
 
     def start(self, p, image, rows, cols):