Record.py

import threading
import queue
import cv2
from Camera import Camera
import time
from colorama import Fore
import os
os.system("color") #fix color in terminal not working
def main(capture_source, eye, start_time):
    time_pre = 0
    cancellation_event = threading.Event()
    capture_event = threading.Event()
    camera_status_outgoing = queue.Queue()
    camera_output_outgoing = queue.Queue(maxsize=1)

    camera = Camera(capture_source, cancellation_event, capture_event, camera_status_outgoing, camera_output_outgoing)
    camera_thread = threading.Thread(target=camera.run)
    camera_thread.start()

    filename = f"output_{eye}.avi"  # Save as .mp4
    fourcc = cv2.VideoWriter_fourcc(*'DIVX')  # Use mp4v for H.264 codec

    output = cv2.VideoWriter(filename, fourcc, 60, (240, 240))


    try:
        while True:
            if not camera_output_outgoing.empty():
                current_time = time.time()
                elapsed_time = current_time - start_time

                frame = camera_output_outgoing.get()
                image, frame_number, fps = frame

                if elapsed_time >5 and elapsed_time < 66:
                    output.write(image)
                    if int(elapsed_time) > int(time_pre):
                        time_pre = int(elapsed_time)
                        print(f"{Fore.GREEN}[INFO] Frame {frame_number} sent to queue Elapsed time: {int(elapsed_time) - 5}/60 seconds")


                if elapsed_time > 68: # wait 10 seconds after video stops to fix stall bug
                    raise KeyboardInterrupt

                cv2.imshow('Camera Frame', image)

                if cv2.waitKey(1) & 0xFF == ord('q'):
                    break

    except KeyboardInterrupt:
        print("[INFO] Shutting down...")
        cv2.destroyAllWindows()
        print(f"{Fore.CYAN} Thank you for contributing <3")
        print("Make sure you DM prohurtz the .avi files generated in the path of the .exe")
        print("[INFO] Close this window to exit the program")
        cancellation_event.set()
        camera_thread.join()

        time.sleep(5)

if __name__ == "__main__":
    src = input('Enter camera address then press enter. (ex COM5, 0, http://192.168.1.1) :>  ')
    eye = input('Enter if this is r for right cam and l for left cam then press enter. :>  ').lower()


    print ('Waiting for camera to warm up for 5 seconds...')
    start_time = time.time()
    main(src, eye, start_time)
   # main("COM15", "r", start_time)