Keep processing output resolution when restoring ignored regions

app.py

@@ -168,11 +168,18 @@ def detect_parts(input_path, part):
 def restore_parts(input_path, output_path, boxes):
     with Image.open(input_path) as original, Image.open(output_path) as cleaned:
         result = cleaned.copy()
-        if result.size != original.size:
-            result = result.resize(original.size, Image.Resampling.LANCZOS)
+        width, height = result.size
+        sx, sy = width / original.width, height / original.height
         original = original.convert(result.mode)
-        for box in boxes:
-            result.paste(original.crop(box), box)
+        if original.size != result.size:
+            original = original.resize(result.size, Image.Resampling.LANCZOS)
+        for x1, y1, x2, y2 in boxes:
+            box = (
+                max(0, math.floor(x1 * sx)), max(0, math.floor(y1 * sy)),
+                min(width, math.ceil(x2 * sx)), min(height, math.ceil(y2 * sy)),
+            )
+            if box[2] > box[0] and box[3] > box[1]:
+                result.paste(original.crop(box), box)
         result.save(output_path)
 
 
@@ -207,9 +214,7 @@ def remove_watermark(file_obj, strength, steps, model, ignore_eyes=True, ignore_
         else:
             model_id = WatermarkRemover.CTRLREGEN_MODEL_ID if model == "CtrlRegen" else None
             run_watermark(input_path, work_path, strength, int(steps), model_id)
-        with Image.open(input_path) as original, Image.open(work_path) as image:
-            if image.size != original.size:
-                image = image.resize(original.size, Image.Resampling.LANCZOS)
+        with Image.open(work_path) as image:
             image.save(output_path)
         boxes = [box for found in regions.values() for box in found]
         if boxes: