Update app.py

app.py

@@ -156,12 +156,12 @@ def save_to_salesforce(input_data, prediction, pdf_buffer):
         # Get the ContentVersion ID
         content_version_id = cv_result["id"]
 
-        # Construct the internal Salesforce URL for the ContentVersion
+        # Construct the correct Salesforce URL for the ContentVersion
         pdf_url = f"{os.environ.get('SF_INSTANCE_URL')}/sfc/servlet.shepherd/version/download/{content_version_id}"
 
         # Update the Delay_Predictor__c record with the PDF URL
         update_result = sf.Delay_Predictor__c.update(record_id, {"PDF_Report__c": pdf_url})
-        if not update_result == 204:  # 204 indicates success for updates
+        if update_result != 204:  # 204 indicates success for updates
             return f"Failed to update PDF_Report__c field: {update_result}"
 
         return None