Update app.py

app.py

@@ -113,6 +113,7 @@ def generate_violation_pdf(violations, score):
 
         c.showPage()
         c.save()
+       _cookie = None
         pdf_file.seek(0)
 
         with open(pdf_path, "wb") as f:
@@ -225,13 +226,20 @@ def process_video(video_data):
         violations, snapshots = [], []
         frame_count = 0
         total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
-        start_time = time.time()
         fps = video.get(cv2.CAP_PROP_FPS)
+        video_duration = total_frames / max(fps, 1)
+        logger.info(f"Video duration: {video_duration:.2f} seconds, Total frames: {total_frames}, FPS: {fps:.2f}")
+
+        start_time = time.time()
         max_frames = int(fps * CONFIG["MAX_PROCESSING_TIME"])  # Max frames for 60 seconds
 
         while True:
             ret, frame = video.read()
-            if not ret or frame_count >= max_frames:
+            if not ret:
+                logger.info(f"End of video reached at frame {frame_count}")
+                break
+            if frame_count >= max_frames:
+                logger.info(f"Reached maximum processing limit at frame {frame_count}")
                 break
 
             results = model(frame, device=device)
@@ -252,6 +260,7 @@ def process_video(video_data):
                         "timestamp": frame_count / max(fps, 1)
                     }
                     violations.append(violation)
+                    logger.debug(f"Violation detected: {label} at frame {frame_count}, timestamp {violation['timestamp']:.2f}s")
 
                     snapshot_path = os.path.join(CONFIG["OUTPUT_DIR"], f"snapshot_{frame_count}_{label}.jpg")
                     cv2.imwrite(snapshot_path, frame)
@@ -265,13 +274,14 @@ def process_video(video_data):
                     })
 
             frame_count += 1
-            if time.time() - start_time > CONFIG["MAX_PROCESSING_TIME"]:
-                logger.warning("Processing time limit of 60 seconds exceeded")
-                break
+
+        processing_time = time.time() - start_time
+        logger.info(f"Processed {frame_count} frames in {processing_time:.2f} seconds")
 
         video.release()
         os.remove(video_path)
-        score = calculate_safety_score(violations, max(total_frames, frame_count))
+        score = calculate_safety_score(violations, frame_count)
+        logger.info(f"Safety score calculated: {score}% with {len(violations)} violations")
         pdf_path, pdf_url, pdf_file = generate_violation_pdf(violations, score)
         report_id, final_pdf_url = push_report_to_salesforce(violations, score, pdf_path, pdf_file)
 
@@ -280,16 +290,20 @@ def process_video(video_data):
             "snapshots": snapshots,
             "score": score,
             "salesforce_record_id": report_id,
-            "violation_details_url": final_pdf_url
+            "violation_details_url": final_pdf_url,
+            "processed_frames": frame_count,
+            "video_duration": video_duration
         }
     except Exception as e:
-        logger.error(f"Error processing video: {e}")
+        logger.error(f"Error processing video: {e}", exc_info=True)
         return {
             "violations": [],
             "snapshots": [],
             "score": 0,
             "salesforce_record_id": None,
-            "violation_details_url": ""
+            "violation_details_url": "",
+            "processed_frames": 0,
+            "video_duration": 0
         }
 
 # ==========================
@@ -297,7 +311,7 @@ def process_video(video_data):
 # ==========================
 def gradio_interface(video_file):
     if not video_file:
-        return "No file uploaded.", "", "No file uploaded.", "", ""
+        return "No file uploaded.", "", "No file uploaded.", "", "", ""
     try:
         with open(video_file, "rb") as f:
             video_data = f.read()
@@ -332,24 +346,26 @@ def gradio_interface(video_file):
             f"Safety Score: {result['score']}%",
             snapshots_text,
             f"Salesforce Record ID: {result['salesforce_record_id'] or 'N/A'}",
-            result["violation_details_url"] or "N/A"
+            result["violation_details_url"] or "N/A",
+            f"Processed {result['processed_frames']} frames (Video duration: {result['video_duration']:.2f} seconds)"
         )
     except Exception as e:
         logger.error(f"Error in Gradio interface: {e}", exc_info=True)
-        return f"Error: {str(e)}", "", "Error in processing.", "", ""
+        return f"Error: {str(e)}", "", "Error in processing.", "", "", ""
 
 interface = gr.Interface(
     fn=gradio_interface,
-    inputs=gr.Video(label="Upload Site Video"),
+    inputs=gr.Video(label="Upload Site Video (Up to 1 Minute)"),
     outputs=[
         gr.Markdown(label="Detected Safety Violations"),
         gr.Textbox(label="Compliance Score"),
         gr.Markdown(label="Snapshots"),
         gr.Textbox(label="Salesforce Record ID"),
-        gr.Textbox(label="Violation Details URL")
+        gr.Textbox(label="Violation Details URL"),
+        gr.Textbox(label="Processing Summary")
     ],
     title="Worksite Safety Violation Analyzer",
-    description="Upload a video (up to 1 minute) to detect safety violations (missing helmet, missing harness, unsafe posture)."
+    description="Upload a video (up to 1 minute) to detect safety violations (missing helmet, missing harness, unsafe posture). Violations are only recorded when safety gear is missing or posture is unsafe."
 )
 
 if __name__ == "__main__":