AI-driven traffic management