AI-based water quality monitoring