Update modules/ui.py

modules/ui.py

@@ -158,7 +158,7 @@ def display_chat_interface():
 
 ##########################################################################
 
-def display_student_progress(username, lang_code='es'):  # Añade el parámetro lang_code
+def display_student_progress(username, lang_code='es'):
     print("lang_code:", lang_code)
     student_data = get_student_data(username)
     
@@ -175,11 +175,6 @@ 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']:
@@ -188,12 +183,15 @@ def display_student_progress(username, lang_code='es'):  # Añade el parámetro
             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)
             
+            print("student_data['word_count']:", student_data['word_count'])
+            print("df:", df)
+            
             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)
+            print("colors:", colors)
+            print("labels:", df['label'].tolist())
+            
             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)
@@ -201,8 +199,32 @@ def display_student_progress(username, lang_code='es'):  # Añade el parámetro
         else:
             st.info("No hay datos de conteo de palabras disponibles.")
         
-        # ... (resto del código para diagramas de arco, diagramas de red y entradas recientes)
-
+        # Diagramas de Arco (consolidados)
+        st.header("Diagramas de Arco")
+        with st.expander("Ver todos los Diagramas de Arco"):
+            for i, entry in enumerate(student_data['entries']):
+                if 'arc_diagrams' in entry and entry['arc_diagrams']:
+                    st.subheader(f"Entrada {i+1} - {entry['timestamp']}")
+                    st.write(entry['arc_diagrams'][0], unsafe_allow_html=True)
+        
+        # Diagramas de Red (consolidados)
+        st.header("Diagramas de Red")
+        with st.expander("Ver todos los Diagramas de Red"):
+            for i, entry in enumerate(student_data['entries']):
+                if 'network_diagram' in entry and entry['network_diagram']:
+                    st.subheader(f"Entrada {i+1} - {entry['timestamp']}")
+                    try:
+                        # Decodificar la imagen base64
+                        image_bytes = base64.b64decode(entry['network_diagram'])
+                        st.image(image_bytes)
+                    except Exception as e:
+                        st.error(f"Error al mostrar el diagrama de red: {str(e)}")
+        
+        # Mostrar entradas recientes
+        st.header("Entradas Recientes")
+        for i, entry in enumerate(student_data['entries'][:5]):  # Mostrar las 5 entradas más recientes
+            with st.expander(f"Entrada {i+1} - {entry['timestamp']}"):
+                st.write(entry['text'])
     else:
         st.warning("No se encontraron entradas para este estudiante.")
         st.info("Intenta realizar algunos análisis de texto primero.")