Update app.py

app.py

@@ -9,25 +9,32 @@ def process_image(image):
         return "No image provided", "", None
     try:
         weight = extract_weight(image)
-
         ist = pytz.timezone('Asia/Kolkata')
-        now_ist = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S IST")
-
-        return weight, now_ist, image
+        timestamp = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S IST")
+        return weight, timestamp, 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 digital weight image. AI detects numeric value and unit (kg or grams).")
+with gr.Blocks(css=".gr-button {background-color: #2e7d32 !important; color: white !important;}") as demo:
+    gr.Markdown("""
+    <h1 style='text-align: center; color: #2e7d32;'>📷 Auto Weight Logger</h1>
+    <p style='text-align: center;'>AI-powered OCR tool to detect weights (kg or grams) from digital balance display images.</p>
+    <hr style='border: 1px solid #ddd;'/>
+    """)
+
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="📁 Upload or Capture Image", elem_id="img_upload")
+
+    detect_btn = gr.Button("🚀 Detect Weight", elem_id="detect_btn")
 
-    image_input = gr.Image(type="pil", label="📁 Upload or Capture Image")
-    detect_btn = gr.Button("🚀 Detect Weight")
+    with gr.Row():
+        weight_out = gr.Textbox(label="📦 Detected Weight", placeholder="e.g., 72.4 kg", show_copy_button=True)
+        time_out = gr.Textbox(label="🕒 Captured At (IST)", placeholder="e.g., 2025-06-30 14:32:10")
 
-    weight_out = gr.Textbox(label="Detected Weight")
-    time_out = gr.Textbox(label="Captured At (IST)")
-    snapshot = gr.Image(label="Snapshot")
+    snapshot = gr.Image(label="📸 Snapshot Preview")
 
     detect_btn.click(fn=process_image, inputs=image_input, outputs=[weight_out, time_out, snapshot])
 
-demo.launch()
+    gr.Markdown("""
+    <br><p style='text-align: center; color: gray;'>Developed by Shalu | Powered by Hugging Face OCR 🚀</p>
+    """)