AI-driven audio enhancement