Update modules/ui/router.py

modules/ui/router.py

@@ -25,28 +25,33 @@ def create_router():
     with gr.Blocks(css="footer {display: none}") as app:
         # Contenedores para cada vista
         with gr.Group(visible=True) as landing_container:
-            landing_view = create_landing_view()
+            landing_view, start_button = create_landing_view()
         
         with gr.Group(visible=False) as login_container:
-            login_view = create_login_view()
+            login_view, login_button = create_login_view()
             
         with gr.Group(visible=False) as dashboard_container:
-            dashboard_view = create_dashboard_view()
+            dashboard_view, logout_button = create_dashboard_view()
 
         # Configurar navegación
         def handle_navigation(view_name, **kwargs):
             return router.switch_view(view_name, **kwargs)
 
         # Conectar eventos de navegación
-        landing_view.submit(
+        start_button.click(
             fn=lambda: handle_navigation("login"),
             outputs=[landing_container, login_container, dashboard_container]
         )
 
-        login_view.submit(
+        login_button.click(
             fn=lambda username, role: handle_navigation("dashboard", username=username, role=role),
             inputs=[login_view.username, login_view.role],
             outputs=[landing_container, login_container, dashboard_container]
         )
 
-    return app
+        logout_button.click(
+            fn=lambda: handle_navigation("landing"),
+            outputs=[landing_container, login_container, dashboard_container]
+        )
+
+    return app