Update app.py

app.py

@@ -4,28 +4,33 @@ from PIL import ImageFilter, ImageEnhance
 import pytz
 import numpy as np
 import cv2
-from ocr_engine import extract_weight_from_image
+from ocr_engine import extract_weight_from_image  # Modify if needed
 
+# Preprocess image to improve OCR accuracy
 def preprocess_image(img):
-    # Enhance sharpness & contrast (PIL)
+    # PIL enhancements
     img = img.filter(ImageFilter.SHARPEN)
     enhancer = ImageEnhance.Contrast(img)
     img = enhancer.enhance(2.0)
 
-    # Convert to OpenCV image
-    img_cv = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
-    return img_cv
+    # Convert to OpenCV format
+    cv_img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
+    return cv_img
 
+# Process image and extract weight using OCR
 def process_image(img):
     if img is None:
         return "No image uploaded", None, None, "No OCR output"
 
     preprocessed_cv = preprocess_image(img)
+
+    # Extract data from your OCR engine (should handle OpenCV input)
     weight, confidence, raw_text = extract_weight_from_image(preprocessed_cv)
 
     ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%d-%m-%Y %I:%M:%S %p")
     return f"{weight} kg (Confidence: {confidence}%)", ist_time, img, raw_text
 
+# Gradio UI setup
 with gr.Blocks(title="⚖️ Auto Weight Logger") as demo:
     gr.Markdown("## ⚖️ Auto Weight Logger")
     gr.Markdown("📷 **Upload or capture an image of a digital weight scale.** The app will auto-detect the weight in kilograms using OCR.")