return the X-GOOEY-GUI-ROUTE header from gui.route

GooeyAI · Jul 26, 2024 · eb93917 · eb93917
1 parent d146dca
commit eb93917
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/gooey_ui/state.py b/gooey_ui/state.py
@@ -6,9 +6,10 @@
 from functools import wraps, partial
 
 from fastapi import Depends
+from fastapi.encoders import jsonable_encoder
 from pydantic import BaseModel
 from starlette.requests import Request
-from starlette.responses import Response, RedirectResponse
+from starlette.responses import Response, RedirectResponse, JSONResponse
 
 from gooey_ui.pubsub import (
     get_subscriptions,
@@ -148,11 +149,16 @@ def runner(
                 return RedirectResponse(e.url, status_code=e.status_code)
             if isinstance(ret, Response):
                 return ret
-            return dict(
-                children=root.children,
-                state=get_session_state(),
-                channels=get_subscriptions(),
-                **(ret or {}),
+            return JSONResponse(
+                jsonable_encoder(
+                    dict(
+                        children=root.children,
+                        state=get_session_state(),
+                        channels=get_subscriptions(),
+                        **(ret or {}),
+                    )
+                ),
+                headers={"X-GOOEY-GUI-ROUTE": "1"},
             )
         except RerunException:
             continue