Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,30 +1,63 @@
 import streamlit as st
 from PIL import Image
+from ocr_engine import extract_weight_from_image
+import urllib.parse
 
-st.title("📸 Auto Weight Logger with Retake Option")
-
-# Session state to track if image is captured
-if 'captured' not in st.session_state:
-    st.session_state.captured = False
-if 'image' not in st.session_state:
-    st.session_state.image = None
-
-# Function to reset camera
-def retake():
-    st.session_state.captured = False
-    st.session_state.image = None
-
-# Only show camera if image is not captured
-if not st.session_state.captured:
-    image = st.camera_input("Take a picture")
-    if image:
-        st.session_state.image = image
-        st.session_state.captured = True
-
-# If image is captured, show options
-if st.session_state.captured and st.session_state.image:
-    st.image(st.session_state.image, caption="Captured Image", use_column_width=True)
-    st.button("Retake", on_click=retake)
-    st.success("✅ Image captured successfully.")
-    # You can add OCR or Salesforce upload button here
+st.set_page_config(page_title="⚖️ Auto Weight Logger", layout="centered")
+st.title("⚖️ Auto Weight Logger")
 
+# ✅ Session state to control camera
+if "retake" not in st.session_state:
+    st.session_state.retake = False
+
+# 🔘 Retake Button
+if st.button("🔁 Clear / Retake Photo"):
+    st.session_state.retake = True
+    st.rerun()
+
+# ✅ If this is the rerun after clearing, reset and skip camera once
+if st.session_state.retake:
+    st.session_state.retake = False
+    st.rerun()
+
+# ✅ At this point, camera will always be shown
+img_data = st.camera_input("📷 Capture the weight display")
+
+if img_data:
+    st.success("✅ Image captured successfully!")
+
+    if len(img_data.getvalue()) > 5 * 1024 * 1024:
+        st.error("❌ Image too large (>5MB). Please try again.")
+        st.stop()
+
+    image = Image.open(img_data)
+    st.image(image, caption="📸 Snapshot", use_column_width=True)
+
+    with st.spinner("🔍 Extracting weight..."):
+        weight, confidence = extract_weight_from_image(image)
+
+    st.write(f"🛠️ DEBUG: weight = {weight}, confidence = {confidence}")
+
+    if not weight or confidence < 80:
+        st.error(f"⚠️ OCR confidence too low ({int(confidence)}%). Please retake the photo.")
+        if st.button("🔁 Retake Photo"):
+            st.session_state.retake = True
+            st.rerun()
+        st.stop()
+
+    st.success(f"✅ Detected Weight: {weight} g (Confidence: {int(confidence)}%)")
+
+    device_id = "BAL-001"
+    image_url = ""  # Optional
+
+    encoded_weight = urllib.parse.quote(str(weight))
+    encoded_device = urllib.parse.quote(device_id)
+    encoded_image_url = urllib.parse.quote(image_url)
+
+    salesforce_url = (
+        "https://autoweightlogger-dev-ed.my.salesforce-sites.com/"
+        f"weight_logger_page?WeightInput={encoded_weight}&DeviceID={encoded_device}&ImageURL={encoded_image_url}"
+    )
+
+    st.markdown("### 📤 Send to Salesforce")
+    st.markdown(f"[✅ Click here to confirm and log in Salesforce]({salesforce_url})", unsafe_allow_html=True)