Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from PIL import Image
 from datetime import datetime
+import pytz
 from ocr_engine import extract_weight
 
 def process_image(image):
@@ -9,18 +10,22 @@ def process_image(image):
 
     try:
         weight = extract_weight(image)
-        timestamp = datetime.now().astimezone().strftime("%Y-%m-%d %H:%M:%S %Z")
-        return weight, timestamp, image
+
+        # Get IST time
+        ist = pytz.timezone('Asia/Kolkata')
+        now_ist = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S IST")
+
+        return weight, now_ist, image
     except Exception as e:
         return f"Error: {str(e)}", "", None
 
 with gr.Blocks() as demo:
     gr.Markdown("## 📷 Auto Weight Logger — Hugging Face OCR Edition")
-    gr.Markdown("Upload a photo of a digital scale. The app auto-detects weight (kg or grams).")
+    gr.Markdown("Upload a digital weight image. Auto-detects value & unit (kg or grams).")
 
     image_input = gr.Image(type="pil", label="📁 Upload or Capture Image")
-
     detect_btn = gr.Button("🚀 Detect Weight")
+
     weight_out = gr.Textbox(label="Detected Weight")
     time_out = gr.Textbox(label="Captured At (IST)")
     snapshot = gr.Image(label="Snapshot")