Update modules/ui.py

modules/ui.py

@@ -192,12 +192,19 @@ def display_student_progress(username, lang_code='es'):
             print("labels:", df['label'].tolist())
             
             # Generar el treemap
-            rects = 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)
+            
+            # Ajustar las etiquetas
+            for text in ax.texts:
+                text.set_visible(False)
             
             # Añadir etiquetas manualmente
-            for r, label in zip(rects, df['label']):
-                x, y, dx, dy = r['x'], r['y'], r['dx'], r['dy']
-                ax.text(x + dx/2, y + dy/2, label, ha='center', va='center', wrap=True)
+            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=8+size*10)
             
             plt.title('Treemap del total de palabras por categoria gramátical')
             plt.axis('off')