File size: 2,709 Bytes
fb1a823 c41b38b ae762c2 82ae7fe fb1a823 ae762c2 d54c470 82ae7fe fec538c c4b9b34 82ae7fe ae762c2 c4b9b34 be301b7 c4b9b34 be301b7 c4b9b34 e9c0e22 c4b9b34 be301b7 fec538c ae762c2 c4b9b34 e9c0e22 c4b9b34 82ae7fe ae762c2 c4b9b34 82ae7fe c4b9b34 ae762c2 c4b9b34 82ae7fe ae762c2 82ae7fe c4b9b34 ae762c2 82ae7fe ae762c2 82ae7fe ae762c2 c4b9b34 82ae7fe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import streamlit as st
from PIL import Image
from ocr_engine import extract_weight_from_image
import urllib.parse
import uuid
st.set_page_config(page_title="⚖️ Auto Weight Logger", layout="centered")
st.title("⚖️ Auto Weight Logger")
# Session state
if "image_data" not in st.session_state:
st.session_state.image_data = None
if "input_mode" not in st.session_state:
st.session_state.input_mode = "Camera"
if "camera_key" not in st.session_state:
st.session_state.camera_key = str(uuid.uuid4())
# Choose input mode
st.radio("📸 Select Image Input Method:", ["Camera", "Upload"], key="input_mode", horizontal=True)
# Reset image
if st.button("🔁 Clear / Retake Photo"):
st.session_state.image_data = None
st.session_state.camera_key = str(uuid.uuid4()) # Reload webcam
# Input section
if st.session_state.image_data is None:
if st.session_state.input_mode == "Camera":
img_data = st.camera_input("📷 Capture the weight display", key=st.session_state.camera_key)
else:
img_data = st.file_uploader("📁 Upload an image of the weight display", type=["jpg", "jpeg", "png"])
if img_data:
st.session_state.image_data = img_data
# OCR and Display
if st.session_state.image_data:
st.success("✅ Image received successfully!")
image = Image.open(st.session_state.image_data)
st.image(image, caption="📸 Snapshot", use_column_width=True)
# Check image size
if len(st.session_state.image_data.getvalue()) > 5 * 1024 * 1024:
st.error("❌ Image too large (>5MB). Please try a smaller file.")
st.stop()
with st.spinner("🔍 Extracting weight..."):
weight, confidence = extract_weight_from_image(image)
st.write(f"🛠️ DEBUG: weight = {weight}, confidence = {confidence}")
if not weight or confidence < 80:
st.error(f"⚠️ OCR confidence too low ({int(confidence)}%). Please try again.")
else:
st.success(f"✅ Detected Weight: {weight} g (Confidence: {int(confidence)}%)")
device_id = "BAL-001"
image_url = "" # Placeholder for now
salesforce_url = (
"https://autoweightlogger-dev-ed.my.salesforce-sites.com/"
f"weight_logger_page?WeightInput={urllib.parse.quote(str(weight))}"
f"&DeviceID={urllib.parse.quote(device_id)}&ImageURL={urllib.parse.quote(image_url)}"
)
st.markdown("### 📤 Send to Salesforce")
st.markdown(f"[✅ Click here to confirm and log in Salesforce]({salesforce_url})", unsafe_allow_html=True)
if st.button("🔁 Retake / Upload Another"):
st.session_state.image_data = None
st.session_state.camera_key = str(uuid.uuid4())
|