Update app.py

app.py

@@ -45,18 +45,23 @@ def tokenize_sentence(text):
     tokens = re.findall(r"[\w\u0370-\u03FF]+|[.,;:·!?]", text)
     return [{"id": str(i+1), "form": t, "lemma": "", "postag": "Pontuação" if t in ".,;:·!?" else "Substantivo", "head": "0" if t in ".,;:·!?" else "", "relation": "AuxK" if (t in ".,;:·!?" and i==len(tokens)-1) else ("AuxX" if t in ".,;:·!?" else ""), "artificial": False} for i, t in enumerate(tokens)]
 
-def render_tree(words, format_type='svg'):
+def render_tree(words):
     if not words: return None
-    dot = Digraph()
+    # Criamos o gráfico pedindo o formato SVG
+    dot = Digraph(format='svg')
     dot.attr(rankdir='TB', nodesep='1.0', ranksep='0.8')
     dot.attr('node', fontname="serif", shape='none')
     dot.node("0", label="ROOT", fontcolor="red", fontsize="24")
+    
     for w in words:
         txt_color = MORPHO_COLORS.get(w.get('postag', 'Substantivo'), "black")
         label = f"<<table border='0' cellborder='0'><tr><td><font point-size='22' color='{txt_color}'><b>{w['form']}</b></font></td></tr><tr><td><font point-size='16' color='gray30'>{w['relation'] or '?'}</font></td></tr></table>>"
-        dot.node(w['id'], label)
-        dot.edge(w['head'] or "0", w['id'], color="#cccccc", penwidth='2.0')
-    return dot.source
+        dot.node(str(w['id']), label)
+        dot.edge(str(w['head'] or "0"), str(w['id']), color="#cccccc", penwidth='2.0')
+    
+    # O SEGREDO: Transformar o gráfico em uma string SVG que o HTML aceita
+    svg_data = dot.pipe(format='svg').decode('utf-8')
+    return svg_data
 
 
 def add_at(words):
@@ -89,7 +94,7 @@ with gr.Blocks(title="Arethusa Lite", css=custom_css) as demo:
                     btn_xml = gr.Button("📦 EXPORTAR XML")
                     out_file = gr.File(label="Download")
                 with gr.Column(scale=2):
-                   svg_out = gr.Graphviz()
+                   svg_out = gr.HTML()
 
     def process_start(text):
         d = tokenize_sentence(text)