Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,18 +1,17 @@
 import streamlit as st
 import cv2
 import tempfile
-import base64
 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 weight display")
+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")
+ 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")