Fix noise

app.py

@@ -32,6 +32,18 @@ def predict(source_img, enlarge_top, enlarge_right, enlarge_bottom, enlarge_left
     if negative_prompt is None or negative_prompt == "":
         raise gr.Error("Please provide a negative prompt input.")
 
+    if enlarge_top is None or enlarge_top == "":
+        raise gr.Error("Please provide a top input.")
+
+    if enlarge_right is None or enlarge_right == "":
+        raise gr.Error("Please provide a right input.")
+
+    if enlarge_bottom is None or enlarge_bottom == "":
+        raise gr.Error("Please provide a bottom input.")
+
+    if enlarge_left is None or enlarge_left == "":
+        raise gr.Error("Please provide a left input.")
+
     if enlarge_top < 0 or enlarge_right < 0 or enlarge_bottom < 0 or enlarge_left < 0:
         raise gr.Error("Please only provide positive margins.")
 
@@ -63,7 +75,7 @@ def predict(source_img, enlarge_top, enlarge_right, enlarge_bottom, enlarge_left
     enlarged_image = Image.new(mode = input_image.mode, size = (original_height, original_width), color = "black")
     enlarged_image.paste(input_image, (0, 0))
     enlarged_image = enlarged_image.resize((output_width, output_height))
-    enlarged_image = enlarged_image.filter(ImageFilter.BoxBlur(25))
+    enlarged_image = enlarged_image.filter(ImageFilter.BoxBlur(20))
 
     enlarged_image.paste(input_image, (enlarge_left, enlarge_top))
 
@@ -81,7 +93,7 @@ def predict(source_img, enlarge_top, enlarge_right, enlarge_bottom, enlarge_left
     enlarged_image.paste(returned_input_image, (enlarge_left + original_width, enlarge_top - (original_height * 2)))
     enlarged_image.paste(returned_input_image, (enlarge_left + original_width, enlarge_top + original_height))
 
-    enlarged_image = enlarged_image.filter(ImageFilter.BoxBlur(25))
+    enlarged_image = enlarged_image.filter(ImageFilter.BoxBlur(20))
 
     # Noise image
     noise_image = Image.new(mode = input_image.mode, size = (output_width, output_height), color = "black")
@@ -90,7 +102,7 @@ def predict(source_img, enlarge_top, enlarge_right, enlarge_bottom, enlarge_left
     for i in range(output_width):
         for j in range(output_height):
             enlarged_pixel = enlarged_pixels[i, j]
-            noise = min(abs(enlarge_left - i), abs(enlarge_left + original_width - i), abs(enlarge_top - j), abs(enlarge_top + original_height - j), 255)
+            noise = min(max(enlarge_left - i, i - (enlarge_left + original_width), enlarge_top - j, j - (enlarge_top + original_height), 0), 255)
             noise_image.putpixel((i, j), (noise_color(enlarged_pixel[0], noise), noise_color(enlarged_pixel[1], noise), noise_color(enlarged_pixel[2], noise), 255))
 
     enlarged_image.paste(noise_image, (0, 0))