Fix Space cockpit JavaScript loading

app.py

@@ -77,8 +77,8 @@ async def blank_redirect():
     return RedirectResponse(url="/")
 
 
-head = getattr(demo, "time_machine_head", None)
-fastapi_app_with_gradio = gr.mount_gradio_app(fastapi_app, demo, path="/app", head=head)
+runtime_js = getattr(demo, "time_machine_js", None)
+fastapi_app_with_gradio = gr.mount_gradio_app(fastapi_app, demo, path="/app", js=runtime_js)
 app = fastapi_app_with_gradio if os.getenv("TIME_MACHINE_FASTAPI_WRAPPER") else demo
 
 
@@ -90,4 +90,4 @@ if __name__ == "__main__":
     if os.getenv("TIME_MACHINE_FASTAPI_WRAPPER"):
         uvicorn.run(fastapi_app_with_gradio, host=host, port=port)
     else:
-        demo.launch(server_name=host, server_port=port)
+        demo.launch(server_name=host, server_port=port, js=runtime_js)

src/time_machine/ui/gradio_app.py

@@ -2,7 +2,6 @@ from __future__ import annotations
 
 from pathlib import Path
 import base64
-import inspect
 import re
 
 import gradio as gr
@@ -30,14 +29,11 @@ def create_app(container: AppContainer) -> gr.Blocks:
 
     cockpit_html = _route_static_urls((ASSET_DIR / "cockpit.html").read_text(encoding="utf-8"))
     intro_html = _route_static_urls((ASSET_DIR / "intro.html").read_text(encoding="utf-8"))
-    intro_js = (ASSET_DIR / "intro.js").read_text(encoding="utf-8")
-    cockpit_js = (ASSET_DIR / "cockpit.js").read_text(encoding="utf-8")
-
-    blocks_kwargs = {"title": "Trans-Temporal Express"}
-    if "head" in inspect.signature(gr.Blocks).parameters:
-        blocks_kwargs["head"] = f"<script>{cockpit_js}</script><script>{intro_js}</script>"
-
-    with gr.Blocks(**blocks_kwargs) as demo:
+    intro_js = (ASSET_DIR / "intro.js").read_text(encoding="utf-8")
+    cockpit_js = (ASSET_DIR / "cockpit.js").read_text(encoding="utf-8")
+    runtime_js = _as_onload_js(f"{cockpit_js}\n{intro_js}")
+
+    with gr.Blocks(title="Trans-Temporal Express") as demo:
         session_state = gr.State(handlers.initial_state())
 
         gr.HTML(f"<style>{css}</style>{intro_html}{cockpit_html}")
@@ -155,8 +151,8 @@ def create_app(container: AppContainer) -> gr.Blocks:
             outputs=[status],
         )
 
-    if "head" not in blocks_kwargs:
-        demo.time_machine_head = f"<script>{cockpit_js}</script><script>{intro_js}</script>"
+    demo.js = runtime_js
+    demo.time_machine_js = runtime_js
     return demo.queue()
 
 
@@ -167,3 +163,7 @@ def _route_static_urls(text: str) -> str:
 def _replace_static_url(match: re.Match[str]) -> str:
     relative_path = match.group("path")
     return f"{match.group('quote')}/gradio_api/file=static/{relative_path}{match.group('quote')}"
+
+
+def _as_onload_js(script: str) -> str:
+    return f"() => {{\n{script}\n}}"