Update app.py

app.py

@@ -1,37 +1,41 @@
 import gradio as gr
+from PIL import Image
+from datetime import datetime
+import pytz
 from ocr_engine import extract_weight_from_image
-from utils import get_ist_time
 
-def process_image(image):
-    try:
-        if image is None:
-            return "No image provided", "-", None, "-"
-        
-        weight, confidence = extract_weight_from_image(image)
-        timestamp = get_ist_time()
+def process_image(img):
+    # Convert image to RGB in case it's RGBA or grayscale
+    img = img.convert("RGB")
 
-        if not weight:
-            return "No weight detected", timestamp, image, "-"
+    # Get IST timestamp
+    ist = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S")
 
-        sf_url = f"https://your-salesforce-site.com/log-weight?weight={weight}&time={timestamp}"
-        return f"{weight} g (Confidence: {confidence}%)", timestamp, image, sf_url
+    # Run OCR
+    weight, confidence = extract_weight_from_image(img)
 
-    except Exception as e:
-        return f"Error: {str(e)}", "-", None, "-"
+    # Format output
+    result = f"📅 Captured At (IST): {ist}\n"
+    if confidence > 0:
+        result += f"⚖️ Detected Weight: **{weight}**\n"
+        result += f"✅ Confidence: `{confidence:.2f}`"
+    else:
+        result += "❌ No weight detected. Try a clearer image."
 
-iface = gr.Interface(
+    return img, result
+
+# Gradio UI
+demo = gr.Interface(
     fn=process_image,
-    inputs=gr.Image(type="pil", label="Upload or Capture Snapshot"),
+    inputs=gr.Image(type="pil", label="Upload or Capture Image"),
     outputs=[
-        gr.Textbox(label="Detected Weight"),
-        gr.Textbox(label="Captured At (IST)"),
-        gr.Image(label="Snapshot Image"),
-        gr.Textbox(label="Salesforce Log URL"),
+        gr.Image(type="pil", label="Snapshot"),
+        gr.Markdown(label="Weight Result")
     ],
-    title="⚖️ Auto Weight Logger",
-    description="Upload or capture a digital scale image to extract the weight.",
-    live=True,
+    title="⚖️ Auto Weight Logger (PaddleOCR)",
+    description="Upload or capture an image of a weighing scale. This app uses PaddleOCR to detect the weight value.",
+    allow_flagging="never"
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()