Update app.py

app.py

@@ -1,37 +1,58 @@
-import gradio as gr
+import streamlit as st
+from PIL import Image, UnidentifiedImageError
+import io
+from datetime import datetime
+import pytz
 from ocr_engine import extract_weight_from_image
-from utils import get_ist_time
 
-def process_image(image):
+st.set_page_config(page_title="⚖️ Auto Weight Logger", layout="centered")
+st.title("⚖️ Auto Weight Logger")
+
+if "camera_key" not in st.session_state:
+    st.session_state["camera_key"] = 0
+
+# UI: Upload or Capture Image
+col1, col2 = st.columns(2)
+with col1:
+    uploaded_file = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])
+with col2:
+    if st.button("📸 Retake Photo"):
+        st.session_state["camera_key"] += 1
+    camera_image = st.camera_input("Capture Image", key=st.session_state["camera_key"])
+
+# Load image
+image = None
+if uploaded_file:
     try:
-        if image is None:
-            return "No image provided", "-", None, "-"
-        
+        image = Image.open(uploaded_file)
+    except UnidentifiedImageError:
+        st.error("Invalid image format.")
+elif camera_image:
+    try:
+        image = Image.open(camera_image)
+    except UnidentifiedImageError:
+        st.error("Invalid camera image.")
+
+# Process image
+if image:
+    with st.spinner("Detecting weight..."):
         weight, confidence = extract_weight_from_image(image)
-        timestamp = get_ist_time()
-
-        if not weight:
-            return "No weight detected", timestamp, image, "-"
-
-        sf_url = f"https://your-salesforce-site.com/log-weight?weight={weight}&time={timestamp}"
-        return f"{weight} g (Confidence: {confidence}%)", timestamp, image, sf_url
-
-    except Exception as e:
-        return f"Error: {str(e)}", "-", None, "-"
-
-iface = gr.Interface(
-    fn=process_image,
-    inputs=gr.Image(type="pil", label="Upload or Capture Snapshot"),
-    outputs=[
-        gr.Textbox(label="Detected Weight"),
-        gr.Textbox(label="Captured At (IST)"),
-        gr.Image(label="Snapshot Image"),
-        gr.Textbox(label="Salesforce Log URL"),
-    ],
-    title="⚖️ Auto Weight Logger",
-    description="Upload or capture a digital scale image to extract the weight.",
-    live=True,
-)
-
-if __name__ == "__main__":
-    iface.launch()
+
+    # Timestamp
+    ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S")
+
+    # Output
+    st.markdown("### 📅 Captured At (IST)")
+    st.info(ist_time)
+
+    st.markdown("### 🖼️ Snapshot")
+    st.image(image, width=400)
+
+    st.markdown("### ⚖️ Detected Weight")
+    if confidence > 0:
+        st.success(f"Detected Weight: **{weight}**  \nConfidence: `{confidence:.2f}`")
+    else:
+        st.error("No weight detected. Try with a clearer image.")
+
+    # (Optional) Send to Salesforce button
+    st.markdown("🔗 [Send to Salesforce](#)", unsafe_allow_html=True)