Update app.py

app.py

@@ -55,30 +55,11 @@ def build_graph_from_jsonld(jsonld_url, selected_name):
     
     return G
 
-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()
+def run_query_and_visualize(selected_location, jsonld_url):
+    G = build_graph_from_jsonld(jsonld_url, selected_location)
 
-    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])
+    if isinstance(G, str):  # Caso de erro
+        return G
     
     # Define posições específicas para os nós importantes
     pos = nx.spring_layout(G)
@@ -90,7 +71,7 @@ def run_query_and_visualize(qtext, 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 SPARQL", size=15)
+    plt.title("Resultado da Consulta", size=15)
     plt.axis('off')
     
     # Salva o gráfico em um arquivo
@@ -105,33 +86,18 @@ def run_query_and_visualize(qtext, 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")
-        query_input = gr.Textbox(label="Consulta SPARQL", value=update_query(names[0]) if names else "", lines=10)
-        run_button = gr.Button("Executar Consulta")
+        run_button = gr.Button("Visualizar Grafo")
 
     graph_output = gr.HTML()
 
-    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)
+    def on_run_button_click(selected_location):
+        return run_query_and_visualize(selected_location, jsonld_url)
 
-    run_button.click(fn=on_run_button_click, inputs=[query_input], outputs=graph_output)
+    run_button.click(fn=on_run_button_click, inputs=[selected_location], outputs=graph_output)
 
 demo.launch()