import gradio as gr
import rdflib
import json
import requests
from pyvis.network import Network

# 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)
    data = response.json()
    
    names = []
    for item in data['@graph']:
        if 'name' in item:
            names.append(item['name'])
    
    return names

# Carregar nomes do arquivo JSON-LD a partir do URL
names = load_names_from_url('https://huggingface.co/spaces/histlearn/ShowGraph/raw/main/datafile.jsonld')

def run_query_and_visualize(qtext, turtle_file):
    # Carrega o arquivo Turtle
    g = rdflib.Graph()
    g.parse(turtle_file, format="turtle")

    # Executa a consulta SPARQL
    qres = g.query(qtext)

    # Prepara o gráfico com recursos remotos
    net = Network(notebook=True, height="400px", width="100%", cdn_resources='remote')
    nodes = set()

    # Processa os resultados da consulta
    for row in qres:
        s, p, o = row
        if str(s) not in nodes:
            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))
            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")
    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]), lines=10)
        turtle_file_input = gr.File(label="Arquivo Turtle")
        run_button = gr.Button("Executar Consulta")

    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, turtle_file):
        return run_query_and_visualize(query, turtle_file.name)

    run_button.click(fn=on_run_button_click, inputs=[query_input, turtle_file_input], outputs=graph_output)

demo.launch()