fix: perspective usage

src/models/ai_engine.py

@@ -481,18 +481,23 @@ def run_donut_ocr(img_array, processor, model, device, model_loaded,
         rgb_for_donut = cv2.cvtColor(thresh, cv2.COLOR_GRAY2RGB) if len(thresh.shape) == 2 else thresh
         print(f"[Donut] Using preprocessed image, shape={rgb_for_donut.shape}")
     else:
-        # Auto-crop to receipt region (handles photos with lots of background)
-        rgb_for_donut = img_array
-        try:
-            bgr = cv2.cvtColor(img_array, cv2.COLOR_RGB2BGR)
-            cropped = perspective_correction(bgr)
-            if cropped is None:
-                cropped = auto_crop_bright_region(bgr)
-            if cropped is not None:
-                rgb_for_donut = cv2.cvtColor(cropped, cv2.COLOR_BGR2RGB)
-                print(f"[Donut] Cropped {bgr.shape[:2]} -> {cropped.shape[:2]}")
-        except Exception as e:
-            print(f"[Donut] crop failed: {e}")
+        # Use RGB image, but apply perspective fix if requested
+        if filter_flags.get("enable_perspective", False):
+            try:
+                bgr = cv2.cvtColor(img_array, cv2.COLOR_RGB2BGR)
+                cropped = perspective_correction(bgr)
+                if cropped is None:
+                    cropped = auto_crop_bright_region(bgr)
+                if cropped is not None:
+                    rgb_for_donut = cv2.cvtColor(cropped, cv2.COLOR_BGR2RGB)
+                    print(f"[Donut] Cropped {bgr.shape[:2]} -> {cropped.shape[:2]}")
+                else:
+                    rgb_for_donut = img_array
+            except Exception as e:
+                rgb_for_donut = img_array
+                print(f"[Donut] crop failed: {e}")
+        else:
+            rgb_for_donut = img_array
 
     pil_img = Image.fromarray(rgb_for_donut).convert("RGB")
     # Let the processor handle resizing to the model's expected input size
@@ -612,17 +617,22 @@ def run_donut_ocr(img_array, processor, model, device, model_loaded,
         rgb_for_donut = cv2.cvtColor(thresh, cv2.COLOR_GRAY2RGB) if len(thresh.shape) == 2 else thresh
         print(f"[Donut] Using preprocessed image, shape={rgb_for_donut.shape}")
     else:
-        rgb_for_donut = img_array
-        try:
-            bgr = cv2.cvtColor(img_array, cv2.COLOR_RGB2BGR)
-            cropped = perspective_correction(bgr)
-            if cropped is None:
-                cropped = auto_crop_bright_region(bgr)
-            if cropped is not None:
-                rgb_for_donut = cv2.cvtColor(cropped, cv2.COLOR_BGR2RGB)
-                print(f"[Donut] Cropped {bgr.shape[:2]} -> {cropped.shape[:2]}")
-        except Exception as e:
-            print(f"[Donut] crop failed: {e}")
+        if filter_flags.get("enable_perspective", False):
+            try:
+                bgr = cv2.cvtColor(img_array, cv2.COLOR_RGB2BGR)
+                cropped = perspective_correction(bgr)
+                if cropped is None:
+                    cropped = auto_crop_bright_region(bgr)
+                if cropped is not None:
+                    rgb_for_donut = cv2.cvtColor(cropped, cv2.COLOR_BGR2RGB)
+                    print(f"[Donut] Cropped {bgr.shape[:2]} -> {cropped.shape[:2]}")
+                else:
+                    rgb_for_donut = img_array
+            except Exception as e:
+                rgb_for_donut = img_array
+                print(f"[Donut] crop failed: {e}")
+        else:
+            rgb_for_donut = img_array
 
     pil_img = Image.fromarray(rgb_for_donut).convert("RGB")
     pixel_values = processor(pil_img, return_tensors="pt").pixel_values.to(device)