Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 import numpy as np
 from simple_salesforce import Salesforce
 
-# Create a connection to Salesforce (you should replace with your credentials)
+# Salesforce connection details (replace these with your actual credentials)
 sf = Salesforce(username='prashanth@safetyanaluzer.com', password='SaiPrash461', security_token='MOA3BXBfGyqnjBneog8a9IcGw')
 
 try:
@@ -101,6 +101,10 @@ def process_video(video_path, frame_skip=5):
 
         video.release()
         score = calculate_safety_score(violations)
+        
+        # Send data to Salesforce
+        send_to_salesforce(violations, score)
+
         return violations, score
 
     except Exception as e:
@@ -122,6 +126,32 @@ 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):
+    # Assuming you have the Safety_Video_Report__c record's Id, you need to get this from the user input or Salesforce
+    report_id = 'your_report_id_here'
+
+    # Prepare the data to update in Salesforce
+    violations_details = "\n".join([f"Frame {v['frame']}: {v['violation']} (Confidence: {v['confidence']})" for v in violations])
+    
+    # Prepare the data payload to update Salesforce record
+    data = {
+        'Compliance_Score__c': score,
+        'Violations_Found__c': len(violations),
+        'Violations_Details__c': violations_details,
+        'Status__c': 'Reviewed',  # Set to 'Reviewed' after processing
+        # 'PDF_Report_URL__c': pdf_url (If you generate and store a PDF URL)
+    }
+
+    try:
+        # Update the Safety_Video_Report__c record in Salesforce
+        sf.Safety_Video_Report__c.update(report_id, data)
+        print("✅ Successfully updated Salesforce with the results!")
+    except Exception as e:
+        print(f"❌ Error updating Salesforce: {e}")
+
 # ==========================
 # Gradio Interface
 # ==========================