Update app.py

app.py

@@ -120,9 +120,12 @@ def upload_file_to_salesforce(file_path, filename, sf_connection, file_type):
         )
         content_document_id = content_document['records'][0]['ContentDocumentId']
         
-        return content_document_id, f"File {filename} uploaded successfully"
+        # Get the URL of the ContentDocument (PDF file)
+        content_document_url = f"https://{sf_connection.sf_instance}.salesforce.com/{content_document_id}"
+        
+        return content_document_id, content_document_url, f"File {filename} uploaded successfully"
     except Exception as e:
-        return None, f"Error uploading {filename} to Salesforce: {str(e)}"
+        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):
@@ -161,6 +164,7 @@ def generate_dpr(files):
     # Salesforce upload
     salesforce_result = ""
     pdf_content_document_id = None
+    pdf_url = None
     image_content_document_ids = []
     
     if sf and pdf_filepath:
@@ -168,14 +172,13 @@ 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']
             salesforce_result += f"Created Daily_Progress_Reports__c record with ID: {dpr_record_id}\n"
             
             # Upload PDF to Salesforce
-            pdf_content_document_id, pdf_upload_result = upload_file_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"
@@ -188,6 +191,13 @@ def generate_dpr(files):
                     'ShareType': 'V'
                 })
             
+            # Update the DPR record with the PDF URL
+            if pdf_url:
+                sf.Daily_Progress_Reports__c.update(dpr_record_id, {
+                    'PDF_URL__c': pdf_url
+                })
+                salesforce_result += f"Updated PDF URL for record ID {dpr_record_id}\n"
+            
             # Upload images to Salesforce
             for file in files:
                 image_filename = os.path.basename(file.name)
@@ -232,4 +242,3 @@ iface = gr.Interface(
 
 if __name__ == "__main__":
     iface.launch()
-