in wrapper return wrappedFunction(*args, **kwargs)

import pyautogui
import time

def center_finder(path):
    location = pyautogui.locateCenterOnScreen(path)
    if location is not None:
        return location
    else:
        return None

def main():
    while True:
        detectEnemy1 = center_finder("drzewiec.png")

        if detectEnemy1 is not None:
            pyautogui.press("3")
            time.sleep(1)

        detectEnemy2 = center_finder("skalny.png")

        if detectEnemy2 is not None:
            pyautogui.press("2")
            time.sleep(1)

if __name__ == "__main__":
    main()

Traceback (most recent call last):
  File "C:\Users\pat\PycharmProjects\PythonProject\.venv\Lib\site-packages\pyautogui\__init__.py", line 172, in wrapper
    return wrappedFunction(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\pat\PycharmProjects\PythonProject\.venv\Lib\site-packages\pyautogui\__init__.py", line 204, in locateCenterOnScreen
    return pyscreeze.locateCenterOnScreen(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\pat\PycharmProjects\PythonProject\.venv\Lib\site-packages\pyscreeze\__init__.py", line 447, in locateCenterOnScreen
    coords = locateOnScreen(image, **kwargs)
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\pat\PycharmProjects\PythonProject\.venv\Lib\site-packages\pyscreeze\__init__.py", line 405, in locateOnScreen
    retVal = locate(image, screenshotIm, **kwargs)
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\pat\PycharmProjects\PythonProject\.venv\Lib\site-packages\pyscreeze\__init__.py", line 383, in locate
    points = tuple(locateAll(needleImage, haystackImage, **kwargs))
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\pat\PycharmProjects\PythonProject\.venv\Lib\site-packages\pyscreeze\__init__.py", line 257, in _locateAll_opencv
    raise ImageNotFoundException('Could not locate the image (highest confidence = %.3f)' % result.max())
pyscreeze.ImageNotFoundException: Could not locate the image (highest confidence = 0.303)

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "C:\Users\pat\PycharmProjects\PythonProject\początki.py", line 29, in <module>
    main()
  File "C:\Users\pat\PycharmProjects\PythonProject\początki.py", line 15, in main
    detectEnemy1 = center_finder("drzewiec.png")
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\pat\PycharmProjects\PythonProject\początki.py", line 6, in center_finder
    location = pyautogui.locateCenterOnScreen(path)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\pat\PycharmProjects\PythonProject\.venv\Lib\site-packages\pyautogui\__init__.py", line 174, in wrapper
    raise ImageNotFoundException  # Raise PyAutoGUI's ImageNotFoundException.
    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
pyautogui.ImageNotFoundException

Process finished with exit code 1

def main():
    try:
        x, y = center_finder("drzewiec.png")
    except yautogui.ImageNotFoundException:
        x, y = center_finder("skalny.png")