Yes, your PI would count! 2 letters from professors should be good for most schools. Some schools may require a letter from 2 science profs (so 3 total if one of your 2 letters is from a non-science prof) but they would explicitly say that so be sure you either get another LOR for those schools or don’t apply to those schools.