Sanjayraju30 commited on
Commit
6b8a810
ยท
verified ยท
1 Parent(s): 385a153

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -5,11 +5,11 @@ 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")
@@ -23,10 +23,7 @@ with gr.Blocks(title="โš–๏ธ Auto Weight Logger") as demo:
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()
 
5
 
6
  def process_image(img):
7
  if img is None:
8
+ return "No image uploaded", None, None
9
 
10
  ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%d-%m-%Y %I:%M:%S %p")
11
+ weight, confidence = extract_weight_from_image(img)
12
+ return f"{weight} kg (Confidence: {confidence}%)", ist_time, img
13
 
14
  with gr.Blocks(title="โš–๏ธ Auto Weight Logger") as demo:
15
  gr.Markdown("## โš–๏ธ Auto Weight Logger")
 
23
  timestamp = gr.Textbox(label="๐Ÿ•’ Captured At (IST)")
24
  snapshot = gr.Image(label="๐Ÿ“ธ Snapshot Image")
25
 
 
 
 
26
  submit = gr.Button("๐Ÿ” Detect Weight")
27
+ submit.click(process_image, inputs=image_input, outputs=[output_weight, timestamp, snapshot])
28
 
29
  demo.launch()