AI powered cell staining