diff --git a/mm_agents/accessibility_tree_wrap/heuristic_retrieve.py b/mm_agents/accessibility_tree_wrap/heuristic_retrieve.py index e37f614..9611ea3 100644 --- a/mm_agents/accessibility_tree_wrap/heuristic_retrieve.py +++ b/mm_agents/accessibility_tree_wrap/heuristic_retrieve.py @@ -40,7 +40,7 @@ def judge_node(node: ET, platform="ubuntu", check_image=False) -> bool: or node.tag.endswith("textfield")\ or node.tag.endswith("textarea")\ or node.tag.endswith("menu")\ - or node.tag in [ "alert", "canvas", "check-box" + or node.tag in { "alert", "canvas", "check-box" , "combo-box", "entry", "icon" , "image", "paragraph", "scroll-bar" , "section", "slider", "static" @@ -48,7 +48,7 @@ def judge_node(node: ET, platform="ubuntu", check_image=False) -> bool: , "netuiribbontab", "start", "trayclockwclass" , "traydummysearchcontrol", "uiimage", "uiproperty" , "uiribboncommandbar" - ] + } keeps = keeps and ( platform=="ubuntu"\ and node.get("{{{:}}}showing".format(state_ns), "false")=="true"\ and node.get("{{{:}}}visible".format(state_ns), "false")=="true"\