Update app.py

app.py

@@ -1,43 +1,33 @@
-import os
-import subprocess
-
-# Force install mmocr at runtime if not available
-try:
-    from mmocr.utils.ocr import MMOCR
-    print("✅ MMOCR module loaded successfully.")
-except ImportError:
-    print("⚠️ MMOCR not found, installing...")
-    subprocess.run(["pip", "install", "mmocr==0.5.0", "--no-deps"])
-    from mmocr.utils.ocr import MMOCR
-    print("✅ MMOCR installed and loaded.")
-
-from ocr_engine import extract_weight_from_image
 import gradio as gr
-from datetime import datetime
-import pytz
-from PIL import Image
+from mmocr.apis import MMOCR
+import cv2
+import os
 
-def process_image(image):
-    if image is None:
-        return "No image provided", "", None, None
+# Initialize MMOCR once
+ocr = MMOCR(det='DB_r18', recog='SAR', device='cpu')  # use 'cuda:0' if GPU is available
 
-    weight, debug_text = extract_weight_from_image(image)
+def detect_weight(image):
+    results = ocr.readtext(image, output=None)
+    
+    detected_texts = [item['text'] for item in results[0]['result']]
+    
+    # Search for weight-like patterns like "25.5 kg", "100kg", etc.
+    weight = "Not detected"
+    for text in detected_texts:
+        if 'kg' in text.lower():
+            weight = text
+            break
 
-    # Get IST time
-    ist = pytz.timezone('Asia/Kolkata')
-    captured_at = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S %Z")
+    return f"Detected Weight: {weight}"
 
-    return f"{weight} kg" if weight else "Not detected", captured_at, image, debug_text
+# Gradio UI
+interface = gr.Interface(
+    fn=detect_weight,
+    inputs=gr.Image(type="filepath", label="Upload Weight Image"),
+    outputs=gr.Textbox(label="OCR Output"),
+    title="Weight Detector using MMOCR",
+    description="Upload an image with weight text (e.g., '25 kg'), and this app will detect it."
+)
 
-gr.Interface(
-    fn=process_image,
-    inputs=gr.Image(type="pil", label="Upload Weight Image"),
-    outputs=[
-        gr.Textbox(label="Detected Weight"),
-        gr.Textbox(label="Captured At (IST)"),
-        gr.Image(label="Snapshot"),
-        gr.Textbox(label="OCR Debug Output")
-    ],
-    title="Auto Weight Logger using MMOCR",
-    description="Upload a weight scale image. This app uses MMOCR to detect the weight."
-).launch()
+if __name__ == "__main__":
+    interface.launch()