The Python extension offers support for Jupyter notebooks via the Jupyter extension to provide you a great Python notebook experience in VS Code. Install the Jupyter extension.

Simple Python package to preview and develop streamlit apps in jupyter notebooks - ddobrinskiy/streamlit-jupyter ...