Update app.py

app.py

@@ -5,6 +5,7 @@ import pytz
 from ocr_engine import extract_weight
 from simple_salesforce import Salesforce
 import base64
+import os
 
 # Salesforce credentials
 SF_USERNAME = "[email protected]"
@@ -25,20 +26,42 @@ def process_image(image):
         if not weight or "No valid" in weight:
             return "❌ Unable to detect. Try again with a clearer image.", "", image, gr.update(visible=True)
 
-        # Save image as base64 (for Snapshot_Image__c)
-        image.save("snapshot.jpg")
-        with open("snapshot.jpg", "rb") as img_file:
-            snapshot_base64 = base64.b64encode(img_file.read()).decode('utf-8')
+        # Save image temporarily
+        image_path = "snapshot.jpg"
+        image.save(image_path)
 
-        # Insert record into Salesforce
-        sf.Weight_Log__c.create({
+        # Create Weight_Log__c record first
+        record = sf.Weight_Log__c.create({
             "Captured_Weight__c": float(weight.replace("kg", "").strip()),
             "Captured_At__c": datetime.now(ist).isoformat(),
-            "Snapshot_Image__c": snapshot_base64,
             "Device_ID__c": "DEVICE-001",
             "Status__c": "Captured"
         })
 
+        # Upload image as ContentVersion
+        with open(image_path, "rb") as f:
+            encoded_image = base64.b64encode(f.read()).decode("utf-8")
+
+        # Create ContentVersion
+        content = sf.ContentVersion.create({
+            "Title": f"Snapshot_{timestamp}",
+            "PathOnClient": "snapshot.jpg",
+            "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'],
+            "ShareType": "V",
+            "Visibility": "AllUsers"
+        })
+
         return weight, timestamp, image, gr.update(visible=False)
     except Exception as e:
         return f"Error: {str(e)}", "", None, gr.update(visible=True)
@@ -46,7 +69,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;'/>
     """)