Update modules/ui.py

modules/ui.py

@@ -6,11 +6,47 @@ from .database import get_student_data, store_analysis_result
 from .morpho_analysis import get_repeated_words_colors, highlight_repeated_words, POS_COLORS, POS_TRANSLATIONS
 from .syntax_analysis import visualize_syntax
 
+##########################################################################
+def login_page():
+    st.title("Iniciar Sesión")
+    username = st.text_input("Usuario")
+    password = st.text_input("Contraseña", type='password')
+    if st.button("Iniciar Sesión"):
+        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")
+
+##########################################################################
+def register_page():
+    st.title("Registrarse")
+    new_username = st.text_input("Nuevo Usuario")
+    new_password = st.text_input("Nueva Contraseña", type='password')
+    role = st.selectbox("Rol", ["Estudiante", "Profesor"])
+    
+    additional_info = {}
+    if role == "Estudiante":
+        additional_info['carrera'] = st.text_input("Carrera")
+    elif role == "Profesor":
+        additional_info['departamento'] = st.text_input("Departamento")
+
+    if st.button("Registrarse"):
+        if register_user(new_username, new_password, role, additional_info):
+            st.success("Registro exitoso. Por favor, inicia sesión.")
+        else:
+            st.error("El usuario ya existe o ocurrió un error durante el registro")
+
+##########################################################################
 def get_chatbot_response(input_text):
     # Esta función debe ser implementada o importada de otro módulo
     # Por ahora, retornamos un mensaje genérico
     return "Lo siento, el chatbot no está disponible en este momento."
 
+##########################################################################
 def display_chat_interface():
     st.markdown("### Chat con AIdeaText")
 
@@ -32,6 +68,7 @@ def display_chat_interface():
             st.session_state.chat_history.append(("bot", response))
             st.experimental_rerun()
 
+##########################################################################
 def display_student_progress(student_data):
     st.success("Datos obtenidos exitosamente")
     
@@ -45,6 +82,7 @@ def display_student_progress(student_data):
     for entry in student_data['entries'][:5]:
         st.text_area(f"Entrada del {entry['timestamp']}", entry['text'], height=100)
 
+##########################################################################
 def display_text_analysis_interface(nlp_models, lang_code):
     translations = {
         'es': {
@@ -135,6 +173,7 @@ def display_text_analysis_interface(nlp_models, lang_code):
                 st.error("Hubo un problema al guardar el análisis. Por favor, inténtelo de nuevo.")
                 st.error(f"Falló el guardado del análisis. Username: {st.session_state.username}")
 
+##########################################################################
 def display_teacher_interface():
     st.write("Bienvenido, profesor. Aquí podrás ver el progreso de tus estudiantes.")
     # Aquí puedes agregar la lógica para mostrar el progreso de los estudiantes