Clean up dynamic routes

flet_extensions.py

@@ -4,8 +4,11 @@ from js_execution import js_execution_router
 def extend_flet_page():
     async def eval_js(self: ft.Page, js_code: str) -> str:
         endpoint, result_future = js_execution_router.create_dynamic_endpoint(js_code)
-        self.launch_url(endpoint, web_window_name="hidden-iframe")
-        return await result_future
+        try:
+            self.launch_url(endpoint, web_window_name="hidden-iframe")
+            return await result_future
+        finally:
+            js_execution_router.cleanup_route(endpoint.split('/')[-1])
 
     if not hasattr(ft.Page, 'eval_js'):
         ft.Page.eval_js = eval_js

js_execution.py

@@ -9,6 +9,7 @@ logger = logging.getLogger(__name__)
 class JSExecutionRouter:
     def __init__(self):
         self.app = None
+        self.active_routes = set()
 
     def init_app(self, app: FastAPI):
         self.app = app
@@ -67,8 +68,14 @@ class JSExecutionRouter:
                 result_future.set_result(data['result'])
             return {"status": "received"}
 
+        self.active_routes.add(result_id)
         return f"/eval-js/{result_id}", result_future
 
+    def cleanup_route(self, result_id: str):
+        if result_id in self.active_routes:
+            self.app.router.routes = [route for route in self.app.router.routes if route.path not in [f"/eval-js/{result_id}", f"/js-result/{result_id}"]]
+            self.active_routes.remove(result_id)
+
 js_execution_router = JSExecutionRouter()
 
 def add_js_execution_routes(app: FastAPI):
@@ -115,7 +122,6 @@ def mount_flet_app(app: FastAPI, route: str, flet_app):
                     var src = currentProtocol + '//' + currentHost + '/flet-app-internal/';
 
                     iframe.src = src;
-                    console.log('Iframe src set to:', src);
 
                     // Set Content-Security-Policy based on the current protocol
                     var meta = document.createElement('meta');