Update ocr_engine.py

ocr_engine.py

@@ -5,25 +5,21 @@ from PIL import Image
 
 def extract_weight_from_image(pil_img):
     try:
-        img = pil_img.convert("L")  # grayscale
+        # Convert PIL image to OpenCV
+        img = pil_img.convert("RGB")
         img = np.array(img)
+        img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
 
-        img = cv2.resize(img, None, fx=2, fy=2, interpolation=cv2.INTER_CUBIC)
-        blur = cv2.GaussianBlur(img, (5, 5), 0)
-        thresh = cv2.adaptiveThreshold(
-            blur, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C,
-            cv2.THRESH_BINARY_INV, 11, 2
-        )
+        # Preprocess
+        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+        blur = cv2.GaussianBlur(gray, (3, 3), 0)
 
-        config = r'--oem 3 --psm 6 -c tessedit_char_whitelist=0123456789.'
-        text = pytesseract.image_to_string(thresh, config=config)
-
-        print("🔍 OCR RAW OUTPUT:", repr(text))
-
-        weight = ''.join(filter(lambda c: c in '0123456789.', text)).strip()
-        confidence = 95 if weight else 0
-        return weight, confidence
+        # OCR
+        text = pytesseract.image_to_string(blur, config='--psm 7 digits')
+        weight = ''.join(filter(lambda c: c in '0123456789.', text))
 
+        confidence = 95  # Replace with real confidence logic if needed
+        return weight.strip(), confidence
     except Exception as e:
-        print("❌ OCR Exception:", str(e))
+        print(f"OCR error: {e}")
         return "", 0