Smart traffic management with AI