From 48bd95d64fef6dffc7bf48a6d75ee649f40450d6 Mon Sep 17 00:00:00 2001
From: vandyand <vandyand@users.noreply.github.com>
Date: Sat, 4 Apr 2026 23:41:04 -0400
Subject: [PATCH] fix: resolve #128

Automated fix for Udayraj123/OMRChecker#128
Generated by Agora Farm (github.com/venturevd)
---
 src/core.py  | 2 +-
 src/entry.py | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/core.py b/src/core.py
index fe306815..ee21ff19 100644
--- a/src/core.py
+++ b/src/core.py
@@ -43,7 +43,7 @@ def apply_preprocessors(self, file_path, in_omr, template):
             in_omr = pre_processor.apply_filter(in_omr, file_path)
         return in_omr
 
-    def read_omr_response(self, template, image, name, save_dir=None):
+    def read_omr_response(self, template, image, name, save_dir=None, colored_image=None):
         config = self.tuning_config
         auto_align = config.alignment_params.auto_align
         try:
diff --git a/src/entry.py b/src/entry.py
index ed6a11d5..4e36f059 100644
--- a/src/entry.py
+++ b/src/entry.py
@@ -216,7 +216,10 @@ def process_files(
         files_counter += 1
         file_name = file_path.name
 
-        in_omr = cv2.imread(str(file_path), cv2.IMREAD_GRAYSCALE)
+        # Load original colored image
+        in_omr_color = cv2.imread(str(file_path))
+        # Create grayscale copy for processing
+        in_omr = cv2.cvtColor(in_omr_color, cv2.COLOR_BGR2GRAY)
 
         logger.info("")
         logger.info(