Update app.py

app.py

@@ -4,33 +4,25 @@ from PIL import ImageFilter, ImageEnhance
 import pytz
 import numpy as np
 import cv2
-from ocr_engine import extract_weight_from_image  # Modify if needed
+from ocr_engine import extract_weight_from_image
 
-# Preprocess image to improve OCR accuracy
 def preprocess_image(img):
-    # PIL enhancements
     img = img.filter(ImageFilter.SHARPEN)
     enhancer = ImageEnhance.Contrast(img)
     img = enhancer.enhance(2.0)
+    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.")