Update modules/ui/ui.py

modules/ui/ui.py

@@ -559,24 +559,39 @@ def display_morphosyntax_analysis_interface(nlp_models, lang_code):
                     morph_df = pd.DataFrame(advanced_analysis['morphological_analysis'])
                     
                     # Traducir las etiquetas POS y otras columnas relevantes
-                    morph_df['pos'] = morph_df['pos'].map(lambda x: POS_TRANSLATIONS[lang_code].get(x, x))
+                    # morph_df['pos'] = morph_df['pos'].map(lambda x: POS_TRANSLATIONS[lang_code].get(x, x))
+                    st.write("Columnas originales:", morph_df.columns.tolist())
                     
                     # Renombrar las columnas para mayor claridad
-                    morph_df = morph_df.rename(columns={
-                        'text': t['word'],
-                        'lemma': t['lemma'],
-                        'pos': t['grammatical_category'],
-                        'tag': t['tag'],
-                        'dep': t['dependency'],
-                        'morph': t['morphology']
-                    })
-        
+                    #morph_df = morph_df.rename(columns={
+                    #    'text': t['word'],
+                    #    'lemma': t['lemma'],
+                    #    'pos': t['grammatical_category'],
+                    #    'tag': t['tag'],
+                    #    'dep': t['dependency'],
+                    #    'morph': t['morphology']
+                    #})
+
+                    # Renombrar solo las columnas que existen
+                    morph_df = morph_df.rename(columns={col: new_name for col, new_name in column_mapping.items() if col in morph_df.columns})
+
+                    #Traducir las etiquetas POS si la columna existe
+                    if 'pos' in morph_df.columns:
+                        morph_df['pos'] = morph_df['pos'].map(lambda x: POS_TRANSLATIONS[lang_code].get(x, x))
+
+                    # Imprimir los nombres de las columnas después del renombramiento para depuración
+                    st.write("Columnas después del renombramiento:", morph_df.columns.tolist())
+                        
+                    # Seleccionar solo las columnas que existen
+                    columns_to_display = [col for col in ['word', 'lemma', 'grammatical_category', 'tag', 'dependency', 'morphology'] if col in morph_df.columns]
+                    
                     # Seleccionar y reordenar las columnas más relevantes
-                    columns_to_display = ['word', 'lemma', 'grammatical_category', 'tag', 'dependency', 'morphology']
-                    morph_df = morph_df[columns_to_display]
+                    #columns_to_display = ['word', 'lemma', 'grammatical_category', 'tag', 'dependency', 'morphology']
+                    #morph_df = morph_df[columns_to_display]
                     
                     # Mostrar el dataframe
-                    st.dataframe(morph_df)
+                    # st.dataframe(morph_df)
+                    st.dataframe(morph_df[columns_to_display]
             
             # Mostrar diagramas de arco (código existente)
             with st.expander(t['arc_diagram'], expanded=True):