Update modules/ui.py

modules/ui.py

@@ -11,6 +11,8 @@ import base64
 from spacy import displacy
 import re
 from .morpho_analysis import POS_COLORS, POS_TRANSLATIONS  # Asegúrate de que esta importación esté presente
+print("POS_COLORS:", POS_COLORS)
+print("POS_TRANSLATIONS:", POS_TRANSLATIONS)
 
 # Importaciones locales
 from .auth import authenticate_user, register_user, get_user_role
@@ -157,6 +159,7 @@ def display_chat_interface():
 ##########################################################################
 
 def display_student_progress(username, lang_code='es'):  # Añade el parámetro lang_code
+    print("lang_code:", lang_code)
     student_data = get_student_data(username)
     
     if student_data is None:
@@ -172,19 +175,28 @@ def display_student_progress(username, lang_code='es'):  # Añade el parámetro
         # Mostrar estadísticas generales
         st.header("Estadísticas Generales")
         st.metric("Total de entradas", student_data['entries_count'])
+
+        print("student_data['word_count']:", student_data['word_count'])
+        print("df:", df)
+        print("colors:", colors)
+        print("labels:", df['label'].tolist())
         
         # Treemap para el conteo de palabras por categoría
         if student_data['word_count']:
-            st.subheader("Conteo Total de Palabras por Categoría Gramatical")
+            st.subheader("Total de palabras por categoria gramatical")
             
             df = pd.DataFrame(list(student_data['word_count'].items()), columns=['category', 'count'])
             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']]
-            squarify.plot(sizes=df['count'], label=df['label'], color=colors, alpha=0.8, ax=ax)
-            plt.title('Treemap de Categorías Gramaticales')
+            #squarify.plot(sizes=df['count'], label=df['label'], color=colors, alpha=0.8, ax=ax)
+            squarify.plot(sizes=df['count'], label=df['label'], color=colors, alpha=0.8, ax=ax, text=df['label'])
+            squarify.plot(sizes=df['count'], label=df['label'], color=list(POS_COLORS.values())[:len(df)], alpha=0.8, ax=ax, text=df['label'])
+            df['label'] = df.apply(lambda x: f"{POS_TRANSLATIONS[lang_code].get(x['category'], x['category'])}\n({x['count']})", axis=1)
+            plt.title('Treemap del total de palabras por categoria gramátical')
             plt.axis('off')
+            print(fig)
             st.pyplot(fig)
         else:
             st.info("No hay datos de conteo de palabras disponibles.")