Ok, so I was on Wikipedia and I saw the following written about these songs. Take them with a grain of salt since it is Wikipedia, but they attribute their source to the new making-of book.
Differences among film and stage versions
The film follows the song order of the Broadway script, except that both "Gee, Officer Krupke" and "Cool" are performed in the first half, with "One Hand, One Heart" appearing in between. Tony sings "Cool" to Riff to get the Jets to wait to fight at the rumble that evening, instead of Riff singing it to encourage the Jets to stay cool during the war council at Doc's Drug Store.
Gee, Officer Krupke
In the 2021 film version, the song is again moved to the first half, performed prior to the rumble, with "Cool" moved to after "One Hand, One Heart" Additionally, instead of in the streets, the song takes place in the 21st Precinct of the New York City Police Department.
Cool
Hours before the rumble, Tony (Ansel Elgort) sings it to Riff (Mike Faist) to get the Jets to wait to fight there that evening.
Somewhere
In the 2021 film, the song takes place after "I Feel Pretty," like in the stage show, except now it is sung by Valentina (Rita Moreno), a reworked version of the character Doc. The ballet is again omitted, like in the 1961 film.