The coulomb interaction term in the kohn sham equation would be replaced by gravitational interaction.
Then the Kohn Sham equation would include the kinetic energy term, gravitational interaction term, interaction potential and the exchange correlatio potential term.
.
