Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,7 +5,7 @@ import uuid
 import urllib.parse
 from datetime import datetime
 import pytz
-from ocr_engine import extract_weight_from_image  # Make sure this returns (weight, confidence)
+from ocr_engine import extract_weight_from_image  # Ensure this function returns (weight, confidence)
 
 # Set up page
 st.set_page_config(page_title="⚖️ Auto Weight Logger", layout="centered")
@@ -38,14 +38,14 @@ image = None
 # Webcam input
 if input_mode == "Camera":
     cam_photo = st.camera_input("📷 Capture Weight Display", key=st.session_state.camera_key)
-    if cam_photo:
+    if cam_photo is not None:
         image_bytes = cam_photo.getvalue()
         st.session_state.captured_time = get_current_ist_time()
 
 # File upload
 elif input_mode == "Upload":
     uploaded_file = st.file_uploader("📁 Upload an image (JPG/PNG)", type=["jpg", "jpeg", "png"])
-    if uploaded_file:
+    if uploaded_file is not None:
         image_bytes = uploaded_file.read()
         st.session_state.captured_time = get_current_ist_time()