Update modules/ui.py

modules/ui.py

@@ -3,6 +3,7 @@
 import streamlit as st
 from spacy import displacy
 import re
+from .morpho_analysis import POS_TRANSLATIONS  # Asegúrate de que esta importación esté presente
 import matplotlib.pyplot as plt
 import matplotlib
 matplotlib.use('Agg')
@@ -94,7 +95,11 @@ def display_student_progress(username):
             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)
+            
+            # Traducir las categorías
+            translated_categories = [POS_TRANSLATIONS.get(cat, cat) for cat in categories]
+            
+            ax.bar(translated_categories, counts)
             ax.set_xlabel('Categoría')
             ax.set_ylabel('Conteo Total')
             ax.set_title('Conteo de Palabras por Categoría Gramatical')
@@ -115,7 +120,7 @@ def display_student_progress(username):
         if evolution_data:
             fig, ax = plt.subplots(figsize=(10, 6))
             for category, counts in evolution_data.items():
-                ax.plot(range(1, len(counts) + 1), counts, label=category)
+                ax.plot(range(1, len(counts) + 1), counts, label=POS_TRANSLATIONS.get(category, category))
             ax.set_xlabel('Número de Entrada')
             ax.set_ylabel('Conteo de Palabras')
             ax.set_title('Evolución del Conteo de Palabras por Categoría')
@@ -129,16 +134,20 @@ def display_student_progress(username):
         for i, entry in enumerate(student_data['entries'][:5]):
             with st.expander(f"Entrada {i+1} - {entry['timestamp']}"):
                 st.write(entry['text'])
-                if 'arc_diagrams' in student_data and len(student_data['arc_diagrams']) > i:
+                if 'arc_diagrams' in entry and entry['arc_diagrams']:
                     st.subheader("Diagrama de Arco")
-                    st.write(student_data['arc_diagrams'][i], unsafe_allow_html=True)
-                if 'network_diagrams' in student_data and len(student_data['network_diagrams']) > i:
+                    st.write(entry['arc_diagrams'][0], unsafe_allow_html=True)
+                if 'network_diagram' in entry and entry['network_diagram']:
                     st.subheader("Diagrama de Red")
-                    st.image(student_data['network_diagrams'][i])
+                    st.image(entry['network_diagram'])
     else:
         st.warning("No se encontraron datos para este estudiante.")
         st.info("Intenta realizar algunos análisis de texto primero.")
 
+    # Agregar información de depuración
+    st.subheader("Información de Depuración")
+    st.json(student_data)
+    
 ##########################################################################
 def display_text_analysis_interface(nlp_models, lang_code):
     translations = {