Update app.py

app.py

@@ -55,11 +55,30 @@ def build_graph_from_jsonld(jsonld_url, selected_name):
     
     return G
 
-def run_query_and_visualize(selected_location, jsonld_url):
-    G = build_graph_from_jsonld(jsonld_url, selected_location)
+def run_query_and_visualize(qtext, jsonld_url):
+    print("Executando consulta SPARQL...")
+    print(f"Consulta SPARQL: {qtext}")
+    
+    # Carrega o arquivo JSON-LD
+    g = rdflib.Graph()
+    g.parse(jsonld_url, format="json-ld")
+    
+    print("Consulta SPARQL carregada...")
+
+    # Executa a consulta SPARQL
+    qres = g.query(qtext)
+
+    # Cria o gráfico de rede
+    G = nx.DiGraph()
 
-    if isinstance(G, str):  # Caso de erro
-        return G
+    print("Processando resultados da consulta...")
+
+    # Processa os resultados da consulta
+    for row in qres:
+        s, p, o = row
+        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 específicas para os nós importantes
     pos = nx.spring_layout(G)
@@ -71,7 +90,7 @@ def run_query_and_visualize(selected_location, jsonld_url):
     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.title("Resultado da Consulta", size=15)
+    plt.title("Resultado da Consulta SPARQL", size=15)
     plt.axis('off')
     
     # Salva o gráfico em um arquivo
@@ -86,18 +105,33 @@ def run_query_and_visualize(selected_location, jsonld_url):
     print("Gráfico gerado com sucesso.")
     return graph_html
 
+def update_query(selected_location):
+    return f"""
+    PREFIX schema: <http://schema.org/>
+    SELECT * WHERE {{
+        ?s schema:name "{selected_location}" .
+        ?s ?p ?o .
+    }}
+    """
+
 with gr.Blocks() as demo:
     gr.Markdown("# Visualização de Query SPARQL")
 
     with gr.Column():
         selected_location = gr.Dropdown(choices=names, label="Selecione o Local")
-        run_button = gr.Button("Visualizar Grafo")
+        query_input = gr.Textbox(label="Consulta SPARQL", value=update_query(names[0]) if names else "", lines=10)
+        run_button = gr.Button("Executar Consulta")
 
     graph_output = gr.HTML()
 
-    def on_run_button_click(selected_location):
-        return run_query_and_visualize(selected_location, jsonld_url)
+    def on_location_change(loc):
+        return update_query(loc)
+
+    selected_location.change(fn=on_location_change, inputs=selected_location, outputs=query_input)
+
+    def on_run_button_click(query):
+        return run_query_and_visualize(query, jsonld_url)
 
-    run_button.click(fn=on_run_button_click, inputs=[selected_location], outputs=graph_output)
+    run_button.click(fn=on_run_button_click, inputs=[query_input], outputs=graph_output)
 
 demo.launch()