Adding description labels and content filters

Inspect_Logic.py

@@ -21,6 +21,8 @@ def get_smem_root_from_file(smem_file):
     parser.parse_file(tokens)
     return parser.get_context_root()
 
+## DEFINE THE FILE UPLOADER ELEMENT
+
 if "col_obj_list" not in st.session_state:
     st.session_state["col_obj_list"] = None
 
@@ -37,13 +39,26 @@ if file is not None:
 else:
     st.session_state["col_obj_list"] = None
 
-# inpath = "/Users/bstearn1/Documents/Projects/OptumGitHub/Intelligent-Solutions/AICarePlan/agent/D-RULES_082222_smem.soar"
-# inpath = "SMEM Generator/test_pim_smem.soar"
-# st.session_state.col_obj_list[0] = get_smem_root_from_file(inpath)
-
 if st.session_state.col_obj_list is None:
     st.stop()
 
+
+## DEFINE THE CONTENT FILTERS
+@st.cache(show_spinner=False, hash_funcs={SMEM_Obj: id})
+def get_filter_features_dict(root_obj):
+    return root_obj.get_referenced_features()
+
+# Get the content to filter on
+features_dict = get_filter_features_dict(st.session_state.col_obj_list[0])
+filters_expander = st.expander(label="Filters")
+filters_cols = filters_expander.columns(len(features_dict))
+filters_dict = {}
+for key, col in zip(features_dict, filters_cols):
+    filters_dict[key] = col.multiselect(label=key, options=["(none)"]+sorted(features_dict[key]))
+
+
+## DEFINE THE KNOWLEDGE INSPECTOR COLUMNS
+
 def add_col(index, obj):
     st.session_state.col_obj_list = st.session_state.col_obj_list[:index+1]
     st.session_state.col_obj_list.append(obj)
@@ -62,10 +77,20 @@ def get_header_str(obj_type):
     else:
         return " "
 
+def get_tested_wmes_from_obj_str(obj_str):
+    attr_list = []
+    val_list = []
+    clauses = str(obj_str).split(" AND ")
+    for clause in clauses:
+        attr,val = clause.replace("(","").replace(")","").split(" is ", maxsplit=1)
+        attr_list += [attr]
+        val_list += [val]
+    return attr_list, val_list
+
 st.subheader("Click the buttons below to select conditions and to inspect resulting actions.")
 
 cols = st.columns(max(MIN_COLS,len(st.session_state.col_obj_list)))
-
+# Iteratively build the columns of navigable knowlege elements
 for i,col in enumerate(cols):
     try:
         if st.session_state.col_obj_list[i] == None:
@@ -85,17 +110,40 @@ for i,col in enumerate(cols):
     col.text(obj_str)
     if obj_desc != None:
         col.markdown("*"+obj_desc+"*")
+
+    # Print the child objects of this object as the items in this column
     for j,sub in enumerate(sub_objs):
         sub_str,sub_label,sub_desc = sub.to_string()
+        if sub_desc != None:
+            button_text = sub_desc
+        else:
+            button_text = sub_label
+        # Check filters
+        if sub.obj_type == ObjType.COND_PRIM or sub.obj_type == ObjType.COND_CONJ:
+            # Get the feature and value for this sub obj
+            keep = True
+            attr_list, val_list = get_tested_wmes_from_obj_str(sub_desc)
+            # Check each filter key for a match
+            for attr in filters_dict:
+                if attr not in attr_list:
+                    # Filter doesn't apply for this object
+                    continue
+                if val_list[attr_list.index(attr)] not in filters_dict[attr] and len(filters_dict[attr]) > 0:
+                    # Value not present
+                    keep = False
+                    break
+            if not keep:
+                continue
+
         if sub.obj_type == ObjType.OP:
             subsub = list(sub.get_child_objects(compact=True))[0]
             _,group_string,_ = subsub.to_string()
             group_string = "\n * "+str(group_string).replace("AND", "\n * ")
             col.markdown(group_string)
-            if col.button(sub_label, key="button"+str(i)+"-"+str(j)):
+            if col.button(button_text, key="button"+str(i)+"-"+str(j)):
                 add_col(i,sub)
         elif sub.obj_type == ObjType.ACT_GROUP:
             col.markdown("*Output Details:* ")
             col.markdown("\n * "+str(sub_str).replace(") AND", ")\n * "))
-        elif col.button(sub_label, key="button"+str(i)+"-"+str(j)):
+        elif col.button(button_text, key="button"+str(i)+"-"+str(j)):
             add_col(i,sub)

smem_obj.py

@@ -1,3 +1,4 @@
+from collections import defaultdict
 from enum import Enum
 
 class SMEM_Obj():
@@ -16,6 +17,8 @@ class SMEM_Obj():
         self.wme_list.append((attr,val))
 
     def get_class(self):
+        if self.class_str != None:
+            return self.class_str
         for (attr,val) in self.wme_list:
             if attr == "^class":
                 self.class_str = val
@@ -32,6 +35,31 @@ class SMEM_Obj():
                 break
         return self.description
 
+    def get_referenced_features(self, dict_of_values=None):
+        # Collect a map of feature fields to possible values, as referenced by descendants from this object
+        if not dict_of_values:
+            dict_of_values = defaultdict(set)
+        children = self.get_child_objects(compact=False)
+        for child in children:
+            _, short_label, desc = child.to_string()
+            # Trim the label based on the object type
+            if child.obj_type == ObjType.COND_PRIM:
+                if desc:
+                    str_to_use = desc
+                else:
+                    str_to_use = short_label
+                ind = str_to_use.index(" is ")
+                feature, value = str_to_use.replace("(","").replace(")","").split(" is ",maxsplit=1)
+                dict_of_values[feature].add(value)
+            # Merge this dict of sets with results from each branch of children
+            child_dict = child.get_referenced_features(dict_of_values)
+            for key, val in child_dict.items():
+                if key in dict_of_values:
+                    dict_of_values[key].update(val)
+                else:
+                    dict_of_values[key] = val
+        return dict_of_values
+
     def get_compact_printable_object(self):
         """ Return this object if printable in compact mode, else return the set of next printable descendants.
         """
@@ -162,17 +190,24 @@ class SMEM_Obj():
                 retval_desc = val
             # if attr == "^supplies-condition":
             #     child_conjs.append(val)
-        
+        compile_component_descs = (retval_desc == None)
+        if compile_component_descs:
+            retval_desc = ""
+
         for i,cond in enumerate(parent_conds):
             if i > 0:
                 retval_long += " \nAND "
                 retval_short += " AND "
+                if compile_component_descs:
+                    retval_desc += " AND "
             if cond.obj_type == ObjType.COND_CONJ:
-                long, short, _ = cond.get_cond_string()
+                long, short, desc = cond.get_cond_string()
             else:
-                long, short, _ = cond.get_prim_cond_string(negate=is_negation)
+                long, short, desc = cond.get_prim_cond_string(negate=is_negation)
             retval_long += long
             retval_short += short
+            if compile_component_descs:
+                retval_desc += "("+desc+")"
         return retval_long, retval_short, retval_desc
     
     def get_op_string(self):