[feat] Update scripts and README

Author: sfc-gh-bnisco

README.md

@@ -17,13 +17,90 @@ Please file [bug reports](https://github.com/streamlit/streamlit/issues/new?temp
 ## 📦 Installation
 
 ```bash
-pip install streamlit-bokeh
+uv pip install streamlit-bokeh
 ```
 
 Ensure you have **Streamlit** and **Bokeh** installed as well:
 
 ```bash
-pip install streamlit bokeh
+uv pip install streamlit bokeh
+```
+
+---
+
+## 🛠️ Development
+
+### Prerequisites
+
+- **Python** 3.10–3.13
+- **Node.js** 24.x.y (see `.nvmrc`)
+- **uv** (fast Python package manager)
+
+### 1) Create and activate a virtual environment
+
+```bash
+uv venv .venv
+source .venv/bin/activate
+```
+
+### 2) Install Python dependencies from `pyproject.toml`
+
+```bash
+# Minimal runtime install (editable)
+uv pip install -e .
+
+# Recommended for development (includes tests/tools)
+uv pip install -e ".[devel]"
+```
+
+### 3) Install and build the frontend
+
+```bash
+cd streamlit_bokeh/frontend
+corepack enable
+yarn install
+yarn build          # one-time build to produce frontend/build assets
+
+# Optional: frontend dev server
+# Use `yarn dev:v2` to utilize the Custom Component v2 frontend (recommended).
+# Use `yarn dev:v1` to utilize the Custom Component v1 frontend (legacy).
+yarn dev:v2
+```
+
+### 4) Run a local demo
+
+```bash
+streamlit run ./e2e_playwright/bokeh_chart_basics.py
+```
+
+### 5) Run tests
+
+Python end-to-end tests (Playwright):
+
+```bash
+# Build the package
+uv build
+# Install the test dependencies
+uv pip install -r e2e_playwright/test-requirements.txt
+# Install browsers (first time only)
+python -m playwright install --with-deps
+# Run tests
+pytest e2e_playwright -n auto
+```
+
+Frontend tests and type checks:
+
+```bash
+cd streamlit_bokeh/frontend
+yarn test
+yarn typecheck
+```
+
+### 6) Build the Python package (optional)
+
+```bash
+uv build
+ls dist/
 ```
 
 ---

scripts/update_bokeh_version.py

@@ -14,11 +14,14 @@
 
 import os
 import re
+import subprocess
+
 import requests
 import semver
-import subprocess
+import toml
 
-SETUP_PY_PATH = "setup.py"
+PYPROJECT_TOML_PATH = "pyproject.toml"
+PACKAGE_PYPROJECT_TOML_PATH = "streamlit_bokeh/pyproject.toml"
 
 
 def get_latest_bokeh_version():
@@ -31,29 +34,31 @@ def get_latest_bokeh_version():
 
 
 def get_component_version():
-    with open(SETUP_PY_PATH, "r") as f:
-        setup_content = f.read()
+    with open(PYPROJECT_TOML_PATH, "r") as f:
+        pyproject_data = toml.load(f)
 
-    # Extract version from setup.py
-    match = re.search(r"version\s*=\s*['\"]([\d\.]+)(['\"])", setup_content)
+    # Extract version from pyproject.toml
+    version = pyproject_data.get("project", {}).get("version")
 
-    if match:
-        return match.group(1)
+    if version:
+        return version
     else:
-        raise ValueError("Bokeh version not found in the file")
+        raise ValueError("Component version not found in pyproject.toml")
 
 
 def get_dependency_bokeh_version():
-    with open(SETUP_PY_PATH, "r") as f:
-        setup_content = f.read()
+    with open(PYPROJECT_TOML_PATH, "r") as f:
+        pyproject_data = toml.load(f)
 
-    # Extract Bokeh version from dependency line (e.g., bokeh==2.4.3)
-    match = re.search(r"bokeh\s*==\s*([\d\.]+)", setup_content)
+    # Extract Bokeh version from dependencies list
+    dependencies = pyproject_data.get("project", {}).get("dependencies", [])
 
-    if match:
-        return match.group(1)
-    else:
-        raise ValueError("Bokeh version not found in the file")
+    for dep in dependencies:
+        match = re.search(r"bokeh\s*==\s*([\d\.]+)", dep)
+        if match:
+            return match.group(1)
+
+    raise ValueError("Bokeh version not found in dependencies")
 
 
 def download_files(new_version, destination):
@@ -67,6 +72,9 @@ def download_files(new_version, destination):
         f"https://raw.githubusercontent.com/bokeh/bokeh/refs/tags/{new_version}/LICENSE.txt",
     ]
 
+    # Ensure destination/bokeh directory exists
+    os.makedirs(os.path.join(destination, "bokeh"), exist_ok=True)
+
     for url in files_to_download:
         filename = os.path.basename(url)
         print(f"Downloading {filename}")
@@ -77,28 +85,30 @@ def download_files(new_version, destination):
                 f.write(chunk)
 
 
-def update_setup_py(new_version, old_bokeh_version, new_bokeh_version):
-    with open(SETUP_PY_PATH, "r") as f:
-        setup_content = f.read()
+def update_pyproject_toml(new_version, old_bokeh_version, new_bokeh_version):
+    for path in (PYPROJECT_TOML_PATH, PACKAGE_PYPROJECT_TOML_PATH):
+        if not os.path.exists(path):
+            raise FileNotFoundError(f"File {path} not found")
 
-    # Replace package version in `version='...'`
-    # This pattern is naive; adapt as needed for your file structure.
-    setup_content = re.sub(
-        r"(version\s*=\s*['\"])([\d\.]+)(['\"])",
-        rf"\g<1>{new_version}\g<3>",
-        setup_content,
-    )
+        with open(path, "r") as f:
+            pyproject_data = toml.load(f)
 
-    # Replace bokeh==old_version with bokeh==new_version
-    if old_bokeh_version:
-        setup_content = re.sub(
-            rf"(bokeh\s*==\s*){old_bokeh_version}",
-            rf"\g<1>{new_bokeh_version}",
-            setup_content,
-        )
+        project = pyproject_data.get("project")
+
+        # Update project version if present
+        if project and "version" in project:
+            project["version"] = new_version
+
+            # Update bokeh dependency version if present in this file
+            if old_bokeh_version and "dependencies" in project:
+                dependencies = project["dependencies"]
+                for i, dep in enumerate(dependencies):
+                    if isinstance(dep, str) and dep.startswith("bokeh=="):
+                        dependencies[i] = f"bokeh=={new_bokeh_version}"
+                        break
 
-    with open(SETUP_PY_PATH, "w") as f:
-        f.write(setup_content)
+        with open(path, "w") as f:
+            toml.dump(pyproject_data, f)
 
 
 def update_test_requirements(
@@ -160,16 +170,56 @@ def update_init_py(old_bokeh_version, new_bokeh_version):
         f.write(init_py_contents)
 
 
-def update_index_html(public_dir, old_version, new_version):
-    index_html_path = os.path.join(public_dir, "index.html")
-    if os.path.exists(index_html_path):
-        with open(index_html_path, "r", encoding="utf-8") as f:
+def update_loader_imports(old_bokeh_version, new_bokeh_version):
+    """
+    Update versioned Bokeh asset import paths in the TypeScript loader to the new version.
+
+    This replaces occurrences like `bokeh-3.8.0.min.js` with `bokeh-<new>.min.js`
+    in `streamlit_bokeh/frontend/src/loaders.ts`.
+    """
+    loader_path = "streamlit_bokeh/frontend/src/loaders.ts"
+    with open(loader_path, "r", encoding="utf-8") as f:
+        contents = f.read()
+
+    suffixes = ["mathjax", "gl", "api", "tables", "widgets", ""]
+    for suffix in suffixes:
+        old_str = (
+            f"bokeh-{suffix}-{old_bokeh_version}.min.js"
+            if suffix
+            else f"bokeh-{old_bokeh_version}.min.js"
+        )
+        new_str = (
+            f"bokeh-{suffix}-{new_bokeh_version}.min.js"
+            if suffix
+            else f"bokeh-{new_bokeh_version}.min.js"
+        )
+        contents = contents.replace(old_str, new_str)
+
+    with open(loader_path, "w", encoding="utf-8") as f:
+        f.write(contents)
+
+
+def update_index_html(frontend_dir, old_version, new_version):
+    """
+    Update the index.html files to reference the new Bokeh asset version.
+    Only the source Vite entry point (`frontend/index.html`) needs to be updated,
+    since build artifacts are regenerated.
+    """
+
+    index_html_paths = [os.path.join(frontend_dir, "index.html")]
+
+    cdn_suffixes = ["mathjax", "gl", "api", "tables", "widgets", ""]
+    found_index = False
+
+    for path in index_html_paths:
+        if not os.path.exists(path):
+            continue
+
+        found_index = True
+        with open(path, "r", encoding="utf-8") as f:
             html_content = f.read()
 
-        # If old_version is known, do a direct replacement
         if old_version:
-            # Replace each script reference with the new version
-            cdn_suffixes = ["mathjax", "gl", "api", "tables", "widgets", ""]
             for suffix in cdn_suffixes:
                 old_str = (
                     f"bokeh-{suffix}-{old_version}.min.js"
@@ -183,10 +233,13 @@ def update_index_html(public_dir, old_version, new_version):
                 )
                 html_content = html_content.replace(old_str, new_str)
 
-        with open(index_html_path, "w", encoding="utf-8") as f:
+        with open(path, "w", encoding="utf-8") as f:
             f.write(html_content)
-    else:
-        print("No index.html found in frontend/public. Skipping HTML update.")
+
+    if not found_index:
+        print(
+            "No index.html found under streamlit_bokeh/frontend. Skipping HTML update."
+        )
 
 
 def check_remote_branch_exists(remote: str, new_version: str) -> bool:
@@ -242,21 +295,24 @@ def check_remote_branch_exists(remote: str, new_version: str) -> bool:
         exit(0)
 
     print("New version available!")
-    public_dir = "streamlit_bokeh/frontend/public"
+    assets_dir = "streamlit_bokeh/frontend/src/assets"
+    frontend_dir = "streamlit_bokeh/frontend"
 
     # Remove original files
-    bokeh_dir = os.path.join(public_dir, "bokeh")
+    bokeh_dir = os.path.join(assets_dir, "bokeh")
+    os.makedirs(bokeh_dir, exist_ok=True)
     for filename in os.listdir(bokeh_dir):
         if "bokeh" in filename and filename.endswith(".js"):
             os.remove(os.path.join(bokeh_dir, filename))
 
-    download_files(new_bokeh_version, public_dir)
-    # Update the bokeh dependency version in index.html and __init__.py
-    update_index_html(public_dir, old_bokeh_version, new_bokeh_version)
+    download_files(new_bokeh_version, assets_dir)
+    # Update the bokeh dependency version in index.html, TS loader and __init__.py
+    update_index_html(frontend_dir, old_bokeh_version, new_bokeh_version)
+    update_loader_imports(old_bokeh_version, new_bokeh_version)
     update_init_py(old_bokeh_version, new_bokeh_version)
 
-    # Update the bokeh dependency version and component version in setup.py and test-requirements.txt
-    update_setup_py(new_version, old_bokeh_version, new_bokeh_version)
+    # Update the bokeh dependency version in pyproject.toml and test-requirements.txt
+    update_pyproject_toml(new_version, old_bokeh_version, new_bokeh_version)
     update_test_requirements(
         old_bokeh_version, new_bokeh_version, old_version, new_version
     )