Update utils.py

utils.py

@@ -1,22 +1,17 @@
 import os
 import cv2
 import numpy as np
+import asyncio
+import logging
 import time
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from reportlab.lib.units import inch
-from io import BytesIO
-import base64
-import asyncio
-import logging
-from simple_salesforce import Salesforce
-import tenacity
 from config import CONFIG
-import dlib
 
 logger = logging.getLogger(__name__)
 
-async def preprocess_frame(frame):
+def preprocess_frame(frame):
     frame = cv2.resize(frame, CONFIG["TARGET_RESOLUTION"], interpolation=cv2.INTER_LINEAR)
     frame = cv2.convertScaleAbs(frame, alpha=1.2, beta=20)
     return frame
@@ -55,7 +50,7 @@ async def generate_violation_pdf(violations, score, output_dir):
     try:
         pdf_filename = f"violations_{int(time.time())}.pdf"
         pdf_path = os.path.join(output_dir, pdf_filename)
-        pdf_file = BytesIO()
+        pdf_file = open(pdf_path, "wb")
         c = canvas.Canvas(pdf_file, pagesize=letter)
         c.setFont("Helvetica-Bold", 16)
         c.drawString(1 * inch, 10 * inch, "Worksite Safety Violation Report")
@@ -80,87 +75,18 @@ async def generate_violation_pdf(violations, score, output_dir):
                     c.showPage()
                     y_position = 10 * inch
         c.save()
-        pdf_file.seek(0)
-        with open(pdf_path, "wb") as f:
-            f.write(pdf_file.getvalue())
-        return pdf_path, f"{CONFIG['PUBLIC_URL_BASE']}{pdf_filename}", pdf_file
+        pdf_file.close()
+        logger.info(f"PDF generated: {pdf_path}")
+        return pdf_path
     except Exception as e:
         logger.error(f"Error generating PDF: {e}")
-        return "", "", None
-
-@tenacity.retry(stop=tenacity.stop_after_attempt(3), wait=tenacity.wait_fixed(2))
-async def connect_to_salesforce():
-    sf = Salesforce(**CONFIG["SF_CREDENTIALS"])
-    logger.info("Connected to Salesforce")
-    return sf
-
-async def upload_pdf_to_salesforce(sf, pdf_file, record_id):
-    try:
-        encoded_pdf = base64.b64encode(pdf_file.getvalue()).decode('utf-8')
-        content_version = sf.ContentVersion.create({
-            "Title": f"Safety_Violation_Report_{int(time.time())}",
-            "PathOnClient": f"safety_violation_{int(time.time())}.pdf",
-            "VersionData": encoded_pdf,
-            "FirstPublishLocationId": record_id
-        })
-        result = sf.query(f"SELECT Id, ContentDocumentId FROM ContentVersion WHERE Id = '{content_version['id']}'")
-        return f"https://{sf.sf_instance}/sfc/servlet.shepherd/version/download/{content_version['id']}"
-    except Exception as e:
-        logger.error(f"Error uploading PDF to Salesforce: {e}")
         return ""
 
-async def push_report_to_salesforce(violations, score, pdf_path, pdf_file):
-    try:
-        sf = await connect_to_salesforce()
-        violations_text = "".join(
-            f"Worker {v.get('worker_id', 'Unknown')}: {CONFIG['DISPLAY_NAMES'].get(v.get('violation', 'Unknown'), 'Unknown')} at {v.get('timestamp', 0.0):.2f}s (Conf: {v.get('confidence', 0.0):.2f})\n"
-            for v in violations
-        ) or "No violations detected."
-        record_data = {
-            "Compliance_Score__c": score,
-            "Violations_Found__c": len(violations),
-            "Violations_Details__c": violations_text,
-            "Status__c": "Pending",
-            "PDF_Report_URL__c": f"{CONFIG['PUBLIC_URL_BASE']}{os.path.basename(pdf_path)}" if pdf_path else ""
-        }
-        record = sf.Safety_Video_Report__c.create(record_data)
-        if pdf_file:
-            uploaded_url = await upload_pdf_to_salesforce(sf, pdf_file, record["id"])
-            if uploaded_url:
-                sf.Safety_Video_Report__c.update(record["id"], {"PDF_Report_URL__c": uploaded_url})
-        return record["id"], uploaded_url or record_data["PDF_Report_URL__c"]
-    except Exception as e:
-        logger.error(f"Salesforce record creation failed: {e}")
-        return "N/A", "Salesforce integration failed."
-
-@tenacity.retry(
-    stop=tenacity.stop_after_attempt(3),
-    wait=tenacity.wait_fixed(1),
-    retry=tenacity.retry_if_exception_type((IOError, OSError))
-)
-def verify_and_open_video(video_path):
-    if not os.path.exists(video_path) or os.path.getsize(video_path) == 0:
-        raise ValueError(f"Invalid video file: {video_path}")
-    cap = cv2.VideoCapture(video_path)
-    if not cap.isOpened():
-        raise ValueError("Could not open video file.")
-    return cap
-
-def blur_faces(frame):
-    detector = dlib.get_frontal_face_detector()
-    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
-    faces = detector(gray)
-    for face in faces:
-        x, y, w, h = face.left(), face.top(), face.width(), face.height()
-        roi = frame[y:y+h, x:x+w]
-        roi = cv2.GaussianBlur(roi, (23, 23), 30)
-        frame[y:y+h, x:x+w] = roi
-    return frame
-
 def clean_output_directory(max_age_seconds=86400):
     output_dir = os.path.join("static", "output")
     if not os.path.exists(output_dir):
         return
+    
     current_time = time.time()
     for filename in os.listdir(output_dir):
         file_path = os.path.join(output_dir, filename)