import streamlit as st
from SPARQLWrapper import SPARQLWrapper, JSON
from streamlit_agraph import agraph, TripleStore, Node, Edge, Config
from layout import footer
import json

def get_inspired():
  sparql = SPARQLWrapper("http://dbpedia.org/sparql")

  query_string = """
  SELECT ?name_pe1_en ?rel_en ?name_pe2_en
  WHERE {
    {
         SELECT ?name_p1 ?rel ?name_p2
         WHERE {
             ?p1 a foaf:Person .
             ?p1 dbo:influencedBy ?p2 .
             ?p2 a foaf:Person .
             ?p1 foaf:name ?name_p1 .
             ?p2 foaf:name ?name_p2 .
            dbo:influencedBy rdfs:label ?rel .
            }
         LIMIT 100
    }
    UNION
    {
         SELECT ?name_p1 ?rel ?name_p2
         WHERE {
            ?p1 a foaf:Person .
            ?p1 dbo:influenced ?p2 .
            ?p2 a foaf:Person .
            ?p1 foaf:name ?name_p1 .
            ?p2 foaf:name ?name_p2 .
            dbo:influenced rdfs:label ?rel .
        }
     LIMIT 100
    }
    FILTER ( LANG(?name_p1) = "en" && LANG(?rel) = "en" && LANG(?name_p2) = "en" )
    BIND ( STR(?name_p1) AS ?name_pe1_en )
    BIND ( STR(?rel) AS ?rel_en )
    BIND ( STR(?name_p2) AS ?name_pe2_en )
  }
  """

  sparql.setQuery(query_string)
  sparql.setReturnFormat(JSON)
  results = sparql.query().convert()
  store = TripleStore()
  for result in results["results"]["bindings"]:
    node1 = result["name_pe1_en"]["value"]
    link = result["rel_en"]["value"]
    node2 = result["name_pe2_en"]["value"]
    store.add_triple(node1, link, node2)
  return store

def app():
  footer()
  st.title("Graph Example")
  st.sidebar.title("Welcome")
  query_type = st.sidebar.selectbox("Query Tpye: ", ["Inspirationals", "Marvel", "Fraud"]) # could add more stuff here later on or add other endpoints in the sidebar.
  config = Config(height=600, width=700, nodeHighlightBehavior=True, highlightColor="#F7A7A6", directed=True,
                  collapsible=True)

  if query_type=="Inspirationals":
    st.subheader("Inspirationals")
    with st.spinner("Loading data"):
      store = get_inspired()
      st.write("Nodes loaded: " + str(len(store.getNodes())))
    st.success("Done")
    agraph(list(store.getNodes()), (store.getEdges() ), config)

  if query_type=="Marvel":
    #based on http://marvel-force-chart.surge.sh/
    with open("data/marvel.json", encoding="utf8") as f:
      marvel_file = json.loads(f.read())
      marvel_store = TripleStore()
      for sub_graph in marvel_file["children"]:
        marvel_store.add_triple(marvel_file["name"], "has_subgroup", sub_graph["name"], picture=marvel_file["img"])
        for node in sub_graph["children"]:
          node1 = node["hero"]
          link = "blongs_to"
          node2 = sub_graph["name"]
          pic = node["img"]
          marvel_store.add_triple(node1, link, node2, picture=pic)
      agraph(list(marvel_store.getNodes()), (marvel_store.getEdges()), config)


  if query_type=="Fraud":
    #based on http://marvel-force-chart.surge.sh/
    with open("data/fraud.json", encoding="utf8") as f:
      marvel_file = json.loads(f.read())
      marvel_store = TripleStore()
      for sub_graph in marvel_file["children"]:
        marvel_store.add_triple(marvel_file["name"], "has_subgroup", sub_graph["name"], picture=marvel_file["img"])
        for node in sub_graph["children"]:
          node1 = node["role"]
          link = "blongs_to"
          node2 = sub_graph["name"]
          pic = node["img"]
          marvel_store.add_triple(node1, link, node2, picture=pic)
      agraph(list(marvel_store.getNodes()), (marvel_store.getEdges()), config)


if __name__ == '__main__':
    app()