Update app.py

app.py

@@ -126,35 +126,35 @@ 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)
-    #}  
+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
+
+    # 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
+
+   # violations_details = "\n".join([f"Frame {v['frame']}: {v['violation']} (Confidence: {v['confidence']})" for v in violations])
+
     data = {
-        'Site__c': 'Sample Site',  # You can replace with dynamic data or capture from user input
-        'Uploaded_By__c': 'User_1',  # Example: Replace with actual user details
-        'Upload_Date__c': '2025-05-08T12:00:00Z',  # Replace with actual upload date
-        'Status__c': 'Reviewed',
+        '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
+        'Status__c': status,
         'Compliance_Score__c': score,
         'Violations_Found__c': len(violations),
-        'Violations_Details__c': "\n".join([f"Frame {v['frame']}: {v['violation']} (Confidence: {v['confidence']})" for v in 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
+       # 'PDF_Report_URL__c': "http://path_to_pdf_report"  # Replace with actual PDF report URL if available
     }
 
-   
     try:
         # Insert new record in Salesforce
         result = sf.Safety_Video_Reports__c.create(data)
@@ -162,7 +162,6 @@ def send_to_salesforce(violations, score):
     except Exception as e:
         print(f"❌ Error creating/updating record in Salesforce: {e}")
 
-
 # ==========================
 # Gradio Interface
 # ==========================