Update app.py

app.py

@@ -297,6 +297,10 @@ def push_report_to_salesforce(violations, score, pdf_path, pdf_file):
 
 def process_video(video_data):
     try:
+        # Ensure output directory exists
+        os.makedirs(CONFIG["OUTPUT_DIR"], exist_ok=True)
+        logger.info(f"Output directory ensured: {CONFIG['OUTPUT_DIR']}")
+
         # Create temp video file
         video_path = os.path.join(CONFIG["OUTPUT_DIR"], f"temp_{int(time.time())}.mp4")
         with open(video_path, "wb") as f:
@@ -306,6 +310,7 @@ def process_video(video_data):
         # Open video file
         cap = cv2.VideoCapture(video_path)
         if not cap.isOpened():
+            os.remove(video_path)
             raise ValueError("Could not open video file")
 
         # Get video properties
@@ -419,7 +424,8 @@ def process_video(video_data):
                     violation_tracker[worker_id][label].append(detection)
 
         cap.release()
-        os.remove(video_path)
+        if os.path.exists(video_path):
+            os.remove(video_path)
         processing_time = time.time() - start_time
         logger.info(f"Processing complete in {processing_time:.2f}s")
 
@@ -482,6 +488,8 @@ def process_video(video_data):
 
     except Exception as e:
         logger.error(f"Error processing video: {e}", exc_info=True)
+        if 'video_path' in locals() and os.path.exists(video_path):
+            os.remove(video_path)
         yield f"Error processing video: {e}", "", "", "", ""
 
 def gradio_interface(video_file):