Update app.py

app.py

@@ -1,53 +1,32 @@
 import gradio as gr
-import numpy as np
-from PIL import Image
-import cv2
-import re
-from paddleocr import PaddleOCR
 from datetime import datetime
+import pytz
+from ocr_engine import extract_weight_from_image
 
-# Load OCR model
-ocr = PaddleOCR(use_angle_cls=True, lang='en')
-
-def preprocess_image(image):
-    img = np.array(image.convert("RGB"))
-    gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
-    _, thresh = cv2.threshold(gray, 150, 255, cv2.THRESH_BINARY)
-    return Image.fromarray(thresh)
-
-def extract_weight_text(image):
-    try:
-        results = ocr.ocr(np.array(image), cls=True)
-        texts = []
-        for line in results[0]:
-            text = line[1][0]
-            texts.append(text)
-            match = re.search(r"\d+\.\d+", text)
-            if match:
-                return match.group(), texts
-        return None, texts
-    except Exception as e:
-        return f"Error: {str(e)}", []
-
-def detect_weight(image):
-    if image is None:
-        return "No image uploaded.", "", None, ""
-    pre_img = preprocess_image(image)
-    weight, texts = extract_weight_text(pre_img)
-    if weight and "Error" not in weight:
-        return f"Detected Weight: {weight} kg", datetime.now().strftime("Captured At: %Y-%m-%d %H:%M:%S"), pre_img, "\n".join(texts)
-    else:
-        return "Weight Not Detected", "", pre_img, "\n".join(texts)
-
-gr.Interface(
-    fn=detect_weight,
-    inputs=gr.Image(type="pil", label="Upload or Capture Image"),
-    outputs=[
-        gr.Textbox(label="Weight"),
-        gr.Textbox(label="Timestamp"),
-        gr.Image(label="Preprocessed Image"),
-        gr.Textbox(label="Detected Text (Debug)")
-    ],
-    title="Auto Weight Logger (Debug Mode)",
-    description="Upload a clear image of a digital scale. This version shows detected text for debugging."
-).launch()
+def process_image(img):
+    if img is None:
+        return "No image uploaded", None, None, "No OCR output"
+
+    ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%d-%m-%Y %I:%M:%S %p")
+    weight, confidence, raw_text = extract_weight_from_image(img)
+    return f"{weight} kg (Confidence: {confidence}%)", ist_time, img, raw_text
+
+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.")
+
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="Upload / Capture Image")
+        output_weight = gr.Textbox(label="⚖️ Detected Weight (in kg)")
+
+    with gr.Row():
+        timestamp = gr.Textbox(label="🕒 Captured At (IST)")
+        snapshot = gr.Image(label="📸 Snapshot Image")
+
+    with gr.Row():
+        debug_output = gr.Textbox(label="🪵 Raw OCR Output")
+
+    submit = gr.Button("🔍 Detect Weight")
+    submit.click(process_image, inputs=image_input, outputs=[output_weight, timestamp, snapshot, debug_output])
+
+demo.launch()