AI-driven stress prediction