Update modules/text_analysis/morpho_analysis.py

modules/text_analysis/morpho_analysis.py

@@ -116,15 +116,28 @@ def highlight_repeated_words(doc, word_colors):
     return ' '.join(highlighted_text)
     
 #################################################################################################
+
+def generate_arc_diagram(doc, lang_code):
+    sentences = list(doc.sents)
+    arc_diagrams = []
+    for sent in sentences:
+        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)
+        arc_diagrams.append(html)
+    return arc_diagrams
+
+"""
 def generate_arc_diagram(doc):
-    """
+
     Genera diagramas de arco para cada oración en el documento.
     
     Args:
         doc: Documento procesado por spaCy
     Returns:
         list: Lista de diagramas en formato HTML
-    """
+
     arc_diagrams = []
     for sent in doc.sents:
         words = [token.text for token in sent]
@@ -181,7 +194,10 @@ def generate_arc_diagram(doc):
 
         arc_diagrams.append(html)
     return arc_diagrams
+    """
+
 
+    
 #################################################################################################
 def get_detailed_pos_analysis(doc):
     """