Update app.py

app.py

@@ -19,7 +19,9 @@ def process_image(image):
     if image is None:
         return "❌ No image provided", "", None, gr.update(visible=True)
     try:
+        # Extract weight using OCR (your updated function should handle any clear image)
         weight = extract_weight(image)
+
         ist = pytz.timezone('Asia/Kolkata')
         timestamp = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S IST")
 
@@ -30,12 +32,12 @@ def process_image(image):
         image_path = "snapshot.jpg"
         image.save(image_path)
 
-        # Create Weight_Log__c record first
+        # Create Weight_Log__c record
         record = sf.Weight_Log__c.create({
             "Captured_Weight__c": float(weight.replace("kg", "").strip()),
             "Captured_At__c": datetime.now(ist).isoformat(),
             "Device_ID__c": "DEVICE-001",
-            "Status__c": "Confirmed"  # ✅ Must match picklist values
+            "Status__c": "Confirmed"
         })
 
         # Upload image as ContentVersion
@@ -48,12 +50,10 @@ def process_image(image):
             "VersionData": encoded_image
         })
 
-        # Get ContentDocumentId
         content_id = sf.query(
             f"SELECT ContentDocumentId FROM ContentVersion WHERE Id = '{content['id']}'"
         )['records'][0]['ContentDocumentId']
 
-        # Link file to Weight_Log__c record via ContentDocumentLink
         sf.ContentDocumentLink.create({
             "ContentDocumentId": content_id,
             "LinkedEntityId": record['id'],
@@ -65,10 +65,11 @@ def process_image(image):
     except Exception as e:
         return f"Error: {str(e)}", "", None, gr.update(visible=True)
 
+# Updated header: Removed "and logs it into Salesforce"
 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 and logs it into Salesforce.</p>
+    <p style='text-align: center;'>Upload or capture a digital weight image. Detects weight using AI OCR.</p>
     <hr style='border: 1px solid #ddd;'/>
     """)