Timothy Chalamet is right age and right level of pretty. I reckon he could pull it off.

Young Christian Slater would have been good.

If the movie was about him being completely blindsided by the wrongful arrest and was just unsure what was happening around him, then young Keanu Reeves would work well.