Update modules/ui.py

modules/ui.py

@@ -187,7 +187,8 @@ def display_chat_interface():
             st.session_state.chat_history.append(("bot", response))
             st.experimental_rerun()
 
-##################################################################################################
+################################################################################
+# Funciones para Cosmos DB MongoDB API (análisis de texto)
 def display_student_progress(username, lang_code='es'):
     student_data = get_student_data(username)
     
@@ -200,14 +201,14 @@ def display_student_progress(username, lang_code='es'):
 
     if student_data['entries_count'] > 0:
         # Mostrar el conteo de palabras
-        if 'word_count' in student_data and student_data['word_count']:
+        if student_data['word_count']:
             with st.expander("Total de palabras por categoría gramatical", expanded=False):
                 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'])}", axis=1)
                 df = df.sort_values('count', ascending=False)
                 
                 fig, ax = plt.subplots(figsize=(12, 6))
-                bars = ax.bar(df['label'], df['count'], color=[x for x in df['category']])  # Usar los colores directamente
+                bars = ax.bar(df['label'], df['count'], color=df['category'])
                 
                 ax.set_xlabel('Categoría Gramatical')
                 ax.set_ylabel('Cantidad de Palabras')
@@ -220,42 +221,44 @@ def display_student_progress(username, lang_code='es'):
                 
                 plt.tight_layout()
                 st.pyplot(fig)
-        
+
         # 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 'arc_diagrams' in entry:
+        morphosyntax_entries = [entry for entry in student_data['entries'] if entry['analysis_type'] == 'morphosyntax']
+        if morphosyntax_entries:
+            with st.expander("Análisis Morfosintáctico - Diagramas de Arco", expanded=False):
+                for i, entry in enumerate(morphosyntax_entries):
                     st.subheader(f"Análisis {i+1} - {entry['timestamp']}")
-                    st.write(entry['text'])  # Mostrar el texto analizado
-                    for j, diagram in enumerate(entry['arc_diagrams']):
+                    st.write(entry['text'])
+                    for j, diagram in enumerate(entry.get('arc_diagrams', [])):
                         st.subheader(f"Diagrama de Arco {j+1}")
                         st.write(diagram, unsafe_allow_html=True)
-        
-        # Mostrar análisis semántico (si existe)
-        if 'semantic_analyses' in student_data:
+
+        # Mostrar análisis semántico
+        if student_data['semantic_analyses']:
             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
+                    st.write(entry['text'])
                     if 'network_diagram' in entry:
                         image_bytes = base64.b64decode(entry['network_diagram'])
                         st.image(image_bytes)
-        
-        # Mostrar análisis del discurso (si existe)
-        if 'discourse_analyses' in student_data:
+
+        # Mostrar análisis del discurso
+        if student_data['discourse_analyses']:
             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(entry.get('text1', 'No disponible'))
                     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)
-        
-        # Mostrar conversaciones del chat (si existe)
-        if 'chat_history' in student_data:
+                    st.write(entry.get('text2', 'No disponible'))
+                    if 'graph1' in entry:
+                        st.image(base64.b64decode(entry['graph1']))
+                    if 'graph2' in entry:
+                        st.image(base64.b64decode(entry['graph2']))
+
+        # Mostrar conversaciones del chat
+        if student_data['chat_history']:
             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']}")
@@ -269,6 +272,11 @@ def display_student_progress(username, lang_code='es'):
         st.warning("No se encontraron entradas para este estudiante.")
         st.info("Intenta realizar algunos análisis de texto primero.")
 
+    # Añadir logs para depuración
+    st.write("Datos del estudiante (para depuración):")
+    st.json(student_data)
+
+
 ##################################################################################################
 def display_morphosyntax_analysis_interface(nlp_models, lang_code):
     translations = {