Update modules/text_analysis/morpho_analysis.py

modules/text_analysis/morpho_analysis.py

@@ -119,7 +119,7 @@ def highlight_repeated_words(doc, word_colors):
 
 def generate_arc_diagram(doc):
     """
-    Genera diagramas de arco para cada oración en el documento.
+    Genera diagramas de arco para cada oración en el documento usando spacy-streamlit.
     
     Args:
         doc: Documento procesado por spaCy
@@ -127,25 +127,34 @@ def generate_arc_diagram(doc):
         list: Lista de diagramas en formato HTML
     """
     arc_diagrams = []
-    for sent in doc.sents:
-        # Configuración básica que sabemos que funciona
-        html = displacy.render(sent, style="dep", options={
-            "distance": 100,
+    try:
+        options = {
+            "compact": False,
             "color": "#ffffff",
             "bg": "#0d6efd",
-            "font": "Arial"
-        })
-        
-        # Ajustes básicos que funcionaban en la versión anterior
-        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)
-        
-        arc_diagrams.append(html)
-    
-    return arc_diagrams
+            "font": "Arial",
+            "offset_x": 50,
+            "distance": 100,
+            "arrow_spacing": 12,
+            "arrow_width": 2,
+            "arrow_stroke": 2,
+            "word_spacing": 25,
+            "maxZoom": 2
+        }
+
+        for sent in doc.sents:
+            try:
+                # Usar el método render de displacy directamente con las opciones
+                html = displacy.render(sent, style="dep", options=options)
+                arc_diagrams.append(html)
+            except Exception as e:
+                logger.error(f"Error al renderizar oración: {str(e)}")
+                continue
 
+        return arc_diagrams
+    except Exception as e:
+        logger.error(f"Error general en generate_arc_diagram: {str(e)}")
+        return None
 
 """
 def generate_arc_diagram(doc):