Patch index error

pages/input.py

@@ -60,25 +60,35 @@ else:
     target_node_type_index = st.session_state.query_options['target_node_type'].index(st.session_state.query['target_node_type'])
     relation_index = st.session_state.query_options['relation'].index(st.session_state.query['relation'])
 
+# Define error catching function
+def catch_index_error(index, index_options):
+    if index >= len(index_options):
+        return 0
+    else:
+        return index
+
 # Select source node type
 source_node_type_options = node_types['node_type']
 source_node_type = st.selectbox("Source Node Type", source_node_type_options,
-                                format_func = lambda x: x.replace("_", " "), index = source_node_type_index)
+                                format_func = lambda x: x.replace("_", " "),
+                                index = catch_index_error(source_node_type_index, source_node_type_options))
 
 # Select source node
 source_node_options = kg_nodes[kg_nodes['node_type'] == source_node_type]['node_name']
 source_node = st.selectbox("Source Node", source_node_options,
-                           index = source_node_index)
+                            index = catch_index_error(source_node_index, source_node_options))
 
 # Select target node type
 target_node_type_options = edge_types[edge_types.x_type == source_node_type].y_type.unique()
 target_node_type = st.selectbox("Target Node Type", target_node_type_options,
-                                format_func = lambda x: x.replace("_", " "), index = target_node_type_index)
+                                format_func = lambda x: x.replace("_", " "),
+                                index = catch_index_error(target_node_type_index, target_node_type_options))
 
 # Select relation
 relation_options = edge_types[(edge_types.x_type == source_node_type) & (edge_types.y_type == target_node_type)].relation.unique()
 relation = st.selectbox("Edge Type", relation_options,
-                        format_func = lambda x: x.replace("_", "-"), index = relation_index)
+                        format_func = lambda x: x.replace("_", "-"),
+                        index = catch_index_error(relation_index, relation_options))
 
 # Button to submit query
 if st.button("Submit Query"):
@@ -115,4 +125,4 @@ if st.button("Submit Query"):
 st.subheader("Knowledge Graph", divider = "red")    
 display_data = kg_nodes[['node_id', 'node_type', 'node_name', 'node_source']].copy()
 display_data = display_data.rename(columns = {'node_id': 'ID', 'node_type': 'Type', 'node_name': 'Name', 'node_source': 'Database'})
-st.dataframe(display_data, use_container_width = True)
+st.dataframe(display_data, use_container_width = True, hide_index = True)