Update src/visualization.py

src/visualization.py

@@ -492,69 +492,52 @@ def display_ontology_stats(ontology_manager):
 
 
 def display_class_hierarchy_tree(ontology_manager, class_hierarchy):
-    """Display class hierarchy as an interactive tree."""
-    # Find root classes (those that aren't subclasses of anything else)
+    """Display class hierarchy using expanders at root level, and plain list for subclasses."""
+    # Find all subcategory sets
     all_subclasses = set()
     for subclasses in class_hierarchy.values():
         all_subclasses.update(subclasses)
-    
+
+    # Root classes are those that are not subclassed by any class
     root_classes = [cls for cls in ontology_manager.get_classes() if cls not in all_subclasses]
-    
-    # Create a recursive function to display the hierarchy
-    def display_subclasses(class_name, indent=0):
-        # Get class info
-        class_info = ontology_manager.ontology_data["classes"].get(class_name, {})
+
+    for root_class in sorted(root_classes):
+        class_info = ontology_manager.ontology_data["classes"].get(root_class, {})
         description = class_info.get("description", "")
-        instance_count = len(ontology_manager.get_instances_of_class(class_name, include_subclasses=False))
-        
-        # Display class with expander for subclasses
-        if indent == 0:
-            # Root level classes are always expanded
-            with st.expander(f"📁 {class_name} ({instance_count} instances)", expanded=True):
-                st.markdown(f"**Description:** {description}")
-                
-                # Show properties if any
-                properties = class_info.get("properties", [])
-                if properties:
-                    st.markdown("**Properties:**")
-                    st.markdown(", ".join(properties))
-                
-                # Display subclasses
-                subclasses = class_hierarchy.get(class_name, [])
-                if subclasses:
-                    st.markdown("**Subclasses:**")
-                    for subclass in sorted(subclasses):
-                        display_subclasses(subclass, indent + 1)
-                else:
-                    st.markdown("*No subclasses*")
-        else:
-            # Nested classes use indentation and only show direct instances
-            if instance_count > 0:
-                class_label = f"📁 {class_name} ({instance_count} instances)"
+        instance_count = len(ontology_manager.get_instances_of_class(root_class, include_subclasses=False))
+        properties = class_info.get("properties", [])
+
+        with st.expander(f"📁 {root_class} ({instance_count} instances)", expanded=True):
+            st.markdown(f"**Description:** {description}")
+            if properties:
+                st.markdown("**Properties:**")
+                st.markdown(", ".join(properties))
+
+            subclasses = class_hierarchy.get(root_class, [])
+            if subclasses:
+                st.markdown("**Subclasses:**")
+                for subclass in sorted(subclasses):
+                    display_subclass_plain(subclass, class_hierarchy, ontology_manager, indent=1)
             else:
-                class_label = f"📁 {class_name}"
-                
-            with st.expander(class_label, expanded=False):
-                st.markdown(f"**Description:** {description}")
-                
-                # Show properties if any
-                properties = class_info.get("properties", [])
-                if properties:
-                    st.markdown("**Properties:**")
-                    st.markdown(", ".join(properties))
-                
-                # Display subclasses
-                subclasses = class_hierarchy.get(class_name, [])
-                if subclasses:
-                    st.markdown("**Subclasses:**")
-                    for subclass in sorted(subclasses):
-                        display_subclasses(subclass, indent + 1)
-                else:
-                    st.markdown("*No subclasses*")
-    
-    # Display each root class
-    for root_class in sorted(root_classes):
-        display_subclasses(root_class)
+                st.markdown("*No subclasses*")
+
+
+def display_subclass_plain(class_name, class_hierarchy, ontology_manager, indent=1):
+    """Use indentation to render subclass info without nested expanders."""
+    indent_str = " " * indent
+    class_info = ontology_manager.ontology_data["classes"].get(class_name, {})
+    description = class_info.get("description", "")
+    properties = class_info.get("properties", [])
+    instance_count = len(ontology_manager.get_instances_of_class(class_name, include_subclasses=False))
+
+    st.markdown(f"{indent_str}🔹 **{class_name}** ({instance_count} instances)")
+    st.markdown(f"{indent_str}> {description}")
+    if properties:
+        st.markdown(f"{indent_str}*Properties:* {', '.join(properties)}")
+
+    subclasses = class_hierarchy.get(class_name, [])
+    for subclass in sorted(subclasses):
+        display_subclass_plain(subclass, class_hierarchy, ontology_manager, indent + 1)
 
 def get_class_depths(ontology_manager) -> Dict[str, int]:
     """Calculate the inheritance depth of each class."""