Update app.py

app.py

@@ -3,28 +3,27 @@ import cv2
 import numpy as np
 from paddleocr import PaddleOCR
 from datetime import datetime
+from pytz import timezone
 import re
 from PIL import Image
 
-# Initialize PaddleOCR once
-ocr = PaddleOCR(use_angle_cls=True, lang='en')  # English OCR model
+# Initialize PaddleOCR
+ocr = PaddleOCR(use_angle_cls=True, lang='en')
 
 def detect_weight(image):
     try:
         if image is None:
             return "No image uploaded", "N/A", None
 
-        # Convert PIL image to RGB numpy array
         image = image.convert("RGB")
         image_np = np.array(image)
 
-        # Convert to grayscale and enhance contrast
         gray = cv2.cvtColor(image_np, cv2.COLOR_RGB2GRAY)
         enhanced = cv2.equalizeHist(gray)
         rgb_image = cv2.cvtColor(enhanced, cv2.COLOR_GRAY2RGB)
 
-        # Perform OCR
-        result = ocr.ocr(rgb_image, cls=True)
+        # OCR (without cls= argument for compatibility)
+        result = ocr.ocr(rgb_image)
 
         best_match = None
         best_conf = 0
@@ -38,16 +37,14 @@ def detect_weight(image):
                     best_conf = conf
 
         if best_match:
-            now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-            return f"Weight: {best_match} kg (Confidence: {round(best_conf * 100, 2)}%)", now, image
+            now_ist = datetime.now(timezone('Asia/Kolkata')).strftime("%Y-%m-%d %H:%M:%S IST")
+            return f"Weight: {best_match} kg (Confidence: {round(best_conf * 100, 2)}%)", now_ist, image
         else:
             return "No weight detected kg (Confidence: 0.0%)", "N/A", image
 
     except Exception as e:
-        # Show error in output instead of crashing
         return f"Error: {str(e)}", "Error", None
 
-# Gradio UI
 gr.Interface(
     fn=detect_weight,
     inputs=gr.Image(type="pil", label="Upload or Capture Weight Image"),