Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,11 +2,15 @@ import streamlit as st
 from PIL import Image
 from ocr_engine import extract_weight_from_image
 import urllib.parse
-from datetime import datetime
 
 st.set_page_config(page_title="⚖️ Auto Weight Logger", layout="centered")
 st.title("⚖️ Auto Weight Logger")
 
+# Show restart button at the top
+if st.button("🔁 Restart App"):
+    st.experimental_rerun()
+
+# Camera input
 img_data = st.camera_input("📷 Capture the weight display")
 
 if img_data:
@@ -26,37 +30,24 @@ if img_data:
 
     if not weight or confidence < 80:
         st.error(f"⚠️ OCR confidence too low ({int(confidence)}%). Please retake the image.")
-        if st.button("🔁 Try Again"):
+        if st.button("🔁 Retry"):
             st.experimental_rerun()
         st.stop()
 
-    # ✅ Prepare data for preview and URL
+    st.success(f"✅ Detected Weight: {weight} g (Confidence: {int(confidence)}%)")
+
+    # Send to Salesforce
     device_id = "BAL-001"
-    image_url = ""  # optional - add later if you host the image
-    captured_at = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    status = "Pending Submission"
-
-    # ✅ Show full preview
-    st.markdown("### 📋 Record Preview (before sending to Salesforce)")
-    st.write({
-        "Auto Number": "Will be generated in Salesforce",
-        "Captured Weight": f"{weight} g",
-        "Captured At": captured_at,
-        "Snapshot Image": "Attached" if image_url else "Not available",
-        "Device ID": device_id,
-        "Status": status
-    })
-
-    # ✅ Encode for Salesforce
+    image_url = ""  # Optional, can be a hosted link if added
+
     encoded_weight = urllib.parse.quote(str(weight))
     encoded_device = urllib.parse.quote(device_id)
     encoded_image_url = urllib.parse.quote(image_url)
 
     salesforce_url = (
-        f"https://autoweightlogger-dev-ed.my.salesforce-sites.com/"
+        "https://autoweightlogger-dev-ed.my.salesforce-sites.com/"
         f"weight_logger_page?WeightInput={encoded_weight}&DeviceID={encoded_device}&ImageURL={encoded_image_url}"
     )
 
-    # ✅ Show Salesforce link
     st.markdown("### 📤 Send to Salesforce")
     st.markdown(f"[✅ Click here to confirm and log in Salesforce]({salesforce_url})", unsafe_allow_html=True)