AI and drug treatment