fix colab runtime

control/mkgui/base/ui/streamlit_ui.py

@@ -18,6 +18,7 @@ from control.mkgui.base import Opyrator
 from control.mkgui.base.core import name_to_title
 from . import schema_utils
 from .streamlit_utils import CUSTOM_STREAMLIT_CSS
+from pyngrok import ngrok
 
 STREAMLIT_RUNNER_SNIPPET = """
 from control.mkgui.base.ui import render_streamlit_ui
@@ -59,7 +60,10 @@ def launch_ui(port: int = 8501) -> None:
             f"""{python_path} "{sys.executable}" -m streamlit run --server.port={port} --server.headless=True --runner.magicEnabled=False --server.maxUploadSize=50  --browser.gatherUsageStats=False {f.name}""",
             shell=True,
         )
+        # Setup a tunnel to the streamlit port 2023
 
+        public_url = ngrok.connect(port="2023")
+        print(public_url)
         f.close()
         unlink(f.name)
 

web.py

@@ -1,7 +1,6 @@
 import os
 import sys
 import typer
-from pyngrok import ngrok
 
 cli = typer.Typer()
 
@@ -19,11 +18,6 @@ def launch(port: int = typer.Option(2023, "--port", "-p")) -> None:
     from control.mkgui.base.ui.streamlit_ui import launch_ui
 
     launch_ui(port)
-
-    # Setup a tunnel to the streamlit port 2023
-
-    public_url = ngrok.connect(port="2023")
-    print(public_url)
     
 
 if __name__ == "__main__":