Upload folder using huggingface_hub

server/app.py

@@ -1,8 +1,8 @@
 """FastAPI application for the OpenCode Environment.
 
-The Gradio web UI is mounted at root (``/``). OpenEnv's standard API
-endpoints (``/health``, ``/reset``, ``/step``, ``/mcp``) are registered
-first and take precedence over Gradio routes.
+Mirrors the ``e2b_desktop`` pattern: pass a ``gradio_builder`` to
+``create_app`` and let OpenEnv handle the Gradio mount (including the
+HF-Space-compatible ``/web`` path). No manual ``gr.mount_gradio_app``.
 
 Usage::
 
@@ -20,40 +20,44 @@ from __future__ import annotations
 
 import os
 
-import gradio as gr
-
 try:
-    from openenv.core.env_server.gradio_theme import (
-        OPENENV_GRADIO_CSS,
-        OPENENV_GRADIO_THEME,
-    )
     from openenv.core.env_server.http_server import create_app
     from openenv.core.env_server.mcp_types import (
         CallToolAction,
         CallToolObservation,
     )
-    from openenv.core.env_server.web_interface import WebInterfaceManager
 
     from .opencode_environment import OpenCodeEnvironment
     from .gradio_ui import opencode_ui_builder
 except ImportError:
-    from openenv.core.env_server.gradio_theme import (
-        OPENENV_GRADIO_CSS,
-        OPENENV_GRADIO_THEME,
-    )
     from openenv.core.env_server.http_server import create_app
     from openenv.core.env_server.mcp_types import (
         CallToolAction,
         CallToolObservation,
     )
-    from openenv.core.env_server.web_interface import WebInterfaceManager
 
     from server.opencode_environment import OpenCodeEnvironment
     from server.gradio_ui import opencode_ui_builder
 
-# Build the HTTP API server with MCP routing. We mount our own Gradio UI
-# below, so disable the built-in web interface from create_app.
-os.environ["ENABLE_WEB_INTERFACE"] = "false"
+
+def _custom_gradio_builder(
+    web_manager,
+    action_fields,
+    metadata,
+    is_chat_env,
+    title,
+    quick_start_md,
+):
+    """Callback invoked by ``create_app`` to build our custom Gradio UI.
+
+    The ``web_manager`` lets the UI drive rollouts through the same
+    ``CallToolAction`` pipeline as MCP clients; we ignore the other hints.
+    """
+    return opencode_ui_builder(web_manager=web_manager)
+
+
+# Enable OpenEnv's built-in Gradio mounting at the standard /web path.
+os.environ.setdefault("ENABLE_WEB_INTERFACE", "true")
 
 app = create_app(
     OpenCodeEnvironment,
@@ -61,24 +65,7 @@ app = create_app(
     CallToolObservation,
     env_name="opencode_env",
     max_concurrent_envs=int(os.getenv("MAX_CONCURRENT_ENVS", "4")),
-)
-
-_web_manager = WebInterfaceManager(
-    OpenCodeEnvironment,
-    CallToolAction,
-    CallToolObservation,
-)
-
-_demo = opencode_ui_builder(
-    web_manager=_web_manager,
-)
-
-app = gr.mount_gradio_app(
-    app,
-    _demo,
-    path="/web",
-    theme=OPENENV_GRADIO_THEME,
-    css=OPENENV_GRADIO_CSS,
+    gradio_builder=_custom_gradio_builder,
 )