Update app.py

app.py

@@ -11,6 +11,7 @@ from reportlab.lib.utils import ImageReader
 import base64
 from PIL import Image
 from simple_salesforce import Salesforce, SalesforceMalformedRequest
+import requests
 
 # ==========================
 # Configuration
@@ -85,8 +86,15 @@ def push_report_to_salesforce(score, violations, pdf_url):
         violations_count = len(violations)
         violations_details = violations_to_text(violations)
 
+        # Verify PDF URL accessibility
+        print(f"🔍 Verifying PDF URL: {pdf_url}")
+        response = requests.head(pdf_url, timeout=5)
+        if response.status_code != 200:
+            print(f"⚠️ PDF URL not accessible: {response.status_code}")
+            pdf_url = ""  # Fallback to empty URL to avoid saving invalid link
+
         # Create the record in the custom object Safety_Video_Report__c
-        print(f"Creating Salesforce record with compliance score: {score} and violations found: {violations_count}")
+        print(f"📝 Creating Salesforce record with compliance score: {score} and violations found: {violations_count}")
         record = sf.Safety_Video_Report__c.create({
             "Compliance_Score__c": score,
             "Violations_Found__c": violations_count,
@@ -155,21 +163,38 @@ def generate_pdf_report(violations, snapshots, score):
                     img = ImageReader(local_img_path)
                     c.drawImage(img, 50, y - 100, width=200, height=150)
                     y -= 170
+                else:
+                    print(f"⚠️ Snapshot not found at {local_img_path}")
 
             if y < 50:
                 c.showPage()
                 y = height - 50
 
         c.save()
-        print(f"PDF generated at {pdf_path}")
+        print(f"✅ PDF generated at {pdf_path}")
 
-        # Verify the file exists
-        if not os.path.exists(pdf_path):
-            raise Exception(f"PDF file not found at {pdf_path}")
+        # Verify file existence with retry
+        for _ in range(5):
+            if os.path.exists(pdf_path):
+                break
+            print(f"⏳ Waiting for PDF at {pdf_path}")
+            time.sleep(1)
+        else:
+            raise Exception(f"PDF file not found at {pdf_path} after retries")
 
         # Generate public URL for the PDF
         pdf_url = f"{BASE_PUBLIC_URL}{pdf_filename}"
-        print(f"Public PDF URL: {pdf_url}")
+        print(f"📎 Public PDF URL: {pdf_url}")
+
+        # Verify URL accessibility
+        try:
+            response = requests.head(pdf_url, timeout=5)
+            if response.status_code == 200:
+                print(f"✅ PDF URL is accessible")
+            else:
+                print(f"⚠️ PDF URL not accessible: HTTP {response.status_code}")
+        except requests.RequestException as e:
+            print(f"⚠️ Failed to verify PDF URL: {e}")
 
         return pdf_path, pdf_url
 
@@ -182,11 +207,11 @@ def generate_pdf_report(violations, snapshots, score):
 # ==========================
 def process_video(video_data, frame_skip=5, max_frames=100):
     try:
-        print("Processing video data...")
+        print("📹 Processing video data...")
         video_path = os.path.join(STATIC_OUTPUT_DIR, f"temp_{int(time.time())}.mp4")
         with open(video_path, "wb") as f:
             f.write(video_data)
-        print(f"Video saved to {video_path}")
+        print(f"💾 Video saved to {video_path}")
 
         video = cv2.VideoCapture(video_path)
         if not video.isOpened():
@@ -228,11 +253,15 @@ def process_video(video_data, frame_skip=5, max_frames=100):
                     snapshot_filename = f"snapshot_{frame_count}_{label}.jpg"
                     snapshot_path = os.path.join(STATIC_OUTPUT_DIR, snapshot_filename)
                     cv2.imwrite(snapshot_path, frame)
-                    snapshots.append({
-                        "violation": label,
-                        "frame": frame_count,
-                        "snapshot_url": f"{BASE_PUBLIC_URL}{snapshot_filename}"
-                    })
+                    if os.path.exists(snapshot_path):
+                        snapshots.append({
+                            "violation": label,
+                            "frame": frame_count,
+                            "snapshot_url": f"{BASE_PUBLIC_URL}{snapshot_filename}"
+                        })
+                        print(f"📸 Snapshot saved at {snapshot_path}")
+                    else:
+                        print(f"⚠️ Snapshot not saved at {snapshot_path}")
 
             frame_count += 1
             processed_frame_count += 1
@@ -246,6 +275,7 @@ def process_video(video_data, frame_skip=5, max_frames=100):
 
         video.release()
         os.remove(video_path)
+        print(f"🗑️ Temporary video file removed: {video_path}")
 
         score = calculate_safety_score(violations)
         pdf_path, pdf_url = generate_pdf_report(violations, snapshots, score)