Update ocr_engine.py

ocr_engine.py

@@ -3,18 +3,12 @@ import numpy as np
 import cv2
 import re
 import logging
-from mmocr.utils.ocr import MMOCR
 
 # Set up logging for debugging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# Initialize OCR engines
+# Initialize EasyOCR
 easyocr_reader = easyocr.Reader(['en'], gpu=False)
-try:
-    mmocr_reader = MMOCR(det='DB_r18', recog='CRNN')
-except:
-    mmocr_reader = None
-    logging.warning("MMOCR initialization failed, falling back to EasyOCR only")
 
 def estimate_blur(img):
     """Estimate image blur using Laplacian variance"""
@@ -75,23 +69,11 @@ def extract_weight_from_image(pil_img):
         best_conf = 0.0
 
         # EasyOCR detection
-        easyocr_results = easyocr_reader.readtext(processed, detail=1, paragraph=False)
-        if not easyocr_results:  # Fallback to original image if no results
-            easyocr_results = easyocr_reader.readtext(img, detail=1, paragraph=False)
-
-        # MMOCR detection (if available)
-        mmocr_results = []
-        if mmocr_reader:
-            try:
-                mmocr_result = mmocr_reader.readtext(processed)
-                mmocr_results = [(bbox, text, score) for bbox, text, score in mmocr_result]
-            except:
-                logging.warning("MMOCR processing failed, using EasyOCR results only")
-
-        # Combine results
-        all_results = easyocr_results + mmocr_results
-
-        for (bbox, text, conf) in all_results:
+        results = easyocr_reader.readtext(processed, detail=1, paragraph=False)
+        if not results:  # Fallback to original image if no results
+            results = easyocr_reader.readtext(img, detail=1, paragraph=False)
+
+        for (bbox, text, conf) in results:
             original_text = text
             text = text.lower().strip()
 
@@ -102,10 +84,10 @@ def extract_weight_from_image(pil_img):
             text = text.replace("g", "9").replace("G", "6")
             text = text.replace("l", "1").replace("I", "1")
             text = text.replace("b", "8").replace("B", "8")
-            text = text.replace("kgs", "").replace("kg", "").replace("k9", "").replace("k", "")
+            text = text.replace("kgs", "").replace("kg", "").replace("k", "")
             text = re.sub(r"[^\d\.]", "", text)
 
-            # Stricter regex for weight (0.0 to 9999.999)
+            # Regex for weight (0.0 to 9999.999)
             if re.fullmatch(r"\d{1,4}(\.\d{0,3})?", text):
                 if conf > best_conf and conf > conf_threshold:
                     best_weight = text