Update app.py

app.py

@@ -11,7 +11,6 @@ from reportlab.lib.utils import ImageReader
 import base64
 from PIL import Image
 from simple_salesforce import Salesforce, SalesforceMalformedRequest
-import requests
 
 # ==========================
 # Configuration
@@ -86,18 +85,12 @@ 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,
+            "Violfilepath: app.py
             "Violations_Details__c": violations_details,
             "Status__c": "Pending",
             "PDF_Report_URL__c": pdf_url
@@ -163,8 +156,6 @@ 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()
@@ -173,11 +164,10 @@ def generate_pdf_report(violations, snapshots, score):
         c.save()
         print(f"✅ PDF generated at {pdf_path}")
 
-        # Verify file existence with retry
-        for _ in range(5):
+        # Verify the file exists with retry
+        for _ in range(3):
             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")
@@ -186,16 +176,6 @@ def generate_pdf_report(violations, snapshots, score):
         pdf_url = f"{BASE_PUBLIC_URL}{pdf_filename}"
         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
 
     except Exception as e:
@@ -259,7 +239,6 @@ def process_video(video_data, frame_skip=5, max_frames=100):
                             "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}")
 
@@ -328,9 +307,9 @@ def gradio_interface(video_file):
         return (
             violation_text,
             f"Safety Score: {result.get('score', 0)}%",
+              f"PDF URL: {result.get('pdf_url', '')}",
             result.get("pdf_base64", ""),
             result.get("snapshots", []),
-            f"PDF URL: {result.get('pdf_url', '')}",
             f"Salesforce Record ID: {result.get('salesforce_record_id', '')}"
         )
     except Exception as e: