import time
from sophon_chakcy.utils.task_manager import TaskManager, TaskArgs


def main():
    task_manager = TaskManager(16)
    start_time = time.time()
    print(start_time)
    for i in range(16):
        task_name = f"yolo_test_{i}"
        args = TaskArgs(
            task_name=task_name,
            algorithms=["yolo_example", "yolo_example", "yolo_example"],
            # input="rtsp://192.168.137.1:8554/stream",
            input="./resources/datasets/test_car_person_1080P.mp4",
            output=f"rtsp://192.168.137.1:8554/stream_{i}",
        )
        task_manager.submit(args)
    
    # time.sleep(30)

    # for i in range(1):
    #     task_name = f"yolo_test_{i}"
    #     task_manager.stop_task(task_name)
        
    # print(time.time() - start_time)



if __name__ == "__main__":
    main()