Nope & big Nope.
Stalin was not a mental guy. He is a soldier & will always be one even if he is not in the force. He would be totally injured taking out Pakistani army single-handedly & doctors fail to remove one bullet which is closer to his heart. He cannot get emotionally stressed which would cause the bullet to get in his heart which would kill him. His chief decides to take him off the force in order to save him. Disappointed he comes back & is still a force to reckon with off his uniform.
Now Jai, is not removed from the force for any of the reasons above. Jai doesn't fight with Pakistani army either. It's an Indian politician played by Danny who does fake terrorist attack which causes Salman's troupe to pay the price. This is thrown in casually.
Believing in Pay it Forward theory makes Chiranjeevi call himself a mad person to hope for people's change. That is where Salman's character goes through when he assumes people aren't changing.
The whole climax of Stalin relies on how the bullet is inching towards his heart while fighting. That part is entirely omitted in well what was easily the best choreographed action climax scene.
Mental wouldn't have matched this film whatsoever. So whatever changes you are saying are mentioned above by me. Other than this, the movie is exactly same as Stalin.
Since the main theme was not subtle, all Jai Ho required was more subtle approach towards the sub plots. The Pay It Forward concept was beautifully established in Telugu when Chiru watched a bunch of differently abled children in a running race & one falls down. All other kids come back to life the kid & finish the line together. This was based on actual incident. But in Jai Ho, Salman thinks of the theory by doing multiplication. That scene fell flat.