Update modules/ui.py

modules/ui.py

@@ -1,9 +1,9 @@
 # modules/ui.py
 # Importaciones estandar de python
 import streamlit as st
-import matplotlib.pyplot as plt
 import matplotlib
 matplotlib.use('Agg')
+import matplotlib.pyplot as plt
 import squarify
 import pandas as pd
 from datetime import datetime
@@ -186,13 +186,19 @@ def display_student_progress(username, lang_code='es'):
             print("student_data['word_count']:", student_data['word_count'])
             print("df:", df)
             
-            fig, ax = plt.subplots(figsize=(8, 6), dpi=80)
+            fig, ax = plt.subplots(figsize=(8, 6), dpi=30)
             colors = [POS_COLORS.get(cat, '#CCCCCC') for cat in df['category']]
             print("colors:", colors)
             print("labels:", df['label'].tolist())
             
             # Generar el treemap
             squarify.plot(sizes=df['count'], label=df['label'], color=colors, alpha=0.8, ax=ax)
+
+            # Después de crear la figura
+            buf = io.BytesIO()
+            fig.savefig(buf, format='png')
+            buf.seek(0)
+            st.image(buf, use_column_width=True)
             
             # Ajustar las etiquetas
             for text in ax.texts: