Update app.py

app.py

@@ -4,25 +4,26 @@ from datetime import datetime
 from ocr_engine import extract_weight
 
 def process_image(image):
-    if image is None:
-        return "No image provided", None, None
-    weight = extract_weight(image)
-    now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    return f"{weight} grams", now, image
+    try:
+        if image is None:
+            return "No image provided", "No timestamp", None
+        weight = extract_weight(image)
+        now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        return f"{weight} grams", now, image
+    except Exception as e:
+        return f"Error: {str(e)}", "Error", None
 
 with gr.Blocks() as demo:
     gr.Markdown("## 📷 Auto Weight Logger — OCR-Based Smart Scale Reader")
-    gr.Markdown("Upload or capture a digital scale snapshot. This app detects the weight using OCR.")
+    gr.Markdown("Upload a photo of a digital scale. The app will detect and log the weight.")
 
-    image_input = gr.Image(type="pil", label="📁 Upload or Capture Image")
+    image_input = gr.Image(type="pil", label="📁 Upload or Capture Digital Scale Image")
 
     detect_btn = gr.Button("🚀 Detect Weight")
 
-    with gr.Row():
-        weight_out = gr.Textbox(label="Detected Weight")
-        time_out = gr.Textbox(label="Captured At (IST)")
-
-    snapshot = gr.Image(label="Snapshot")
+    weight_out = gr.Textbox(label="📦 Detected Weight")
+    time_out = gr.Textbox(label="⏱️ Captured At (IST)")
+    snapshot = gr.Image(label="🖼️ Snapshot Preview")
 
     detect_btn.click(fn=process_image, inputs=image_input, outputs=[weight_out, time_out, snapshot])