Expose server host and port in dataset viewer CLI

Signed-off-by: Ahmad Kiswani <kiswani.ahmad@gmail.com>

Author: ahmadki

docs/get-started/rollout-collection.md

@@ -122,7 +122,21 @@ Launch the rollout viewer:
 ng_viewer +jsonl_fpath=results/simple_weather_rollouts.jsonl
 ```
 
-Then visit <http://127.0.0.1:7860>
+The viewer starts on port 7860 and accepts requests from anywhere by default. Visit <http://SERVER_ADDRESS:7860> in your browser.
+
+:::{tip}
+**Configuring Network Access**
+
+By default, the viewer accepts requests from anywhere (`server_host=0.0.0.0`), making it accessible on your local network. To restrict access:
+
+```bash
+# Accept requests only from localhost (more secure)
+ng_viewer +jsonl_fpath=results/simple_weather_rollouts.jsonl +server_host=127.0.0.1
+
+# Use a custom port
+ng_viewer +jsonl_fpath=results/simple_weather_rollouts.jsonl +server_port=8080
+```
+:::
 
 The viewer shows each rollout with:
 

docs/reference/cli-commands.md

@@ -243,12 +243,25 @@ Launch a Gradio interface to view and explore dataset rollouts interactively.
 * - `jsonl_fpath`
   - str
   - Filepath to a local JSONL file to view.
+* - `server_host`
+  - str
+  - Network address where the viewer accepts requests. Set to `"0.0.0.0"` to accept requests from anywhere (default), or `"127.0.0.1"` to accept requests only from localhost. Can be set by environment variable `GRADIO_SERVER_NAME`.
+* - `server_port`
+  - int
+  - Port where the viewer accepts requests. Default: `7860`. If the specified port is unavailable, Gradio will search for the next available port. Can be set by environment variable `GRADIO_SERVER_PORT`.
 ```
 
-**Example**
+**Examples**
 
 ```bash
+# Launch viewer with default settings (accepts requests from anywhere)
 ng_viewer +jsonl_fpath=weather_rollouts.jsonl
+
+# Accept requests only from localhost (more secure)
+ng_viewer +jsonl_fpath=weather_rollouts.jsonl +server_host=127.0.0.1
+
+# Use a custom port
+ng_viewer +jsonl_fpath=weather_rollouts.jsonl +server_port=8080
 ```
 
 ---

nemo_gym/dataset_viewer.py

@@ -212,11 +212,26 @@ class JsonlDatasetViewerConfig(BaseNeMoGymCLIConfig):
     Examples:
 
     ```bash
+    # Accept requests from anywhere (default)
     ng_viewer +jsonl_fpath=weather_rollouts.jsonl
+
+    # Accept requests only from localhost
+    ng_viewer +jsonl_fpath=weather_rollouts.jsonl +server_host=127.0.0.1
+
+    # Use a custom port
+    ng_viewer +jsonl_fpath=weather_rollouts.jsonl +server_port=8080
     ```
     """
 
     jsonl_fpath: str = Field(description="Filepath to a local jsonl file to view.")
+    server_host: str = Field(
+        default="0.0.0.0",
+        description='Network address where the viewer accepts requests. Set to "0.0.0.0" to accept requests from anywhere (default), or "127.0.0.1" to accept requests only from localhost. Can be set by environment variable GRADIO_SERVER_NAME.',
+    )
+    server_port: int = Field(
+        default=7860,
+        description="Port where the viewer accepts requests. If the specified port is unavailable, Gradio will search for the next available port. Can be set by environment variable GRADIO_SERVER_PORT.",
+    )
 
 
 def get_aggregate_metrics(data: List[DatasetViewerVerifyResponse]) -> Dict[str, Any]:
@@ -281,4 +296,4 @@ def select_item(value: int):
 def main():  # pragma: no cover
     config = JsonlDatasetViewerConfig.model_validate(get_global_config_dict())
     demo = build_jsonl_dataset_viewer(config)
-    demo.launch(enable_monitoring=False)
+    demo.launch(server_name=config.server_host, server_port=config.server_port, enable_monitoring=False)