Update app.py

app.py

@@ -6,8 +6,14 @@ import numpy as np
 from simple_salesforce import Salesforce
 import time
 
-# Salesforce connection details (replace these with your actual credentials)
-sf = Salesforce(username='prashanth@safetyanaluzer.com', password='SaiPrash461', security_token='MOA3BXBfGyqnjBneog8a9IcGw')
+# Test Salesforce connection
+try:
+    sf = Salesforce(username='prashanth@safetyanaluzer.com', password='SaiPrash461', security_token='MOA3BXBfGyqnjBneog8a9IcGw')
+    # Simple query to verify the connection
+    result = sf.query("SELECT Id FROM Safety_Video_Reports__c LIMIT 1")
+    print("✅ Salesforce connection successful.")
+except Exception as e:
+    print(f"❌ Error connecting to Salesforce: {e}")
 
 try:
     from ultralytics import YOLO
@@ -144,8 +150,10 @@ def send_to_salesforce(violations, score, video_path):
     upload_date = "2025-05-08T12:00:00Z"  # Replace with actual upload date
     status = "Reviewed"  # Update based on video processing result
 
+    print("Starting file upload to Salesforce...")
     # Video file upload
     file_id = upload_video_to_salesforce(video_path)  # Upload the video and get the file ID
+    print(f"File uploaded to Salesforce with ID: {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])
@@ -164,7 +172,7 @@ def send_to_salesforce(violations, score, video_path):
     }
 
     try:
-        # Insert a new record in Salesforce
+        print(f"Creating Salesforce record with data: {data}")  # Log data being sent to 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:
@@ -174,13 +182,11 @@ def send_to_salesforce(violations, score, video_path):
 # 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
+        'Title': 'Safety Video',
         '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)