Update modules/ui.py

modules/ui.py

@@ -191,12 +191,13 @@ def display_student_progress(username, lang_code='es'):
             print("colors:", colors)
             print("labels:", df['label'].tolist())
             
-            # Corrección en la función squarify.plot
-            squarify.plot(sizes=df['count'], label=df['label'], color=colors, alpha=0.8, ax=ax)
+            # Generar el treemap
+            rects = squarify.plot(sizes=df['count'], label=df['label'], color=colors, alpha=0.8, ax=ax)
             
             # Añadir etiquetas manualmente
-            for i, (x, y, dx, dy) in enumerate(zip(ax.patches[0].get_xy(), ax.patches[0].get_width(), ax.patches[0].get_height())):
-                ax.text(x + dx/2, y + dy/2, df['label'].iloc[i], ha='center', va='center')
+            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)
             
             plt.title('Treemap del total de palabras por categoria gramátical')
             plt.axis('off')