Refact AI

AI frontend code refactoring