Update app.py

app.py

@@ -99,18 +99,17 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
         print('If you encounter CUDA out of memory, try to set --tile with a smaller number.')
     else:
         # Save restored image and return it to the output Image component
-        #if img_mode == 'RGBA':  # RGBA images should be saved in png format
+        if img_mode == 'RGBA':  # RGBA images should be saved in png format
             extension = 'png'
-        #else:
-        #    extension = 'jpg'
+        else:
+            extension = 'jpg'
 
         #out_filename = f"output_{rnd_string(8)}.{extension}"
-        #cv2.imwrite(out_filename, output)
-        cv2.imwrite("output.png", output)
+        out_filename = f"output_{rnd_string(8)}.png"
+        cv2.imwrite(out_filename, output)
         global last_file
-        #last_file = out_filename
-        #return out_filename
-        return "output.png"
+        last_file = out_filename
+        return out_filename
 
 
 def rnd_string(x):
@@ -163,10 +162,10 @@ def image_properties(img):
     """
     global img_mode
     if img:
-        #if has_transparency(img):
+        if has_transparency(img):
             img_mode = "RGBA"
-        #else:
-        #    img_mode = "RGB"
+        else:
+            img_mode = "RGB"
         properties = f"Width: {img.size[0]}, Height: {img.size[1]}  |  Color Mode: {img_mode}"
         return properties