Update app.py

app.py

@@ -95,19 +95,22 @@ def save_dpr_to_pdf(dpr_text, filename):
         return f"Error saving PDF: {str(e)}", None
 
 # Function to upload a file to Salesforce as ContentVersion
-def upload_file_to_salesforce(file_path, filename, sf_connection):
+def upload_file_to_salesforce(file_path, filename, sf_connection, file_type):
     try:
         # Read file content and encode in base64
         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 file_type == "pdf" else "Site Image"
+        
         # Create ContentVersion
         content_version = sf_connection.ContentVersion.create({
             'Title': filename,
             'PathOnClient': filename,
             'VersionData': file_content_b64,
-            'Description': f'Uploaded for Daily Progress Report'
+            'Description': description
         })
         
         # Get ContentDocumentId
@@ -165,6 +168,7 @@ def generate_dpr(files):
             # Create Daily_Progress_Reports__c record
             report_description = "; ".join(captions)[:255]  # Concatenate captions, limit to 255 chars
             dpr_record = sf.Daily_Progress_Reports__c.create({
+
                 'Report_Description__c': report_description
             })
             dpr_record_id = dpr_record['id']
@@ -172,7 +176,7 @@ def generate_dpr(files):
             
             # Upload PDF to Salesforce
             pdf_content_document_id, pdf_upload_result = upload_file_to_salesforce(
-                pdf_filepath, pdf_filename, sf
+                pdf_filepath, pdf_filename, sf, "pdf"
             )
             salesforce_result += pdf_upload_result + "\n"
             
@@ -188,7 +192,7 @@ def generate_dpr(files):
             for file in files:
                 image_filename = os.path.basename(file.name)
                 image_content_document_id, image_upload_result = upload_file_to_salesforce(
-                    file.name, image_filename, sf
+                    file.name, image_filename, sf, "image"
                 )
                 if image_content_document_id:
                     image_content_document_ids.append(image_content_document_id)