src/streamlit_app.py

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

st.set_page_config(page_title="Auto Weight Logger", layout="centered")
st.title("⚖️ Auto Weight Logger")

img_data = st.camera_input("📷 Capture the 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")