Update modules/ui.py

modules/ui.py

@@ -20,14 +20,35 @@ 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 initialize_session_state():
+    if 'initialized' not in st.session_state:
+        st.session_state.clear()
+        st.session_state.initialized = True
+        st.session_state.logged_in = False
+        st.session_state.page = 'login'
+        st.session_state.username = None
+        st.session_state.role = None
+
+##################################################################################################
+def main():
+    initialize_session_state()
+    
+    if st.session_state.page == 'login':
+        login_register_page()
+    elif st.session_state.page == 'admin':
+        admin_page()
+    elif st.session_state.page == 'user':
+        user_page()
+    
+    print(f"Estado actual de la sesión: {st.session_state}")  # Log para depuración
+
 ##################################################################################################
 def login_register_page():
     st.title("AIdeaText")
 
-    # Dividir la pantalla en dos columnas
-    left_column, right_column = st.columns([1, 3])  # 25% izquierda, 75% derecha
+    left_column, right_column = st.columns([1, 3])
 
-    # Sección izquierda para login y registro
     with left_column:
         tab1, tab2 = st.tabs(["Iniciar Sesión", "Registrarse"])
         
@@ -37,123 +58,96 @@ def login_register_page():
         with tab2:
             register_form()
 
-    # Sección derecha para videos de YouTube
     with right_column:
-        st.header("Videos: pitch, demos, entrevistas, otros")
-        
-        # Diccionario de videos de YouTube con títulos amigables
-        videos = {
-            "Intro AideaText": "https://www.youtube.com/watch?v=UA-md1VxaRc",
-            "Pitch que facilitó acceder a la segunda fase de IFE Explora del TEC de Monterrey": "https://www.youtube.com/watch?v=Fqi4Di_Rj_s",
-            "Entrevista con el doctor Guillermo Ruíz, EduMate Lima Perú" : "https://www.youtube.com/watch?v=_ch8cRja3oc",
-            "Demo versión desktop de AIdeaText": "https://www.youtube.com/watch?v=nP6eXbog-ZY"
-        }
-        
-        # Selector de video
-        video_titles = list(videos.keys())
-        if video_titles:
-            selected_title = st.selectbox("Selecciona un video tutorial:", video_titles)
-            
-            # Obtener la URL correspondiente al título seleccionado
-            if selected_title in videos:
-                selected_video = videos[selected_title]
-                
-                # Reproductor de YouTube
-                try:
-                    st_player(selected_video)
-                except Exception as e:
-                    st.error(f"Error al cargar el video: {str(e)}")
-            else:
-                st.warning("El video seleccionado no está disponible.")
-        else:
-            st.warning("No hay videos disponibles para mostrar.")
-
-        # Información adicional
-        st.markdown("""
-        ## Novedades de la versión actual
-        - Nueva función de análisis semántico
-        - Soporte para múltiples idiomas
-        - Interfaz mejorada para una mejor experiencia de usuario
-        """)
-
-##################################################################################################
-
-#def main():
-#    if 'logged_in' not in st.session_state:
-#        st.session_state.logged_in = False
-
-#    print(f"Estado de sesión al inicio de main: {st.session_state}")  # Nuevo log
-
-#    if not st.session_state.logged_in:
-#        login_register_page()
-#    else:
-#        print(f"Usuario autenticado en main: {st.session_state.username}, Rol: {st.session_state.role}")  # Log modificado
-#       if st.session_state.role == 'Administrador':
-#            print("Intentando mostrar página de administrador")  # Nuevo log
-#            admin_page()
-#        else:
-#            print(f"Mostrando página de usuario para rol: {st.session_state.role}")  # Nuevo log
-#            user_page()
-
-#    print(f"Estado de sesión al final de main: {st.session_state}")  # Nuevo log
-
-def main():
-    if 'page' not in st.session_state:
-        st.session_state.page = 'login'
-
-    if st.session_state.page == 'login':
-        login_register_page()
-    elif st.session_state.page == 'admin':
-        admin_page()
-    elif st.session_state.page == 'user':
-        user_page()
-
+        display_videos_and_info()
 
 ##################################################################################################
 
 def login_form():
-    username = st.text_input("Correo electrónico")
-    password = st.text_input("Contraseña", type="password")
+    username = st.text_input("Correo electrónico", key="login_username")
+    password = st.text_input("Contraseña", type="password", key="login_password")
     
-    if st.button("Iniciar Sesión"):
+    if st.button("Iniciar Sesión", key="login_button"):
         success, role = authenticate_user(username, password)
         if success:
-            st.session_state.clear()  # Limpia la sesión antes de establecer nuevos valores
             st.session_state.logged_in = True
             st.session_state.username = username
             st.session_state.role = role
             st.session_state.page = 'admin' if role == 'Administrador' else 'user'
             print(f"Inicio de sesión exitoso. Usuario: {username}, Rol: {role}")
-            print(f"Estado de sesión después de login: {st.session_state}")  # Nuevo log
-            st.rerun()  # Usa st.rerun() en lugar de st.experimental_rerun()
+            print(f"Estado de sesión después de login: {st.session_state}")
+            st.rerun()
         else:
             st.error("Credenciales incorrectas")
 
-
-
-
 ##################################################################################################
 def admin_page():
     st.title("Panel de Administración")
     st.write(f"Bienvenido, {st.session_state.username}")
     
-    # Crear nuevo usuario estudiante
     st.header("Crear Nuevo Usuario Estudiante")
-    new_username = st.text_input("Correo electrónico del nuevo usuario")
-    new_password = st.text_input("Contraseña", type="password")
-    if st.button("Crear Usuario"):
+    new_username = st.text_input("Correo electrónico del nuevo usuario", key="admin_new_username")
+    new_password = st.text_input("Contraseña", type="password", key="admin_new_password")
+    if st.button("Crear Usuario", key="admin_create_user"):
         if create_student_user(new_username, new_password):
             st.success(f"Usuario estudiante {new_username} creado exitosamente")
         else:
             st.error("Error al crear el usuario estudiante")
 
+    # Aquí puedes añadir más funcionalidades para el panel de administración
+
+##################################################################################################
+def user_page():
+    st.title("Bienvenido a AIdeaText")
+    st.write(f"Hola, {st.session_state.username}")
+
+    # Aquí puedes añadir las funcionalidades para el usuario estudiante
+    # Por ejemplo:
+    tabs = st.tabs(["Análisis Morfosintáctico", "Análisis Semántico", "Análisis del Discurso", "Chat", "Mi Progreso"])
+    
+    with tabs[0]:
+        display_morphosyntax_analysis_interface(nlp_models, 'es')  # Asumiendo que 'es' es el idioma por defecto
+    with tabs[1]:
+        display_semantic_analysis_interface(nlp_models, 'es')
+    with tabs[2]:
+        display_discourse_analysis_interface(nlp_models, 'es')
+    with tabs[3]:
+        display_chatbot_interface('es')
+    with tabs[4]:
+        display_student_progress(st.session_state.username, 'es')
+
+##################################################################################################
+def display_videos_and_info():
+    st.header("Videos: pitch, demos, entrevistas, otros")
+    
+    videos = {
+        "Intro AideaText": "https://www.youtube.com/watch?v=UA-md1VxaRc",
+        "Pitch IFE Explora": "https://www.youtube.com/watch?v=Fqi4Di_Rj_s",
+        "Entrevista Dr. Guillermo Ruíz": "https://www.youtube.com/watch?v=_ch8cRja3oc",
+        "Demo versión desktop": "https://www.youtube.com/watch?v=nP6eXbog-ZY"
+    }
+    
+    selected_title = st.selectbox("Selecciona un video tutorial:", list(videos.keys()))
+    
+    if selected_title in videos:
+        try:
+            st_player(videos[selected_title])
+        except Exception as e:
+            st.error(f"Error al cargar el video: {str(e)}")
+    
+    st.markdown("""
+    ## Novedades de la versión actual
+    - Nueva función de análisis semántico
+    - Soporte para múltiples idiomas
+    - Interfaz mejorada para una mejor experiencia de usuario
+    """)
+
 ##################################################################################################
 def register_form():
     ##
     ##
     pass
 ##################################################################################################
-
 def display_chat_interface():
     st.markdown("### Chat con AIdeaText")
 
@@ -176,7 +170,6 @@ def display_chat_interface():
             st.experimental_rerun()
 
 ##################################################################################################
-
 def display_student_progress(username, lang_code='es'):
     student_data = get_student_data(username)
     
@@ -240,8 +233,11 @@ def display_student_progress(username, lang_code='es'):
         st.warning("No se encontraron entradas para este estudiante.")
         st.info("Intenta realizar algunos análisis de texto primero.")
 
-##################################################################################################
 
+
+
+
+##################################################################################################
 def display_morphosyntax_analysis_interface(nlp_models, lang_code):
     translations = {
         'es': {