import streamlit as st
import cv2
import tempfile
import base64
from ocr_engine import extract_weight

st.title("Auto Weight Logger")
img_data = st.camera_input("Capture weight display")

if img_data:
    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as f:
        f.write(img_data.getvalue())
        f.flush()
        weight = extract_weight(f.name)
        st.image(f.name, caption="Snapshot")
        st.success(f"Detected Weight: {weight} g")