Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,25 +7,31 @@ st.set_page_config(page_title="⚖️ Auto Weight Logger", layout="centered")
 st.title("⚖️ Auto Weight Logger")
 
 # ------------------------------
-# ✅ Session state for clearing input
+# ✅ Session state management
 # ------------------------------
-if "reset" not in st.session_state:
-    st.session_state.reset = False
+if "show_camera" not in st.session_state:
+    st.session_state.show_camera = True
 
-# 🔁 Auto rerun if reset is active
-if st.session_state.reset:
-    st.session_state.reset = False
-    st.rerun()
+# 🔘 Clear or Retake Button pressed
+if st.button("🔁 Clear / Retake Photo"):
+    st.session_state.show_camera = False  # hide camera for one cycle
+    st.experimental_rerun()  # or st.rerun() depending on streamlit version
 
-# 🔘 Manual clear/reset button
-if st.button("🔁 Clear Photo"):
-    st.session_state.reset = True
+# 🔄 Reset camera on next run
+if not st.session_state.show_camera:
+    st.session_state.show_camera = True
     st.rerun()
 
-# 📷 Capture webcam image
-img_data = st.camera_input("📷 Capture the weight display")
+# ------------------------------
+# 📷 Show camera input
+# ------------------------------
+img_data = None
+if st.session_state.show_camera:
+    img_data = st.camera_input("📷 Capture the weight display")
 
-# 🔍 Process image only if captured
+# ------------------------------
+# ✅ Process if image is captured
+# ------------------------------
 if img_data:
     st.success("✅ Image captured successfully!")
 
@@ -41,18 +47,15 @@ if img_data:
 
     st.write(f"🛠️ DEBUG: weight = {weight}, confidence = {confidence}")
 
-    # ❌ If confidence is low or result missing — show Retake
+    # ❌ Handle low confidence
     if not weight or confidence < 80:
-        st.error(f"⚠️ OCR confidence too low ({int(confidence)}%). Please retake the image.")
-        if st.button("🔁 Retake Photo"):
-            st.session_state.reset = True
-            st.rerun()
+        st.error(f"⚠️ OCR confidence too low ({int(confidence)}%). Please retake the photo.")
         st.stop()
 
-    # ✅ Valid result
+    # ✅ Success
     st.success(f"✅ Detected Weight: {weight} g (Confidence: {int(confidence)}%)")
 
-    # 🔗 Salesforce URL
+    # Salesforce URL setup
     device_id = "BAL-001"
     image_url = ""  # Optional