Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,29 +6,23 @@ import urllib.parse
 st.set_page_config(page_title="⚖️ Auto Weight Logger", layout="centered")
 st.title("⚖️ Auto Weight Logger")
 
-# ✅ Session state to manage retake and image
-if "image_captured" not in st.session_state:
-    st.session_state.image_captured = False
+# Initialize session state
 if "image_data" not in st.session_state:
     st.session_state.image_data = None
 
-# 🔘 Retake / Clear Button
+# Clear button to retake image
 if st.button("🔁 Clear / Retake Photo"):
-    st.session_state.image_captured = False
     st.session_state.image_data = None
-    st.rerun()
 
-# ✅ Show camera only if no image captured
-if not st.session_state.image_captured:
+# Show camera if no image captured
+if st.session_state.image_data is None:
     img_data = st.camera_input("📷 Capture the weight display")
-
     if img_data:
         st.session_state.image_data = img_data
-        st.session_state.image_captured = True
-        st.rerun()
+        st.experimental_rerun()  # NOTE: Works fine in most Streamlit cloud spaces
 
-# ✅ If image is already captured
-if st.session_state.image_captured and st.session_state.image_data:
+# Show captured image and process
+if st.session_state.image_data:
     st.success("✅ Image captured successfully!")
 
     img_data = st.session_state.image_data
@@ -48,16 +42,15 @@ if st.session_state.image_captured and st.session_state.image_data:
     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.image_captured = False
             st.session_state.image_data = None
-            st.rerun()
+            st.experimental_rerun()
         st.stop()
 
     st.success(f"✅ Detected Weight: {weight} g (Confidence: {int(confidence)}%)")
 
-    # 🌐 Salesforce link logic
+    # 🌐 Salesforce URL
     device_id = "BAL-001"
-    image_url = ""  # Optional
+    image_url = ""
 
     encoded_weight = urllib.parse.quote(str(weight))
     encoded_device = urllib.parse.quote(device_id)