add support for root function in ui.run_with

Author: falkoschindler

nicegui/ui_run.py

@@ -83,7 +83,7 @@ def run(root: Optional[Callable] = None, *,
     You can call `ui.run()` with optional arguments.
     Most of them only apply after stopping and fully restarting the app and do not apply with auto-reloading.
 
-    :param root: root page function
+    :param root: root page function (*added in version 3.0.0*)
     :param host: start server with this host (defaults to `'127.0.0.1` in native mode, otherwise `'0.0.0.0'`)
     :param port: use this port (default: 8080 in normal mode, and an automatically determined open port in native mode)
     :param title: page title (default: `'NiceGUI'`, can be overwritten per page)

nicegui/ui_run_with.py

@@ -1,6 +1,6 @@
 from contextlib import asynccontextmanager
 from pathlib import Path
-from typing import Literal, Optional, Union
+from typing import Callable, Literal, Optional, Union
 
 from fastapi import FastAPI
 from fastapi.middleware.gzip import GZipMiddleware
@@ -14,6 +14,7 @@
 
 def run_with(
     app: FastAPI, *,
+    root: Optional[Callable] = None,
     title: str = 'NiceGUI',
     viewport: str = 'width=device-width, initial-scale=1',
     favicon: Optional[Union[str, Path]] = None,
@@ -33,6 +34,7 @@ def run_with(
     """Run NiceGUI with FastAPI.
 
     :param app: FastAPI app
+    :param root: root page function (*added in version 3.0.0*)
     :param title: page title (default: `'NiceGUI'`, can be overwritten per page)
     :param viewport: page meta viewport content (default: `'width=device-width, initial-scale=1'`, can be overwritten per page)
     :param favicon: relative filepath, absolute URL to a favicon (default: `None`, NiceGUI icon will be used) or emoji (e.g. `'🚀'`, works for most browsers)
@@ -64,6 +66,7 @@ def run_with(
         show_welcome_message=show_welcome_message,
         cache_control_directives=cache_control_directives,
     )
+    core.root = root
     storage.set_storage_secret(storage_secret)
     core.app.add_middleware(GZipMiddleware)
     core.app.add_middleware(RedirectWithPrefixMiddleware)