Rename src/ocr_engine.py to ocr_engine.py

ocr_engine.py

@@ -0,0 +1,9 @@
+import pytesseract
+import re
+
+def extract_weight_from_image(image):
+text = pytesseract.image_to_string(image)
+match = re.search(r"\d+.\d{2}", text)
+if match:
+return match.group()
+return "No weight found"

src/ocr_engine.py

@@ -1,9 +0,0 @@
-import cv2
-import pytesseract
-from PIL import Image
-
-def extract_weight(img_path):
-img = cv2.imread(img_path)
-gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-text = pytesseract.image_to_string(gray, config='--psm 7 digits')
-return ''.join(filter(lambda x: x in '0123456789.', text))