Unverified Commit 4b5b5d19 authored by K900's avatar K900 Committed by GitHub
Browse files

nixos/test-driver: fix OCR (#375091)

parents a9aa816c 84b216c2
Loading
Loading
Loading
Loading
+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}")