Update ocr_engine.py

ocr_engine.py

@@ -9,6 +9,7 @@ def extract_weight_from_image(pil_img):
         img = pil_img.convert("RGB")
         img_np = np.array(img)
         img_cv = cv2.cvtColor(img_np, cv2.COLOR_RGB2BGR)
+        print("🧼 Image converted to OpenCV format.")
 
         # Convert to grayscale
         gray = cv2.cvtColor(img_cv, cv2.COLOR_BGR2GRAY)
@@ -21,12 +22,15 @@ def extract_weight_from_image(pil_img):
         # Resize to enhance small text
         resized = cv2.resize(processed, None, fx=2, fy=2, interpolation=cv2.INTER_LINEAR)
 
-        # OCR config tuned for digit blocks
+        # Optional: Apply dilation to enhance contours
+        kernel = np.ones((2, 2), np.uint8)
+        dilated = cv2.dilate(resized, kernel, iterations=1)
+
+        # OCR config tuned for digits
         config = r'--oem 3 --psm 7 -c tessedit_char_whitelist=0123456789.'
 
         # Run OCR
-        text = pytesseract.image_to_string(resized, config=config)
-
+        text = pytesseract.image_to_string(dilated, config=config)
         print("🔍 RAW OCR OUTPUT:", repr(text))
 
         # Clean the text
@@ -38,4 +42,4 @@ def extract_weight_from_image(pil_img):
 
     except Exception as e:
         print("❌ OCR Error:", str(e))
-        return "", 0
+        return "", 0