Update app.py

app.py

@@ -1,22 +1,24 @@
 import gradio as gr
 from PIL import Image
-from datetime import datetime
 from ocr_engine import extract_weight_from_image
 from utils import get_ist_time
 
 def process_image(image):
-    if image is None:
-        return "No image provided", None, None, None
+    try:
+        if image is None:
+            return "No image provided", "-", None, "-"
 
-    weight, confidence = extract_weight_from_image(image)
-    timestamp = get_ist_time()
-    if not weight:
-        return "No weight detected", timestamp, image, "-"
-    
-    return f"{weight} g (Confidence: {confidence}%)", timestamp, image, f"https://your-salesforce-site.com/log-weight?weight={weight}&time={timestamp}"
+        weight, confidence = extract_weight_from_image(image)
+        timestamp = get_ist_time()
 
-title = "⚖️ Auto Weight Logger"
-description = "Upload or capture an image to extract the weight and log the result."
+        if not weight:
+            return "No weight detected", timestamp, image, "-"
+
+        log_url = f"https://your-salesforce-site.com/log-weight?weight={weight}&time={timestamp}"
+        return f"{weight} g (Confidence: {confidence}%)", timestamp, image, log_url
+
+    except Exception as e:
+        return f"Error: {str(e)}", "-", None, "-"
 
 iface = gr.Interface(
     fn=process_image,
@@ -27,8 +29,8 @@ iface = gr.Interface(
         gr.Image(label="Snapshot Image"),
         gr.Textbox(label="Salesforce Log URL"),
     ],
-    title=title,
-    description=description,
+    title="⚖️ Auto Weight Logger",
+    description="Upload or capture an image to extract the weight and log the result.",
     live=True,
 )