Update app.py

app.py

@@ -99,10 +99,13 @@ def save_record_to_salesforce(annotated_image_url, coverage_percent, original_im
         domain=os.environ.get('SF_DOMAIN', 'login')  # 'test' for sandbox
     )
     
-    # Encode original image to base64 string for storage
-    buffered = io.BytesIO()
-    original_image_pil.save(buffered, format="JPEG")
-    original_img_b64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
+    # Save original image temporarily, upload it, get URL
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_orig_img_file:
+        original_image_pil.save(temp_orig_img_file.name, format="JPEG")
+        temp_orig_img_path = temp_orig_img_file.name
+    
+    original_image_url = upload_image_and_get_url(temp_orig_img_path)
+    os.unlink(temp_orig_img_path)
 
     compliance_status = 'Pass' if coverage_percent >= compliance_threshold else 'Fail'
     technician_id = os.environ.get('SF_TECHNICIAN_ID')  # Salesforce UserId lookup
@@ -113,7 +116,7 @@ def save_record_to_salesforce(annotated_image_url, coverage_percent, original_im
         'Name': record_name,
         'Annotated_Image__c': annotated_image_url,
         'Coverage_Percentage__c': round(coverage_percent, 2),
-        'Original_Image__c': original_img_b64,
+        'Original_Image_URL__c': original_image_url,  # Changed to URL field instead of base64 text
         'Compliance_Status__c': compliance_status,
         'Technician_ID__c': technician_id,
         'Verified_On__c': datetime.utcnow().isoformat()