# 시스템 생성 — 이 파일은 커스텀 UI 입력에 따라 자동 생성됩니다. 직접 수정하지 마세요.

import argparse

from clearml import Task

from start import run

def upload_acai_task(task: Task, output: dict) -> None:
    """UI output 정의에 맞춰 업로드"""
    result_data = {"output": output, "status": "PASS"}
    task.upload_artifact(name="final_result", artifact_object=result_data)


if __name__ == "__main__":
    task = Task.init(
        project_name="Person_Attribute_Recognition",
        task_name="model-yolo-person-classify",
    )

    parser = argparse.ArgumentParser()
    parser.add_argument("--image_url", type=str, required=True)
    parser.add_argument("--xywh", type=str, required=True)
    args = parser.parse_args()
    task.connect(parser)

    output = run(args.image_url, args.xywh)
    upload_acai_task(task, output)