Integrating What-If-Tool Widget in Streamlit App

Hi, I have been developing a Streamlit app for Machine Learning models and want to integrate the Widget provided by the What-If Tool (link: What-If Tool).

For an example code of the Widget that I want to integrate please see the link: Example Code of Widget on Google Colab

Can anyone please help me on how to include the exact same widget in streamlit application with all the interactive functionalities maintained.

Thanks in Advance!!