Introducing team-specific defaults, immediate about page loading, and validation display on predict page

app.py

@@ -155,4 +155,6 @@ def check_password():
 menu() # Render the dynamic menu!
 
 if not check_password():
-    st.stop()
+    st.stop()
+
+st.switch_page("pages/about.py")

pages/input.py

@@ -43,6 +43,17 @@ if "query" not in st.session_state:
     source_node_index = 0
     target_node_type_index = 0
     relation_index = 0
+
+    if st.session_state.team == "Clalit":
+        source_node_type_index = 2
+        source_node_index = 0
+        target_node_type_index = 3
+        relation_index = 2
+
+    if st.session_state.team == "ASAP":
+        source_node_type_index = 2
+        source_node_index = 10255
+
 else:
     source_node_type_index = st.session_state.query_options['source_node_type'].index(st.session_state.query['source_node_type'])
     source_node_index = st.session_state.query_options['source_node'].index(st.session_state.query['source_node'])
@@ -88,6 +99,10 @@ if st.button("Submit Query"):
         "relation": list(relation_options)
     }
 
+    # Delete validation from session state
+    if "validation" in st.session_state:
+        del st.session_state.validation
+
     # # Write query to console
     # st.write("Current Query:")
     # st.write(st.session_state.query)

pages/predict.py

@@ -143,6 +143,37 @@ with st.spinner('Computing predictions...'):
     # Add URLs to database column
     display_data['Database'] = display_data.apply(lambda x: map_dbs[target_node_type](x['ID']), axis = 1)
 
+    # Check if validation data exists
+    if 'validation' in st.session_state:
+
+        # Checkbox to allow reverse edges
+        show_val = st.checkbox("Show Ground Truth Validation?", value = False)
+
+        if show_val:
+
+            # Get validation data
+            val_results = st.session_state.validation.copy()
+
+            # Merge with predictions
+            val_display_data = pd.merge(display_data, val_results, left_on = 'ID', right_on = 'y_id', how='left')
+            val_display_data = val_display_data.fillna(0).drop(columns='y_id')
+
+            # Get new columns
+            val_relations = val_display_data.columns.difference(display_data.columns).tolist()
+
+            # Replace 0 with blank and 1 with check emoji in new columns
+            for col in val_relations:
+                val_display_data[col] = val_display_data[col].replace({0: '', 1: '✅'})
+
+            # Define a function to apply styles
+            def style_val(val):
+                if val == '✅':
+                    return 'background-color: #C2EABD;' #  text-align: center;
+                return 'background-color: #F5F5F5;' # text-align: center;
+
+    else:
+        show_val = False
+
 
     # NODE SEARCH
 
@@ -152,11 +183,16 @@ with st.spinner('Computing predictions...'):
 
     # Filter nodes
     if len(selected_nodes) > 0:
-        selected_display_data = display_data[display_data.Name.isin(selected_nodes)]
+
+        if show_val:
+            # selected_display_data = val_display_data[val_display_data.Name.isin(selected_nodes)]
+            selected_display_data = val_display_data[val_display_data.Name.isin(selected_nodes)].style.map(style_val, subset=val_relations)
+        else:
+            selected_display_data = display_data[display_data.Name.isin(selected_nodes)]
 
         # Show filtered nodes
         if target_node_type not in ['disease', 'anatomy']:
-            st.dataframe(selected_display_data, use_container_width = True,
+            st.dataframe(selected_display_data, use_container_width = True, hide_index = True,
                         column_config={"Database": st.column_config.LinkColumn(width = "small",
                                                                                help = "Click to visit external database.",
                                                                                display_text = display_database)})
@@ -185,14 +221,18 @@ with st.spinner('Computing predictions...'):
     # Show top ranked nodes
     st.subheader("Model Predictions", divider = "blue")
     top_k = st.slider('Select number of top ranked nodes to show.', 1, target_nodes.shape[0], min(500, target_nodes.shape[0])) 
+
+    # Show full results
+    # full_results = val_display_data.iloc[:top_k] if show_val else display_data.iloc[:top_k]
+    full_results = val_display_data.iloc[:top_k].style.map(style_val, subset=val_relations) if show_val else display_data.iloc[:top_k]
     
     if target_node_type not in ['disease', 'anatomy']:
-        st.dataframe(display_data.iloc[:top_k], use_container_width = True,
+        st.dataframe(full_results, use_container_width = True, hide_index = True,
                     column_config={"Database": st.column_config.LinkColumn(width = "small",
                                                                            help = "Click to visit external database.",
                                                                            display_text = display_database)})
     else:
-        st.dataframe(display_data.iloc[:top_k], use_container_width = True)
+        st.dataframe(full_results, use_container_width = True, hide_index = True,)
 
     # Save to session state
     st.session_state.predictions = display_data

pages/validate.py

@@ -65,6 +65,16 @@ with st.spinner('Searching known relationships...'):
 # If there exist edges in KG
 if len(edges_in_kg) > 0:
 
+    with st.spinner('Saving validation results...'):
+
+        # Cast long to wide
+        val_results = edge_subset[['relation', 'y_id']].pivot_table(index='y_id', columns='relation', aggfunc='size', fill_value=0)
+        val_results = (val_results > 0).astype(int).reset_index()
+        val_results.columns = [val_results.columns[0]] + [x.replace('_', ' ').title() for x in val_results.columns[1:]]
+
+        # Save validation results to session state
+        st.session_state.validation = val_results
+
     with st.spinner('Plotting known relationships...'):
 
         # Define a color map for different relations