Update app.py

app.py

@@ -1,55 +1,40 @@
-import streamlit as st
+import gradio as gr
 from PIL import Image, UnidentifiedImageError
 from datetime import datetime
 import pytz
 from ocr_engine import extract_weight_from_image
 
-# Streamlit page config
-st.set_page_config(page_title="⚖️ Auto Weight Logger", layout="centered")
-st.title("⚖️ Auto Weight Logger")
-
-# Debug: Confirm app starts
-st.write("✅ Streamlit app initialized.")
-
-if "camera_key" not in st.session_state:
-    st.session_state["camera_key"] = 0
-
-# Upload or Capture
-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:
-        image = Image.open(uploaded_file)
-    except UnidentifiedImageError:
-        st.error("Invalid image format.")
-elif camera_image:
+def detect_weight(image: Image.Image):
     try:
-        image = Image.open(camera_image)
-    except UnidentifiedImageError:
-        st.error("Invalid camera image.")
-
-# OCR Process
-if image:
-    st.markdown("### 🕒 Captured At (IST)")
-    ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S")
-    st.info(ist_time)
+        # Get IST timestamp
+        ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S")
 
-    st.markdown("### 🖼️ Snapshot Image")
-    st.image(image, width=400)
-
-    with st.spinner("🔍 Detecting weight..."):
+        # Perform OCR
         weight, confidence = extract_weight_from_image(image)
 
-    st.markdown("### ⚖️ Captured Weight & Confidence")
-    if confidence > 0:
-        st.success(f"Detected Weight: **{weight}**  \nConfidence: `{confidence:.2f}`")
-    else:
-        st.error("No weight detected. Please upload a clearer image.")
+        if confidence > 0:
+            result = f"✅ **Detected Weight:** {weight}\n\n📊 **Confidence:** {confidence:.2f}\n\n🕒 **Captured At (IST):** {ist_time}"
+        else:
+            result = "❌ No weight detected. Please upload a clearer image."
+        return image, result
+    except UnidentifiedImageError:
+        return None, "❌ Invalid image format."
+
+# Gradio UI
+title = "⚖️ Auto Weight Logger"
+description = "Upload or capture an image of a digital weighing scale to automatically extract the weight using OCR."
+
+demo = gr.Interface(
+    fn=detect_weight,
+    inputs=gr.Image(type="pil", label="Upload or Capture Image"),
+    outputs=[
+        gr.Image(type="pil", label="Snapshot"),
+        gr.Markdown(label="Result")
+    ],
+    title=title,
+    description=description,
+    allow_flagging="never"
+)
+
+if __name__ == "__main__":
+    demo.launch()