Spaces:
Sleeping
Sleeping
| 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() |