Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -172,19 +172,37 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
         logger.error(f"Error: {str(e)}")
 
 
-def display_morphosyntax_results(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)}")
+def display_morphosyntax_results(result, lang_code, morpho_t):
+    """
+    Muestra solo el diagrama de arco.
+    Args:
+        result: Diccionario con el documento procesado y su análisis
+        lang_code: Código del idioma
+        morpho_t: Diccionario de traducciones
+    """
+    if result is None:
+        return
+        
+    try:
+        doc = result['doc']
+        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,
+                    "arrow_spacing": 20,
+                    "word_spacing": 30
+                })
+                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)}")
+                continue
+    except Exception as e:
+        logger.error(f"Error en display_morphosyntax_results: {str(e)}")