Arduino coloured ball counter PLEASE HELP FOR SCHOOL PROJECT!! Answered

I have a school project where i have to make a colour sorter with a arduino i got it far enough that it sorts it phisically but I also have to sort is digitally

This is how it's supposed to work:

1) the laptop where the arduino is connected to has to show "press button to start colour sorter"

2) you put in the balls and it sorts it

3) then the laptop shows "press button to stop sorting"

4) you press the button again the machine stops and shows

-total amount of balls

-the amount of red balls,blue balls and white balls (only colours I need to sort)

-and show "press button to restart again"

this is what it's supposed to look like

https://www.youtube.com/watch?v=g3i51hdfLaw