Diddy's lawyer shares insight on rapper's "state of mind" in prison