Update modules/ui/views/login_ui.py

modules/ui/views/login_ui.py

@@ -1,37 +1,43 @@
-#modules/ui/login_ui.py
-
+# modules/ui/views/login.py
 import gradio as gr
+from ...auth.auth import authenticate_user
+
+class LoginView:
+    def __init__(self):
+        with gr.Blocks() as self.view:
+            with gr.Column():
+                gr.Markdown("# Iniciar Sesión")
+                self.username = gr.Textbox(label="Usuario")
+                self.password = gr.Textbox(
+                    label="Contraseña",
+                    type="password"
+                )
+                self.role = gr.State()
+                self.message = gr.Markdown()
+                
+                with gr.Row():
+                    self.back_button = gr.Button("Volver")
+                    self.login_button = gr.Button(
+                        "Iniciar Sesión",
+                        variant="primary"
+                    )
+
+            def handle_login(username, password):
+                success, role = authenticate_user(username, password)
+                if success:
+                    self.role.value = role
+                    return username, role, ""
+                return None, None, "Credenciales incorrectas"
+
+            self.login_button.click(
+                fn=handle_login,
+                inputs=[self.username, self.password],
+                outputs=[self.username, self.role, self.message]
+            )
+
+    @property
+    def submit(self):
+        return self.login_button.click
 
-def create_login_interface():
-    """
-    Interfaz de inicio de sesión.
-    """
-    with gr.Blocks() as login_interface:
-        gr.Markdown("# Iniciar Sesión")
-        
-        with gr.Row():
-            username = gr.Textbox(label="Usuario")
-        with gr.Row():
-            password = gr.Textbox(label="Contraseña", type="password")
-            
-        with gr.Row():
-            back_button = gr.Button("Volver")
-            login_button = gr.Button("Iniciar Sesión", variant="primary")
-            
-        message = gr.Markdown()
-        
-    def render(navigate_back, handle_login):
-        back_button.click(
-            fn=navigate_back,
-            inputs=[],
-            outputs=[]
-        )
-        
-        login_button.click(
-            fn=handle_login,
-            inputs=[username, password],
-            outputs=[message]
-        )
-    
-    login_interface.render = render
-    return login_interface
+def create_login_view():
+    return LoginView().view