Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import gradio as gr
 from PIL import Image
 from datetime import datetime
@@ -17,6 +16,13 @@ SF_TOKEN = "UgiHKWT0aoZRX9gvTYDjAiRY"
 # Connect to Salesforce
 sf = Salesforce(username=SF_USERNAME, password=SF_PASSWORD, security_token=SF_TOKEN)
 
+def restore_decimal(text):
+    if re.fullmatch(r"\d{5}", text):
+        return f"{text[:2]}.{text[2:]}"
+    elif re.fullmatch(r"\d{4}", text):
+        return f"{text[:2]}.{text[2:]}"
+    return text
+
 def process_image(image):
     if image is None:
         return "❌ No image provided", "", None, gr.update(visible=True)
@@ -30,12 +36,19 @@ def process_image(image):
         if not weight or weight.startswith("Error"):
             return f"❌ OCR Error: {weight}", "", image, gr.update(visible=True)
 
-        numeric_match = re.search(r'\d{1,5}(\.\d{1,3})?', weight)
-        if not numeric_match:
-            return f"❌ Could not extract number | OCR: {weight}", "", image, gr.update(visible=True)
-
-        numeric_value = float(numeric_match.group())
-        unit = "kg" if "kg" in weight.lower() else "g"
+        match = re.search(r'(\d{1,3}\.\d{1,3})\s*(kg|g)?', weight)
+        if match:
+            numeric_value = float(match.group(1))
+            unit = match.group(2) if match.group(2) else "g"
+        else:
+            # fallback try to fix raw number like 53255 → 52.255
+            cleaned = re.sub(r"[^\d]", "", weight)
+            decimal_fixed = restore_decimal(cleaned)
+            try:
+                numeric_value = float(decimal_fixed)
+                unit = "g"
+            except:
+                return f"❌ Could not extract number | OCR: {weight}", "", image, gr.update(visible=True)
 
         image_path = "snapshot.jpg"
         image.save(image_path)
@@ -68,7 +81,7 @@ def process_image(image):
             "Visibility": "AllUsers"
         })
 
-        return weight, timestamp, image, gr.update(visible=False)
+        return f"{numeric_value} {unit}", timestamp, image, gr.update(visible=False)
     except Exception as e:
         return f"Error: {str(e)}", "", None, gr.update(visible=True)