Update modules/ui.py

modules/ui.py

@@ -20,6 +20,23 @@ 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")
@@ -78,7 +95,7 @@ def login_register_page():
 
 ##################################################################################################
 def login_form():
-    username = st.text_input("Usuario")
+    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?")
     
@@ -96,22 +113,27 @@ def login_form():
             st.error("Captcha incorrecto")
 
 ##################################################################################################
-def register_form():
-    new_username = st.text_input("Nuevo Usuario")
-    new_password = st.text_input("Nueva Contraseña", type='password')
-    carrera = st.text_input("Carrera")
-    captcha_answer = st.text_input("Captcha: ¿Cuánto es 3 + 4?")
+# En ui.py
+def admin_page():
+    st.title("Panel de Administración")
     
-    if st.button("Registrarse"):
-        if captcha_answer == "7":
-            additional_info = {'carrera': carrera}
-            if register_user(new_username, new_password, 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")
+    # Crear nuevo usuario
+    st.header("Crear Nuevo Usuario")
+    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"):
+        if create_invited_user(new_username, new_password):
+            st.success(f"Usuario {new_username} creado exitosamente")
         else:
-            st.error("Captcha incorrecto")
+            st.error("Error al crear el usuario")
 
+    # Aquí puedes añadir más funcionalidades CRUD en el futuro
+
+##################################################################################################
+def register_form():
+    ##
+    ##
+    pass
 ##################################################################################################
 def display_chat_interface():
     st.markdown("### Chat con AIdeaText")
@@ -459,4 +481,5 @@ def display_chatbot_interface(lang_code):
         store_chat_history(st.session_state.username, st.session_state.messages)
 
     # Scroll al final del chat
-    st.markdown('<script>window.scrollTo(0,document.body.scrollHeight);</script>', unsafe_allow_html=True)
+    st.markdown('<script>window.scrollTo(0,document.body.scrollHeight);</script>', unsafe_allow_html=True)
+