Update app.py

app.py

@@ -1,7 +1,10 @@
 import gradio as gr
 import rdflib
 import requests
-from pyvis.network import Network
+import matplotlib.pyplot as plt
+import networkx as nx
+from io import BytesIO
+import base64
 
 # Função para carregar e extrair os nomes do arquivo JSON-LD a partir de uma URL
 def load_names_from_url(jsonld_url):
@@ -32,28 +35,38 @@ def run_query_and_visualize(qtext, jsonld_url):
     # Executa a consulta SPARQL
     qres = g.query(qtext)
 
-    # Prepara o gráfico com recursos remotos
-    net = Network(notebook=False, height="400px", width="100%", cdn_resources='remote')
-    nodes = set()
+    # Cria o gráfico de rede
+    G = nx.DiGraph()
 
     print("Processando resultados da consulta...")
 
     # Processa os resultados da consulta
     for row in qres:
         s, p, o = row
-        if str(s) not in nodes:
-            net.add_node(str(s), label=str(s))
-            nodes.add(str(s))
-        if str(o) not in nodes:
-            net.add_node(str(o), label=str(o))
-            nodes.add(str(o))
-        net.add_edge(str(s), str(o), title=str(p))
-
-    # Gera o gráfico e salva em um arquivo HTML
-    net.show("graph.html")
-    with open("graph.html", "r") as file:
-        graph_html = file.read()
+        G.add_node(str(s))
+        G.add_node(str(o))
+        G.add_edge(str(s), str(o), label=str(p))
     
+    # Desenha o gráfico usando NetworkX e Matplotlib
+    pos = nx.spring_layout(G, k=0.5)
+    plt.figure(figsize=(15, 10))
+    
+    nx.draw(G, pos, with_labels=True, node_size=5000, node_color="skyblue", font_size=10, font_color="black", font_weight="bold", edge_color="gray", width=2, alpha=0.7)
+    edge_labels = nx.get_edge_attributes(G, 'label')
+    nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels, font_color="red", font_size=8)
+
+    plt.title("Resultado da Consulta SPARQL", size=15)
+    plt.box(False)
+    
+    # Salva o gráfico em um arquivo
+    buf = BytesIO()
+    plt.savefig(buf, format='png')
+    buf.seek(0)
+    img_str = base64.b64encode(buf.read()).decode()
+    graph_html = f'<img src="data:image/png;base64,{img_str}"/>'
+    
+    plt.close()
+
     print("Gráfico gerado com sucesso.")
     return graph_html