Update modules/ui.py

modules/ui.py

@@ -10,8 +10,7 @@ from datetime import datetime
 import base64
 from spacy import displacy
 import re
-from .morpho_analysis import POS_TRANSLATIONS  # Asegúrate de que esta importación esté presente
-from .morpho_analysis import POS_COLORS  # Asegúrate de que esta importación esté presente
+from .morpho_analysis import POR_COLORS, POS_TRANSLATIONS  # Asegúrate de que esta importación esté presente
 
 # Importaciones locales
 from .auth import authenticate_user, register_user, get_user_role
@@ -79,7 +78,7 @@ def display_chat_interface():
 
 ##########################################################################
 
-def display_student_progress(username):
+def display_student_progress(username, lang_code='es'):  # Añade el parámetro lang_code
     student_data = get_student_data(username)
     
     if student_data is None:
@@ -101,7 +100,7 @@ def display_student_progress(username):
             st.subheader("Conteo Total de Palabras por Categoría Gramatical")
             
             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)
+            df['label'] = df.apply(lambda x: f"{POS_TRANSLATIONS[lang_code].get(x['category'], 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']]
@@ -112,32 +111,8 @@ def display_student_progress(username):
         else:
             st.info("No hay datos de conteo de palabras disponibles.")
         
-        # Diagramas de Arco (consolidados)
-        st.header("Diagramas de Arco")
-        with st.expander("Ver todos los Diagramas de Arco"):
-            for i, entry in enumerate(student_data['entries']):
-                if 'arc_diagrams' in entry and entry['arc_diagrams']:
-                    st.subheader(f"Entrada {i+1} - {entry['timestamp']}")
-                    st.write(entry['arc_diagrams'][0], unsafe_allow_html=True)
-        
-        # Diagramas de Red (consolidados)
-        st.header("Diagramas de Red")
-        with st.expander("Ver todos los Diagramas de Red"):
-            for i, entry in enumerate(student_data['entries']):
-                if 'network_diagram' in entry and entry['network_diagram']:
-                    st.subheader(f"Entrada {i+1} - {entry['timestamp']}")
-                    try:
-                        # Decodificar la imagen base64
-                        image_bytes = base64.b64decode(entry['network_diagram'])
-                        st.image(image_bytes)
-                    except Exception as e:
-                        st.error(f"Error al mostrar el diagrama de red: {str(e)}")
-        
-        # Mostrar entradas recientes
-        #st.header("Entradas Recientes")
-        #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'])
+        # ... (resto del código para diagramas de arco, diagramas de red y entradas recientes)
+
     else:
         st.warning("No se encontraron entradas para este estudiante.")
         st.info("Intenta realizar algunos análisis de texto primero.")