Update app.py

app.py

@@ -46,14 +46,14 @@ except Exception as e:
     raise
 
 # ==========================
-# Salesforce connection
+# Salesforce connection with hardcoded credentials
 # ==========================
 def connect_to_salesforce():
     sf = Salesforce(
-        username=os.getenv("prashanth1ai@safety.com"),
-        password=os.getenv("SaiPrash461"),
-        security_token=os.getenv("AP4AQnPoidIKPvSvNEfAHyoK"),
-        domain=os.getenv("SF_DOMAIN", "login")
+        username="prashanth1ai@safety.com",
+        password="SaiPrash461",
+        security_token="AP4AQnPoidIKPvSvNEfAHyoK",
+        domain="login"  # Use "test" if sandbox
     )
     return sf
 
@@ -70,20 +70,22 @@ def push_report_to_salesforce(score, violations, pdf_filename):
     violations_count = len(violations)
     violations_details = violations_to_text(violations)
 
-    # Public URL base for your PDF reports on HuggingFace space or other hosting
     PUBLIC_PDF_BASE_URL = "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/output/"
-
     pdf_url = f"{PUBLIC_PDF_BASE_URL}{os.path.basename(pdf_filename)}"
 
-    record = sf.Safety_Video_Report__c.create({
-        "Compliance_Score__c": score,
-        "Violations_Found__c": violations_count,
-        "Violations_Details__c": violations_details,
-        "Status__c": "Pending",
-        "PDF_Report_URL__c": pdf_url
-    })
-
-    return record.get("id")
+    try:
+        record = sf.Safety_Video_Report__c.create({
+            "Compliance_Score__c": score,
+            "Violations_Found__c": violations_count,
+            "Violations_Details__c": violations_details,
+            "Status__c": "Pending",
+            "PDF_Report_URL__c": pdf_url
+        })
+        print(f"✅ Salesforce record created with Id: {record.get('id')}")
+        return record.get("id")
+    except Exception as e:
+        print(f"❌ Salesforce record creation failed: {e}")
+        return None
 
 # ==========================
 # Safety Score Calculation
@@ -109,15 +111,12 @@ def generate_pdf_report(violations, snapshots, score):
         c = canvas.Canvas(pdf_path, pagesize=letter)
         width, height = letter
 
-        # Title
         c.setFont("Helvetica-Bold", 16)
         c.drawString(50, height - 50, "Worksite Safety Compliance Report")
 
-        # Compliance Score
         c.setFont("Helvetica", 12)
         c.drawString(50, height - 80, f"Compliance Score: {score}%")
 
-        # Violations Table
         y = height - 120
         c.setFont("Helvetica-Bold", 12)
         c.drawString(50, y, "Detected Violations:")
@@ -129,7 +128,6 @@ def generate_pdf_report(violations, snapshots, score):
             c.drawString(50, y, text)
             y -= 20
 
-            # Add snapshot if available and exists locally
             snapshot = next((s for s in snapshots if s["frame"] == v["frame"] and s["violation"] == v["violation"]), None)
             if snapshot:
                 local_img_path = os.path.join(OUTPUT_DIR, os.path.basename(snapshot["snapshot_url"]))
@@ -157,7 +155,6 @@ def generate_pdf_report(violations, snapshots, score):
 def process_video(video_data, frame_skip=5, max_frames=100):
     try:
         print("Processing video data...")
-        # Save uploaded video data to a temporary file
         video_path = os.path.join(OUTPUT_DIR, f"temp_{int(time.time())}.mp4")
         with open(video_path, "wb") as f:
             f.write(video_data)
@@ -182,7 +179,6 @@ def process_video(video_data, frame_skip=5, max_frames=100):
                 frame_count += 1
                 continue
 
-            # Model inference
             results = model(frame, device=device)
 
             for result in results:
@@ -201,7 +197,6 @@ def process_video(video_data, frame_skip=5, max_frames=100):
                     }
                     violations.append(violation)
 
-                    # Save snapshot
                     snapshot_path = os.path.join(OUTPUT_DIR, f"snapshot_{frame_count}_{label}.jpg")
                     cv2.imwrite(snapshot_path, frame)
                     snapshots.append({
@@ -226,14 +221,12 @@ def process_video(video_data, frame_skip=5, max_frames=100):
         score = calculate_safety_score(violations)
         pdf_path = generate_pdf_report(violations, snapshots, score)
 
-        # Push report to Salesforce with public PDF URL
         if pdf_path:
             report_id = push_report_to_salesforce(score, violations, pdf_path)
             print(f"Salesforce record created with Id: {report_id}")
         else:
             report_id = None
 
-        # Convert PDF to base64 for UI display (optional)
         with open(pdf_path, "rb") as f:
             pdf_base64 = base64.b64encode(f.read()).decode('utf-8')