Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,64 +1,30 @@
 import streamlit as st
 from PIL import Image
-from ocr_engine import extract_weight_from_image
-import urllib.parse
 
-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)
+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