AI-driven image enhancement