Update app.py

app.py

@@ -46,17 +46,29 @@ def run_query_and_visualize(qtext, jsonld_url):
         G.add_node(str(s), label=str(s).split('/')[-1])  # Usar o último segmento da URI como rótulo
         G.add_node(str(o), label=str(o).split('/')[-1])
         G.add_edge(str(s), str(o), label=str(p).split('/')[-1])
-    
-    # Define posições para os nós
-    pos = nx.spring_layout(G)
-    
+
+    # Define posições específicas para os nós importantes
+    pos = {
+        "https://huggingface.co/spaces/histlearn/MachadodeAssis/Aden": (0, 0.5),
+        "https://huggingface.co/spaces/histlearn/MachadodeAssis/Aden-geo": (-0.6, -0.5),
+        "http://dados.literaturabrasileira.ufsc.br/resource/O_imortal_(Machado_de_Assis)#o-imortal": (0.6, -0.5)
+    }
+
+    # Preencher posições para todos os nós se não definidos
+    for node in G.nodes():
+        if node not in pos:
+            pos[node] = (0, 0)  # Posição padrão para nós não especificados
+
     # Desenha o gráfico usando NetworkX e Matplotlib
-    plt.figure(figsize=(15, 10))
+    plt.figure(figsize=(10, 8))
     nx.draw_networkx_nodes(G, pos, node_size=3000, node_color="skyblue", alpha=0.9)
     nx.draw_networkx_edges(G, pos, width=2, alpha=0.5, edge_color='gray')
     nx.draw_networkx_labels(G, pos, labels=nx.get_node_attributes(G, 'label'), font_size=9, font_color="black")
     nx.draw_networkx_edge_labels(G, pos, edge_labels=nx.get_edge_attributes(G, 'label'), font_size=9, font_color="red")
 
+    plt.xlim(-1, 1)
+    plt.ylim(-1, 1)
+
     plt.title("Resultado da Consulta SPARQL", size=15)
     plt.axis('off')