Update app.py

app.py

@@ -4,9 +4,8 @@ from google.oauth2 import service_account
 from google.cloud import language_v1
 
 # Adding checkbox options for entity types
-entity_types_to_show = [
-    "UNKNOWN", "PERSON", "LOCATION", "ORGANIZATION", "EVENT", "WORK_OF_ART", "CONSUMER_GOOD", "OTHER"
-]
+entity_types_to_show = ["PERSON", "ORGANIZATION", "EVENT"]
+
 selected_types = st.multiselect('Select entity types to show:', entity_types_to_show)
 
 # Header and intro
@@ -36,7 +35,7 @@ def sample_analyze_entities(text_content, your_query=""):
     entities_list = []
     for entity in response.entities:
         entity_type_name = language_v1.Entity.Type(entity.type_).name
-        if not selected_types or entity_type_name in selected_types:
+        if entity_type_name in ["PERSON", "ORGANIZATION", "EVENT"]:
             entity_details = {
                 "Name": entity.name,
                 "Type": entity_type_name,