Update modules/ui.py

modules/ui.py

@@ -20,23 +20,6 @@ from .semantic_analysis import visualize_semantic_relations, perform_semantic_an
 from .discourse_analysis import compare_semantic_analysis, perform_discourse_analysis
 from .chatbot import initialize_chatbot, get_chatbot_response
 
-
-##################################################################################################
-def main():
-    if 'logged_in' not in st.session_state:
-        st.session_state.logged_in = False
-
-    if not st.session_state.logged_in:
-        login_register_page()
-    else:
-        if st.session_state.role == 'admin':
-            admin_page()
-        else:
-            user_page()  # Esta sería la página normal para estudiantes
-            
-if __name__ == "__main__":
-    main()
-
 ##################################################################################################
 def login_register_page():
     st.title("AIdeaText")
@@ -96,24 +79,36 @@ def login_register_page():
 ##################################################################################################
 def login_form():
     username = st.text_input("Correo electrónico")
-    password = st.text_input("Contraseña", type='password')
-    captcha_answer = st.text_input("Captcha: ¿Cuánto es 2 + 3?")
+    password = st.text_input("Contraseña", type="password")
     
     if st.button("Iniciar Sesión"):
-        if captcha_answer == "5":
-            if authenticate_user(username, password):
-                st.success(f"Bienvenido, {username}!")
-                st.session_state.logged_in = True
-                st.session_state.username = username
-                st.session_state.role = get_user_role(username)
-                st.experimental_rerun()
-            else:
-                st.error("Usuario o contraseña incorrectos")
+        success, role = authenticate_user(username, password)
+        if success:
+            st.session_state.logged_in = True
+            st.session_state.username = username
+            st.session_state.role = role
+            st.experimental_rerun()
+        else:
+            st.error("Credenciales incorrectas")
+
+##################################################################################################
+def main():
+    if 'logged_in' not in st.session_state:
+        st.session_state.logged_in = False
+
+    if not st.session_state.logged_in:
+        login_register_page()
+    else:
+        if st.session_state.role == 'admin':
+            admin_page()
         else:
-            st.error("Captcha incorrecto")
+            user_page()  # Esta sería la página normal para estudiantes
+            
+if __name__ == "__main__":
+    main()
+
 
 ##################################################################################################
-# En ui.py
 def admin_page():
     st.title("Panel de Administración")