AI-powered energy management