File size: 1,088 Bytes
0590b95
d55b56b
 
 
0590b95
d55b56b
91bcc4e
 
 
 
 
 
 
 
d55b56b
 
 
91bcc4e
136c114
91bcc4e
d55b56b
f32159a
d55b56b
91bcc4e
 
 
f32159a
 
0590b95
d55b56b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import gradio as gr
from PIL import Image
from datetime import datetime
from ocr_engine import extract_weight

def process_image(image):
    try:
        if image is None:
            return "No image provided", "No timestamp", None
        weight = extract_weight(image)
        now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        return f"{weight} grams", now, image
    except Exception as e:
        return f"Error: {str(e)}", "Error", None

with gr.Blocks() as demo:
    gr.Markdown("## ๐Ÿ“ท Auto Weight Logger โ€” OCR-Based Smart Scale Reader")
    gr.Markdown("Upload a photo of a digital scale. The app will detect and log the weight.")

    image_input = gr.Image(type="pil", label="๐Ÿ“ Upload or Capture Digital Scale Image")

    detect_btn = gr.Button("๐Ÿš€ Detect Weight")

    weight_out = gr.Textbox(label="๐Ÿ“ฆ Detected Weight")
    time_out = gr.Textbox(label="โฑ๏ธ Captured At (IST)")
    snapshot = gr.Image(label="๐Ÿ–ผ๏ธ Snapshot Preview")

    detect_btn.click(fn=process_image, inputs=image_input, outputs=[weight_out, time_out, snapshot])

demo.launch()