Sanjayraju30 commited on
Commit
c4904a4
·
verified ·
1 Parent(s): 34ca6d7

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +32 -0
app.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from datetime import datetime
3
+ import pytz
4
+ from ocr_engine import extract_weight_from_image
5
+
6
+ def process_image(img):
7
+ if img is None:
8
+ return "No image uploaded", None, None, "No OCR output"
9
+
10
+ ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%d-%m-%Y %I:%M:%S %p")
11
+ weight, confidence, raw_text = extract_weight_from_image(img)
12
+ return f"{weight} kg (Confidence: {confidence}%)", ist_time, img, raw_text
13
+
14
+ with gr.Blocks(title="⚖️ Auto Weight Logger") as demo:
15
+ gr.Markdown("## ⚖️ Auto Weight Logger")
16
+ gr.Markdown("📷 **Upload or capture an image of a digital weight scale.** The app will auto-detect the weight in kilograms using OCR.")
17
+
18
+ with gr.Row():
19
+ image_input = gr.Image(type="pil", label="Upload / Capture Image")
20
+ output_weight = gr.Textbox(label="⚖️ Detected Weight (in kg)")
21
+
22
+ with gr.Row():
23
+ timestamp = gr.Textbox(label="🕒 Captured At (IST)")
24
+ snapshot = gr.Image(label="📸 Snapshot Image")
25
+
26
+ with gr.Row():
27
+ debug_output = gr.Textbox(label="🪵 Raw OCR Output")
28
+
29
+ submit = gr.Button("🔍 Detect Weight")
30
+ submit.click(process_image, inputs=image_input, outputs=[output_weight, timestamp, snapshot, debug_output])
31
+
32
+ demo.launch()