Update app.py

app.py

@@ -26,11 +26,11 @@ def process_image(image):
         ist = pytz.timezone('Asia/Kolkata')
         timestamp = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S IST")
 
-        # Check for valid weight
+        # Handle missing or invalid output
         if not weight or ("No valid" in weight and "OCR:" not in weight):
             return "❌ Unable to detect. Try again with a clearer image.", "", image, gr.update(visible=True)
 
-        # Extract numeric part and unit
+        # Extract number and unit
         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)
@@ -42,16 +42,16 @@ def process_image(image):
         image_path = "snapshot.jpg"
         image.save(image_path)
 
-        # Create Salesforce record
+        # Create Weight_Log__c record
         record = sf.Weight_Log__c.create({
             "Captured_Weight__c": numeric_value,
-            "Captured_Unit__c": unit,  # ✅ make sure this exists in Salesforce
+            "Captured_Unit__c": unit,  # ✅ Must exist in Salesforce
             "Captured_At__c": datetime.now(ist).isoformat(),
             "Device_ID__c": "DEVICE-001",
             "Status__c": "Confirmed"
         })
 
-        # Upload image to Salesforce
+        # Upload image
         with open(image_path, "rb") as f:
             encoded_image = base64.b64encode(f.read()).decode("utf-8")