Update app.py

app.py

@@ -9,19 +9,25 @@ def process_image(img):
 
     # Get IST time
     ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%d-%m-%Y %I:%M:%S %p")
-
-    # Extract weight and confidence
+    
+    # Run OCR
     weight, confidence = extract_weight_from_image(img)
 
-    # Format output in kg
-    return f"{weight} kg (Confidence: {confidence}%)", ist_time, img
+    # Format output with "kg"
+    if weight.replace('.', '', 1).isdigit():
+        formatted_weight = f"{weight} kg (Confidence: {confidence}%)"
+    else:
+        formatted_weight = f"{weight} (Confidence: {confidence}%)"
+
+    return formatted_weight, ist_time, img
 
+# Gradio UI
 with gr.Blocks(title="⚖️ Auto Weight Logger") as demo:
-    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.")
+    gr.Markdown("# ⚖️ Auto Weight Logger")
+    gr.Markdown("Upload or capture an image of a **digital scale display**. It will auto-detect the weight in kg.")
 
     with gr.Row():
-        image_input = gr.Image(type="pil", label="Upload / Capture Image")
+        image_input = gr.Image(type="pil", label="📷 Upload or Capture Image")
         output_weight = gr.Textbox(label="⚖️ Detected Weight (in kg)")
 
     with gr.Row():