Hey @jetilton,
I’ve implemented yet another session state which tackles your exact problem : Multi-page app with session state
I solved the issue by doing a rerun each time a value is set in the session state. Feel free to use it as is, or adapt OP’s version. Mine doesn’t feature python typing.
Tell me how it goes!