Нужно добавить проверку на то что если input не array в самом начале, тогда если функция получает не массив то возвращает пустой массив.
И дальше если нет еще, условия на 0, если элемент массива равен 0. В целом загвоздка для меня была только в первом условии, что было не прописано, что делать если функция получает не массив.
/////
It is necessary to add a check about what if input is not an array at the very beginning, then if the function does not receive an array, it returns an empty array. And then, a condition for 0, if the array element is 0 what you want function to do.
In general, the only snag for me was in the first condition, what to do if the function does not receive an array.