Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,6 @@
 import streamlit as st
-import cv2
-import tempfile
-from ocr_engine import extract_weight
+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")
@@ -9,9 +8,8 @@ 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: " + str(weight) + " g")
+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")