Update ocr_engine.py

ocr_engine.py

@@ -1,10 +1,16 @@
-from paddleocr import PaddleOCR
-import re
-
-# Initialize PaddleOCR once
-ocr = PaddleOCR(use_angle_cls=True, lang='en')
+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
 
 def extract_weight_from_image(image):
+    if ocr is None:
+        return ("OCR not initialized", 0.0)
+
     result = ocr.ocr(image, cls=True)
     debug_texts = []
 
@@ -14,16 +20,14 @@ def extract_weight_from_image(image):
     for line in result[0]:
         text, confidence = line[1][0], line[1][1]
         debug_texts.append(f"{text} (Conf: {confidence:.2f})")
-
-        # Find number with optional decimal + optional "kg", "g"
         match = re.search(r'(\d+\.?\d*)\s*(kg|g)?', text.lower())
         if match:
-            unit = match.group(2) if match.group(2) else "g"
             weight = match.group(1)
+            unit = match.group(2) if match.group(2) else "g"
             return (f"{weight} {unit}", confidence)
 
-    print("DEBUG OCR OUTPUT:")
-    for d in debug_texts:
-        print(d)
+    print("OCR Debug Output:")
+    for t in debug_texts:
+        print(t)
 
     return ("No weight detected", 0.0)