Update app.py

app.py

@@ -5,33 +5,32 @@ from ocr_engine import extract_weight
 
 def process_image(image):
     if image is None:
-        return "No image received", None, None
-
+        return "No image provided", None, None
     weight = extract_weight(image)
     now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     return f"{weight} grams", now, image
 
 with gr.Blocks() as demo:
     gr.Markdown("## 📷 Auto Weight Logger — OCR-Based Smart Scale Reader")
-    gr.Markdown("Upload a snapshot of a digital weight display or use webcam. The app extracts weight using OCR.")
+    gr.Markdown("Upload a snapshot or use webcam. The app extracts weight using OCR.")
 
     with gr.Row():
-        webcam = gr.Image(source="webcam", type="pil", label="📸 Capture from Webcam")
-        upload = gr.Image(source="upload", type="pil", label="📁 Or Upload Image")
+        camera_input = gr.Camera(label="📸 Capture via Webcam", type="pil")
+        upload_input = gr.Image(label="📁 Upload Image", type="pil")
 
-    image_input = gr.State()
+    selected_image = gr.State()
 
-    def choose_image(w, u):
-        return w if w else u
+    def choose_latest_image(cam_img, upload_img):
+        return cam_img if cam_img is not None else upload_img
 
-    webcam.change(fn=choose_image, inputs=[webcam, upload], outputs=image_input)
-    upload.change(fn=choose_image, inputs=[webcam, upload], outputs=image_input)
+    camera_input.change(fn=choose_latest_image, inputs=[camera_input, upload_input], outputs=selected_image)
+    upload_input.change(fn=choose_latest_image, inputs=[camera_input, upload_input], outputs=selected_image)
 
-    submit_btn = gr.Button("🚀 Detect Weight")
+    btn = gr.Button("🚀 Detect Weight")
     weight_out = gr.Textbox(label="Detected Weight")
     time_out = gr.Textbox(label="Captured At (IST)")
-    snapshot_out = gr.Image(label="Snapshot")
+    preview = gr.Image(label="Snapshot")
 
-    submit_btn.click(fn=process_image, inputs=image_input, outputs=[weight_out, time_out, snapshot_out])
+    btn.click(fn=process_image, inputs=selected_image, outputs=[weight_out, time_out, preview])
 
 demo.launch()