video_utils.py

def extract_frames_from_stream(rtmp_url):
    cap = cv2.VideoCapture(rtmp_url)
    if not cap.isOpened():
        raise ValueError("Unable to open video stream.")
    
    while True:
        ret, frame = cap.read()
        if not ret:
            break
        yield frame
    
    cap.release()

def process_frame(frame, model):
    results = model(frame)
    annotated_frame = results.render()[0]
    return annotated_frame