Update app.py

app.py

@@ -22,54 +22,56 @@ def load_names_from_url(jsonld_url):
 jsonld_url = 'https://huggingface.co/spaces/histlearn/ShowGraph/raw/main/datafile.jsonld'
 names = load_names_from_url(jsonld_url)
 
-def run_query_and_visualize(qtext, jsonld_url):
-    print("Executando consulta SPARQL...")
-    print(f"Consulta SPARQL: {qtext}")
+def build_graph_from_jsonld(jsonld_url, selected_name):
+    response = requests.get(jsonld_url)
+    data = response.json()
     
-    # Carrega o arquivo JSON-LD
-    g = rdflib.Graph()
-    g.parse(jsonld_url, format="json-ld")
+    # Filtrar o local selecionado
+    selected_data = next((item for item in data if item['name'] == selected_name), None)
+    
+    if not selected_data:
+        return "Local não encontrado."
     
-    print("Consulta SPARQL carregada...")
-
-    # Executa a consulta SPARQL
-    qres = g.query(qtext)
-
-    # Cria o gráfico de rede
     G = nx.DiGraph()
+    
+    # Adicionar nó do Place
+    place_id = selected_data['@id']
+    place_label = f"schema:Place\nName: {selected_data['name']}\nDescription: {selected_data['description'][:30]}..."
+    G.add_node(place_id, label=place_label)
+    
+    # Adicionar nó de GeoCoordinates
+    geo_data = selected_data['geo']
+    geo_id = geo_data['@id']
+    geo_label = f"geo:SpatialThing\nLat: {geo_data['lat']}\nLong: {geo_data['long']}\nFeatureCode: {geo_data['gn:featureCode']}\nFeatureCodeName: {geo_data['gn:featureCodeName']}\nName: {geo_data['gn:name']}"
+    G.add_node(geo_id, label=geo_label)
+    G.add_edge(place_id, geo_id, label="schema:geo")
+    
+    # Adicionar nós de CreativeWork
+    for work in selected_data.get('subjectOf', []):
+        work_id = work['@id']
+        work_label = f"schema:CreativeWork\nHeadline: {work['headline']}\nGenre: {work['genre']}\nDatePublished: {work['datePublished']}\nText: {work['text'][:30]}...\nLanguage: {work['inLanguage']}"
+        G.add_node(work_id, label=work_label)
+        G.add_edge(place_id, work_id, label="schema:subjectOf")
+    
+    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])
+def run_query_and_visualize(selected_location, jsonld_url):
+    G = build_graph_from_jsonld(jsonld_url, selected_location)
 
+    if isinstance(G, str):  # Caso de erro
+        return 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
-
+    pos = nx.spring_layout(G)
+    
     # Desenha o gráfico usando NetworkX e Matplotlib
-    plt.figure(figsize=(10, 8))
+    plt.figure(figsize=(15, 10))
     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.title("Resultado da Consulta", size=15)
     plt.axis('off')
     
     # Salva o gráfico em um arquivo
@@ -84,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()