Update src/ocr_engine.py

src/ocr_engine.py

@@ -1,12 +1,12 @@
-def extract_weight_from_image(pil_img):
-    import os
-    os.system("apt-get update && apt-get install -y libgl1-mesa-glx")
+import os
+os.system("apt-get update && apt-get install -y libgl1-mesa-glx")
 
-    import cv2
-    import numpy as np
-    import pytesseract
-    from PIL import Image
+import cv2
+import numpy as np
+import pytesseract
+from PIL import Image
 
+def extract_weight_from_image(pil_img):
     image = np.array(pil_img.convert("RGB"))
     gray = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
     blur = cv2.GaussianBlur(gray, (3, 3), 0)
@@ -17,6 +17,6 @@ def extract_weight_from_image(pil_img):
 
     extracted_text = ''.join(filter(lambda x: x in '0123456789.', ''.join(data['text'])))
     confidences = [int(conf) for conf in data['conf'] if conf.isdigit()]
-    avg_conf = sum(confidences) / len(confidences) if confidences else 0
+    avg_conf = sum(confidences)/len(confidences) if confidences else 0
 
     return extracted_text.strip(), avg_conf