Update app.py

app.py

@@ -3,24 +3,27 @@ from datetime import datetime
 from PIL import ImageFilter, ImageEnhance
 import pytz
 from ocr_engine import extract_weight_from_image
+from PIL import Image
+
+def enhance_image(pil_img):
+    pil_img = pil_img.filter(ImageFilter.SHARPEN)
+    enhancer = ImageEnhance.Contrast(pil_img)
+    return enhancer.enhance(2.0)
 
 def process_image(img):
     if img is None:
         return "No image uploaded", None, None, "No OCR output"
 
-    # 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)
+    enhanced = enhance_image(img)
+    weight, confidence, raw_text, processed_img_cv = extract_weight_from_image(enhanced)
+    processed_img = Image.fromarray(processed_img_cv)
 
     ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%d-%m-%Y %I:%M:%S %p")
-    return f"{weight} kg (Confidence: {confidence}%)", ist_time, deblurred_img, raw_text
+    return f"{weight} kg (Confidence: {confidence}%)", ist_time, processed_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.** This version uses Restormer to deblur before OCR.")
+    gr.Markdown("📷 **Upload or capture an image of a digital weight scale.** This app sharpens images and extracts weights using EasyOCR.")
 
     with gr.Row():
         image_input = gr.Image(type="pil", label="Upload / Capture Image")
@@ -28,16 +31,14 @@ with gr.Blocks(title="⚖️ Auto Weight Logger") as demo:
 
     with gr.Row():
         timestamp = gr.Textbox(label="🕒 Captured At (IST)")
-        snapshot = gr.Image(label="📸 Deblurred Snapshot")
+        snapshot = gr.Image(label="🧪 Processed 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()