Update app.py

app.py

@@ -27,12 +27,12 @@ def process_image(image):
         ist = pytz.timezone('Asia/Kolkata')
         timestamp = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S IST")
 
-        if not weight or ("No valid" in weight and "OCR:" in weight):
-            return f"❌ OCR Failed: {weight}", "", image, gr.update(visible=True)
+        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 "❌ Could not extract number", "", image, gr.update(visible=True)
+            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"