Update app.py

app.py

@@ -3,10 +3,6 @@ import streamlit as st
 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"
-]
 selected_types = st.multiselect('Select entity types to show:', entity_types_to_show)
 
 # Header and intro
@@ -36,7 +32,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", "LOCATION", "ORGANIZATION", "EVENT"]:
             entity_details = {
                 "Name": entity.name,
                 "Type": entity_type_name,