import subprocess
from local_config import CHEXBERT_ENV_PATH, CHEXBERT_PATH


def run_chexbert_labeler(reports_path, output_path):
    # Path to a Python interpreter that runs any Python script
    # under the virtualenv /path/to/virtualenv/
    python_bin = CHEXBERT_ENV_PATH

    # Path to the script that must run under the virtualenv
    script_file = "label"

    print("Starting to extract Chexbert labels...")

    process = subprocess.run([python_bin, "-m", script_file, "-c", "checkpoint/chexbert.pth", "-d", reports_path, "-o", output_path],
                             stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=CHEXBERT_PATH)

    # Print the output
    print(process.stdout.decode())

    # Print the errors if there are any
    if process.stderr:
        print("Errors:")
        print(process.stderr.decode())

    print("Finished extracting Chexbert labels.")