Update app.py

app.py

@@ -156,7 +156,7 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
             "Status__c": status,
             "Summarized_Missing_Items__c": summarized_missing,
             "Alert_Sent__c": alert_sent,
-            "Client_PDF_Pack_URL__c": "PDF attached to this record",  # Placeholder message
+            "Client_PDF_Pack_URL__c": "",  # Will be updated after attachment is created
             "Closure_Pack_URL__c": "",  # Placeholder; update if you have a closure pack URL
             "Escalation_Flag__c": escalation_flag,
             "Evaluated_At__c": evaluated_at,
@@ -172,7 +172,8 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
         record_id = result.get('id')
         logging.info(f"Record ID: {record_id}")
 
-        # Attach the PDF to the record if pdf_path exists
+        # Attach the PDF to the record if pdf_path exists and update the URL
+        pdf_download_url = ""
         if pdf_path and os.path.exists(pdf_path):
             logging.info(f"Attaching PDF to Salesforce record ID: {record_id}")
             with open(pdf_path, "rb") as pdf_file:
@@ -188,17 +189,29 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
             
             attachment_result = sf.Attachment.create(attachment)
             logging.info(f"Successfully attached PDF to record: {attachment_result}")
-            logging.info(f"Attachment ID: {attachment_result.get('id')}")
+            attachment_id = attachment_result.get('id')
+            logging.info(f"Attachment ID: {attachment_id}")
+
+            # Construct the direct download URL for the attachment
+            pdf_download_url = f"{SF_INSTANCE_URL}/servlet/servlet.FileDownload?file={attachment_id}"
+            logging.info(f"Generated PDF download URL: {pdf_download_url}")
+
+            # Update the Project_Closure_Handover__c record with the PDF download URL
+            update_data = {
+                "Client_PDF_Pack_URL__c": pdf_download_url
+            }
+            sf.Project_Closure_Handover__c.update(record_id, update_data)
+            logging.info(f"Updated record {record_id} with Client_PDF_Pack_URL__c: {pdf_download_url}")
         else:
-            logging.warning(f"No PDF file found at {pdf_path}. Skipping attachment.")
+            logging.warning(f"No PDF file found at {pdf_path}. Skipping attachment and URL update.")
 
-        return f"Record created successfully. Record ID: {record_id}. PDF attached to the record."
+        return f"Record created successfully. Record ID: {record_id}. PDF attached and URL set to: {pdf_download_url}"
     except SalesforceError as e:
         logging.error(f"Salesforce error while creating Project_Closure_Handover__c record: {str(e)}")
         logging.error("Possible issues: Object permissions, field-level security, validation rules, or required fields.")
         logging.error("Check the following in your Salesforce org:")
-        logging.error("- Ensure the user has Create permission on Project_Closure_Handover__c.")
-        logging.error("- Ensure the user has permission to create Attachments.")
+        logging.error("- Ensure the user has Create and Edit permission on Project_Closure_Handover__c.")
+        logging.error("- Ensure the user has permission to create and read Attachments.")
         logging.error("- Verify field-level security for all fields in the record.")
         logging.error("- Check for validation rules or required fields that might be failing.")
         return f"Salesforce error: {str(e)}"