Update app.py

app.py

@@ -27,6 +27,34 @@ def process_results(results, image):
 
     return image
 
+def process_video(uploaded_file):
+    # Read the video file
+    video = cv2.VideoCapture(uploaded_file)
+    frames = []
+    
+    while True:
+        ret, frame = video.read()
+        if not ret:
+            break  # Break the loop if there are no frames left
+
+        # Run YOLO model on the current frame
+        results = run_yolo(frame)
+        
+        # Process the results and draw boxes on the current frame
+        processed_frame = process_results(results, frame)
+        frames.append(processed_frame)  # Save the processed frame
+
+    video.release()
+    
+    # Create a video writer to save the processed frames
+    height, width, _ = frames[0].shape
+    out = cv2.VideoWriter('processed_video.mp4', cv2.VideoWriter_fourcc(*'mp4v'), 30, (width, height))
+
+    for frame in frames:
+        out.write(frame)  # Write each processed frame to the video
+
+    out.release()
+
 def main():
     st.title("Motorbike Violation Detection")
 
@@ -47,8 +75,8 @@ def main():
 
         elif uploaded_file.type == "video/mp4":
             # Process the video
-            video_bytes = uploaded_file.read()
-            st.video(video_bytes)
+            process_video(uploaded_file)  # Process the video and save the output
+            st.video('processed_video.mp4')  # Display the processed video
 
 if __name__ == "__main__":
     main()