from agent.cdss.capbility import CDSSCapability
from agent.cdss.models.schemas import CDSSInput, CDSSOutput, CDSSInt, CDSSText

capability = CDSSCapability()

record = CDSSInput(
    pat_age=CDSSInt(type="month", value=24),
    pat_sex=CDSSText(type="sex", value="男"),
    #chief_complaint=["右下腹痛","恶心","呕吐"],
    #chief_complaint=["胸痛","左肩放射痛","下颌放射痛","呼吸困难","冷汗","恶心"],
    #chief_complaint=["呕血", "黑便", "头晕", "心悸"],
    chief_complaint=["大汗"],

    department=CDSSText(type='department',value="急诊医学科")
    )

if __name__ == "__main__":
    output = capability.process(input=record)

    for item in output.diagnosis.value:
        print(f"DIAG {item}  {output.diagnosis.value[item]} ")
    for item in output.departments.value:
        print(f"DEPARTMENT {item}")


    # for item in output.count_diagnosis.value:
    #     print(f"count DIAG {item}  {output.count_diagnosis.value[item]} ")
    # for item in output.checks.value:
    #     print(f"CHECK {item}  {output.checks.value[item]} ")
    # for item in output.drugs.value:
    #     print(f"DRUG {item}  {output.drugs.value[item]} ")