Update app.py

app.py

@@ -225,16 +225,20 @@ def generate_dpr(files):
                 )
                 
                 if image_content_document_id:
-                    image_content_document_ids.append(image_content_document_id)
-                
-                # Now, link the image to the DPR record in the correct field (e.g., Site Images)
-                # Assuming you have a field like `Site_Images__c` in the DPR record that accepts ContentDocumentId
-                sf.Daily_Progress_Reports__c.update(dpr_record_id, {
-                    'Site_Images__c': image_content_document_id  # Store ContentDocumentId in Site_Images__c field
-                })
-                
-                salesforce_result += image_upload_result + "\n"
-            
+                    # Link image to the Daily Progress Report record (DPR) using ContentDocumentLink
+                    sf.ContentDocumentLink.create({
+                        'ContentDocumentId': image_content_document_id,
+                        'LinkedEntityId': dpr_record_id,  # Link image to DPR record
+                        'ShareType': 'V'  # 'V' means Viewer access
+                    })
+                    
+                    # Now, update the DPR record with the ContentDocumentId in the Site_Images field (if it's a text or URL field)
+                    sf.Daily_Progress_Reports__c.update(dpr_record_id, {
+                        'Site_Images__c': image_content_document_id  # Storing the ContentDocumentId directly
+                    })
+                    
+                    salesforce_result += image_upload_result + "\n"
+                        
         except Exception as e:
             salesforce_result += f"Error interacting with Salesforce: {str(e)}\n"
     else: