import streamlit as st
from PIL import Image
from ocr_engine import extract_weight_from_image

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:
    image = Image.open(img_data)
    st.image(image, caption="📸 Snapshot", use_column_width=True)

    with st.spinner("Extracting weight..."):
        weight = extract_weight_from_image(image)

    st.success(f"✅ Detected Weight: {weight} g")