Update ocr_engine.py

ocr_engine.py

@@ -1,32 +1,27 @@
-from mmocr.apis import MMOCRInferencer
+from mmocr.utils.ocr import MMOCR
 import numpy as np
 import cv2
 import re
 from PIL import Image
 
-# Initialize MMOCR
-ocr = MMOCRInferencer(det='DBNet', recog='SAR', device='cpu')  # or 'cuda' if GPU available
+# Load MMOCR (det + recog)
+ocr = MMOCR(det='DBPANet', recog='SAR', device='cpu')  # CPU mode for Hugging Face
 
 def extract_weight_from_image(pil_img):
     try:
-        # Convert PIL to OpenCV image (BGR)
         img = np.array(pil_img.convert("RGB"))[:, :, ::-1]
 
-        # Run MMOCR inference
-        result = ocr(img)
+        result = ocr.readtext(img, print_result=False, output=None)[0]['result']
 
         raw_texts = []
         weight_candidates = []
         fallback_weight = None
         fallback_conf = 0.0
 
-        for item in result['predictions'][0]:
-            text = item['text']
-            conf = item.get('score', 0.8)  # Fallback confidence
+        for text, conf in result:
             original = text
             cleaned = text.lower().strip()
 
-            # Fix common OCR misreads
             cleaned = cleaned.replace(",", ".")
             cleaned = cleaned.replace("o", "0").replace("O", "0")
             cleaned = cleaned.replace("s", "5").replace("S", "5")