Update ocr_engine.py

ocr_engine.py

@@ -1,14 +1,9 @@
+import cv2
 import pytesseract
 from PIL import Image
-import re
 
-def extract_weight_from_image(image):
-"""Extract weight as a decimal number (e.g., 53.76) from the image."""
-try:
-text = pytesseract.image_to_string(image)
-match = re.search(r"\d+.\d{2}", text)
-if match:
-return match.group()
-return "No weight found"
-except Exception as e:
-return f"Error: {str(e)}"
+def extract_weight(img_path):
+    img = cv2.imread(img_path)
+    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+    text = pytesseract.image_to_string(gray, config='--psm 7 digits')
+    return ''.join(filter(lambda x: x in '0123456789.', text))