Update app.py

app.py

@@ -18,14 +18,29 @@
 g.bind("bigmap", BIGMAP)
 g.bind("schema", SCHEMA)
 
+# Define specific nodes
+specific_nodes = [
+    BIGMAP["bigmap_94526240_5e05_40e2_8461_a1cdb77d2018"],
+    BIGMAP["bigmap_cc70b46b_0401_409d_af37_c78685e98f76"],
+    BIGMAP["bigmap_66db2cb6_df8f_4e27_9aa2_6a0062344568"],
+    BIGMAP["bigmap_9338cde6_8924_4d7e_a02d_2d3e0701a7ac"],
+    BIGMAP["bigmap_7c99c738_c7b7_4297_894e_38ed5ae83b50"],
+    BIGMAP["bigmap_66642d47_58c4_4241_ab06_030feffa0aaf"],
+    BIGMAP["bigmap_8866787e_893b_4430_af03_8b21c9a63449"],
+    BIGMAP["bigmap_a46f3676_3bdb_424d_b864_bd933dfeb68c"],
+    BIGMAP["bigmap_dc4aacec_4357_4486_a1bd_18ec894ebed1"],
+    BIGMAP["bigmap_b08c7a35_bdb7_4aca_83be_77f9554a462b"],
+    BIGMAP["bigmap_166d46d7_78ac_4b21_9a03_d8be72f9830e"],
+    BIGMAP["bigmap_e62dc54b_059b_4b72_9f4c_5fe52a2b1740"]
+]
+
 # Edge types for sidebar
 edge_types = {
     "work packages": "bigmap:hasWorkPackage",
     "wp leaders": "bigmap:hasLeadPartner",
     "publications": "schema:citation",
     "results": "eurio:isResultOf",
-    "presentations": "bigmap:hasPresentation",
-    "people": "eurio:author"
+    "presentations": "bigmap:hasPresentation"
 }
 
 # Edge color dictionary
@@ -46,12 +61,36 @@ def get_unique_predicates(graph):
 def main():
     st.title("BIG-MAP Key Demonstrator Explorer")
     kd_options = [f"KD{i}" for i in range(1, 12)]
-    selected_kds = st.multiselect("Select KD Numbers", options=kd_options, default=kd_options)
+    if 'selected_kds' not in st.session_state:
+        st.session_state.selected_kds = kd_options
+
+    def select_all():
+        st.session_state.selected_kds = kd_options
+
+    def deselect_all():
+        st.session_state.selected_kds = []
+
+    col1, col2 = st.columns([1, 1])
+    with col1:
+        if st.button("Select All"):
+            select_all()
+    with col2:
+        if st.button("Deselect All"):
+            deselect_all()
+
+    selected_kds = st.multiselect("Select KD Numbers", options=kd_options, default=st.session_state.selected_kds)
+
 
     # Sidebar for predefined edge types
     with st.sidebar:
         st.header("Predefined Edge Options")
-        edge_selections = {label: st.toggle(label, True) for label in edge_types.keys()}
+        edge_selections = {}
+        for label in edge_types.keys():
+            if label == "results":
+                edge_selections[label] = st.toggle(label, False)
+            else:
+                edge_selections[label] = st.toggle(label, True)
+        show_project_deliverables = st.toggle("deliverables", True)
 
     # Collect predicates for selected edges from sidebar
     selected_edges = [edge_types[label] for label, selected in edge_selections.items() if selected]
@@ -66,13 +105,13 @@ def main():
     selected_edges.extend(selected_additional_edges)
 
     if selected_kds and selected_edges:
-        nodes, edges = extract_graph(selected_kds, selected_edges)
-        config = Config(width=1000, height=800, directed=True, nodeHighlightBehavior=True, highlightColor="#F7A7A6", collapsible=True)
+        nodes, edges = extract_graph(selected_kds, selected_edges, show_project_deliverables)
+        config = Config(width=1200, height=800, directed=True, nodeHighlightBehavior=True, highlightColor="#F7A7A6", collapsible=True)
         agraph(nodes=nodes, edges=edges, config=config)
     else:
         st.write("Please select at least one KD number and enable edge types to visualize.")
 
-def extract_graph(focus_labels, selected_edges):
+def extract_graph(focus_labels, selected_edges, show_project_deliverables):
     nodes = []
     edges = []
     node_dict = {}  # Use a dictionary to track added nodes for easier lookup
@@ -122,16 +161,33 @@ def extract_graph(focus_labels, selected_edges):
                     width=3
                 ))
 
-    return nodes, edges
-
-
+    # Add project deliverables and their connections if the toggle is enabled
+    if show_project_deliverables:
+        for s in g.subjects(RDF.type, EURIO.ProjectDeliverable):
+            if any(o in specific_nodes for _, o in g.predicate_objects(subject=s) if g.qname(_) == "eurio:isResultOf"):
+                if str(s) not in node_dict:
+                    node_dict[str(s)] = create_node(s, is_deliverable=True)
+                    nodes.append(node_dict[str(s)])
+                for p, o in g.predicate_objects(subject=s):
+                    if g.qname(p) == "eurio:isResultOf":
+                        if str(o) not in node_dict:
+                            node_dict[str(o)] = create_node(o)
+                            nodes.append(node_dict[str(o)])
+                        edges.append(Edge(
+                            source=str(s),
+                            target=str(o),
+                            color=edge_colors.get("eurio:isResultOf", "#32CD32"),
+                            width=3
+                        ))
 
+    return nodes, edges
 
-def create_node(subject):
+def create_node(subject, is_deliverable=False):
     label = str(g.value(subject, SKOS.prefLabel) or g.value(subject, RDFS.label) or "")
     url = str(g.value(subject, SCHEMA.url) or "#")
     presentation_icon = "https://raw.githubusercontent.com/BIG-MAP/ProjectKnowledgeGraph/main/assets/img/icon/presentation_icon.png"
     publication_icon = "https://raw.githubusercontent.com/BIG-MAP/ProjectKnowledgeGraph/main/assets/img/icon/publication_icon.png"
+    deliverable_icon = "https://raw.githubusercontent.com/BIG-MAP/ProjectKnowledgeGraph/main/assets/img/icon/deliverable_icon.png"
 
     # Check if this node is an object of the relations bigmap:hasPresentation or schema:citation
     is_presentation_node = any(
@@ -144,18 +200,19 @@ def create_node(subject):
     )
 
     # Decide on image URL based on node's relationship
-    if is_presentation_node:
+    if is_deliverable:
+        image_url = deliverable_icon
+    elif is_presentation_node:
         image_url = presentation_icon
     elif is_publication_node:
         image_url = publication_icon
     else:
         image_url = str(g.value(subject, SCHEMA.logo) or None)
 
     shape = "circularImage" if image_url else "dot"
-    color = None if image_url else "blue"
-
-    return Node(id=str(subject), label=label, url=url, color=color, image=image_url, shape=shape, font_color="#FAFAFA")
+    color = None if image_url else "white"  # Set default color to white if no image
 
+    return Node(id=str(subject), label=label, url=url, color=color, image=image_url, shape=shape, font_color="#000000")
 
 
 if __name__ == "__main__":