Update app.py

app.py

@@ -1,37 +1,37 @@
 import gradio as gr
-from PIL import Image, UnidentifiedImageError
-from datetime import datetime
-import pytz
 from ocr_engine import extract_weight_from_image
+from utils import get_ist_time
 
-def detect_weight(image: Image.Image):
+def process_image(image):
     try:
-        # Get IST time
-        ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S")
+        if image is None:
+            return "No image provided", "-", None, "-"
 
-        # OCR extraction
         weight, confidence = extract_weight_from_image(image)
+        timestamp = get_ist_time()
 
-        if confidence > 0:
-            result = f"✅ **Detected Weight:** {weight}  \n📊 **Confidence:** {confidence:.2f}  \n🕒 **Captured At (IST):** {ist_time}"
-        else:
-            result = "❌ No weight detected. Please upload a clearer image."
+        if not weight:
+            return "No weight detected", timestamp, image, "-"
 
-        return image, result
-    except UnidentifiedImageError:
-        return None, "❌ Invalid image format."
+        log_url = f"https://your-salesforce-site.com/log-weight?weight={weight}&time={timestamp}"
+        return f"{weight} g (Confidence: {confidence}%)", timestamp, image, log_url
 
-demo = gr.Interface(
-    fn=detect_weight,
-    inputs=gr.Image(type="pil", label="Upload or Capture Image"),
+    except Exception as e:
+        return f"Error: {str(e)}", "-", None, "-"
+
+iface = gr.Interface(
+    fn=process_image,
+    inputs=gr.Image(type="pil", label="Upload or Capture Snapshot"),
     outputs=[
-        gr.Image(type="pil", label="Snapshot"),
-        gr.Markdown(label="Result")
+        gr.Textbox(label="Detected Weight"),
+        gr.Textbox(label="Captured At (IST)"),
+        gr.Image(label="Snapshot Image"),
+        gr.Textbox(label="Salesforce Log URL"),
     ],
     title="⚖️ Auto Weight Logger",
-    description="Upload or capture a digital scale image to detect weight automatically.",
-    allow_flagging="never"
+    description="Upload or capture a digital scale snapshot to detect weight automatically.",
+    live=True,
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    iface.launch()