Update app.py

app.py

@@ -6,53 +6,48 @@ import re
 from datetime import datetime
 import pytz
 
-# Load EasyOCR model (no Tesseract needed)
+# Load EasyOCR model (CPU only)
 reader = easyocr.Reader(['en'], gpu=False)
 
-# Enhance image before OCR
+# Image enhancement
 def enhance_image(image):
     image = image.convert("L")  # Grayscale
-    image = ImageOps.invert(image)  # Invert black ↔ white
-    image = ImageEnhance.Contrast(image).enhance(2.0)
+    image = ImageOps.invert(image)
+    image = ImageEnhance.Contrast(image).enhance(2.5)
     image = ImageEnhance.Sharpness(image).enhance(2.5)
-    image = image.resize((image.width * 2, image.height * 2))  # Enlarge
+    image = image.resize((image.width * 2, image.height * 2))
     return image
 
-# Main function to extract weight
+# Detect weight from image
 def detect_weight(image):
     try:
-        # Enhance
         processed_image = enhance_image(image)
-
-        # Convert to array
         np_image = np.array(processed_image)
 
-        # OCR
+        # Perform OCR
         result = reader.readtext(np_image, detail=0)
-        text = " ".join(result)
+        full_text = " ".join(result)
+        print("OCR Output:", full_text)  # DEBUGGING LINE
 
-        # Extract decimal weight like 52.75
-        match = re.search(r"\d{1,4}\.\d{1,4}", text)
+        # Match both integer and decimal
+        match = re.search(r"\d{1,4}(?:\.\d{1,4})?", full_text)
         weight = match.group(0) if match else "Not detected"
 
-        # IST Time
+        # Get IST time
         ist = pytz.timezone("Asia/Kolkata")
-        timestamp = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S")
+        current_time = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S")
 
-        return f"Weight: {weight} kg\nCaptured At: {timestamp} (IST)", image
+        return f"Weight: {weight} kg\nCaptured At: {current_time} (IST)\n\nRaw OCR: {full_text}", image
     except Exception as e:
         return f"Error: {str(e)}", image
 
-# Gradio Interface
+# Gradio UI
 interface = gr.Interface(
     fn=detect_weight,
     inputs=gr.Image(type="pil", label="Upload or Capture Image"),
-    outputs=[
-        gr.Textbox(label="Weight Info"),
-        gr.Image(label="Snapshot")
-    ],
-    title="⚖️ Auto Weight Detector (Decimal Accurate)",
-    description="Detects exact weight including decimals (e.g., 52.75 kg) using EasyOCR. Shows IST time."
+    outputs=[gr.Textbox(label="Weight Info"), gr.Image(label="Snapshot")],
+    title="⚖️ Accurate Auto Weight Detector (with Decimal)",
+    description="Detects weight from digital scale image using EasyOCR (no Tesseract). Shows IST timestamp."
 )
 
 interface.launch()