|
import streamlit as st |
|
from PIL import Image |
|
import io |
|
import uuid |
|
import urllib.parse |
|
from ocr_engine import extract_weight_from_image |
|
|
|
|
|
st.set_page_config(page_title="βοΈ Auto Weight Logger", layout="centered") |
|
st.title("βοΈ Auto Weight Logger") |
|
|
|
|
|
if "camera_key" not in st.session_state: |
|
st.session_state.camera_key = str(uuid.uuid4()) |
|
|
|
|
|
input_mode = st.radio("πΈ Select Input Method", ["Camera", "Upload"], horizontal=True) |
|
|
|
|
|
if st.button("π Clear / Retake"): |
|
st.session_state.camera_key = str(uuid.uuid4()) |
|
st.experimental_rerun() |
|
|
|
|
|
image_bytes = None |
|
image = None |
|
|
|
|
|
if input_mode == "Camera": |
|
cam_photo = st.camera_input("π· Take a photo of the weight display", key=st.session_state.camera_key) |
|
if cam_photo is not None: |
|
image_bytes = cam_photo.getvalue() |
|
|
|
|
|
elif input_mode == "Upload": |
|
uploaded_file = st.file_uploader("π Upload an image (JPG/PNG)", type=["jpg", "jpeg", "png"]) |
|
if uploaded_file is not None: |
|
image_bytes = uploaded_file.read() |
|
|
|
|
|
if image_bytes: |
|
try: |
|
image = Image.open(io.BytesIO(image_bytes)) |
|
st.image(image, caption="πΈ Preview", use_column_width=True) |
|
|
|
if len(image_bytes) > 5 * 1024 * 1024: |
|
st.error("β Image is too large (>5MB). Please use a smaller image.") |
|
st.stop() |
|
|
|
with st.spinner("π Extracting weight using OCR..."): |
|
weight, confidence = extract_weight_from_image(image) |
|
|
|
st.write(f"π οΈ DEBUG: Weight = {weight}, Confidence = {confidence}") |
|
|
|
if not weight or confidence < 80: |
|
st.error(f"β οΈ Low OCR confidence ({int(confidence)}%). Try again.") |
|
else: |
|
st.success(f"β
Detected Weight: {weight} g (Confidence: {int(confidence)}%)") |
|
|
|
|
|
device_id = "BAL-001" |
|
image_url = "" |
|
|
|
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"[β
Confirm and Log in Salesforce]({salesforce_url})", unsafe_allow_html=True) |
|
|
|
except Exception as e: |
|
st.error("β Error processing the image.") |
|
st.exception(e) |
|
|