Upload folder using huggingface_hub

server/app.py

@@ -55,10 +55,11 @@ def _custom_gradio_builder(
 ):
     """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.
+    We ignore ``web_manager`` (its public API is ``reset_environment`` /
+    ``step_environment`` / ``connect_websocket`` — not an env instance) and
+    hand the UI the env class directly, matching e2b_desktop's pattern.
     """
-    return opencode_ui_builder(web_manager=web_manager)
+    return opencode_ui_builder(env_factory=OpenCodeEnvironment)
 
 
 # Enable OpenEnv's built-in Gradio mounting at the standard /web path.

server/gradio_ui.py

@@ -58,16 +58,29 @@ _EXAMPLE_VLLM_URLS = [
 
 def opencode_ui_builder(
     *,
-    web_manager: Any,
+    env_factory: Any = None,
+    web_manager: Any = None,   # kept for backward compat; unused
     title: str = "OpenCode Env",
     **_: Any,
 ) -> gr.Blocks:
-    """Build the Gradio Blocks UI bound to ``web_manager``.
+    """Build the Gradio Blocks UI.
 
-    The web manager is a thin wrapper around ``OpenCodeEnvironment`` —
-    calling ``call_tool("run_rollout", ...)`` on it drives one rollout.
+    ``env_factory`` is a zero-arg callable that returns a fresh
+    :class:`OpenCodeEnvironment` on first click (lazy, so the Space's
+    cold-start path doesn't pay the E2B cost until someone hits Run).
     """
 
+    _env_cache: dict[str, Any] = {"instance": None}
+
+    def _get_env():
+        inst = _env_cache.get("instance")
+        if inst is None:
+            if env_factory is None:
+                raise RuntimeError("opencode_ui_builder needs env_factory=...")
+            inst = env_factory()
+            _env_cache["instance"] = inst
+        return inst
+
     with gr.Blocks(title=title, analytics_enabled=False) as demo:
         gr.Markdown(f"# {title}\nRun one OpenCode rollout against any OpenAI-compatible endpoint.")
 
@@ -168,25 +181,32 @@ def opencode_ui_builder(
             setup_shell_v: str,
         ):
             try:
-                env = web_manager.get_environment()
+                from openenv.core.env_server.mcp_types import CallToolAction
+
+                env = _get_env()
                 env.reset()
-                result_raw = env.call_tool(
-                    "run_rollout",
-                    vllm_url=vllm_url_v,
-                    model=model_v,
-                    instruction=instruction_v,
-                    test_script=test_script_v,
-                    task_id=task_id_v,
-                    setup_shell=setup_shell_v,
-                    upload_files={},
-                    provider=provider_v,
-                    api_key=api_key_v,
-                    mode=mode_v,
-                    disable_thinking=bool(disable_thinking_v),
-                    max_tokens_cap=int(max_tokens_cap_v),
-                    agent_timeout_s=float(agent_timeout_s_v),
+                obs = env.step(
+                    CallToolAction(
+                        tool_name="run_rollout",
+                        arguments={
+                            "vllm_url": vllm_url_v,
+                            "model": model_v,
+                            "instruction": instruction_v,
+                            "test_script": test_script_v,
+                            "task_id": task_id_v,
+                            "setup_shell": setup_shell_v,
+                            "upload_files": {},
+                            "provider": provider_v,
+                            "api_key": api_key_v,
+                            "mode": mode_v,
+                            "disable_thinking": bool(disable_thinking_v),
+                            "max_tokens_cap": int(max_tokens_cap_v),
+                            "agent_timeout_s": float(agent_timeout_s_v),
+                        },
+                    ),
+                    timeout_s=float(agent_timeout_s_v) + 300,
                 )
-                result = _parse_result(result_raw)
+                result = _parse_result(obs)
             except Exception as exc:
                 msg = f"**Error:** `{type(exc).__name__}: {exc}`"
                 return (msg, None, None, None, None, "", [], "", "", {"error": str(exc)})