Update app.py

app.py

@@ -9,6 +9,7 @@ from huggingface_hub import InferenceClient
 from retry import retry
 import time
 from datetime import datetime
+import base64
 
 # Print statement to confirm script initialization
 print("Starting Project Closure Readiness Evaluator app...")
@@ -147,21 +148,7 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
         # Explicitly convert escalated to lowercase boolean string for Escalation_Flag__c
         escalation_flag = str(bool(escalated)).lower()  # Ensure this is also a proper boolean string
 
-        # Handle the PDF URL for Client_PDF_Pack_URL__c
-        pdf_url = ""
-        if pdf_path:
-            # In Hugging Face Spaces, the pdf_path is a local filepath (e.g., readiness_report.pdf)
-            # Gradio may serve this file, but we don't have a direct public URL
-            # For demonstration, we'll construct a placeholder URL assuming the file is hosted
-            # In a real production environment, you'd need to upload the PDF to a public location (e.g., S3)
-            logging.info(f"PDF path provided: {pdf_path}")
-            # Simulate a URL for Hugging Face Spaces (this is a placeholder and may not work)
-            # Replace 'your-huggingface-space-url' with the actual Space URL if known
-            pdf_url = f"https://your-huggingface-space-url.hf.space/file={pdf_path}"
-            logging.warning("Using a placeholder URL for Client_PDF_Pack_URL__c. In a production environment, upload the PDF to a publicly accessible location (e.g., AWS S3) and use that URL.")
-            logging.warning(f"Generated placeholder URL: {pdf_url}")
-
-        # Create a record in the custom object Project_Closure_Handover__c
+        # Create the record in Project_Closure_Handover__c without the PDF URL for now
         record = {
             "Readiness_Score__c": score,
             "Checklist_Summary__c": checklist_summary,
@@ -169,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_url,  # Use the placeholder URL
+            "Client_PDF_Pack_URL__c": "PDF attached to this record",  # Placeholder message
             "Closure_Pack_URL__c": "",  # Placeholder; update if you have a closure pack URL
             "Escalation_Flag__c": escalation_flag,
             "Evaluated_At__c": evaluated_at,
@@ -182,13 +169,36 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
         logging.debug(f"Attempting to create Salesforce record in Project_Closure_Handover__c with data: {record}")
         result = sf.Project_Closure_Handover__c.create(record)
         logging.info(f"Successfully created Salesforce record: {result}")
-        logging.info(f"Record ID: {result.get('id')}")
-        return f"Record created successfully. Record ID: {result.get('id')}"
+        record_id = result.get('id')
+        logging.info(f"Record ID: {record_id}")
+
+        # Attach the PDF to the record if pdf_path exists
+        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:
+                pdf_content = pdf_file.read()
+                pdf_base64 = base64.b64encode(pdf_content).decode('utf-8')
+            
+            attachment = {
+                "ParentId": record_id,
+                "Name": "Readiness_Report.pdf",
+                "Body": pdf_base64,
+                "ContentType": "application/pdf"
+            }
+            
+            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')}")
+        else:
+            logging.warning(f"No PDF file found at {pdf_path}. Skipping attachment.")
+
+        return f"Record created successfully. Record ID: {record_id}. PDF attached to the record."
     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("- 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)}"