Update ocr_engine.py

ocr_engine.py

@@ -1,21 +1,16 @@
-import cv2
-import pytesseract
 from PIL import Image
+import pytesseract
 import re
 
 def extract_weight(img_path):
-    img = cv2.imread(img_path)
-    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-    
-    # OCR
-    text = pytesseract.image_to_string(gray, config='--psm 6')
+    img = Image.open(img_path).convert("L")  # Grayscale
 
-    # Clean and lower text
+    # OCR
+    text = pytesseract.image_to_string(img, config='--psm 6')
     text = text.lower().replace('\n', ' ').strip()
 
-    # Regex to find pattern like 52.25 g or 75.8 kg
+    # Find weight + unit (e.g., 52.25 g, 75.8 kg)
     match = re.search(r'(\d+\.\d+|\d+)\s*(kg|g)', text)
-
     if match:
         number = match.group(1)
         unit = match.group(2)