Good direction, well written story of a biologist

Rating: 10/10 (28 June 2017)

All Spoilers, just writing for my future self. So that I will be able to know myself when I will re watch he will re watch it.
A biologist who is married (but has lost his engagement ring somehow) is about to loss his wife due to some sort of cancer (tumor). She is losing her senses (only shown to have lost sense for sensing temperature). She has a seizure and was admitted to the hospital. Meanwhile, our biologist is trying to find cure for tumor by testing different drugs on monkeys(the test subjects). While he was upset and thinking about her wife condition, nurse warned the doctor the the test subject has been open for some time and is about to die if nothing is done soon. So they tried everything and nothing worked. Then he realized (which is shown as looking up to peep in his mind palace), that some extract from some tree could help stabilize her (monkey). And it worked. The monkey is better than before except her tumor. But he wanted to cure tumor so that he can save his wife. His team tested the same herb on different subject and it worked except the tumor is there, which later on starts decreasing. Now there are two other sets are there in the movie. One is the doctors mind palace, shown as a shown in an invisible bubble traveling in space. Inside it is dead tree of life and the doctor as a monk. The other is the fountain the book which her wife was writing. He has to complete the last chapter of the book as she could do it. So in the last chapter he writes the person who attacked him somehow realizes that he is not the second one to come to him after his after but he is the faster only. So he offer his throat. The lead character of the story moves on to find the tree of life. He stabbed his dagger in the tree trunk. A white shampoo like liquid comes out it. Which when dropped on ground kick started the growth of weed (life). Then he healed his wounds withe the same liquid. Then for unknown reason started drinking the liquid. Which kick started the growth of weed throughout his body thus killing the body. Meanwhile the mind palace guy is going into an about to die start. Which die with a supernova thus killing him and reviving the tree of life.