Update src/streamlit_app.py

src/streamlit_app.py

@@ -14,8 +14,8 @@ if "show_camera" not in st.session_state:
 
 # 🔘 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
+    st.session_state.show_camera = False  # hide camera temporarily
+    st.rerun()
 
 # 🔄 Reset camera on next run
 if not st.session_state.show_camera:
@@ -50,6 +50,9 @@ if img_data:
     # ❌ Handle low 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.show_camera = False
+            st.rerun()
         st.stop()
 
     # ✅ Success
@@ -57,7 +60,7 @@ if img_data:
 
     # Salesforce URL setup
     device_id = "BAL-001"
-    image_url = ""  # Optional
+    image_url = ""  # Optional: add image hosting logic later
 
     encoded_weight = urllib.parse.quote(str(weight))
     encoded_device = urllib.parse.quote(device_id)