Update app.py

app.py

@@ -13,13 +13,6 @@ from dotenv import load_dotenv
 import base64
 import io
 import concurrent.futures
-import logging
-
-
-
-# Configure logging to show detailed messages
-logging.basicConfig(level=logging.DEBUG)
-logger = logging.getLogger(__name__)
 
 # Load environment variables from .env file
 load_dotenv()
@@ -118,60 +111,40 @@ def save_dpr_to_pdf(dpr_text, image_paths, captions, filename):
     except Exception as e:
         return f"Error saving PDF: {str(e)}", None
 
-
-
-import base64
-
-def upload_pdf_to_salesforce(pdf_file, project_title, record_id=None, sf_connection=None):
+# Function to upload a file to Salesforce as ContentVersion
+def upload_file_to_salesforce(file_path, filename, sf_connection, file_type):
     try:
-        if not sf_connection:
-            logger.error("Salesforce connection failed. Cannot upload PDF.")
-            return None, None
-
         # Read file content and encode in base64
-        encoded_pdf_data = base64.b64encode(pdf_file.getvalue()).decode('utf-8')
-        logger.debug(f"Uploading PDF for project: {project_title}, record ID: {record_id}")
-
+        with open(file_path, 'rb') as f:
+            file_content = f.read()
+        file_content_b64 = base64.b64encode(file_content).decode('utf-8')
+        
         # Set description based on file type
-        description = "Daily Progress Report PDF" if pdf_file.name.endswith(".pdf") else "Site Image"
-
-        # Prepare ContentVersion data
-        content_version_data = {
-            "Title": filename,
-            "PathOnClient": filename,
-            "VersionData": encoded_pdf_data,
-            "Description": description
-        }
-
-        # If a record ID is provided, associate the file with a Salesforce record
-        if record_id:
-            content_version_data["FirstPublishLocationId"] = record_id
-
-        # Create ContentVersion in Salesforce
-        content_version = sf_connection.ContentVersion.create(content_version_data)
-        content_version_id = content_version["id"]
-        logger.info(f"PDF uploaded to Salesforce with ContentVersion ID: {content_version_id}")
-
-        # Query to get ContentDocumentId from the ContentVersion record
-        result = sf_connection.query(f"SELECT Id, ContentDocumentId FROM ContentVersion WHERE Id = '{content_version_id}'")
-        if not result['records']:
-            logger.error("No records returned for ContentVersion query")
-            return content_version_id, None
-
-        # Retrieve ContentDocumentId
-        content_document_id = result['records'][0]['ContentDocumentId']
+        description = "Daily Progress Report PDF" if file_type == "pdf" else "Site Image"
+        
+        # Create ContentVersion
+        content_version = sf_connection.ContentVersion.create({
+            'Title': filename,
+            'PathOnClient': filename,
+            'VersionData': file_content_b64,
+            'Description': description
+        })
         
-        # Generate a valid Salesforce URL for the file (Download link)
-        instance_url = sf_connection.base_url.split('/')[2]  # Extract instance URL (e.g., 'na1', 'ap1', etc.)
-        file_url = f"https://{instance_url}.salesforce.com/sfc/servlet.shepherd/version/download/{content_version_id}"
+        # Get ContentDocumentId
+        content_version_id = content_version['id']
+        content_document = sf_connection.query(
+            f"SELECT ContentDocumentId FROM ContentVersion WHERE Id = '{content_version_id}'"
+        )
+        content_document_id = content_document['records'][0]['ContentDocumentId']
+        
+        # Generate a valid Salesforce URL for the ContentDocument
+        content_document_url = f"https://{sf_connection.sf_instance}.salesforce.com/{content_document_id}"
 
-        logger.debug(f"Generated PDF URL: {file_url}")
-        return content_version_id, file_url
-    
+        # Ensure the link is valid
+        return content_document_id, content_document_url, f"File {filename} uploaded successfully"
     except Exception as e:
-        logger.error(f"Error uploading PDF to Salesforce: {str(e)}", exc_info=True)
-        return None, None
-        
+        return None, None, f"Error uploading {filename} to Salesforce: {str(e)}"
+
 # Function to generate the daily progress report (DPR), save as PDF, and upload to Salesforce
 def generate_dpr(files):
     dpr_text = []
@@ -225,7 +198,7 @@ def generate_dpr(files):
             salesforce_result += f"Created Daily_Progress_Reports__c record with ID: {dpr_record_id}\n"
             
             # Upload PDF to Salesforce
-            pdf_content_document_id, pdf_url, pdf_upload_result = upload_pdf_to_salesforce(
+            pdf_content_document_id, pdf_url, pdf_upload_result = upload_file_to_salesforce(
                 pdf_filepath, pdf_filename, sf, "pdf"
             )
             salesforce_result += pdf_upload_result + "\n"