Update app.py

app.py

@@ -1,17 +1,20 @@
 import streamlit as st
 from PIL import Image, UnidentifiedImageError
-import io
 from datetime import datetime
 import pytz
 from ocr_engine import extract_weight_from_image
 
+# Streamlit page config
 st.set_page_config(page_title="⚖️ Auto Weight Logger", layout="centered")
 st.title("⚖️ Auto Weight Logger")
 
+# Debug: Confirm app starts
+st.write("✅ Streamlit app initialized.")
+
 if "camera_key" not in st.session_state:
     st.session_state["camera_key"] = 0
 
-# UI: Upload or Capture Image
+# Upload or Capture
 col1, col2 = st.columns(2)
 with col1:
     uploaded_file = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])
@@ -33,26 +36,20 @@ elif camera_image:
     except UnidentifiedImageError:
         st.error("Invalid camera image.")
 
-# Process image
+# OCR Process
 if image:
-    with st.spinner("Detecting weight..."):
-        weight, confidence = extract_weight_from_image(image)
-
-    # Timestamp
+    st.markdown("### 🕒 Captured At (IST)")
     ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S")
-
-    # Output
-    st.markdown("### 📅 Captured At (IST)")
     st.info(ist_time)
 
-    st.markdown("### 🖼️ Snapshot")
+    st.markdown("### 🖼️ Snapshot Image")
     st.image(image, width=400)
 
-    st.markdown("### ⚖️ Detected Weight")
+    with st.spinner("🔍 Detecting weight..."):
+        weight, confidence = extract_weight_from_image(image)
+
+    st.markdown("### ⚖️ Captured Weight & Confidence")
     if confidence > 0:
         st.success(f"Detected Weight: **{weight}**  \nConfidence: `{confidence:.2f}`")
     else:
-        st.error("No weight detected. Try with a clearer image.")
-
-    # (Optional) Send to Salesforce button
-    st.markdown("🔗 [Send to Salesforce](#)", unsafe_allow_html=True)
+        st.error("No weight detected. Please upload a clearer image.")