Update modules/ui.py

modules/ui.py

@@ -207,7 +207,7 @@ def display_student_progress(username, lang_code='es'):
                 df = df.sort_values('count', ascending=False)
                 
                 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']])
+                bars = ax.bar(df['label'], df['count'], color=[x for x in df['category']])  # Usar los colores directamente
                 
                 ax.set_xlabel('Categoría Gramatical')
                 ax.set_ylabel('Cantidad de Palabras')
@@ -224,38 +224,38 @@ def display_student_progress(username, lang_code='es'):
         # Mostrar análisis morfosintáctico
         with st.expander("Análisis Morfosintáctico - Diagramas de Arco", expanded=False):
             for i, entry in enumerate(student_data['entries']):
-                if entry.get('analysis_type') == 'morphosyntax' and 'arc_diagrams' in entry:
+                if 'arc_diagrams' in entry:
                     st.subheader(f"Análisis {i+1} - {entry['timestamp']}")
-                    st.write(entry['arc_diagrams'][0], unsafe_allow_html=True)
+                    st.write(entry['text'])  # Mostrar el texto analizado
+                    for j, diagram in enumerate(entry['arc_diagrams']):
+                        st.subheader(f"Diagrama de Arco {j+1}")
+                        st.write(diagram, unsafe_allow_html=True)
         
-        # Mostrar análisis semántico
-        with st.expander("Análisis Semántico - Diagramas de Red", expanded=False):
-            for i, entry in enumerate(student_data['entries']):
-                if entry.get('analysis_type') == 'semantic' and 'network_diagram' in entry:
-                    st.subheader(f"Análisis {i+1} - {entry['timestamp']}")
-                    try:
+        # Mostrar análisis semántico (si existe)
+        if 'semantic_analyses' in student_data:
+            with st.expander("Análisis Semántico - Diagramas de Red", expanded=False):
+                for i, entry in enumerate(student_data['semantic_analyses']):
+                    st.subheader(f"Análisis Semántico {i+1} - {entry['timestamp']}")
+                    st.write(entry['text'])  # Mostrar el texto analizado
+                    if 'network_diagram' in entry:
                         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 análisis del discurso
-        with st.expander("Análisis del Discurso - Comparación de Grafos", expanded=False):
-            for i, entry in enumerate(student_data['entries']):
-                if entry.get('analysis_type') == 'discourse' and 'combined_graph' in entry:
-                    st.subheader(f"Análisis {i+1} - {entry['timestamp']}")
-                    try:
+        # Mostrar análisis del discurso (si existe)
+        if 'discourse_analyses' in student_data:
+            with st.expander("Análisis del Discurso - Comparación de Grafos", expanded=False):
+                for i, entry in enumerate(student_data['discourse_analyses']):
+                    st.subheader(f"Análisis del Discurso {i+1} - {entry['timestamp']}")
+                    st.write("Texto del documento patrón:")
+                    st.write(entry['text1'])
+                    st.write("Texto del documento comparado:")
+                    st.write(entry['text2'])
+                    if 'combined_graph' in entry:
                         image_bytes = base64.b64decode(entry['combined_graph'])
                         st.image(image_bytes)
-                        st.write("Texto del documento patrón:")
-                        st.write(entry['text1'])
-                        st.write("Texto del documento comparado:")
-                        st.write(entry['text2'])
-                    except Exception as e:
-                        st.error(f"Error al mostrar el gráfico combinado: {str(e)}")
         
-        # Mostrar conversaciones del chat
-        if 'chat_history' in student_data and student_data['chat_history']:
+        # Mostrar conversaciones del chat (si existe)
+        if 'chat_history' in student_data:
             with st.expander("Historial de Conversaciones del Chat", expanded=False):
                 for i, chat in enumerate(student_data['chat_history']):
                     st.subheader(f"Conversación {i+1} - {chat['timestamp']}")