Update app.py

app.py

@@ -7,17 +7,22 @@ def process_image(img):
     if img is None:
         return "No image uploaded", None, None
 
+    # Get IST time
     ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%d-%m-%Y %I:%M:%S %p")
+
+    # Extract weight and confidence
     weight, confidence = extract_weight_from_image(img)
-    return f"{weight} g (Confidence: {confidence}%)", ist_time, img
+
+    # Format output in kg
+    return f"{weight} kg (Confidence: {confidence}%)", ist_time, img
 
 with gr.Blocks(title="⚖️ Auto Weight Logger") as demo:
-    gr.Markdown("# ⚖️ Auto Weight Logger")
-    gr.Markdown("Upload or capture an image of a **digital scale display**. It will auto-detect the weight.")
+    gr.Markdown("## ⚖️ Auto Weight Logger")
+    gr.Markdown("📷 **Upload or capture an image of a digital weight scale.** The app will auto-detect the weight in kilograms using OCR.")
 
     with gr.Row():
-        image_input = gr.Image(type="pil", label="📷 Upload or Capture Image")
-        output_weight = gr.Textbox(label="⚖️ Detected Weight")
+        image_input = gr.Image(type="pil", label="Upload / Capture Image")
+        output_weight = gr.Textbox(label="⚖️ Detected Weight (in kg)")
 
     with gr.Row():
         timestamp = gr.Textbox(label="🕒 Captured At (IST)")