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")
 
-        # Allow valid results even if debug message is included
+        # Check for valid weight
         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 value
+        # Extract numeric part 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)
@@ -38,20 +38,20 @@ def process_image(image):
         numeric_value = float(numeric_match.group())
         unit = "kg" if "kg" in weight.lower() else "g"
 
-        # Save image temporarily
+        # Save image
         image_path = "snapshot.jpg"
         image.save(image_path)
 
-        # Create Weight_Log__c record
+        # Create Salesforce record
         record = sf.Weight_Log__c.create({
             "Captured_Weight__c": numeric_value,
-            "Captured_Unit__c": unit,  # Optional: create this field in Salesforce
+            "Captured_Unit__c": unit,  # ✅ make sure this exists in Salesforce
             "Captured_At__c": datetime.now(ist).isoformat(),
             "Device_ID__c": "DEVICE-001",
             "Status__c": "Confirmed"
         })
 
-        # Upload image as ContentVersion
+        # Upload image to Salesforce
         with open(image_path, "rb") as f:
             encoded_image = base64.b64encode(f.read()).decode("utf-8")
 
@@ -79,7 +79,7 @@ def process_image(image):
 with gr.Blocks(css=".gr-button {background-color: #2e7d32 !important; color: white !important;}") as demo:
     gr.Markdown("""
     <h1 style='text-align: center; color: #2e7d32;'>📷 Auto Weight Logger</h1>
-    <p style='text-align: center;'>Upload or capture a digital weight image. Detects weight using AI OCR.</p>
+    <p style='text-align: center;'>Upload or capture a digital weight image. Detects weight using AI OCR and logs it into Salesforce.</p>
     <hr style='border: 1px solid #ddd;'/>
     """)