Update app.py

app.py

@@ -236,20 +236,11 @@ def run_ontology_structure_analysis():
 def run_entity_exploration():
     st.title("Entity Exploration")
 
-    # 永久修正：從 class_name 與圖中的 class_type 做不區分大小寫比對
-    def safe_match(class_name, class_type):
-        return class_name.strip().lower() == class_type.strip().lower() if class_type else False
-
-    # 重新抓取實體：正確比對 class_type
-    entities = []
-    for class_name in ontology_manager.get_classes():
-        class_entities = [
-            node for node, attr in ontology_manager.graph.nodes(data=True)
-            if attr.get("type") == "instance" and safe_match(class_name, attr.get("class_type"))
-        ]
-        if class_entities:
-            entities.extend(class_entities)
-
+    # 從圖中抓出所有實體（不依賴 ontology["classes"] 中的列表）
+    entities = [
+        node for node, attr in ontology_manager.graph.nodes(data=True)
+        if attr.get("type") == "instance"
+    ]
     entities = sorted(set(entities))
 
     # Debug info
@@ -276,9 +267,8 @@ def run_entity_exploration():
             max_distance=max_distance,
             include_classes=True
         )
-    elif not entities:
+    else:
         st.warning("No entities found in the ontology.")
-
     
     if selected_entity:
         # Get entity information