Update app.py

app.py

@@ -3,8 +3,6 @@ import streamlit as st
 from google.oauth2 import service_account
 from google.cloud import language_v1
 
-
-
 # Header and intro
 st.title("Google Cloud NLP Entity Analyzer")
 st.write("## Introduction to the Knowledge Graph API")
@@ -31,16 +29,14 @@ 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 entity_type_name in ["PERSON", "LOCATION", "ORGANIZATION", "EVENT"]:
-            entity_details = {
-                "Name": entity.name,
-                "Type": entity_type_name,
-                "Salience Score": entity.salience,
-                "Metadata": entity.metadata,
-                "Mentions": [mention.text.content for mention in entity.mentions]
-            }
-            entities_list.append(entity_details)
+        entity_details = {
+            "Name": entity.name,
+            "Type": language_v1.Entity.Type(entity.type_).name,
+            "Salience Score": entity.salience,
+            "Metadata": entity.metadata,
+            "Mentions": [mention.text.content for mention in entity.mentions]
+        }
+        entities_list.append(entity_details)
 
     if your_query:
         st.write(f"### We found {len(entities_list)} results for your query of **{your_query}**")