Update modules/ui.py

modules/ui.py

@@ -36,16 +36,12 @@ 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")
+    additional_info['carrera'] = st.text_input("Carrera")
 
     if st.button("Registrarse"):
-        if register_user(new_username, new_password, role, additional_info):
+        if register_user(new_username, new_password, "Estudiante", 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")
@@ -94,20 +90,21 @@ def display_student_progress(username):
         st.success(f"Datos obtenidos exitosamente para {username}")
         
         # Mostrar estadísticas generales
-        st.header("Inventario de palabras")
+        st.header("Estadísticas Generales")
         st.metric("Total de entradas", student_data['entries_count'])
         
-        # Gráfico de barras para el conteo de palabras por categoría
+        # Treemap para el conteo de palabras por categoría
         if student_data['word_count']:
             st.subheader("Conteo Total de Palabras por Categoría Gramatical")
-            fig, ax = plt.subplots(figsize=(10, 6))
-            categories = list(student_data['word_count'].keys())
-            counts = list(student_data['word_count'].values())
-            ax.bar(categories, counts)
-            ax.set_xlabel('Categoría')
-            ax.set_ylabel('Conteo Total')
-            ax.set_title('Conteo de Palabras por Categoría Gramatical')
-            plt.xticks(rotation=45, ha='right')
+            
+            df = pd.DataFrame(list(student_data['word_count'].items()), columns=['category', 'count'])
+            df['label'] = df.apply(lambda x: f"{x['category']}\n({x['count']})", axis=1)
+            
+            fig, ax = plt.subplots(figsize=(12, 8), dpi=80)
+            colors = [POS_COLORS.get(cat, '#CCCCCC') for cat in df['category']]
+            squarify.plot(sizes=df['count'], label=df['label'], color=colors, alpha=0.8, ax=ax)
+            plt.title('Treemap de Categorías Gramaticales')
+            plt.axis('off')
             st.pyplot(fig)
         else:
             st.info("No hay datos de conteo de palabras disponibles.")
@@ -138,9 +135,9 @@ def display_student_progress(username):
         #for i, entry in enumerate(student_data['entries'][:5]):  # Mostrar las 5 entradas más recientes
         #    with st.expander(f"Entrada {i+1} - {entry['timestamp']}"):
         #        st.write(entry['text'])
-    #else:
-    #    st.warning("No se encontraron entradas para este estudiante.")
-    #    st.info("Intenta realizar algunos análisis de texto primero.")
+    else:
+        st.warning("No se encontraron entradas para este estudiante.")
+        st.info("Intenta realizar algunos análisis de texto primero.")
     
 ##########################################################################
 def display_text_analysis_interface(nlp_models, lang_code):