Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 from simple_salesforce import Salesforce
 
 # Salesforce connection details (replace these with your actual credentials)
-sf = Salesforce(username='prashanth@safetyanaluzer.com', password='SaiPrash461', security_token='MOA3BXBfGyqnjBneog8a9IcGw')
+sf = Salesforce(username='your_username', password='your_password', security_token='your_security_token')
 
 try:
     from ultralytics import YOLO
@@ -101,9 +101,9 @@ def process_video(video_path, frame_skip=5):
 
         video.release()
         score = calculate_safety_score(violations)
-        
+
         # Send data to Salesforce
-        send_to_salesforce(violations, score)
+        send_to_salesforce(violations, score, video_path)
 
         return violations, score
 
@@ -126,7 +126,7 @@ 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):
@@ -138,9 +138,9 @@ def send_to_salesforce(violations, score, video_path):
     status = "Reviewed"  # Update this based on the processing state
 
     # 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_url = "http://path_to_video"  # Replace with the actual URL or path if available
 
-   violations_details = "\n".join([f"Frame {v['frame']}: {v['violation']} (Confidence: {v['confidence']})" for v in violations])
+    violations_details = "\n".join([f"Frame {v['frame']}: {v['violation']} (Confidence: {v['confidence']})" for v in violations])
 
     data = {
         'Site__c': site_name,  # Replace with dynamic data