Update utils.py

utils.py

@@ -1,6 +1,7 @@
 import os
 import cv2
 import numpy as np
+import time
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from reportlab.lib.units import inch
@@ -52,14 +53,14 @@ def calculate_safety_score(violations):
 
 async def generate_violation_pdf(violations, score, output_dir):
     try:
-        pdf_filename = f"violations_{int(asyncio.get_event_loop().time())}.pdf"
+        pdf_filename = f"violations_{int(time.time())}.pdf"
         pdf_path = os.path.join(output_dir, pdf_filename)
         pdf_file = BytesIO()
         c = canvas.Canvas(pdf_file, pagesize=letter)
         c.setFont("Helvetica-Bold", 16)
         c.drawString(1 * inch, 10 * inch, "Worksite Safety Violation Report")
         c.setFont("Helvetica", 12)
-        c.drawString(1 * inch, 9.5 * inch, f"Date: {asyncio.get_event_loop().time():.0f}")
+        c.drawString(1 * inch, 9.5 * inch, f"Date: {time.strftime('%Y-%m-%d')}")
         c.setFont("Helvetica-Bold", 14)
         c.drawString(1 * inch, 8.7 * inch, f"Safety Compliance Score: {score}%")
         c.setFont("Helvetica", 10)
@@ -97,8 +98,8 @@ 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(asyncio.get_event_loop().time())}",
-            "PathOnClient": f"safety_violation_{int(asyncio.get_event_loop().time())}.pdf",
+            "Title": f"Safety_Violation_Report_{int(time.time())}",
+            "PathOnClient": f"safety_violation_{int(time.time())}.pdf",
             "VersionData": encoded_pdf,
             "FirstPublishLocationId": record_id
         })
@@ -154,4 +155,20 @@ def blur_faces(frame):
         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
+    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)
+        if os.path.isfile(file_path):
+            file_age = current_time - os.path.getmtime(file_path)
+            if file_age > max_age_seconds:
+                try:
+                    os.remove(file_path)
+                    logger.info(f"Removed old output file: {file_path}")
+                except Exception as e:
+                    logger.error(f"Failed to remove {file_path}: {e}")