Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -114,7 +114,7 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     arc_diagrams=analysis['arc_diagrams']
                 ):
                     # Mostrar diagrama original
-                    display_arc_diagram(doc, morpho_t)
+                    display_morphosyntax_results({'doc': doc, 'analysis': analysis}, lang_code, morpho_t)
 
                     st.markdown("---")
 
@@ -157,7 +157,7 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                                 text=iteration_text,
                                 arc_diagrams=analysis['arc_diagrams']
                             ):
-                                display_arc_diagram(doc, morpho_t)
+                                display_morphosyntax_results({'doc': doc, 'analysis': analysis}, lang_code, morpho_t)
 
                         except Exception as e:
                             st.error("Error al procesar iteración")
@@ -171,55 +171,19 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
         st.error("Error general en la interfaz")
         logger.error(f"Error: {str(e)}")
 
-#########################################################################3
-
-def display_morphosyntax_results(result, lang_code, morpho_t):
-    """
-    Muestra solo el análisis sintáctico con diagramas de arco.
-    Args:
-        result: Resultado del análisis que contiene el documento procesado
-        lang_code: Código del idioma actual
-        morpho_t: Diccionario de traducciones
-    """
-    if result is None:
-        logger.warning("No hay resultados para mostrar")
-        return
-
-    try:
-        doc = result['doc']
-        
-        with st.container():
-            sentences = list(doc.sents)
-            for i, sent in enumerate(sentences):
-                try:
-                    st.subheader(f"{morpho_t.get('sentence', 'Sentence')} {i+1}")
-                    
-                    # Renderizar con opciones optimizadas
-                    html = displacy.render(sent, style="dep", options={
-                        "distance": 100,
-                        "arrow_spacing": 20,
-                        "word_spacing": 30
-                    })
-                    
-                    # Ajustar dimensiones del SVG
-                    html = html.replace('height="375"', 'height="200"')
-                    html = re.sub(r'<svg[^>]*>', 
-                                lambda m: m.group(0).replace('height="450"', 'height="300"'), 
-                                html)
-                    html = re.sub(r'<g [^>]*transform="translate\((\d+),(\d+)\)"', 
-                                lambda m: f'<g transform="translate({m.group(1)},50)"', 
-                                html)
-                    
-                    # Envolver en contenedor con clase CSS
-                    html = f'<div class="arc-diagram-container">{html}</div>'
-                    
-                    # Mostrar diagrama
-                    st.write(html, unsafe_allow_html=True)
-                    
-                except Exception as e:
-                    logger.error(f"Error al renderizar oración {i}: {str(e)}")
-                    continue  # Continuar con la siguiente oración si hay error
-                    
-    except Exception as e:
-        logger.error(f"Error general en display_morphosyntax_results: {str(e)}")
+def display_arc_diagram(doc, morpho_t):
+    """Muestra solo el diagrama de arco"""
+    sentences = list(doc.sents)
+    for i, sent in enumerate(sentences):
+        try:
+            st.subheader(f"{morpho_t.get('sentence', 'Sentence')} {i+1}")
+            html = displacy.render(sent, style="dep", options={"distance": 100})
+            html = html.replace('height="375"', 'height="200"')
+            html = re.sub(r'<svg[^>]*>', lambda m: m.group(0).replace('height="450"', 'height="300"'), html)
+            html = re.sub(r'<g [^>]*transform="translate\((\d+),(\d+)\)"', 
+                        lambda m: f'<g transform="translate({m.group(1)},50)"', html)
+            html = f'<div class="arc-diagram-container">{html}</div>'
+            st.write(html, unsafe_allow_html=True)
+        except Exception as e:
+            logger.error(f"Error en diagrama {i}: {str(e)}")