Add arg for max websocket message size

README.md

@@ -8,4 +8,27 @@ app_port: 7860
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Viser + Gradio
+
+Demo for integrating [viser](https://github.com/nerfstudio-project/viser) 3D
+visualizations into a [Gradio](https://www.gradio.app/) application.
+
+- Uses Gradio's session management to create isolated 3D visualization contexts.
+- Exposes both Gradio and Viser over the same port.
+
+## Deploying on HuggingFace Spaces
+
+**[ [Live example](https://brentyi-viser-gradio-embed.hf.space/) ]**
+
+This repository should work out-of-the-box with HF Spaces via Docker.
+
+- Unlike a vanilla Gradio Space, this is unfortunately not supported by [ZeroGPU](https://huggingface.co/docs/hub/en/spaces-zerogpu).
+
+## Local Demo
+
+```bash
+pip install -r requirements.txt
+python app.py
+```
+
+https://github.com/user-attachments/assets/b94a117a-b9e5-4854-805a-8666941c7816

viser_proxy_manager.py

@@ -19,6 +19,7 @@ class ViserProxyManager:
             These ports are used only for internal communication and don't need to be publicly exposed.
         max_local_port: Maximum local port number to use for Viser servers. Defaults to 9000.
             These ports are used only for internal communication and don't need to be publicly exposed.
+        max_websocket_message_size_bytes: Maximum size of WebSocket messages in bytes. Defaults to 32 MB.
     """
 
     def __init__(
@@ -26,11 +27,13 @@ class ViserProxyManager:
         app: FastAPI,
         min_local_port: int = 8000,
         max_local_port: int = 9000,
+        max_websocket_message_size_bytes: int = 32_000_000,
     ) -> None:
         self._min_port = min_local_port
         self._max_port = max_local_port
         self._server_from_session_hash: dict[str, viser.ViserServer] = {}
         self._last_port = self._min_port - 1  # Track last port tried
+        self._max_websocket_message_size_bytes = max_websocket_message_size_bytes
 
         @app.get("/viser/{server_id}/{proxy_path:path}")
         async def proxy(request: Request, server_id: str, proxy_path: str):
@@ -96,7 +99,9 @@ class ViserProxyManager:
 
             try:
                 # Connect to the target WebSocket
-                async with websockets.connect(target_ws_url) as ws_target:
+                async with websockets.connect(
+                    target_ws_url, max_size=self._max_websocket_message_size_bytes
+                ) as ws_target:
                     # Create tasks for bidirectional communication
                     async def forward_to_target():
                         """Forward messages from the client to the target WebSocket."""