Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,8 +4,6 @@ import cv2
 from paddleocr import PaddleOCR
 from PIL import Image
 import re
-from datetime import datetime
-import pytz
 
 ocr = PaddleOCR(use_angle_cls=True, lang='en')
 
@@ -24,10 +22,7 @@ def extract_weight_text(image):
             return match.group()
     return None
 
-st.set_page_config(page_title="Auto Weight Logger", layout="centered")
 st.title("📦 Auto Weight Logger")
-st.write("Upload or click image of weight display. App will read the weight.")
-
 uploaded_file = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])
 camera_image = st.camera_input("Or Capture Image")
 
@@ -35,16 +30,12 @@ input_image = uploaded_file or camera_image
 
 if input_image:
     image = Image.open(input_image)
-    st.image(image, caption="Original", use_column_width=True)
-
-    processed = preprocess_image(image)
-    st.image(processed, caption="Processed", use_column_width=True)
+    st.image(image, caption="Original Image", use_column_width=True)
 
-    weight = extract_weight_text(processed)
+    processed_image = preprocess_image(image)
+    weight = extract_weight_text(processed_image)
 
     if weight:
-        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("❌ Weight not detected. Try clearer image.")
+        st.error("❌ Weight not detected.")