Update modules/ui.py

modules/ui.py

@@ -172,28 +172,30 @@ def display_student_progress(username, lang_code='es'):
 
     if student_data['entries_count'] > 0:
         if 'word_count' in student_data and student_data['word_count']:
-            st.subheader("Total de palabras por categoria gramatical")
+            st.subheader("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"{POS_TRANSLATIONS[lang_code].get(x['category'], x['category'])}\n({x['count']})", axis=1)
+            df['label'] = df.apply(lambda x: f"{POS_TRANSLATIONS[lang_code].get(x['category'], x['category'])}", axis=1)
             
-            fig, ax = plt.subplots(figsize=(8, 6), dpi=100)  # Tamaño reducido
-            colors = [POS_COLORS.get(cat, '#CCCCCC') for cat in df['category']]
+            # Ordenar el DataFrame por conteo de palabras, de mayor a menor
+            df = df.sort_values('count', ascending=False)
             
-            squarify.plot(sizes=df['count'], label=df['label'], color=colors, alpha=0.8, ax=ax)
+            fig, ax = plt.subplots(figsize=(12, 6))
+            bars = ax.bar(df['label'], df['count'], color=[POS_COLORS.get(cat, '#CCCCCC') for cat in df['category']])
             
-            for text in ax.texts:
-                text.set_visible(False)
+            ax.set_xlabel('Categoría Gramatical')
+            ax.set_ylabel('Cantidad de Palabras')
+            ax.set_title('Total de palabras por categoría gramatical')
+            plt.xticks(rotation=45, ha='right')
             
-            norm = plt.Normalize(df['count'].min(), df['count'].max())
-            for rect, label in zip(ax.patches, df['label']):
-                x = rect.get_x() + rect.get_width()/2
-                y = rect.get_y() + rect.get_height()/2
-                size = norm(rect.get_height() * rect.get_width())
-                ax.text(x, y, label, ha='center', va='center', fontsize=6+size*8)  # Tamaño de fuente reducido
+            # Añadir etiquetas de valor en las barras
+            for bar in bars:
+                height = bar.get_height()
+                ax.text(bar.get_x() + bar.get_width()/2., height,
+                        f'{height}',
+                        ha='center', va='bottom')
             
-            plt.title('Treemap del total de palabras por categoria gramátical')
-            plt.axis('off')
+            plt.tight_layout()
             
             buf = io.BytesIO()
             fig.savefig(buf, format='png')
@@ -226,7 +228,6 @@ def display_student_progress(username, lang_code='es'):
         st.warning("No se encontraron entradas para este estudiante.")
         st.info("Intenta realizar algunos análisis de texto primero.")
 
-
 ##############################################################Mostrar entradas recientes######################################################################
         #st.header("Entradas Recientes")
         #for i, entry in enumerate(student_data['entries'][:5]):  # Mostrar las 5 entradas más recientes