Update app.py

app.py

@@ -235,27 +235,45 @@ def run_ontology_structure_analysis():
 
 def run_entity_exploration():
     st.title("Entity Exploration")
-    
-    # Get all entities
+
+    # Get all entities from graph based on instance type
     entities = []
     for class_name in ontology_manager.get_classes():
-        entities.extend(ontology_manager.get_instances_of_class(class_name))
-    
+        class_entities = ontology_manager.get_instances_of_class(class_name)
+        if class_entities:
+            entities.extend(class_entities)
+
     # Remove duplicates and sort
     entities = sorted(set(entities))
 
-    # Debug output: Confirm loaded entities
+    # Debug: print full entity list and their types
     st.write("Loaded entities:", entities)
-    
     st.write("Ontology classes:", ontology_manager.get_classes())
-    
-    # 額外 debug: 顯示所有節點類型與 class_type
     for node, attr in ontology_manager.graph.nodes(data=True):
         if attr.get("type") == "instance":
             st.write("Found instance:", node, "of class:", attr.get("class_type"))
-    
-    # Create a dropdown selection box
-    selected_entity = st.selectbox("Select Entity", entities)
+
+    if entities:
+        selected_entity = st.selectbox("Select Entity", entities)
+
+        if selected_entity:
+            entity_info = ontology_manager.get_entity_info(selected_entity)
+            display_entity_details(entity_info, ontology_manager)
+
+            if st.button("View this Entity in the Knowledge Graph"):
+                st.session_state.central_entity = selected_entity
+                st.rerun()
+
+            st.subheader("Entity Neighborhood")
+            max_distance = st.slider("Maximum Neighborhood Distance", 1, 3, 1)
+            neighborhood = knowledge_graph.get_entity_neighborhood(
+                selected_entity,
+                max_distance=max_distance,
+                include_classes=True
+            )
+            # Optional: visualize neighborhood or print debug info
+    else:
+        st.warning("No entities found in the ontology.")
     
     if selected_entity:
         # Get entity information