Spaces:
Running
Running
from PIL import Image | |
import pytesseract | |
import re | |
def extract_weight(img_path): | |
img = Image.open(img_path).convert("L") # Grayscale | |
# OCR | |
text = pytesseract.image_to_string(img, config='--psm 6') | |
text = text.lower().replace('\n', ' ').strip() | |
# Find weight + unit (e.g., 52.25 g, 75.8 kg) | |
match = re.search(r'(\d+\.\d+|\d+)\s*(kg|g)', text) | |
if match: | |
number = match.group(1) | |
unit = match.group(2) | |
return f"{number} {unit}" | |
else: | |
return "Weight not detected" | |