Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,30 +7,28 @@ st.set_page_config(page_title="⚖️ Auto Weight Logger", layout="centered")
 st.title("⚖️ Auto Weight Logger")
 
 # ------------------------------
-# ✅ Session state management
+# ✅ Setup session state for reset
 # ------------------------------
-if "show_camera" not in st.session_state:
-    st.session_state.show_camera = True
+if "reset" not in st.session_state:
+    st.session_state.reset = False
 
-# 🔘 Clear or Retake Button pressed
-if st.button("🔁 Clear / Retake Photo"):
-    st.session_state.show_camera = False  # hide camera temporarily
+# 🔁 Handle reset if needed (immediately rerun and then show camera again)
+if st.session_state.reset:
+    st.session_state.reset = False  # Clear flag
     st.rerun()
 
-# 🔄 Reset camera on next run
-if not st.session_state.show_camera:
-    st.session_state.show_camera = True
+# 🔘 Clear / Retake button
+if st.button("🔁 Clear / Retake Photo"):
+    st.session_state.reset = True
     st.rerun()
 
 # ------------------------------
 # 📷 Show camera input
 # ------------------------------
-img_data = None
-if st.session_state.show_camera:
-    img_data = st.camera_input("📷 Capture the weight display")
+img_data = st.camera_input("📷 Capture the weight display")
 
 # ------------------------------
-# ✅ Process if image is captured
+# ✅ Handle image if captured
 # ------------------------------
 if img_data:
     st.success("✅ Image captured successfully!")
@@ -42,25 +40,26 @@ if img_data:
     image = Image.open(img_data)
     st.image(image, caption="📸 Snapshot", use_column_width=True)
 
+    # 🔍 Extract weight via OCR
     with st.spinner("🔍 Extracting weight..."):
         weight, confidence = extract_weight_from_image(image)
 
     st.write(f"🛠️ DEBUG: weight = {weight}, confidence = {confidence}")
 
-    # ❌ Handle low confidence
+    # ❌ If low confidence, ask to retake
     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.show_camera = False
+            st.session_state.reset = True
             st.rerun()
         st.stop()
 
     # ✅ Success
     st.success(f"✅ Detected Weight: {weight} g (Confidence: {int(confidence)}%)")
 
-    # Salesforce URL setup
+    # Salesforce parameters
     device_id = "BAL-001"
-    image_url = ""  # Optional: add image hosting logic later
+    image_url = ""  # Optional image hosting URL
 
     encoded_weight = urllib.parse.quote(str(weight))
     encoded_device = urllib.parse.quote(device_id)