Update ocr_engine.py

ocr_engine.py

@@ -1,33 +1,23 @@
-try:
-    from paddleocr import PaddleOCR
-    import re
-    ocr = PaddleOCR(use_angle_cls=True, lang='en')
-except Exception as e:
-    import streamlit as st
-    st.error(f"❌ OCR Engine Load Failed: {e}")
-    ocr = None
+import easyocr
+import re
 
-def extract_weight_from_image(image):
-    if ocr is None:
-        return ("OCR not initialized", 0.0)
+# Load OCR reader once
+reader = easyocr.Reader(['en'])
 
-    result = ocr.ocr(image, cls=True)
-    debug_texts = []
+def extract_weight_from_image(pil_image):
+    results = reader.readtext(pil_image)
 
-    if not result or not result[0]:
-        return ("No weight detected", 0.0)
+    weight = None
+    confidence = 0.0
 
-    for line in result[0]:
-        text, confidence = line[1][0], line[1][1]
-        debug_texts.append(f"{text} (Conf: {confidence:.2f})")
-        match = re.search(r'(\d+\.?\d*)\s*(kg|g)?', text.lower())
+    for (bbox, text, conf) in results:
+        match = re.search(r'(\d+(\.\d+)?)\s?g', text.lower())
         if match:
-            weight = match.group(1)
-            unit = match.group(2) if match.group(2) else "g"
-            return (f"{weight} {unit}", confidence)
+            weight = match.group(1) + " g"
+            confidence = conf
+            break
 
-    print("OCR Debug Output:")
-    for t in debug_texts:
-        print(t)
-
-    return ("No weight detected", 0.0)
+    if weight:
+        return weight, confidence
+    else:
+        return "No weight detected", 0.0