Update app.py

app.py

@@ -1,51 +1,58 @@
 import gradio as gr
 from PIL import Image, ImageEnhance, ImageOps
+import numpy as np
 import easyocr
 import re
 from datetime import datetime
 import pytz
 
-# Initialize EasyOCR reader
-reader = easyocr.Reader(['en'])
+# Load EasyOCR model (no Tesseract needed)
+reader = easyocr.Reader(['en'], gpu=False)
 
-# Preprocessing to enhance image for OCR
+# Enhance image before OCR
 def enhance_image(image):
     image = image.convert("L")  # Grayscale
-    image = ImageOps.invert(image)  # Invert
+    image = ImageOps.invert(image)  # Invert black ↔ white
     image = ImageEnhance.Contrast(image).enhance(2.0)
     image = ImageEnhance.Sharpness(image).enhance(2.5)
     image = image.resize((image.width * 2, image.height * 2))  # Enlarge
     return image
 
-# Weight detection function
+# Main function to extract weight
 def detect_weight(image):
     try:
+        # Enhance
         processed_image = enhance_image(image)
 
-        # OCR using EasyOCR
-        result = reader.readtext(np.array(processed_image), detail=0)
-        full_text = " ".join(result)
+        # Convert to array
+        np_image = np.array(processed_image)
 
-        # Extract number with decimal point
-        match = re.search(r"(\d{1,4}\.\d{1,4})", full_text)
-        weight = match.group(1) if match else "Not detected"
+        # OCR
+        result = reader.readtext(np_image, detail=0)
+        text = " ".join(result)
 
-        # Get IST time
-        ist = pytz.timezone('Asia/Kolkata')
-        current_time = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S")
+        # Extract decimal weight like 52.75
+        match = re.search(r"\d{1,4}\.\d{1,4}", text)
+        weight = match.group(0) if match else "Not detected"
 
-        return f"Weight: {weight} kg\nCaptured At: {current_time} (IST)", image
+        # IST Time
+        ist = pytz.timezone("Asia/Kolkata")
+        timestamp = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S")
 
+        return f"Weight: {weight} kg\nCaptured At: {timestamp} (IST)", image
     except Exception as e:
         return f"Error: {str(e)}", image
 
-# Gradio interface
+# Gradio Interface
 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="⚖️ Accurate Auto Weight Detector",
-    description="Detects full decimal weight (e.g., 52.75 kg) using EasyOCR and shows timestamp (IST)"
+    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."
 )
 
 interface.launch()