Update app.py

app.py

@@ -1,35 +1,28 @@
 import gradio as gr
 import rdflib
-import json
 import requests
 from pyvis.network import Network
 from IPython.display import HTML
 
-# Função para carregar e extrair os nomes do arquivo JSON-LD a partir de um URL
-def load_names_from_url(url):
-    response = requests.get(url)
+# Função para carregar e extrair os nomes do arquivo JSON-LD a partir de uma URL
+def load_names_from_url(jsonld_url):
+    response = requests.get(jsonld_url)
     data = response.json()
     
-    print("Data received from URL:", data)  # Adicionando impressão para depuração
-    
     names = []
     for item in data:
-        if isinstance(item, dict) and '@graph' in item:
-            for entry in item['@graph']:
-                if 'name' in entry:
-                    names.append(entry['name'])
+        if 'name' in item:
+            names.append(item['name'])
     
     return names
 
-# Carregar nomes do arquivo JSON-LD a partir do URL
+# Carregar nomes do arquivo JSON-LD
 names = load_names_from_url('https://huggingface.co/spaces/histlearn/ShowGraph/raw/main/datafile.jsonld')
 
-print("Loaded names:", names)  # Adicionando impressão para depuração
-
-def run_query_and_visualize(qtext, jsonld_file):
+def run_query_and_visualize(qtext, jsonld_url):
     # Carrega o arquivo JSON-LD
     g = rdflib.Graph()
-    g.parse(jsonld_file, format="json-ld")
+    g.parse(jsonld_url, format="json-ld")
 
     # Executa a consulta SPARQL
     qres = g.query(qtext)
@@ -45,15 +38,13 @@ def run_query_and_visualize(qtext, jsonld_file):
             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))
+            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 o exibe diretamente na célula do notebook
     net.show("graph.html")
-    with open("graph.html", "r") as file:
-        graph_html = file.read()
-    return graph_html
+    return HTML("graph.html")
 
 def update_query(selected_location):
     return f"""
@@ -69,10 +60,7 @@ with gr.Blocks() as demo:
 
     with gr.Column():
         selected_location = gr.Dropdown(choices=names, label="Selecione o Local")
-        if names:  # Verifica se a lista de nomes não está vazia
-            query_input = gr.Textbox(label="Consulta SPARQL", value=update_query(names[0]), lines=10)
-        else:
-            query_input = gr.Textbox(label="Consulta SPARQL", value="", lines=10)
+        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()