Update app.py

app.py

@@ -106,29 +106,28 @@ def process_video_and_save(uploaded_file):
     # Dictionary to track violations per license plate
     violations_dict = {}
 
-    # Video path (input)
-    video_path = "/kaggle/working/uploaded_video.mp4"  # Save the uploaded video file to this path
+    # Paths for saving violation images and videos
+    video_path = "/kaggle/working/uploaded_video.mp4"
+    output_video_path = '/kaggle/working/output_violation.mp4'
+    
+    # Save the uploaded video file to this path
     with open(video_path, "wb") as f:
         f.write(uploaded_file.getbuffer())
 
     cap = cv2.VideoCapture(video_path)
-
-    # Check if the video file opened successfully
+    
     if not cap.isOpened():
-        print("Error opening video file")
+        st.error("Error opening video file.")
         return None
 
-    # Define codec and output video settings
+    # Codec and output settings
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    output_video_path = '/kaggle/working/output_violation.mp4'
     fps = cap.get(cv2.CAP_PROP_FPS)
-    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))  # Frame width
-    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))  # Frame height
+    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     out = cv2.VideoWriter(output_video_path, fourcc, fps, (width, height))
 
-    margin_y = 50
-
-    # Process the video frame by frame
+    # Process frames
     while cap.isOpened():
         ret, frame = cap.read()
         if not ret:
@@ -223,6 +222,8 @@ def process_video_and_save(uploaded_file):
     # Release resources when done
     cap.release()
     out.release()
+    if not os.path.exists(output_video_path):
+        st.error("Error: Processed video was not created.")
     return output_video_path  # Return the path of the processed video