Update app.py

app.py

@@ -126,43 +126,65 @@ def calculate_safety_score(violations):
         base_score -= penalties.get(v["violation"], 0)
     return max(base_score, 0)
 
-# ==========================
-# Send Results to Salesforce
-# ==========================
 def send_to_salesforce(violations, score, video_path):
-    # Assuming you have dynamic values for the Site, Uploaded_By, etc.
-    # These values can be collected from user input or a UI
-    site_name = "Sample Site"  # Replace with dynamic data
-    uploaded_by = "User_1"  # Replace with dynamic data (e.g., user info)
-    upload_date = "2025-05-08T12:00:00Z"  # Replace with dynamic upload date
-    status = "Reviewed"  # Update this based on the processing state
+    # Dynamic values (e.g., from user input)
+    site_name = "Construction Site 1"  # Replace with dynamic data from your app
+    uploaded_by = "JohnDoe"  # Replace with actual user info (e.g., from session or UI)
+    upload_date = "2025-05-08T12:00:00Z"  # Replace with actual upload date
+    status = "Reviewed"  # Update based on video processing result
 
-    # If you have a file or video, you can add it here as well
-    video_file_url = "http://path_to_video"  # Replace with the actual URL or path if available
+    # Video file upload
+    file_id = upload_video_to_salesforce(video_path)  # Upload the video and get the file ID
 
+    # Violations details: Frame number, violation type, confidence level
     violations_details = "\n".join([f"Frame {v['frame']}: {v['violation']} (Confidence: {v['confidence']})" for v in violations])
 
+    # Data to be inserted into Salesforce
     data = {
-        'Site__c': site_name,  # Replace with dynamic data
-        'Uploaded_By__c': uploaded_by,  # Replace with actual user details
-        'Upload_Date__c': upload_date,  # Replace with actual upload date
+        'Site__c': site_name,
+        'Uploaded_By__c': uploaded_by,
+        'Upload_Date__c': upload_date,
         'Status__c': status,
         'Compliance_Score__c': score,
         'Violations_Found__c': len(violations),
         'Violations_Details__c': violations_details,
-        'Video_File__c': video_file_url,  # Include the file URL here if you want to store the file
-        # Optionally add a URL for the PDF report
+        'Video_File__c': file_id,  # This is the file ID from ContentVersion
         'PDF_Report_URL__c': "http://path_to_pdf_report"  # Replace with actual PDF report URL if available
     }
 
     try:
-        # Insert new record in Salesforce
+        # Insert a new record in Salesforce
         result = sf.Safety_Video_Reports__c.create(data)
         print(f"✅ Successfully created new report in Salesforce with ID: {result['id']}")
     except Exception as e:
         print(f"❌ Error creating/updating record in Salesforce: {e}")
 
 # ==========================
+# Function to Upload Video to Salesforce
+# ==========================
+def upload_video_to_salesforce(video_path):
+    # Open the video file and read it as a binary file
+    with open(video_path, 'rb') as file:
+        video_data = file.read()
+
+    # Upload the video as a ContentVersion record in Salesforce
+    content_version_data = {
+        'Title': 'Safety Video',  # Title of the file
+        'PathOnClient': 'safety_video.mp4',  # Path of the file on the client's system
+        'VersionData': video_data,  # Video file content as binary
+        'FirstPublishLocationId': 'your_salesforce_library_id',  # Library ID (if required, or leave blank)
+    }
+
+    try:
+        # Create a new ContentVersion record to upload the video
+        content_version = sf.ContentVersion.create(content_version_data)
+        file_id = content_version['Id']  # This is the ContentDocument ID
+        print(f"✅ Video uploaded to Salesforce with File ID: {file_id}")
+        return file_id
+    except Exception as e:
+        print(f"❌ Error uploading video to Salesforce: {e}")
+        return None
+# ==========================
 # Gradio Interface
 # ==========================
 def gradio_interface(video_file):