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

# 安装导入器
packer = PythonToSQLite("plugins.mbank")
finder = packer.install_importer()


def main():
    task_manager = TaskManager(4)
    start_time = time.time()
    print(start_time)
    for i in range(1):
        task_name = f"yolo_test_{i}"
        args = TaskArgs(
            task_name=task_name,
            algorithms=["yolo_example", "yolo_example"],
            input="rtsp://192.168.137.1:8554/stream",
            is_vfs=True
        )
        task_manager.submit(args)
    
    # time.sleep(5)

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



if __name__ == "__main__":
    main()