Ensure JS runs on Gradio 6

app.py

@@ -661,7 +661,7 @@ CSS = """
 }
 """
 
-JS = """
+JS_FUNC = """
 () => {
   if (window.__rosaDemoInit) return;
   window.__rosaDemoInit = true;
@@ -1450,11 +1450,13 @@ JS = """
 }
 """
 
+JS_BOOT = f"(function(){{ const init = {JS_FUNC}; init(); return init; }})()"
+
 
 if GRADIO_MAJOR >= 6:
     demo_context = gr.Blocks()
 else:
-    demo_context = gr.Blocks(css=CSS, js=JS)
+    demo_context = gr.Blocks(css=CSS, js=JS_BOOT)
 
 with demo_context as demo:
     gr.HTML(
@@ -1501,5 +1503,5 @@ with demo_context as demo:
 if __name__ == "__main__":
     launch_kwargs = {}
     if GRADIO_MAJOR >= 6:
-        launch_kwargs = {"css": CSS, "js": JS, "ssr_mode": False}
+        launch_kwargs = {"css": CSS, "js": JS_BOOT, "ssr_mode": False}
     demo.launch(**launch_kwargs)