Update app.py

app.py

@@ -9,6 +9,9 @@ from basicsr.utils.download_util import load_file_from_url
 from realesrgan import RealESRGANer
 from realesrgan.archs.srvgg_arch import SRVGGNetCompact
 
+from PIL import Image
+from PIL.PngImagePlugin import PngInfo
+
 
 last_file = None
 img_mode = "RGBA"
@@ -84,6 +87,11 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale, ext):
             channel_multiplier=2,
             bg_upsampler=upsampler)
 
+    try:
+        parameters = img.text['parameters']
+    except:
+        parameters = ''
+
     # Convert the input PIL image to cv2 image, so that it can be processed by realesrgan
     cv_img = numpy.array(img)
     img = cv2.cvtColor(cv_img, cv2.COLOR_RGBA2BGRA)
@@ -106,7 +114,10 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale, ext):
 
         #out_filename = f"output_{rnd_string(8)}.{extension}"
         out_filename = f"output_{rnd_string(8)}." + ext
-        cv2.imwrite(out_filename, output)
+        metadata = PngInfo()
+        if parameters != '':
+            metadata.add_text("parameters", parameters)
+        cv2.imwrite(out_filename, output, pnginfo=metadata)
         global last_file
         last_file = out_filename
         return out_filename