Launching Jupyter on Modal programmatically

launch_jupyter.py

import os
import secrets
import subprocess
import time

import modal

app = modal.App()

image = (
    modal.Image.debian_slim()
    .pip_install("jupyter", "bing-image-downloader~=1.1.2")
    .pip_install("pandas")  # Any other deps
)


@app.function(image=image, timeout=1_500)
def run_jupyter(queue, timeout=1_500):
    token = secrets.token_urlsafe(13)
    jupyter_port = 8888
    with modal.forward(jupyter_port) as tunnel:
        jupyter_process = subprocess.Popen(
            [
                "jupyter",
                "notebook",
                "--no-browser",
                "--allow-root",
                "--ip=0.0.0.0",
                f"--port={jupyter_port}",
                "--NotebookApp.allow_origin='*'",
                "--NotebookApp.allow_remote_access=1",
            ],
            env={**os.environ, "JUPYTER_TOKEN": token},
        )

        url = f"{tunnel.url}/?token={token}"
        queue.put(url)

        try:
            end_time = time.time() + timeout
            while time.time() < end_time:
                time.sleep(5)
            print(f"Reached end of {timeout} second timeout period. Exiting...")
        except KeyboardInterrupt:
            print("Exiting...")
        finally:
            jupyter_process.kill()


@app.local_entrypoint()
def main():
    with modal.Queue.ephemeral() as queue:
        run_jupyter.spawn(queue)
        tunnel_url = queue.get()
        print(f"Jupyter notebook is running at: {tunnel_url}")

    time.sleep(1000)