Update app.py

app.py

@@ -1,14 +1,13 @@
 import gradio as gr
-import cv2
-import numpy as np
 from paddleocr import PaddleOCR
+from PIL import Image
+import numpy as np
+import cv2
+import re
 from datetime import datetime
 from pytz import timezone
-import re
-from PIL import Image
 
-# Initialize PaddleOCR
-ocr = PaddleOCR(use_angle_cls=True, lang='en')
+ocr = PaddleOCR(use_angle_cls=False, lang='en')  # cls removed
 
 def detect_weight(image):
     try:
@@ -22,7 +21,6 @@ def detect_weight(image):
         enhanced = cv2.equalizeHist(gray)
         rgb_image = cv2.cvtColor(enhanced, cv2.COLOR_GRAY2RGB)
 
-        # OCR (without cls= argument for compatibility)
         result = ocr.ocr(rgb_image)
 
         best_match = None
@@ -30,11 +28,12 @@ def detect_weight(image):
 
         for line in result:
             for box in line:
-                text, conf = box[1]
-                match = re.search(r"\d+\.\d+", text)
-                if match and conf > best_conf:
-                    best_match = match.group()
-                    best_conf = conf
+                if len(box) > 1 and isinstance(box[1], tuple):
+                    text, conf = box[1]
+                    match = re.search(r"\d+\.\d+", text)
+                    if match and conf > best_conf:
+                        best_match = match.group()
+                        best_conf = conf
 
         if best_match:
             now_ist = datetime.now(timezone('Asia/Kolkata')).strftime("%Y-%m-%d %H:%M:%S IST")
@@ -45,14 +44,16 @@ def detect_weight(image):
     except Exception as e:
         return f"Error: {str(e)}", "Error", None
 
-gr.Interface(
-    fn=detect_weight,
-    inputs=gr.Image(type="pil", label="Upload or Capture Weight Image"),
-    outputs=[
-        gr.Text(label="Detected Weight"),
-        gr.Text(label="Captured At (IST)"),
-        gr.Image(label="Snapshot")
-    ],
-    title="Auto Weight Logger",
-    description="Upload or capture a digital scale image. This app detects the weight automatically using AI."
-).launch()
+with gr.Blocks() as demo:
+    gr.Markdown("## Auto Weight Logger\nUpload or capture a digital scale image. This app detects the weight automatically using AI.")
+
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="Upload or Capture Weight Image", tool="editor", sources=["upload", "camera"])
+        with gr.Column():
+            output_text = gr.Textbox(label="Detected Weight")
+            output_time = gr.Textbox(label="Captured At (IST)")
+            snapshot_output = gr.Image(label="📷 Snapshot")
+
+    image_input.change(fn=detect_weight, inputs=image_input, outputs=[output_text, output_time, snapshot_output])
+
+demo.launch()