Update src/streamlit_app.py

src/streamlit_app.py

@@ -25,26 +25,26 @@ def extract_weight_text(image):
     return None
 
 st.set_page_config(page_title="Auto Weight Logger", layout="centered")
-st.title("📦 Auto Weight Logger (Streamlit)")
-st.write("Upload or capture an image of a digital weight display to extract weight.")
+st.title("📦 Auto Weight Logger")
+st.write("Upload or click image of weight display. App will read the weight.")
 
-uploaded_img = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-camera_img = st.camera_input("Or click an image")
+uploaded_file = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])
+camera_image = st.camera_input("Or Capture Image")
 
-input_img = uploaded_img or camera_img
+input_image = uploaded_file or camera_image
 
-if input_img is not None:
-    image = Image.open(input_img)
-    st.image(image, caption="Original Image", use_column_width=True)
+if input_image:
+    image = Image.open(input_image)
+    st.image(image, caption="Original", use_column_width=True)
 
-    pre_img = preprocess_image(image)
-    st.image(pre_img, caption="Preprocessed Image", use_column_width=True)
+    processed = preprocess_image(image)
+    st.image(processed, caption="Processed", use_column_width=True)
 
-    weight = extract_weight_text(pre_img)
+    weight = extract_weight_text(processed)
 
     if weight:
-        ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S")
-        st.success(f"✅ Weight Detected: **{weight} kg**")
-        st.info(f"🕒 Captured At (IST): {ist_time}")
+        time_now = datetime.now(pytz.timezone('Asia/Kolkata')).strftime('%Y-%m-%d %H:%M:%S')
+        st.success(f"✅ Weight Detected: {weight} kg")
+        st.info(f"⏱️ Captured At: {time_now}")
     else:
-        st.error("❌ Could not detect weight. Please try with a clearer image.")
+        st.error("❌ Weight not detected. Try clearer image.")