Update app.py

app.py

@@ -111,7 +111,7 @@ def detect_roi(img_cv):
         return img_cv
 
 def extract_weight(img):
-    """Extract weight from image using Tesseract OCR with multiple PSM modes."""
+    """Extract weight from image using Tesseract OCR with improved configuration."""
     try:
         if img is None:
             logging.error("No image provided for OCR")
@@ -124,29 +124,26 @@ def extract_weight(img):
         # Preprocess the ROI
         processed_img = preprocess_image(roi_img)
         
-        # Try multiple PSM modes for better detection
-        psm_modes = [
-            ('--psm 7 digits', 'Single line, digits only'),
-            ('--psm 6 digits', 'Single block, digits only'),
-            ('--psm 10 digits', 'Single character, digits only'),
-            ('--psm 8 digits', 'Single word, digits only')
-        ]
+        # OCR configuration for digit extraction
+        custom_config = r'--oem 3 --psm 6 -c tessedit_char_whitelist=0123456789.'
         
-        for config, desc in psm_modes:
-            text = pytesseract.image_to_string(processed_img, config=config)
-            logging.info(f"OCR attempt with {desc}: Raw text = '{text}'")
-            weight = ''.join(filter(lambda x: x in '0123456789.', text.strip()))
+        # Run OCR
+        text = pytesseract.image_to_string(processed_img, config=custom_config)
+        logging.info(f"OCR result: '{text}'")
+        
+        # Extract valid weight from OCR result
+        weight = ''.join(filter(lambda x: x in '0123456789.', text.strip()))
+        if weight:
             try:
                 weight_float = float(weight)
-                if weight_float >= 0:  # Allow zero weights
-                    confidence = 95.0  # Simplified confidence for valid numbers
+                if weight_float >= 0:  # Only accept valid weights
+                    confidence = 95.0  # Assume high confidence if we have a valid weight
                     logging.info(f"Weight detected: {weight} (Confidence: {confidence:.2f}%)")
                     return weight, confidence
             except ValueError:
                 logging.warning(f"Invalid number format: {weight}")
-                continue
         
-        logging.error("All OCR attempts failed to detect a valid weight")
+        logging.error("OCR failed to detect a valid weight")
         return "Not detected", 0.0
     except Exception as e:
         logging.error(f"OCR processing failed: {str(e)}")