Update app.py

app.py

@@ -5,9 +5,8 @@ import pytz
 from ocr_engine import extract_weight
 from simple_salesforce import Salesforce
 import base64
-import os
 
-# Salesforce credentials (for safety, you can later store these in environment variables)
+# Salesforce credentials
 SF_USERNAME = "[email protected]"
 SF_PASSWORD = "autoweight@32"
 SF_TOKEN = "UgiHKWT0aoZRX9gvTYDjAiRY"
@@ -22,21 +21,21 @@ def process_image(image):
         weight = extract_weight(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:
             return "❌ Unable to detect. Try again with a clearer image.", "", image, gr.update(visible=True)
 
-        # Save snapshot image to base64 (optional field, depends on usage)
-        buffered_image = image.copy()
-        buffered_image.save("snapshot.jpg")
+        # 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')
 
-        # Create record in Salesforce
+        # Insert record into Salesforce
         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",   # Static or dynamic based on your design
+            "Device_ID__c": "DEVICE-001",
             "Status__c": "Captured"
         })