Update app.py

app.py

@@ -2,30 +2,25 @@ import gradio as gr
 from datetime import datetime
 from PIL import ImageFilter, ImageEnhance
 import pytz
-import numpy as np
-import cv2
 from ocr_engine import extract_weight_from_image
 
-def preprocess_image(img):
-    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
-
 def process_image(img):
     if img is None:
         return "No image uploaded", None, None, "No OCR output"
 
-    preprocessed_cv = preprocess_image(img)
-    weight, confidence, raw_text = extract_weight_from_image(preprocessed_cv)
+    # Enhance image before deblurring
+    img = img.filter(ImageFilter.SHARPEN)
+    enhancer = ImageEnhance.Contrast(img)
+    img = enhancer.enhance(2.0)
+
+    weight, confidence, raw_text, deblurred_img = extract_weight_from_image(img)
 
     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
+    return f"{weight} kg (Confidence: {confidence}%)", ist_time, deblurred_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.")
+    gr.Markdown("📷 **Upload or capture an image of a digital weight scale.** This version uses Restormer to deblur before OCR.")
 
     with gr.Row():
         image_input = gr.Image(type="pil", label="Upload / Capture Image")
@@ -33,12 +28,16 @@ with gr.Blocks(title="⚖️ Auto Weight Logger") as demo:
 
     with gr.Row():
         timestamp = gr.Textbox(label="🕒 Captured At (IST)")
-        snapshot = gr.Image(label="📸 Snapshot Image")
+        snapshot = gr.Image(label="📸 Deblurred Snapshot")
 
     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])
+    submit.click(
+        process_image,
+        inputs=image_input,
+        outputs=[output_weight, timestamp, snapshot, debug_output]
+    )
 
 demo.launch()