Update app.py

app.py

@@ -111,16 +111,20 @@ def save_dpr_to_pdf(dpr_text, image_paths, captions, filename):
     except Exception as e:
         return f"Error saving PDF: {str(e)}", None
 
+# Function to upload a file to Salesforce as ContentVersion and get its download URL
 # Function to upload a file to Salesforce as ContentVersion and get its download URL
 def upload_pdf_to_salesforce(pdf_file, project_title, record_id=None):
     try:
-        sf = get_salesforce_connection()
+        sf = get_salesforce_connection()  # Get Salesforce connection
         if not sf:
             print("Salesforce connection failed. Cannot upload PDF.")
             return None, None
 
+        # Encoding the file in base64
         encoded_pdf_data = base64.b64encode(pdf_file.getvalue()).decode('utf-8')
         print(f"Uploading PDF for project: {project_title}, record ID: {record_id}")
+        
+        # Create ContentVersion data to upload to Salesforce
         content_version_data = {
             "Title": f"{project_title} - Gantt Chart PDF",
             "PathOnClient": f"{project_title}_Gantt_Chart.pdf",
@@ -130,10 +134,12 @@ def upload_pdf_to_salesforce(pdf_file, project_title, record_id=None):
         if record_id:
             content_version_data["FirstPublishLocationId"] = record_id
 
+        # Uploading the PDF to Salesforce
         content_version = sf.ContentVersion.create(content_version_data)
         content_version_id = content_version["id"]
         print(f"PDF uploaded to Salesforce with ContentVersion ID: {content_version_id}")
 
+        # Querying the ContentDocumentId to get the download URL
         result = sf.query(f"SELECT Id, ContentDocumentId FROM ContentVersion WHERE Id = '{content_version_id}'")
         if not result['records']:
             print("No records returned for ContentVersion query")
@@ -147,7 +153,7 @@ def upload_pdf_to_salesforce(pdf_file, project_title, record_id=None):
         print(f"Error uploading PDF to Salesforce: {str(e)}", exc_info=True)
         return None, None
 
-# Function to upload files and generate the daily progress report (DPR), save as PDF, and upload to Salesforce
+# Correct the call to upload_pdf_to_salesforce() in generate_dpr
 def generate_dpr(files):
     dpr_text = []
     captions = []
@@ -199,9 +205,9 @@ def generate_dpr(files):
             dpr_record_id = dpr_record['id']
             salesforce_result += f"Created Daily_Progress_Reports__c record with ID: {dpr_record_id}\n"
             
-            # Upload PDF to Salesforce
+            # Corrected call to upload_pdf_to_salesforce
             pdf_content_document_id, pdf_url, pdf_upload_result = upload_pdf_to_salesforce(
-                pdf_filepath, pdf_filename, sf, "pdf"
+                pdf_filepath, project_title=pdf_filename  # Pass project title as the title
             )
             salesforce_result += pdf_upload_result + "\n"