Commit 84b216c2 authored by K900's avatar K900
Browse files

nixos/test-driver: fix OCR 

I don't know why it doesn't want to do TIFF now, but there's also
absolutely no reason for it to be TIFF anyway, so let's just use
an image format that is actually sane.
parent a3a980d7

nixos/lib/test-driver/src/test_driver/machine.py

+5 −3
Original line number Diff line number Diff line
@@ -105,14 +105,16 @@ def _perform_ocr_on_screenshot( 


    tess_args = "-c debug_file=/dev/null --psm 11"



    cmd = f"convert {magick_args} '{screenshot_path}' 'tiff:{screenshot_path}.tiff'"

    cmd = f"magick convert {magick_args} '{screenshot_path}' '{screenshot_path}.png'"

    ret = subprocess.run(cmd, shell=True, capture_output=True)

    if ret.returncode != 0:

        raise Exception(f"TIFF conversion failed with exit code {ret.returncode}")

        raise Exception(

            f"Image processing failed with exit code {ret.returncode}, stdout: {ret.stdout.decode()}, stderr: {ret.stderr.decode()}"

        )



    model_results = []

    for model_id in model_ids:

        cmd = f"tesseract '{screenshot_path}.tiff' - {tess_args} --oem '{model_id}'"

        cmd = f"tesseract '{screenshot_path}.png' - {tess_args} --oem '{model_id}'"

        ret = subprocess.run(cmd, shell=True, capture_output=True)

        if ret.returncode != 0:

            raise Exception(f"OCR failed with exit code {ret.returncode}")