AI diagnosis assistant